fix: multi llm test bug

Author: CaralHsi

src/memos/llms/deepseek.py

@@ -1,3 +1,5 @@
+from collections.abc import Generator
+
 from memos.configs.llm import DeepSeekLLMConfig
 from memos.llms.openai import OpenAILLM
 from memos.llms.utils import remove_thinking_tags
@@ -43,7 +45,6 @@ def generate_stream(self, messages: MessageList, **kwargs) -> Generator[str, Non
             extra_body=self.config.extra_body,
         )
         # Streaming chunks of text
-        answer_parts = ""
         for chunk in response:
             delta = chunk.choices[0].delta
             if hasattr(delta, "reasoning_content") and delta.reasoning_content:

tests/llms/test_deepseek.py

@@ -38,7 +38,7 @@ def test_deepseek_llm_generate_with_and_without_think_prefix(self):
         self.assertEqual(output_with_think, full_content)
 
         # Config with think tag removed
-        config_without_think = config_with_think.copy(update={"remove_think_prefix": True})
+        config_without_think = config_with_think.model_copy(update={"remove_think_prefix": True})
         llm_without_think = DeepSeekLLM(config_without_think)
         llm_without_think.client.chat.completions.create = MagicMock(return_value=mock_response)
 

tests/llms/test_qwen.py

@@ -40,7 +40,7 @@ def test_qwen_llm_generate_with_and_without_think_prefix(self):
         self.assertEqual(response_with_think, full_content)
 
         # Create config with remove_think_prefix = True
-        config_without_think = config_with_think.copy(update={"remove_think_prefix": True})
+        config_without_think = config_with_think.model_copy(update={"remove_think_prefix": True})
 
         # Instance with think tag removed
         llm_without_think = QwenLLM(config_without_think)