andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 08:38:30 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refactoring task execution

Browse Source
This commit is contained in:
João Moura
2024-02-10 11:28:08 -08:00
parent 00206a62ab
commit e79da7bc05
5 changed files with 62 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/agent.py
View File

@@ -125,7 +125,7 @@ class Agent(BaseModel):
def execute_task(
self,
task: str,
task: Any,
context: Optional[str] = None,
tools: Optional[List[Any]] = None,
) -> str:
@@ -140,17 +140,20 @@ class Agent(BaseModel):
Output of the agent
"""
task_prompt = task.prompt()
if context:
task = self.i18n.slice("task_with_context").format(
task=task, context=context
task_prompt = self.i18n.slice("task_with_context").format(
task=task_prompt, context=context
)
tools = tools or self.tools
self.agent_executor.tools = tools
self.agent_executor.task = task
result = self.agent_executor.invoke(
{
"input": task,
"input": task_prompt,
"tool_names": self.__tools_names(tools),
"tools": render_text_description(tools),
}