[Improvement] Fix Reasoning with Ollama

Author: amidabuddha

console_gpt/unichat_handler.py

@@ -36,6 +36,11 @@ def handle_streaming_completion(model_name, response_stream, conversation):
                 reasoning_content += delta.reasoning_content
                 rmd = Markdown(reasoning_content, code_theme="dracula")
                 live.update(rmd)
+            
+            if hasattr(delta, "reasoning") and delta.reasoning:
+                reasoning_content += delta.reasoning
+                rmd = Markdown(reasoning_content, code_theme="dracula")
+                live.update(rmd)
 
             if hasattr(delta, "content") and delta.content:
                 current_content += delta.content
@@ -130,6 +135,10 @@ def handle_non_streaming_completion(model_name, response, conversation):
     if reasoning_content:
         assistance_reply(reasoning_content, f"{model_name} Reasoning")
 
+    reasoning = getattr(message, "reasoning", None)
+    if reasoning:
+        assistance_reply(reasoning, f"{model_name} Reasoning")
+
     # Handle content
     content = getattr(message, "content", None)
     if not content and not has_previous_tool_calls: