tried to get stepan changes working

hookla · Oct 23, 2023 · ad02e19 · ad02e19
1 parent f726c26
commit ad02e19
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/agents/agent.py b/agents/agent.py
@@ -24,8 +24,12 @@ def __init__(
         self.name = name
 
         self.client = client
+        self.system_prompt = system_prompt
+        self.user_prompt = user_prompt
         self.client.common_instructions = system_prompt
         self.client.user_prompt = user_prompt
 
     def query_gpt(self, transcript: str) -> str:
+        self.client.system_instructions = self.system_prompt
+        self.client.user_prompt = self.user_prompt
         return self.client.query(transcript)
diff --git a/agents/chairman.py b/agents/chairman.py
@@ -48,7 +48,7 @@ def decide_next_speaker(self, transcript_list: list[str]) -> SME:
             if next_executive is not None:
                 next_executive.spoken_count += 1  # Update the frequency count
                 self.user_prompt = self.update_user_prompt(self.executives)
-                self.gpt_client.user_prompt = self.user_prompt
+                self.client.user_prompt = self.user_prompt
                 return next_executive
 
             logger.info(f"{next_speaker} is not a valid exec...")