From ffa3253aef1be3a30e57e59bf28a1e7f037bb6e4 Mon Sep 17 00:00:00 2001 From: cobycloud <25079070+cobycloud@users.noreply.github.com> Date: Fri, 28 Jun 2024 12:34:38 -0500 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9B=200.3.5.dev19=20-=20F821?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- swarmauri/__init__.py | 2 +- swarmauri/standard/agents/concrete/RagAgent.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) 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):