Compare commits

...

1 Commits

Author SHA1 Message Date
Brandon Hancock
040e5a78d2 Add back in Gui's tool_usage fix 2024-07-15 09:21:21 -04:00
2 changed files with 4 additions and 9 deletions

View File

@@ -181,7 +181,6 @@ class Agent(BaseAgent):
self.agent_executor.tools = parsed_tools self.agent_executor.tools = parsed_tools
self.agent_executor.task = task self.agent_executor.task = task
# TODO: COMPARE WITH ARGS AND WITHOUT ARGS
self.agent_executor.tools_description = self._render_text_description_and_args( self.agent_executor.tools_description = self._render_text_description_and_args(
parsed_tools parsed_tools
) )

View File

@@ -151,16 +151,12 @@ class ToolUsage:
for k, v in calling.arguments.items() for k, v in calling.arguments.items()
if k in acceptable_args if k in acceptable_args
} }
result = tool._run(**arguments) result = tool.invoke(input=arguments)
except Exception: except Exception:
if tool.args_schema: arguments = calling.arguments
arguments = calling.arguments result = tool.invoke(input=arguments)
result = tool._run(**arguments)
else:
arguments = calling.arguments.values() # type: ignore # Incompatible types in assignment (expression has type "dict_values[str, Any]", variable has type "dict[str, Any]")
result = tool._run(*arguments)
else: else:
result = tool._run() result = tool.invoke(input={})
except Exception as e: except Exception as e:
self._run_attempts += 1 self._run_attempts += 1
if self._run_attempts > self._max_parsing_attempts: if self._run_attempts > self._max_parsing_attempts: