1) Improve Build-in agent "Tracy Wang" 2) fix bug

2023-09-18 23:25:44 -07:00
parent 0d8ee2a7af
commit 874aa98e21
8 changed files with 81 additions and 26 deletions
@@ -239,13 +239,12 @@ class AIAgent:
    
    async def _get_prompt_from_session(self,chatsession:AIChatSession,is_groupchat=False) -> AgentPrompt:
        # TODO: get prompt from group chat is different from single chat
-        messages = chatsession.read_history() # read last 10 message
+        messages = chatsession.read_history() # read 
        result_prompt = AgentPrompt()
        for msg in reversed(messages):
-            if msg.target == chatsession.owner_id:
-                result_prompt.messages.append({"role":"user","content":msg.body})
-            elif msg.sender == chatsession.owner_id:
+            if msg.sender == self.agent_id:
                result_prompt.messages.append({"role":"assistant","content":msg.body})
-        
+            else:
+                result_prompt.messages.append({"role":"user","content":msg.body})
        return result_prompt