refactor: think

This commit is contained in:
莘权 马 2023-09-05 22:20:39 +08:00
parent f450b61bc2
commit cb17a17b4a

View file

@ -129,10 +129,9 @@ 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
last_talk = await self._llm.rewrite(sentence=last_talk, context=history_text)
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
return last_talk