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

Merge pull request #27 from send18/feature-zhanglei

Browse source 
update:修改get_summary，加上是否保持语言的配置
This commit is contained in:
seeker-jie 2023-09-04 10:57:53 +08:00 • committed by GitHub
commit ada6cc7124
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
metagpt/provider/openai_api.py
View file

@ -221,18 +221,18 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
return CONFIG.max_tokens_rsp
return get_max_completion_tokens(messages, self.model, CONFIG.max_tokens_rsp)
async def get_summary(self, text: str, max_words=200):
async def get_summary(self, text: str, max_words=200, keep_language: bool = False):
max_token_count = DEFAULT_MAX_TOKENS
max_count = 100
while max_count > 0:
if len(text) < max_token_count:
return await self._get_summary(text, max_words=max_words)
return await self._get_summary(text=text, max_words=max_words,keep_language=keep_language)
padding_size = 20 if max_token_count > 20 else 0
text_windows = self.split_texts(text, window_size=max_token_count - padding_size)
summaries = []
for ws in text_windows:
response = await self._get_summary(ws, max_words=max_words)
response = await self._get_summary(text=ws, max_words=max_words,keep_language=keep_language)
summaries.append(response)
if len(summaries) == 1:
return summaries[0]
@ -243,11 +243,14 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
max_count -= 1 # safeguard
raise openai.error.InvalidRequestError("text too long")
async def _get_summary(self, text: str, max_words=20):
async def _get_summary(self, text: str, max_words=20, keep_language: bool = False):
"""Generate text summary"""
if len(text) < max_words:
return text
command = f"Translate the above content into a summary of less than {max_words} words."
if keep_language:
command = f".Translate the above content into a summary of less than {max_words} words in language of the content."
else:
command = f"Translate the above content into a summary of less than {max_words} words."
msg = text + "\n\n" + command
logger.info(f"summary ask:{msg}")
response = await self.aask(msg=msg, system_msgs=[])