swarmauri-standard 0.6.0.dev111__tar.gz → 0.6.0.dev113__tar.gz

{swarmauri_standard-0.6.0.dev111 → swarmauri_standard-0.6.0.dev113}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: swarmauri-standard
-Version: 0.6.0.dev111
+Version: 0.6.0.dev113
 Summary: This repository includes standard components within the Swarmauri framework.
 Home-page: http://github.com/swarmauri/swarmauri-sdk
 License: Apache-2.0
@@ -31,8 +31,8 @@ Requires-Dist: pydantic (>=2.9.2,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: scikit-learn (>=1.4.2,<2.0.0) ; extra == "ml-toolkits" or extra == "full"
 Requires-Dist: scipy (>=1.7.0,<1.14.0)
-Requires-Dist: swarmauri_base (>=0.6.0.dev111,<0.7.0)
-Requires-Dist: swarmauri_core (>=0.6.0.dev111,<0.7.0)
+Requires-Dist: swarmauri_base (>=0.6.0.dev113,<0.7.0)
+Requires-Dist: swarmauri_core (>=0.6.0.dev113,<0.7.0)
 Requires-Dist: toml (>=0.10.2,<0.11.0)
 Requires-Dist: typing_extensions
 Requires-Dist: yake (==0.4.8) ; extra == "nlp" or extra == "full"

{swarmauri_standard-0.6.0.dev111 → swarmauri_standard-0.6.0.dev113}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "swarmauri-standard"
-version = "0.6.0.dev111"
+version = "0.6.0.dev113"
 description = "This repository includes standard components within the Swarmauri framework."
 authors = ["Jacob Stewart <jacob@swarmauri.com>"]
 license = "Apache-2.0"
@@ -17,8 +17,8 @@ classifiers = [
 python = ">=3.10,<3.13"
 
 # Swarmauri
-swarmauri_core = { version = "^0.6.0.dev111" }
-swarmauri_base = { version = "^0.6.0.dev111" }
+swarmauri_core = { version = "^0.6.0.dev113" }
+swarmauri_base = { version = "^0.6.0.dev113" }
 
 # Dependencies
 toml = "^0.10.2"

{swarmauri_standard-0.6.0.dev111 → swarmauri_standard-0.6.0.dev113}/swarmauri_standard/agents/QAAgent.py

@@ -2,8 +2,9 @@ from typing import Any, Optional, Dict, Literal
 from swarmauri_base.agents.AgentBase import AgentBase
 from swarmauri_standard.conversations.MaxSystemContextConversation import MaxSystemContextConversation
 from swarmauri_standard.messages.HumanMessage import HumanMessage
+from swarmauri_core.ComponentBase import ComponentBase
 
-
+@ComponentBase.register_type(AgentBase, 'QAAgent')
 class QAAgent(AgentBase):
     conversation: MaxSystemContextConversation = MaxSystemContextConversation(max_size=2)
     type: Literal['QAAgent'] = 'QAAgent'

{swarmauri_standard-0.6.0.dev111 → swarmauri_standard-0.6.0.dev113}/swarmauri_standard/agents/RagAgent.py

@@ -11,11 +11,12 @@ from swarmauri_standard.messages.HumanMessage import HumanMessage
 from swarmauri_standard.messages.SystemMessage import SystemMessage
 from swarmauri_standard.messages.AgentMessage import AgentMessage
 
-from swarmauri_core.typing import SubclassUnion
 from swarmauri_base.llms.LLMBase import LLMBase
 from swarmauri_base.conversations.ConversationBase import ConversationBase
 from swarmauri_base.vector_stores.VectorStoreBase import VectorStoreBase
+from swarmauri_core.ComponentBase import SubclassUnion, ComponentBase
 
+@ComponentBase.register_type(AgentBase, 'RagAgent')
 class RagAgent(AgentRetrieveMixin, 
                AgentVectorStoreMixin, 
                AgentSystemContextMixin, 

{swarmauri_standard-0.6.0.dev111 → swarmauri_standard-0.6.0.dev113}/swarmauri_standard/agents/SimpleConversationAgent.py

@@ -1,15 +1,13 @@
 from typing import Any, Optional, Dict, Literal, List, Union
 
+from swarmauri_standard.messages.HumanMessage import contentItem
+from swarmauri_standard.messages.HumanMessage import HumanMessage
 from swarmauri_base.agents.AgentBase import AgentBase
 from swarmauri_base.agents.AgentConversationMixin import AgentConversationMixin
-from swarmauri_standard.messages.HumanMessage import HumanMessage
-
-from swarmauri_core.typing import SubclassUnion
 from swarmauri_base.conversations.ConversationBase import ConversationBase
+from swarmauri_core.ComponentBase import SubclassUnion, ComponentBase
 
-from swarmauri_standard.messages.HumanMessage import contentItem
-
-
+@ComponentBase.register_type(AgentBase, 'SimpleConversationAgent')
 class SimpleConversationAgent(AgentConversationMixin, AgentBase):
     conversation: SubclassUnion[ConversationBase]  #
     type: Literal["SimpleConversationAgent"] = "SimpleConversationAgent"

{swarmauri_standard-0.6.0.dev111 → swarmauri_standard-0.6.0.dev113}/swarmauri_standard/agents/ToolAgent.py

@@ -2,18 +2,19 @@ from pydantic import ConfigDict
 from typing import Any, Optional, Union, Dict, Literal
 import json
 import logging
-from swarmauri_core.messages import IMessage
 
 from swarmauri_base.llms.LLMBase import LLMBase
 from swarmauri_base.agents.AgentBase import AgentBase
 from swarmauri_base.agents.AgentConversationMixin import AgentConversationMixin
 from swarmauri_base.agents.AgentToolMixin import AgentToolMixin
-from swarmauri_standard.messages.HumanMessage import HumanMessage
-
-from swarmauri_core.typing import SubclassUnion
 from swarmauri_base.toolkits.ToolkitBase import ToolkitBase
 from swarmauri_base.conversations.ConversationBase import ConversationBase
+from swarmauri_standard.messages.HumanMessage import HumanMessage
+
+from swarmauri_core.messages import IMessage
+from swarmauri_core.ComponentBase import SubclassUnion, ComponentBase
 
+@ComponentBase.register_type(AgentBase, 'ToolAgent')
 class ToolAgent(AgentToolMixin, AgentConversationMixin, AgentBase):
     llm: SubclassUnion[LLMBase]
     toolkit: SubclassUnion[ToolkitBase]