Claude API fix

app.py

@@ -20,8 +20,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 class ClaudeServerModel:
     """
-    ClaudeServerModel acts as a wrapper to make Anthropic Claude models callable
-    in agentic frameworks like smolagents.
+    ClaudeServerModel wraps Anthropic Claude API for smolagents-style usage.
     """
 
     def __init__(self, api_key: str, model_id: str = "claude-3-opus-20240229", temperature: float = 0.0):
@@ -29,7 +28,7 @@ class ClaudeServerModel:
         self.model_id = model_id
         self.temperature = temperature
 
-    def complete(self, prompt: str) -> str:
+    def complete(self, prompt: str, stop_sequences: list[str] = None) -> str:
         headers = {
             "x-api-key": self.api_key,
             "anthropic-version": "2023-06-01",
@@ -43,12 +42,16 @@ class ClaudeServerModel:
             "prompt": f"\n\nHuman: {prompt}\n\nAssistant:"
         }
 
+        # Claude expects stop_sequences as "stop_sequences", if passed
+        if stop_sequences:
+            body["stop_sequences"] = stop_sequences
+
         response = requests.post("https://api.anthropic.com/v1/complete", headers=headers, json=body)
         response.raise_for_status()
         return response.json()["completion"].strip()
 
-    def __call__(self, prompt: str) -> str:
-        return self.complete(prompt)
+    def __call__(self, prompt: str, stop_sequences: list[str] = None) -> str:
+        return self.complete(prompt, stop_sequences=stop_sequences)
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"