Update space

app.py

@@ -27,17 +27,20 @@ def respond(
 
     response = ""
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-
-        response += token
-        yield response
+    try:
+        for chunk in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            # The API returns the text directly in streaming mode
+            if chunk:
+                response += chunk
+                yield response
+    except Exception as e:
+        yield f"Error: {str(e)}"
 
 
 """