feat: + summary

metagpt/provider/metagpt_llm_api.py

@@ -5,10 +5,13 @@
 @File    : metagpt_llm_api.py
 @Desc    : MetaGPT LLM related APIs
 """
+import json
 
 import openai
+from pydantic import BaseModel
 
 from metagpt.config import CONFIG
+from metagpt.memory.brain_memory import BrainMemory
 from metagpt.provider import OpenAIGPTAPI
 from metagpt.provider.openai_api import RateLimiter
 
@@ -31,3 +34,21 @@ class MetaGPTLLMAPI(OpenAIGPTAPI):
             openai.api_type = CONFIG.METAGPT_API_TYPE
             openai.api_version = CONFIG.METAGPT_API_VERSION
         self.rpm = int(CONFIG.RPM) if CONFIG.RPM else 10
+
+    async def get_summary(self, memory: BrainMemory, max_words=200, keep_language: bool = False, **kwargs):
+        summary = []
+
+        class QuweryAnswerPair(BaseModel):
+            ask: str
+            answer: str
+
+        rh = reversed(memory.history)
+        ix = 0
+        while ix < len(rh):
+            t = rh[ix]
+            print(t)
+            # 如果 t是ask, continue
+            pass
+
+        data = json.dumps(summary)
+        return data

metagpt/roles/assistant.py

@@ -121,7 +121,9 @@ class Assistant(Role):
             return None
         if history_text == "":
             return last_talk
-        history_summary = await self._llm.get_summary(history_text, max_words=800, keep_language=True)
+        history_summary = await self._llm.get_summary(
+            history_text, max_words=800, keep_language=True, memory=self.memory
+        )
         await self.memory.set_history_summary(
             history_summary=history_summary, redis_key=CONFIG.REDIS_KEY, redis_conf=CONFIG.REDIS
         )