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

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 2b06dd263f
commit 8547d4651b
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"] )["output"]
def set_cache_handler(self, cache_handler) -> None: def set_cache_handler(self, cache_handler) -> None:
print(f"cache_handler: {cache_handler}")
self.cache_handler = cache_handler self.cache_handler = cache_handler
self.tools_handler = ToolsHandler(cache=self.cache_handler) self.tools_handler = ToolsHandler(cache=self.cache_handler)
self.__create_agent_executor() self.__create_agent_executor()

71
crewai/prompts.py
View File

@@ -12,76 +12,73 @@ class Prompts(BaseModel):
TASK_SLICE: ClassVar[str] = dedent( TASK_SLICE: ClassVar[str] = dedent(
"""\ """\
Begin! This is VERY important to you, your job depends on it! Begin! This is VERY important to you, your job depends on it!
Current Task: {input} Current Task: {input}"""
{agent_scratchpad}
"""
) )
SCRATCHPAD_SLICE: ClassVar[str] = "\n{agent_scratchpad}"
MEMORY_SLICE: ClassVar[str] = dedent( MEMORY_SLICE: ClassVar[str] = dedent(
"""\ """\
This is the summary of your work so far: This is the summary of your work so far:
{chat_history} {chat_history}"""
"""
) )
ROLE_PLAYING_SLICE: ClassVar[str] = dedent( ROLE_PLAYING_SLICE: ClassVar[str] = dedent(
"""\ """\
You are {role}. You are {role}.
{backstory} {backstory}
Your personal goal is: {goal} Your personal goal is: {goal}"""
"""
) )
TOOLS_SLICE: ClassVar[str] = dedent( TOOLS_SLICE: ClassVar[str] = dedent(
"""\ """\
TOOLS:
------
You have access to the following tools:
{tools} TOOLS:
------
You have access to the following tools:
To use a tool, please use the exact following format: {tools}
``` To use a tool, please use the exact following format:
Thought: Do I need to use a tool? Yes
Action: the action to take, should be one of [{tool_names}], just the name.
Action Input: the input to the action
Observation: the result of the action
```
When you have a response for your task, or if you do not need to use a tool, you MUST use the format: ```
Thought: Do I need to use a tool? Yes
Action: the action to take, should be one of [{tool_names}], just the name.
Action Input: the input to the action
Observation: the result of the action
```
``` When you have a response for your task, or if you do not need to use a tool, you MUST use the format:
Thought: Do I need to use a tool? No
Final Answer: [your response here] ```
``` Thought: Do I need to use a tool? No
""" Final Answer: [your response here]
```"""
) )
VOTING_SLICE: ClassVar[str] = dedent( VOTING_SLICE: ClassVar[str] = dedent(
"""\ """\
You are working on a crew with your co-workers and need to decide who will execute the task. You are working on a crew with your co-workers and need to decide who will execute the task.
These are your format instructions: These are your format instructions:
{format_instructions} {format_instructions}
These are your co-workers and their roles: These are your co-workers and their roles:
{coworkers} {coworkers}"""
"""
) )
TASK_EXECUTION_WITH_MEMORY_PROMPT: ClassVar[str] = PromptTemplate.from_template( 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( 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( CONSENSUNS_VOTING_PROMPT: ClassVar[str] = PromptTemplate.from_template(
ROLE_PLAYING_SLICE + VOTING_SLICE + TASK_SLICE ROLE_PLAYING_SLICE + VOTING_SLICE + TASK_SLICE + SCRATCHPAD_SLICE
) )