Update app.py

app.py

@@ -203,22 +203,26 @@ def respond_with_enhanced_streaming(message, history):
         tool_call_args_str = ""
 
         for chunk in stream:
-            if isinstance(chunk, CompletionChunk):
+            # Check if chunk has choices and handle accordingly
+            if hasattr(chunk, 'choices') and chunk.choices and len(chunk.choices) > 0:
+                choice = chunk.choices[0]
+                
                 # Handle text chunks
-                if chunk.choices and chunk.choices[0].delta.content:
-                    text_chunk = chunk.choices[0].delta.content
+                if hasattr(choice, 'delta') and hasattr(choice.delta, 'content') and choice.delta.content:
+                    text_chunk = choice.delta.content
                     full_response += text_chunk
                     yield full_response
                 
                 # Handle tool call chunks
-                if chunk.choices and chunk.choices[0].delta.tool_calls:
-                    tool_call_delta = chunk.choices[0].delta.tool_calls[0]
+                if hasattr(choice, 'delta') and hasattr(choice.delta, 'tool_calls') and choice.delta.tool_calls:
+                    tool_call_delta = choice.delta.tool_calls[0]
                     
                     # Accumulate name and arguments from stream
-                    if tool_call_delta.function.name:
-                        tool_call_name = tool_call_delta.function.name
-                    if tool_call_delta.function.arguments:
-                        tool_call_args_str += tool_call_delta.function.arguments
+                    if hasattr(tool_call_delta, 'function'):
+                        if hasattr(tool_call_delta.function, 'name') and tool_call_delta.function.name:
+                            tool_call_name = tool_call_delta.function.name
+                        if hasattr(tool_call_delta.function, 'arguments') and tool_call_delta.function.arguments:
+                            tool_call_args_str += tool_call_delta.function.arguments
                     
                     # Check if we have received the full tool call
                     # This is a simple heuristic and might need refinement