Update app.py

app.py

@@ -10,7 +10,10 @@ from txagent import TxAgent
 from tooluniverse import ToolUniverse
 
 # Patch PyTorch to allow loading old numpy pickles
-torch.serialization.add_safe_globals([numpy.core.multiarray._reconstruct])
+torch.serialization.add_safe_globals([
+    numpy.core.multiarray._reconstruct,
+    numpy.ndarray
+])
 
 logging.basicConfig(
     level=logging.INFO,
@@ -128,7 +131,12 @@ def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_ag
         )
         collected = ""
         for chunk in response_generator:
-            collected += chunk.get("content", "") if isinstance(chunk, dict) else str(chunk)
+            if chunk is None:
+                continue
+            elif isinstance(chunk, dict):
+                collected += chunk.get("content", "")
+            else:
+                collected += str(chunk)
         chat_history.append({"role": "assistant", "content": collected})
     except Exception as e:
         chat_history.append({"role": "assistant", "content": f"Error: {e}"})
@@ -159,4 +167,4 @@ def main():
     demo.launch(share=False)
 
 if __name__ == "__main__":
-    main()
+    main()