refactor: think

This commit is contained in:
莘权 马 2023-09-05 22:04:25 +08:00
parent 508fff6920
commit a9b56a6f56

View file

@ -129,9 +129,10 @@ class Assistant(Role):
await self.memory.set_history_summary(
history_summary=history_summary, redis_key=CONFIG.REDIS_KEY, redis_conf=CONFIG.REDIS
)
if last_talk and await self._llm.is_related(last_talk, history_summary): # Merge relevant content.
last_talk = await self._llm.rewrite(sentence=last_talk, context=history_text)
return last_talk
# if last_talk and await self._llm.is_related(last_talk, history_summary): # Merge relevant content.
# last_talk = await self._llm.rewrite(sentence=last_talk, context=history_text)
# return last_talk
last_talk = await self._llm.rewrite(sentence=last_talk, context=history_text)
return last_talk