Skip to content

Commit 6b8583d

Browse files
CaralHsiCopilot
andauthored
Update src/memos/llms/deepseek.py
Co-authored-by: Copilot <[email protected]>
1 parent 95fcd48 commit 6b8583d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/memos/llms/deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def generate(self, messages: MessageList) -> str:
3131
else:
3232
return response_content
3333

34-
def generate_stream(self, messages: MessageList, **kwargs):
34+
def generate_stream(self, messages: MessageList, **kwargs) -> Generator[str, None, None]:
3535
"""Stream response from DeepSeek."""
3636
response = self.client.chat.completions.create(
3737
model=self.config.model_name_or_path,

0 commit comments

Comments
 (0)