standard - explicit conversation type

swarmauri · Jan 6, 2025 · d6b90ed · d6b90ed
1 parent 4eb29f9
commit d6b90ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkgs/standards/swarmauri_standard/swarmauri_standard/agents/RagAgent.py b/pkgs/standards/swarmauri_standard/swarmauri_standard/agents/RagAgent.py
@@ -3,6 +3,7 @@
 from swarmauri_standard.messages.HumanMessage import HumanMessage
 from swarmauri_standard.messages.SystemMessage import SystemMessage
 from swarmauri_standard.messages.AgentMessage import AgentMessage
+from swarmauri_standard.conversations.MaxSystemContextConversation import MaxSystemContextConversation
 
 from swarmauri_base.llms.LLMBase import LLMBase
 from swarmauri_base.agents.AgentBase import AgentBase
@@ -26,7 +27,7 @@ class RagAgent(AgentRetrieveMixin,
     specialized in retrieving documents based on input queries and generating responses.
     """
     llm: SubclassUnion[LLMBase]
-    conversation: SubclassUnion[ConversationBase]
+    conversation: MaxSystemContextConversation = MaxSystemContextConversation(system_context="")
     vector_store: SubclassUnion[VectorStoreBase]
     system_context:  Union[SystemMessage, str]
     type: Literal['RagAgent'] = 'RagAgent'