fixbug: last_talk

This commit is contained in:
莘权 马 2023-09-05 00:23:36 +08:00
parent bc52a674e7
commit dfc189510e

View file

@ -233,9 +233,10 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
padding_size = 20 if max_token_count > 20 else 0
text_windows = self.split_texts(text, window_size=max_token_count - padding_size)
part_max_words = min(int(max_words / len(text_windows)) + 1, 200)
summaries = []
for ws in text_windows:
response = await self._get_summary(text=ws, max_words=200, keep_language=keep_language)
response = await self._get_summary(text=ws, max_words=part_max_words, keep_language=keep_language)
summaries.append(response)
if len(summaries) == 1:
return summaries[0]