From d2a3f83bd65cf41e06fc6e5a7fd29c911a15b953 Mon Sep 17 00:00:00 2001
From: tcmofashi <mofashiforzbx@qq.com>
Date: Fri, 23 May 2025 13:48:23 +0800
Subject: [PATCH] fix: fit gemini

---
 src/chat/message_receive/chat_stream.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/chat/message_receive/chat_stream.py b/src/chat/message_receive/chat_stream.py
index ef7998752..1f2ebbf89 100644
--- a/src/chat/message_receive/chat_stream.py
+++ b/src/chat/message_receive/chat_stream.py
@@ -261,6 +261,8 @@ class ChatManager:
     def get_stream(self, stream_id: str) -> Optional[ChatStream]:
         """通过stream_id获取聊天流"""
         stream = self.streams.get(stream_id)
+        if not stream:
+            return None
         if stream_id in self.last_messages:
             stream.set_context(self.last_messages[stream_id])
         return stream