Update crewai/prompts.py

crewai/prompts.py

@@ -1,84 +1,57 @@
-"""Prompts for generic agent."""
-
-from textwrap import dedent
-from typing import ClassVar
+import json
+import os
+from typing import ClassVar, Dict, Optional
 
 from langchain.prompts import PromptTemplate
-from pydantic import BaseModel
+from pydantic import BaseModel, Field, PrivateAttr, ValidationError, model_validator
 
 
 class Prompts(BaseModel):
-    """Prompts for generic agent."""
-
-    TASK_SLICE: ClassVar[str] = dedent(
-        """\
-	Begin! This is VERY important to you, your job depends on it!
-
-	Current Task: {input}"""
-    )
+    """Manages and generates prompts for a generic agent with support for different languages."""
+
+    _prompts: Optional[Dict[str, str]] = PrivateAttr()
+    language: Optional[str] = Field(
+        default="en",
+        description="Language of the prompts.",
+    )
+
+    @model_validator(mode="after")
+    def load_prompts(self) -> "Prompts":
+        """Load prompts from a JSON file based on the specified language."""
+        try:
+            dir_path = os.path.dirname(os.path.realpath(__file__))
+            prompts_path = os.path.join(dir_path, f"prompts/{self.language}.json")
+
+            with open(prompts_path, "r") as f:
+                self._prompts = json.load(f)["slices"]
+        except FileNotFoundError:
+            raise ValidationError(
+                f"Prompt file for language '{self.language}' not found."
+            )
+        except json.JSONDecodeError:
+            raise ValidationError(f"Error decoding JSON from the prompts file.")
+        return self
 
     SCRATCHPAD_SLICE: ClassVar[str] = "\n{agent_scratchpad}"
 
-    MEMORY_SLICE: ClassVar[str] = dedent(
-        """\
-	This is the summary of your work so far:
-	{chat_history}"""
-    )
-
-    ROLE_PLAYING_SLICE: ClassVar[str] = dedent(
-        """\
-	You are {role}.
-	{backstory}
-
-	Your personal goal is: {goal}"""
-    )
-
-    TOOLS_SLICE: ClassVar[str] = dedent(
-        """\
-
-
-	TOOLS:
-	------
-	You have access to the following tools:
-
-	{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? No
-	Final Answer: [your response here]
-	```"""
-    )
-
-    VOTING_SLICE: ClassVar[str] = dedent(
-        """\
-	You are working on a crew with your co-workers and need to decide who will execute the task.
-
-	These are your format instructions:
-	{format_instructions}
-
-	These are your co-workers and their roles:
-	{coworkers}"""
-    )
-
-    TASK_EXECUTION_WITH_MEMORY_PROMPT: ClassVar[str] = PromptTemplate.from_template(
-        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 + SCRATCHPAD_SLICE
-    )
-
-    CONSENSUNS_VOTING_PROMPT: ClassVar[str] = PromptTemplate.from_template(
-        ROLE_PLAYING_SLICE + VOTING_SLICE + TASK_SLICE + SCRATCHPAD_SLICE
-    )
+    def task_execution_with_memory(self) -> str:
+        """Generate a prompt for task execution with memory components."""
+        return self._build_prompt(["role_playing", "tools", "memory", "task"])
+
+    def task_execution_without_tools(self) -> str:
+        """Generate a prompt for task execution without tools components."""
+        return self._build_prompt(["role_playing", "task"])
+
+    def task_execution(self) -> str:
+        """Generate a standard prompt for task execution."""
+        return self._build_prompt(["role_playing", "tools", "task"])
+
+    def _build_prompt(self, components: [str]) -> str:
+        """Constructs a prompt string from specified components."""
+        prompt_parts = [
+            self._prompts[component]
+            for component in components
+            if component in self._prompts
+        ]
+        prompt_parts.append(self.SCRATCHPAD_SLICE)
+        return PromptTemplate.from_template("".join(prompt_parts))