updated ClaudeServerModel with stop sequence

app.py

@@ -25,28 +25,11 @@ class ClaudeServerModel:
     """
 
     def __init__(self, api_key: str, model_id: str = "claude-3-opus-20240229", temperature: float = 0.0):
-        """
-        Initialize the ClaudeServerModel.
-
-        Args:
-            api_key (str): Your Claude API key.
-            model_id (str): Claude model ID (e.g., "claude-3-opus-20240229").
-            temperature (float): Sampling temperature for completion generation.
-        """
         self.api_key = api_key
         self.model_id = model_id
         self.temperature = temperature
 
-    def complete(self, prompt: str) -> str:
-        """
-        Sends a prompt to the Claude API and returns the text response.
-
-        Args:
-            prompt (str): Prompt to send to the Claude model.
-
-        Returns:
-            str: The Claude model's response text.
-        """
+    def complete(self, prompt: str, stop_sequences: list[str] = None) -> str:
         headers = {
             "x-api-key": self.api_key,
             "anthropic-version": "2023-06-01",
@@ -58,24 +41,15 @@ class ClaudeServerModel:
             "temperature": self.temperature,
             "messages": [{"role": "user", "content": prompt}]
         }
+        if stop_sequences:
+            body["stop_sequences"] = stop_sequences
+
         response = requests.post("https://api.anthropic.com/v1/messages", headers=headers, json=body)
         response.raise_for_status()
-        response_data = response.json()
-        if "content" in response_data and isinstance(response_data["content"], list):
-            return response_data["content"][0].get("text", "").strip()
-        return "Claude API did not return valid text content."
-
-    def __call__(self, prompt: str) -> str:
-        """
-        Enables model to be used as a callable by agent frameworks.
-
-        Args:
-            prompt (str): Prompt to send.
+        return response.json()["content"][0]["text"].strip()
 
-        Returns:
-            str: Claude-generated response.
-        """
-        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"