standard - utilize SubclassUnion[ConversationBase]

swarmauri · Jan 9, 2025 · ae1f749 · ae1f749
1 parent 5ab73f7
commit ae1f749
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkgs/standards/swarmauri_standard/swarmauri_standard/agents/QAAgent.py b/pkgs/standards/swarmauri_standard/swarmauri_standard/agents/QAAgent.py
@@ -5,11 +5,12 @@
 
 from swarmauri_base.llms.LLMBase import LLMBase
 from swarmauri_base.agents.AgentBase import AgentBase
+from swarmauri_base.conversations.ConversationBase import ConversationBase
 from swarmauri_core.ComponentBase import SubclassUnion, ComponentBase
 
 @ComponentBase.register_type(AgentBase, 'QAAgent')
 class QAAgent(AgentBase):
-    conversation: MaxSystemContextConversation = MaxSystemContextConversation(max_size=2)
+    conversation: SubclassUnion[ConversationBase] = MaxSystemContextConversation(max_size=2)
     type: Literal['QAAgent'] = 'QAAgent'
 
     def exec(self,