none

chain-ml · Aug 21, 2023 · 2da2ce8 · 2da2ce8
1 parent 4bef845
commit 2da2ce8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/python_agent/agent.py b/src/python_agent/agent.py
@@ -26,8 +26,6 @@
 )
 from python_agent.controller import LLMInstructController
 from python_agent.evaluator import BasicEvaluatorWithSource
-from python_agent.llm_fallback import LLMFallback
-
 
 class AgentApp:
     def __init__(self, work_dir="./python_agent"):
@@ -174,7 +172,8 @@ def init_controller(self):
             llm=self.llm,
             top_k_execution_plan=1,
             hints=[
-                "When you use the 'direct_to_user' chain, don't respond with instructions, but instead respond with a message that directly addresses the user."
+                "When you use the 'direct_to_user' chain, don't respond with instructions, but instead respond with a message that directly addresses the user.",
+                "Whenever graphical changes are being considered, always make sure you give instructions to draw graphics 'manually' in pygame. We can't load any external files and chains need to be reminded of this."
             ],
         )
 

diff --git a/src/python_agent/prompts/python_code_generation.toml b/src/python_agent/prompts/python_code_generation.toml
@@ -13,7 +13,7 @@ If present, always make sure the SOLUTION begins with the following code snippet
 $code_header
 
 ## EXISTING CODE
-If present and relevant, base your SOLUTION on the following code:
+Use the following code as the basis of your SOLUTION. Ensure that all of this code is part of your SOLUTION unless the user is specifcally asking for parts to be removed.
 ```python
 $existing_code
 ```

diff --git a/src/python_agent/prompts/python_error_correction.toml b/src/python_agent/prompts/python_error_correction.toml
@@ -9,6 +9,7 @@ prompt_template = """
 Your role is to review ERRORS and correct EXISTING PYTHON CODE according to your TASK.
 
 ## EXISTING PYTHON CODE
+Use the following code as the basis of your SOLUTION. Ensure that all of this code is part of your SOLUTION unless the user is specifcally asking for parts to be removed.
 $existing_code
 
 ## ERRORS