apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-25 00:36:55 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update deepseek reasoner

Browse source
This commit is contained in:
better629 2025-02-27 16:12:27 +08:00
parent 78a9cd3a4c
commit 9245e8ab80
2 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
metagpt/provider/base_llm.py
View file

@ -215,7 +215,10 @@ class BaseLLM(ABC):
def get_choice_text(self, rsp: dict) -> str:
"""Required to provide the first text of choice"""
return rsp.get("choices")[0]["message"]["content"]
message = rsp.get("choices")[0]["message"]
if "reasoning_content" in message:
self.reasoning_content = message["reasoning_content"]
return message["content"]
def get_choice_delta_text(self, rsp: dict) -> str:
"""Required to provide the first text of stream choice"""

6
metagpt/provider/openai_api.py
View file

@ -91,8 +91,12 @@ class OpenAILLM(BaseLLM):
)
usage = None
collected_messages = []
collected_reasoning_messages = []
has_finished = False
async for chunk in response:
if hasattr(chunk.choices[0].delta, "reasoning_content"):
collected_reasoning_messages.append(chunk.choices[0].delta.reasoning_content) # for deepseek
continue
chunk_message = chunk.choices[0].delta.content or "" if chunk.choices else "" # extract the message
finish_reason = (
chunk.choices[0].finish_reason if chunk.choices and hasattr(chunk.choices[0], "finish_reason") else None
@ -118,6 +122,8 @@ class OpenAILLM(BaseLLM):
log_llm_stream("\n")
full_reply_content = "".join(collected_messages)
if collected_reasoning_messages:
self.reasoning_content = "".join(collected_reasoning_messages)
if not usage:
# Some services do not provide the usage attribute, such as OpenAI or OpenLLM
usage = self._calc_usage(messages, full_reply_content)