andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 08:12:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Small updates to the code formatting

Browse Source
This commit is contained in:
João Moura
2023-12-30 10:53:10 -03:00
parent d6c60f873a
commit f887d9bd79
2 changed files with 34 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
crewai/agent.py
View File

@@ -101,7 +101,6 @@ class Agent(BaseModel):
)["output"]
def set_cache_handler(self, cache_handler) -> None:
print(f"cache_handler: {cache_handler}")
self.cache_handler = cache_handler
self.tools_handler = ToolsHandler(cache=self.cache_handler)
self.__create_agent_executor()

25
crewai/prompts.py
View File

@@ -14,16 +14,15 @@ class Prompts(BaseModel):
"""\
Begin! This is VERY important to you, your job depends on it!
Current Task: {input}
{agent_scratchpad}
"""
Current Task: {input}"""
)
SCRATCHPAD_SLICE: ClassVar[str] = "\n{agent_scratchpad}"
MEMORY_SLICE: ClassVar[str] = dedent(
"""\
This is the summary of your work so far:
{chat_history}
"""
{chat_history}"""
)
ROLE_PLAYING_SLICE: ClassVar[str] = dedent(
@@ -31,13 +30,13 @@ class Prompts(BaseModel):
You are {role}.
{backstory}
Your personal goal is: {goal}
"""
Your personal goal is: {goal}"""
)
TOOLS_SLICE: ClassVar[str] = dedent(
"""\
TOOLS:
------
You have access to the following tools:
@@ -58,8 +57,7 @@ class Prompts(BaseModel):
```
Thought: Do I need to use a tool? No
Final Answer: [your response here]
```
"""
```"""
)
VOTING_SLICE: ClassVar[str] = dedent(
@@ -70,18 +68,17 @@ class Prompts(BaseModel):
{format_instructions}
These are your co-workers and their roles:
{coworkers}
"""
{coworkers}"""
)
TASK_EXECUTION_WITH_MEMORY_PROMPT: ClassVar[str] = PromptTemplate.from_template(
ROLE_PLAYING_SLICE + TOOLS_SLICE + MEMORY_SLICE + TASK_SLICE
ROLE_PLAYING_SLICE + TOOLS_SLICE + MEMORY_SLICE + TASK_SLICE + SCRATCHPAD_SLICE
)
TASK_EXECUTION_PROMPT: ClassVar[str] = PromptTemplate.from_template(
ROLE_PLAYING_SLICE + TOOLS_SLICE + TASK_SLICE
ROLE_PLAYING_SLICE + TOOLS_SLICE + TASK_SLICE + SCRATCHPAD_SLICE
)
CONSENSUNS_VOTING_PROMPT: ClassVar[str] = PromptTemplate.from_template(
ROLE_PLAYING_SLICE + VOTING_SLICE + TASK_SLICE
ROLE_PLAYING_SLICE + VOTING_SLICE + TASK_SLICE + SCRATCHPAD_SLICE
)