Update app.py

app.py

@@ -7,7 +7,7 @@ app = Server()
 
 client = OpenAI(
     base_url="https://router.huggingface.co/v1",
-    api_key=os.environ.get("HF_TOKEN", ""),
+    api_key=os.environ["HF_TOKEN"],
     default_headers={
         "X-HF-Bill-To": "huggingface"
     }
@@ -15,18 +15,17 @@ client = OpenAI(
 
 @app.api()
 def chat(message: str, history_json: list) -> str:
-    # history_json should be a list of dictionaries like [{"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]
     messages = history_json + [{"role": "user", "content": message}]
     
-    completion = client.chat.completions.create(
+    stream = client.chat.completions.create(
         model="deepseek-ai/DeepSeek-V4-Pro:together",
         messages=messages,
-        stream=True
+        stream=True,
     )
     
     full_response = ""
-    for chunk in completion:
-        if chunk.choices and chunk.choices[0].delta.content:
+    for chunk in stream:
+        if chunk.choices[0].delta.content is not None:
             full_response += chunk.choices[0].delta.content
             yield full_response