refactor: brain memory

This commit is contained in:
莘权 马 2023-09-08 00:00:41 +08:00
parent 20fb71b0a3
commit 42d0281fbb

View file

@ -218,7 +218,7 @@ class BrainMemory(pydantic.BaseModel):
async def get_title(self, llm, max_words=5, **kwargs) -> str:
"""Generate text title"""
summary = await self.summarize(max_words=500)
summary = await self.summarize(llm=llm, max_words=500)
language = CONFIG.language or DEFAULT_LANGUAGE
command = f"Translate the above summary into a {language} title of less than {max_words} words."