updating RPM and max_inter logic

This commit is contained in:
João Moura
2024-02-05 23:14:22 -08:00
parent 856981e0ed
commit ab5647145f
4 changed files with 7 additions and 8 deletions

View File

@@ -33,7 +33,7 @@ class CrewAgentExecutor(AgentExecutor):
def _force_answer(self, output: AgentAction):
return AgentStep(
action=output, observation=self.i18n.errors("used_too_many_tools")
action=output, observation=self.i18n.errors("force_final_answer")
)
def _call(
@@ -106,7 +106,7 @@ class CrewAgentExecutor(AgentExecutor):
**inputs,
)
if self._should_force_answer():
if isinstance(output, AgentAction):
if isinstance(output, AgentAction) or isinstance(output, AgentFinish):
output = output
elif isinstance(output, CacheHit):
output = output.action