Prepping for Claude3

blazickjp · Mar 5, 2024 · 16f879f · 16f879f
1 parent c37042b
commit 16f879f
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/backend/agent/agent_prompts.py b/backend/agent/agent_prompts.py
@@ -49,7 +49,7 @@
 Assistant for Providing Directly Usable Code Snippets
 
 ### **Instructions**
-Generate clear, precise, and adaptable code for direct use in the user's programming tasks.
+Generate clear, precise, and adaptable code for direct use in the user's programming tasks. Use the code provided by the user as a reference to ensure compatibility and consistency.
 
 ### **Workflow**
 1. Understand the specific programming task.
@@ -73,6 +73,7 @@
 - **Efficiency**: Focus on practical, ready-to-use code generation.
 - **User-Centric**: Tailor the response to the user's skill level and needs.
 - **Expressiveness**: Use emojis to maintain a friendly and engaging interaction.
+- **Consistency**: Ensure your responses align with the user's codebase and style!
 
 Ready to assist with your coding tasks! 🚀
 What shall we start with? 

diff --git a/backend/agent/coding_agent.py b/backend/agent/coding_agent.py
@@ -82,9 +82,10 @@ def __init__(
             functions (Optional[List[dict]]): A list of functions that the agent can call.
             callables (Optional[List[Callable]]): A list of callable functions.
         """
+        # TODO Fix model saying it's checking systems when its not
         self.memory_manager = memory_manager
         self.function_map = function_map
-        self.GPT_MODEL = None
+        self.GPT_MODEL = 'gpt-4-0125-preview'
         self.codebase = codebase
         self.max_tokens = 4000
         self.temperature = 0.75

diff --git a/backend/app_setup.py b/backend/app_setup.py
@@ -129,7 +129,7 @@ def setup_codebase() -> MyCodebase:
 def setup_app() -> CodingAgent:
     print("Setting up app")
     codebase = setup_codebase()
-    memory = setup_memory_manager(tree=codebase.tree(), identity=LSA)
+    memory = setup_memory_manager(tree=codebase.tree(), identity=DEFAULT_SYSTEM_PROMPT)
     agent = CodingAgent(
         memory_manager=memory, function_map=[_OP_LIST], codebase=codebase
     )

diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -1,18 +1,15 @@
 fastapi==0.100.0
 numpy==1.23.5
 openai==1.2.2
-psycopg2==2.9.6
 pydantic==2.0.3
 python-dotenv==1.0.0
 PyYAML==6.0
 tiktoken==0.4.0
 urllib3==2.0.3
 uvicorn==0.23.0
-watchdog==3.0.0
 instructor==0.4.2
 scikit-learn==1.2.2
 pytest==7.4.0
 tenacity==8.2.2
-pexpect==4.8.0
 boto3==1.28.58
 astor==0.8.1