Method _call_tools

backend/app/agents/scraping_agent.py:89–131

Execute the selected tools.

(self, state: AgentState)

        }
    
    async def _call_tools(self, state: AgentState) -> AgentState:
        """Execute the selected tools."""
        last_message = state["messages"][-1]
        
        if not last_message.tool_calls:
            return state
        
        # Execute each tool call
        for tool_call in last_message.tool_calls:
            tool_name = tool_call["name"]
            tool_args = tool_call["args"]
            
            # Find and execute the tool
            tool = next((t for t in SCRAPING_TOOLS if t.name == tool_name), None)
            if tool:
                if asyncio.iscoroutinefunction(tool.func):
                    result = await tool.func(**tool_args)
                else:
                    result = tool.func(**tool_args)
                
                # Update state based on tool
                if tool_name == "add_url" and result["success"]:
                    state["urls"].append(tool_args["url"])
                elif tool_name == "remove_url" and result["success"]:
                    state["urls"].remove(tool_args["url"])
                elif tool_name == "define_schema" and result["success"]:
                    state["schema"] = result["schema"]
                elif tool_name == "generate_code" and result["success"]:
                    state["generated_code"] = result["code"]
                elif tool_name == "clear_pipeline" and result["success"]:
                    state["urls"] = []
                    state["schema"] = {}
                    state["generated_code"] = ""
                
                # Add tool response as message
                tool_message = AIMessage(
                    content=json.dumps(result),
                    name=tool_name,
                    additional_kwargs={"tool_call_id": tool_call["id"]}
                )
                state["messages"].append(tool_message)
        
        return state
    
    async def _execute_scraping(self, state: AgentState) -> AgentState:
        """Execute the actual scraping with ScrapeGraphAI."""