updated claude server model

app.py

@@ -19,12 +19,34 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- 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.
+    """
+
     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.
+        """
         headers = {
             "x-api-key": self.api_key,
             "anthropic-version": "2023-06-01",
@@ -34,13 +56,26 @@ class ClaudeServerModel:
             "model": self.model_id,
             "max_tokens": 1024,
             "temperature": self.temperature,
-            "messages": [
-                {"role": "user", "content": prompt}
-            ]
+            "messages": [{"role": "user", "content": prompt}]
         }
         response = requests.post("https://api.anthropic.com/v1/messages", headers=headers, json=body)
         response.raise_for_status()
-        return response.json()["content"][0]["text"].strip()
+        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.
+
+        Returns:
+            str: Claude-generated response.
+        """
+        return self.complete(prompt)
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"