refactor: Remove unused pandas DataFrame from ResourceManager and update CustomTool conversion

Author: BinglanLi

BaseAgent/resource_manager.py

@@ -7,7 +7,6 @@
 
 import importlib
 from typing import Optional
-import pandas as pd
 from BaseAgent.resources import (
     Tool,
     DataLakeItem,
@@ -33,10 +32,7 @@ class ResourceManager:
     def __init__(self):
         """Initialize the resource manager."""
         self.collection = ResourceCollection()
-        
-        # Tool registry features
         self.next_tool_id: int = 0
-        self.tool_document_df: pd.DataFrame = pd.DataFrame(columns=["docid", "document_content"])
 
     # ==========================================================================
     # Tool Management
@@ -54,12 +50,7 @@ def add_tool(self, tool: Tool) -> None:
         # Always assign a fresh sequential ID (internal counter)
         tool.id = self.next_tool_id
         self.next_tool_id += 1
-        
         self.collection.tools.append(tool)
-        
-        # Add to dataframe for retrieval
-        new_row = pd.DataFrame([[tool.id, tool]], columns=["docid", "document_content"])
-        self.tool_document_df = pd.concat([self.tool_document_df, new_row], ignore_index=True)
 
     def add_custom_tool(self, custom_tool: CustomTool) -> None:
         """Add a custom tool to the collection.
@@ -149,10 +140,6 @@ def remove_tool_by_id(self, tool_id: int) -> bool:
         tool = self.find_tool_by_id(tool_id)
         if tool:
             self.collection.tools = [t for t in self.collection.tools if t.id != tool_id]
-            # Remove from dataframe
-            self.tool_document_df = self.tool_document_df[
-                self.tool_document_df['docid'] != tool_id
-            ].reset_index(drop=True)
             return True
         return False
 
@@ -167,12 +154,7 @@ def remove_tool_by_name(self, name: str) -> bool:
         """
         tool = self.get_tool_by_name(name)
         if tool and tool.id is not None:
-            tool_id = tool.id
             self.collection.tools = [t for t in self.collection.tools if t.name != name]
-            # Remove from dataframe
-            self.tool_document_df = self.tool_document_df[
-                self.tool_document_df['docid'] != tool_id
-            ].reset_index(drop=True)
             return True
         return False
 

BaseAgent/resources.py

@@ -16,8 +16,6 @@
     - CustomTool: User-defined custom tools
     - CustomData: User-defined custom datasets
     - CustomSoftware: User-defined custom software/libraries
-
-Date: 2025-10-30
 """
 
 from typing import Any, Literal, Optional
@@ -245,13 +243,14 @@ def validate_name(cls, v: str) -> str:
         return v
 
     def to_tool(self) -> Tool:
-        """Convert to standard Tool model."""
+        """Convert to standard Tool model, preserving selection state."""
         return Tool(
             name=self.name,
             description=self.description,
             required_parameters=self.required_parameters,
             optional_parameters=self.optional_parameters,
-            module=self.module
+            module=self.module,
+            selected=self.selected,
         )