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

bug fixing

Browse Source
This commit is contained in:
João Moura
2024-02-20 10:40:16 -03:00
parent e24f4867df
commit 8f5d735b2f
6 changed files with 38 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/crewai/utilities/prompts.py
View File

@@ -15,12 +15,7 @@ class Prompts(BaseModel):
def task_execution_with_memory(self) -> BasePromptTemplate:
"""Generate a prompt for task execution with memory components."""
slices = ["role_playing"]
if len(self.tools) > 0:
slices.append("tools")
else:
slices.append("no_tools")
slices.extend(["memory", "task"])
slices = ["role_playing", "tools", "memory", "task"]
return self._build_prompt(slices)
def task_execution_without_tools(self) -> BasePromptTemplate:
@@ -29,12 +24,7 @@ class Prompts(BaseModel):
def task_execution(self) -> BasePromptTemplate:
"""Generate a standard prompt for task execution."""
slices = ["role_playing"]
if len(self.tools) > 0:
slices.append("tools")
else:
slices.append("no_tools")
slices.append("task")
slices = ["role_playing", "tools", "task"]
return self._build_prompt(slices)
def _build_prompt(self, components: list[str]) -> BasePromptTemplate: