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

Merge pull request #10 from send18/dev

Browse source 
Dev
This commit is contained in:
Guess 2023-09-08 15:26:20 +08:00 • committed by GitHub
commit 86e96e6eec
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/memory/brain_memory.py
View file

@ -156,7 +156,9 @@ class BrainMemory(pydantic.BaseModel):
part_max_words = min(int(max_words / len(text_windows)) + 1, 100)
summaries = []
for ws in text_windows:
response = await self._get_summary(text=ws, max_words=part_max_words, keep_language=keep_language)
response = await self._get_summary(
text=ws, llm=llm, max_words=part_max_words, keep_language=keep_language
)
summaries.append(response)
if len(summaries) == 1:
summary = summaries[0]

9
metagpt/provider/openai_api.py
View file

@ -263,15 +263,6 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
raise openai.error.OpenAIError("Exceeds the maximum retries")
async def get_summary(self, text: str, max_words=200, keep_language: bool = False, **kwargs) -> str:
"""
Return string in the following format：
[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Knock knock."},
{"role": "assistant", "content": "Who's there?"},
{"role": "user", "content": "Orange."},
]
"""
from metagpt.memory.brain_memory import BrainMemory
memory = BrainMemory(llm_type=LLMType.OPENAI.value, historical_summary=text, cacheable=False)