Update core/prompt_generation_agent.py

core/prompt_generation_agent.py

@@ -11,47 +11,47 @@ class PromptGenerationAgent:
         # Template patterns for different techniques
         self.technique_templates = {
             "zero_shot": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nOutput:",
-                "instruction": "Please complete the following task:"
+                "template": "{user_prompt}",
+                "instruction": ""
             },
             "few_shot": {
-                "template": "{instruction}\n\n{examples}\n\nInput: {user_prompt}\n\nOutput:",
-                "instruction": "Here are some examples of the desired format:",
-                "examples": "Example 1: [Input] -> [Output]\nExample 2: [Input] -> [Output]"
+                "template": "Study the following examples and apply the same pattern.\n\n{examples}\n\nNow complete this:\n{user_prompt}",
+                "instruction": ""
             },
             "chain_of_thought": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nLet's think step by step:\n1.",
-                "instruction": "Solve this problem by breaking it down into clear, logical steps."
+                "template": "You're a helpful assistant. Answer the following by reasoning step by step:\n\nQ: {user_prompt}\n\nA:\nLet's think it through:\n1.",
+                "instruction": ""
             },
             "react": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nThought: I need to think about this step by step.\nAction: [What action to take]\nObservation: [What I observe]\nThought: [Next thought based on observation]\n...",
-                "instruction": "Use the ReAct framework (Reasoning and Acting) to solve this problem. Alternate between Thought, Action, and Observation."
+                "template": "You're a reasoning agent who can both think and act using tools. Follow the ReAct format.\n\nTask: {user_prompt}\n\nThought: Let's analyze the problem.\nAction: [what action to take]\nObservation: [result of action]\nThought: [next reasoning step]",
+                "instruction": ""
             },
             "tree_of_thoughts": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nLet me explore multiple approaches:\n\nApproach 1:\n- [First approach]\n\nApproach 2:\n- [Second approach]\n\nApproach 3:\n- [Third approach]\n\nEvaluation: [Compare approaches and select the best one]",
-                "instruction": "Explore multiple solution paths and evaluate them to find the best approach."
+                "template": "Let's explore several possible strategies to solve this task:\n\nProblem: {user_prompt}\n\nApproach 1:\n- ...\n\nApproach 2:\n- ...\n\nApproach 3:\n- ...\n\nNow compare the approaches and choose the best one.",
+                "instruction": ""
             },
             "self_consistency": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nSolution 1:\n[First reasoning path]\n\nSolution 2:\n[Second reasoning path]\n\nSolution 3:\n[Third reasoning path]\n\nFinal Answer: [Most consistent answer across solutions]",
-                "instruction": "Generate multiple reasoning paths and select the most consistent answer."
+                "template": "Answer the following using three independent reasoning paths. Then choose the most consistent answer.\n\nPrompt: {user_prompt}\n\nPath 1:\n...\n\nPath 2:\n...\n\nPath 3:\n...\n\nFinal Answer:",
+                "instruction": ""
             },
             "generated_knowledge": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nFirst, let me generate relevant knowledge:\nKnowledge: [Generate relevant facts and information]\n\nNow, using this knowledge:\nAnswer: [Answer based on generated knowledge]",
-                "instruction": "First generate relevant knowledge, then use it to answer the question."
+                "template": "Before answering, first generate helpful background knowledge. Then use it to respond.\n\nPrompt: {user_prompt}\n\nRelevant Knowledge:\n...\n\nAnswer:",
+                "instruction": ""
             },
             "prompt_chaining": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nStep 1: [Break down the task]\nSubtask 1: [First subtask]\nResult 1: [Result of first subtask]\n\nStep 2: [Next step]\nSubtask 2: [Second subtask]\nResult 2: [Result of second subtask]\n\nFinal Integration: [Combine results]",
-                "instruction": "Break this complex task into smaller subtasks and solve them sequentially."
+                "template": "Break the task below into logical steps and solve each one. Then combine the results.\n\nPrompt: {user_prompt}\n\nStep 1:\nSubtask: ...\nResult: ...\n\nStep 2:\nSubtask: ...\nResult: ...\n\nFinal Answer:",
+                "instruction": ""
             },
             "meta_prompting": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nStructural Analysis:\n- Pattern: [Identify the pattern]\n- Format: [Identify the format]\n- Requirements: [Identify requirements]\n\nSolution: [Apply the identified structure]",
-                "instruction": "Focus on the structural and syntactical aspects of this task."
+                "template": "Analyze the prompt structure and respond in a format that matches its intent.\n\nPrompt: {user_prompt}\n\nStructure Analysis:\n- Pattern:\n- Format:\n- Constraints:\n\nResponse:",
+                "instruction": ""
             },
             "pal": {
-                "template": "{instruction}\n\nInput: {user_prompt}\n\nLet me solve this step by step with code:\n\n```python\n# Step 1: [Describe what this code does]\n[code]\n\n# Step 2: [Describe next step]\n[code]\n\n# Final result\nprint(result)\n```\n\nAnswer: [Interpret the code result]",
-                "instruction": "Solve this problem by writing and executing code."
+                "template": "Write Python code to solve the following problem. Explain the logic as comments.\n\nTask: {user_prompt}\n\n```python\n# Step 1:\n# ...\n\n# Step 2:\n# ...\n\n# Final step:\nprint(...)\n```\n\nAnswer:",
+                "instruction": ""
             }
         }
+   
     
     def generate_prompt(self, user_prompt: str, technique_key: str, 
                        analysis_result: Dict[str, Any]) -> Dict[str, Any]: