Update code

model_client.py

@@ -1,5 +1,5 @@
 import requests
-from typing import Optional, Tuple
+from typing import Tuple
 
 from config import settings
 
@@ -12,7 +12,7 @@ class ModelClient:
         self.temperature = settings.DEFAULT_TEMPERATURE
         self.top_p = settings.DEFAULT_TOP_P
 
-    def _build_payload(self, prompt: str, model_name: str) -> dict:
+    def _build_payload(self, prompt: str) -> dict:
         return {
             "inputs": prompt,
             "parameters": {
@@ -24,15 +24,13 @@ class ModelClient:
                 "wait_for_model": True,
                 "use_cache": False,
             },
-            "model": model_name,
         }
 
     def _extract_text(self, response_json) -> str:
         if isinstance(response_json, list) and len(response_json) > 0:
             first_item = response_json[0]
-            if isinstance(first_item, dict):
-                if "generated_text" in first_item:
-                    return str(first_item["generated_text"]).strip()
+            if isinstance(first_item, dict) and "generated_text" in first_item:
+                return str(first_item["generated_text"]).strip()
 
         if isinstance(response_json, dict):
             if "generated_text" in response_json:
@@ -50,7 +48,7 @@ class ModelClient:
         if hf_token:
             headers["Authorization"] = f"Bearer {hf_token}"
 
-        payload = self._build_payload(prompt, model_name)
+        payload = self._build_payload(prompt)
 
         response = requests.post(
             api_url,
@@ -58,7 +56,17 @@ class ModelClient:
             json=payload,
             timeout=self.timeout,
         )
-        response.raise_for_status()
+
+        try:
+            response.raise_for_status()
+        except requests.HTTPError:
+            try:
+                error_json = response.json()
+                if isinstance(error_json, dict) and "error" in error_json:
+                    raise RuntimeError(str(error_json["error"]).strip())
+            except ValueError:
+                pass
+            raise
 
         return self._extract_text(response.json())