update

Author: Joseph Blazick

backend/agent/coding_agent.py

@@ -296,8 +296,9 @@ def call_model_streaming(self, command: Optional[str] | None = None, **kwargs):
                     modelId="anthropic.claude-3-sonnet-20240229-v1:0",
                     body=json.dumps(
                         {
+
                             "messages": kwargs["messages"][1:],
-                            "system": kwargs["messages"][0]["content"],
+                            "system": self.generate_anthropic_prompt(sys_only=True),
                             "max_tokens": max(kwargs["max_tokens"], 2000),
                             "temperature": kwargs["temperature"],
                             "anthropic_version": "bedrock-2023-05-31",
@@ -344,7 +345,7 @@ def call_model_streaming(self, command: Optional[str] | None = None, **kwargs):
                     print("UnboundLocalError")
                     break
 
-    def generate_anthropic_prompt(self, include_messages=True) -> str:
+    def generate_anthropic_prompt(self, include_messages: Optional[bool]=True, sys_only: Optional[bool]=None) -> str:
         """
         Generates a prompt for the Gaive model.
 
@@ -406,6 +407,9 @@ def generate_anthropic_prompt(self, include_messages=True) -> str:
         else:
             sys_prompt = self.memory_manager.identity + "\n\n" + tree + file_context
 
+        if sys_only:
+            return sys_prompt
+        
         return (
             "\n\nHuman: The folllowing is your system prompt: "
             + sys_prompt

backend/app_setup.py

@@ -75,7 +75,7 @@ def flush(self):
 # from agent.agent_functions.changes import Changes
 
 IGNORE_DIRS = ["node_modules", ".next", ".venv", "__pycache__", ".git"]
-FILE_EXTENSIONS = [".js", ".py", ".md", "Dockerfile", '.txt']
+FILE_EXTENSIONS = [".js", ".py", ".md", "Dockerfile", '.txt', '.ts', '.yaml']
 
 def create_database_connection() -> sqlite3.Connection:
     try: