diff --git a/application/agents/tool_executor.py b/application/agents/tool_executor.py index 0fde060d..69739076 100644 --- a/application/agents/tool_executor.py +++ b/application/agents/tool_executor.py @@ -185,7 +185,10 @@ class ToolExecutor: target_dict[param] = value # Load tool (with caching) - tool = self._get_or_load_tool(tool_data, tool_id, action_name) + tool = self._get_or_load_tool( + tool_data, tool_id, action_name, + headers=headers, query_params=query_params, + ) resolved_arguments = ( {"query_params": query_params, "headers": headers, "body": body} @@ -238,7 +241,10 @@ class ToolExecutor: return result, call_id - def _get_or_load_tool(self, tool_data: Dict, tool_id: str, action_name: str): + def _get_or_load_tool( + self, tool_data: Dict, tool_id: str, action_name: str, + headers: Optional[Dict] = None, query_params: Optional[Dict] = None, + ): """Load a tool, using cache when possible.""" cache_key = f"{tool_data['name']}:{tool_id}:{self.user or ''}" if cache_key in self._loaded_tools: @@ -251,8 +257,8 @@ class ToolExecutor: tool_config = { "url": action_config["url"], "method": action_config["method"], - "headers": {}, - "query_params": {}, + "headers": headers or {}, + "query_params": query_params or {}, } if "body_content_type" in action_config: tool_config["body_content_type"] = action_config.get( diff --git a/extensions/react-widget/package.json b/extensions/react-widget/package.json index 7a9b80df..588c825e 100644 --- a/extensions/react-widget/package.json +++ b/extensions/react-widget/package.json @@ -1,6 +1,6 @@ { "name": "docsgpt", - "version": "0.6.1", + "version": "0.6.2", "private": false, "description": "DocsGPT 🦖 is an innovative open-source tool designed to simplify the retrieval of information from project documentation using advanced GPT models 🤖.", "source": "./src/index.html",