Update agent.py

agent.py

@@ -259,6 +259,9 @@ def analyze_excel_file(file_path: str, query: str) -> str:
     except Exception as e:
         return f"Error analyzing Excel file: {str(e)}"
 
+# --------------------------------------------------------------------------- #
+# Custom LiteLLM model with rate limiting and error handling
+# --------------------------------------------------------------------------- #
 # --------------------------------------------------------------------------- #
 # Custom LiteLLM model with rate limiting and error handling
 # --------------------------------------------------------------------------- #
@@ -303,17 +306,23 @@ class RateLimitedClaudeModel:
             temperature=temperature
         )
     
-    def __call__(self, prompt: str, system_instruction: str, **kwargs) -> str:
+    def __call__(self, prompt: str, **kwargs) -> str:
         """
         Call the model with rate limiting and error handling
         
         Args:
             prompt: The prompt to generate from
-            system_instruction: The system instruction to use
             
         Returns:
             The generated text
         """
+        # Make sure system_instruction is always present
+        if "system_instruction" not in kwargs:
+            system_instruction = """You are a concise, highly accurate assistant specialized in solving challenges.
+Your answers should be precise, direct, and exactly match the expected format.
+All answers are graded by exact string match, so format carefully!"""
+            kwargs["system_instruction"] = system_instruction
+        
         retries = 0
         while True:
             try:
@@ -321,7 +330,7 @@ class RateLimitedClaudeModel:
                 RATE_LIMITER.wait()
                 
                 # Call the model
-                return self.model(prompt, system_instruction=system_instruction, **kwargs)
+                return self.model(prompt, **kwargs)
             
             except Exception as e:
                 # Check if it's a rate limit error