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

Log newline character after receiving llm stream response

Browse source
This commit is contained in:
shenchucheng 2024-01-05 16:50:03 +08:00
parent fba730390c
commit 9ce0182fab
4 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
metagpt/provider/google_gemini_api.py
View file

@ -120,6 +120,7 @@ class GeminiLLM(BaseLLM):
content = chunk.text
log_llm_stream(content)
collected_content.append(content)
log_llm_stream("\n")
full_content = "".join(collected_content)
usage = await self.aget_usage(messages, full_content)

1
metagpt/provider/ollama_api.py
View file

@ -119,6 +119,7 @@ class OllamaLLM(BaseLLM):
else:
# stream finished
usage = self.get_usage(chunk)
log_llm_stream("\n")
self._update_costs(usage)
full_content = "".join(collected_content)

1
metagpt/provider/openai_api.py
View file

@ -134,6 +134,7 @@ class OpenAILLM(BaseLLM):
async for i in resp:
log_llm_stream(i)
collected_messages.append(i)
log_llm_stream("\n")
full_reply_content = "".join(collected_messages)
usage = self._calc_usage(messages, full_reply_content)

1
metagpt/provider/zhipuai_api.py
View file

@ -118,6 +118,7 @@ class ZhiPuAILLM(BaseLLM):
usage = meta.get("usage")
else:
print(f"zhipuapi else event: {event.data}", end="")
log_llm_stream("\n")
self._update_costs(usage)
full_content = "".join(collected_content)