Dhruv Pawar
commited on
Commit
·
9d5bde0
1
Parent(s):
5e0ae28
added simple mode
Browse files- src/config/constants.py +3 -1
- src/core/prompt_engine.py +9 -3
src/config/constants.py
CHANGED
|
@@ -15,6 +15,7 @@ class ReasoningMode(Enum):
|
|
| 15 |
REFLEXION = "Reflexion + Self-Correction"
|
| 16 |
DEBATE = "Multi-Agent Debate"
|
| 17 |
ANALOGICAL = "Analogical Reasoning"
|
|
|
|
| 18 |
|
| 19 |
def __str__(self) -> str:
|
| 20 |
return self.value
|
|
@@ -28,7 +29,8 @@ class ReasoningMode(Enum):
|
|
| 28 |
cls.SELF_CONSISTENCY: "Generates multiple independent solutions and identifies the most consistent answer.",
|
| 29 |
cls.REFLEXION: "Solves problems, critiques reasoning, and refines solutions iteratively.",
|
| 30 |
cls.DEBATE: "Presents multiple perspectives and synthesizes the strongest arguments.",
|
| 31 |
-
cls.ANALOGICAL: "Finds similar problems and applies their solutions to the current problem."
|
|
|
|
| 32 |
}
|
| 33 |
return descriptions.get(mode, "Advanced reasoning mode")
|
| 34 |
|
|
|
|
| 15 |
REFLEXION = "Reflexion + Self-Correction"
|
| 16 |
DEBATE = "Multi-Agent Debate"
|
| 17 |
ANALOGICAL = "Analogical Reasoning"
|
| 18 |
+
SIMPLE = "Simple (Direct Response)"
|
| 19 |
|
| 20 |
def __str__(self) -> str:
|
| 21 |
return self.value
|
|
|
|
| 29 |
cls.SELF_CONSISTENCY: "Generates multiple independent solutions and identifies the most consistent answer.",
|
| 30 |
cls.REFLEXION: "Solves problems, critiques reasoning, and refines solutions iteratively.",
|
| 31 |
cls.DEBATE: "Presents multiple perspectives and synthesizes the strongest arguments.",
|
| 32 |
+
cls.ANALOGICAL: "Finds similar problems and applies their solutions to the current problem.",
|
| 33 |
+
cls.SIMPLE: "Direct response without structured reasoning - fastest and most natural."
|
| 34 |
}
|
| 35 |
return descriptions.get(mode, "Advanced reasoning mode")
|
| 36 |
|
src/core/prompt_engine.py
CHANGED
|
@@ -78,7 +78,9 @@ Show multiple viewpoints and reasoned synthesis.""",
|
|
| 78 |
4. Adapt to current problem context
|
| 79 |
5. Verify applicability
|
| 80 |
|
| 81 |
-
Draw meaningful analogies and explain transfer logic."""
|
|
|
|
|
|
|
| 82 |
}
|
| 83 |
|
| 84 |
# Pre-built templates for common tasks
|
|
@@ -149,7 +151,11 @@ Include:
|
|
| 149 |
2. Core principles
|
| 150 |
3. Examples and analogies
|
| 151 |
4. Common misconceptions
|
| 152 |
-
5. Practical applications"""
|
|
|
|
|
|
|
|
|
|
|
|
|
| 153 |
}
|
| 154 |
|
| 155 |
@classmethod
|
|
@@ -157,7 +163,7 @@ Include:
|
|
| 157 |
"""
|
| 158 |
✅ GET SYSTEM PROMPT FOR REASONING MODE
|
| 159 |
"""
|
| 160 |
-
prompt = cls.SYSTEM_PROMPTS.get(mode, cls.SYSTEM_PROMPTS[ReasoningMode.
|
| 161 |
logger.debug(f"📝 Retrieved system prompt for mode: {mode}")
|
| 162 |
return prompt
|
| 163 |
|
|
|
|
| 78 |
4. Adapt to current problem context
|
| 79 |
5. Verify applicability
|
| 80 |
|
| 81 |
+
Draw meaningful analogies and explain transfer logic.""",
|
| 82 |
+
|
| 83 |
+
ReasoningMode.SIMPLE: """You are a helpful AI assistant. Provide clear, direct, and accurate responses to user queries."""
|
| 84 |
}
|
| 85 |
|
| 86 |
# Pre-built templates for common tasks
|
|
|
|
| 151 |
2. Core principles
|
| 152 |
3. Examples and analogies
|
| 153 |
4. Common misconceptions
|
| 154 |
+
5. Practical applications""",
|
| 155 |
+
|
| 156 |
+
"Simple": """Answer the following question directly:
|
| 157 |
+
|
| 158 |
+
{query}"""
|
| 159 |
}
|
| 160 |
|
| 161 |
@classmethod
|
|
|
|
| 163 |
"""
|
| 164 |
✅ GET SYSTEM PROMPT FOR REASONING MODE
|
| 165 |
"""
|
| 166 |
+
prompt = cls.SYSTEM_PROMPTS.get(mode, cls.SYSTEM_PROMPTS[ReasoningMode.SIMPLE])
|
| 167 |
logger.debug(f"📝 Retrieved system prompt for mode: {mode}")
|
| 168 |
return prompt
|
| 169 |
|