diff --git a/swarmauri/__init__.py b/swarmauri/__init__.py index 86e3281fb..6a50ff5fa 100644 --- a/swarmauri/__init__.py +++ b/swarmauri/__init__.py @@ -1,4 +1,4 @@ -__version__ = "0.3.5.dev18" +__version__ = "0.3.5.dev19" __long_desc__ = """ # swarmaURI sdk diff --git a/swarmauri/standard/agents/concrete/RagAgent.py b/swarmauri/standard/agents/concrete/RagAgent.py index 1e03a0d18..f8587801b 100644 --- a/swarmauri/standard/agents/concrete/RagAgent.py +++ b/swarmauri/standard/agents/concrete/RagAgent.py @@ -12,6 +12,7 @@ AgentMessage) from swarmauri.core.typing import SubclassUnion +from swarmauri.standard.llms.base.LLMBase import LLMBase from swarmauri.standard.conversations.base.ConversationBase import ConversationBase from swarmauri.standard.vector_stores.base.VectorStoreBase import VectorStoreBase @@ -26,8 +27,8 @@ class RagAgent(AgentRetrieveMixin, """ llm: SubclassUnion[LLMBase] conversation: SubclassUnion[ConversationBase] - system_context: Union[SystemMessage, str] vector_store: SubclassUnion[VectorStoreBase] + system_context: Union[SystemMessage, str] type: Literal['RagAgent'] = 'RagAgent' def _create_preamble_context(self):