From 16f879f359d6baaaa00d7de463c57b266ca10216 Mon Sep 17 00:00:00 2001
From: Joseph Blazick <jblazick@amazon.com>
Date: Mon, 4 Mar 2024 17:05:50 +0000
Subject: [PATCH] Prepping for Claude3

---
 backend/agent/agent_prompts.py | 3 ++-
 backend/agent/coding_agent.py  | 3 ++-
 backend/app_setup.py           | 2 +-
 backend/requirements.txt       | 3 ---
 4 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/backend/agent/agent_prompts.py b/backend/agent/agent_prompts.py
index 8c18490..8820bad 100644
--- 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
index e749dda..8b9e963 100644
--- 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
index 08f8eb0..19e2b14 100644
--- 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
index 3970345..1d8487c 100644
--- 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
\ No newline at end of file