Change additional Tools to Tools

src/codegen/agents/chat_agent.py

@@ -29,7 +29,7 @@ def __init__(self, codebase: "Codebase", model_provider: str = "anthropic", mode
                 - max_tokens: Maximum number of tokens to generate
         """
         self.codebase = codebase
-        self.agent = create_chat_agent(self.codebase, model_provider=model_provider, model_name=model_name, memory=memory, additional_tools=tools, **kwargs)
+        self.agent = create_chat_agent(self.codebase, model_provider=model_provider, model_name=model_name, memory=memory, tools=tools, **kwargs)
 
     def run(self, prompt: str, thread_id: Optional[str] = None) -> str:
         """Run the agent with a prompt.

src/codegen/agents/code_agent.py

@@ -70,7 +70,7 @@
             model_provider=model_provider,
             model_name=model_name,
             memory=memory,
-            additional_tools=tools,
+            tools=tools,
             config=agent_config,
             **kwargs,
         )
@@ -87,7 +87,7 @@
         print(f"Using LangSmith project: {self.project_name}")
 
         # Store SWEBench metadata if provided
         self.run_id = metadata.get("run_id")
         self.instance_id = metadata.get("instance_id")
         # Extract difficulty value from "difficulty_X" format
         difficulty_str = metadata.get("difficulty", "")

src/codegen/extensions/langchain/agent.py

@@ -40,7 +40,7 @@
     system_message: SystemMessage = SystemMessage(REASONER_SYSTEM_MESSAGE),
     memory: bool = True,
     debug: bool = False,
-    additional_tools: list[BaseTool] | None = None,
+    tools: list[BaseTool] | None = None,
     config: AgentConfig | None = None,
     **kwargs,
 ) -> CompiledGraph:
@@ -63,39 +63,32 @@
     """
     llm = LLM(model_provider=model_provider, model_name=model_name, **kwargs)
 
-    # Get all codebase tools
-    tools = [
-        ViewFileTool(codebase),
-        ListDirectoryTool(codebase),
-        RipGrepTool(codebase),
-        # EditFileTool(codebase),
-        CreateFileTool(codebase),
-        DeleteFileTool(codebase),
-        RenameFileTool(codebase),
-        # MoveSymbolTool(codebase),
-        # RevealSymbolTool(codebase),
-        # SemanticEditTool(codebase),
-        ReplacementEditTool(codebase),
-        RelaceEditTool(codebase),
-        ReflectionTool(codebase),
-        SearchFilesByNameTool(codebase),
-        GlobalReplacementEditTool(codebase),
-        # SemanticSearchTool(codebase),
-        # =====[ Github Integration ]=====
-        # Enable Github integration
-        # GithubCreatePRTool(codebase),
-        # GithubViewPRTool(codebase),
-        # GithubCreatePRCommentTool(codebase),
-        # GithubCreatePRReviewCommentTool(codebase),
-    ]
-
-    # Add additional tools if provided
-    if additional_tools:
-        tools.extend(additional_tools)
+    if not tools:
+        tools = [
+            ViewFileTool(codebase),
+            ListDirectoryTool(codebase),
+            RipGrepTool(codebase),
+            # EditFileTool(codebase),
+            CreateFileTool(codebase),
+            DeleteFileTool(codebase),
+            RenameFileTool(codebase),
+            ReplacementEditTool(codebase),
+            RelaceEditTool(codebase),
+            ReflectionTool(codebase),
+            SearchFilesByNameTool(codebase),
+            GlobalReplacementEditTool(codebase),
+            # SemanticSearchTool(codebase),
+            # =====[ Github Integration ]=====
+            # Enable Github integration
+            # GithubCreatePRTool(codebase),
+            # GithubViewPRTool(codebase),
+            # GithubCreatePRCommentTool(codebase),
+            # GithubCreatePRReviewCommentTool(codebase),
+        ]
 
     memory = MemorySaver() if memory else None
 
     return create_react_agent(model=llm, tools=tools, system_message=system_message, checkpointer=memory, debug=debug, config=config)
 
 
 def create_chat_agent(
@@ -105,7 +98,7 @@
     system_message: SystemMessage = SystemMessage(REASONER_SYSTEM_MESSAGE),
     memory: bool = True,
     debug: bool = False,
-    additional_tools: list[BaseTool] | None = None,
+    tools: list[BaseTool] | None = None,
     config: dict[str, Any] | None = None,  # over here you can pass in the max length of the number of messages
     **kwargs,
 ) -> CompiledGraph:
@@ -128,24 +121,22 @@
     """
     llm = LLM(model_provider=model_provider, model_name=model_name, **kwargs)
 
-    tools = [
-        ViewFileTool(codebase),
-        ListDirectoryTool(codebase),
-        RipGrepTool(codebase),
-        CreateFileTool(codebase),
-        DeleteFileTool(codebase),
-        RenameFileTool(codebase),
-        MoveSymbolTool(codebase),
-        RevealSymbolTool(codebase),
-        RelaceEditTool(codebase),
-    ]
-
-    if additional_tools:
-        tools.extend(additional_tools)
+    if not tools:
+        tools = [
+            ViewFileTool(codebase),
+            ListDirectoryTool(codebase),
+            RipGrepTool(codebase),
+            CreateFileTool(codebase),
+            DeleteFileTool(codebase),
+            RenameFileTool(codebase),
+            MoveSymbolTool(codebase),
+            RevealSymbolTool(codebase),
+            RelaceEditTool(codebase),
+        ]
 
     memory = MemorySaver() if memory else None
 
     return create_react_agent(model=llm, tools=tools, system_message=system_message, checkpointer=memory, debug=debug, config=config)
 
 
 def create_codebase_inspector_agent(
@@ -182,8 +173,8 @@
         RevealSymbolTool(codebase),
     ]
 
     memory = MemorySaver() if memory else None
     return create_react_agent(model=llm, tools=tools, system_message=system_message, checkpointer=memory, debug=debug, config=config)
 
 
 def create_agent_with_tools(
@@ -216,6 +207,6 @@
     """
     llm = LLM(model_provider=model_provider, model_name=model_name, **kwargs)
 
     memory = MemorySaver() if memory else None
 
     return create_react_agent(model=llm, tools=tools, system_message=system_message, checkpointer=memory, debug=debug, config=config)