swarmauri_base 0.10.1.dev2

This repository includes base classes and mixins for the Swarmauri framework.

---

Swarmauri Base

swarmauri_base is where Swarmauri's dynamic component schema machinery lives. ComponentBase combines Pydantic JSON support, YAML/TOML serialization mixins, and DynamicBase registration so components can preserve concrete subclass identity across configuration files, APIs, factories, queues, and databases.

When a subclass is registered with ComponentBase.register_type(...), fields typed with SubclassUnion[...] can hydrate that concrete subclass from a serialized type discriminator. The same component can be dumped to JSON, YAML, or TOML and loaded again without becoming an untyped dictionary.

from typing import Literal
from pydantic import BaseModel
from swarmauri_base.ComponentBase import ComponentBase
from swarmauri_base.DynamicBase import SubclassUnion


@ComponentBase.register_model()
class StoreConfig(ComponentBase):
    type: Literal["StoreConfig"] = "StoreConfig"
    name: str


@ComponentBase.register_type(StoreConfig, "SqliteStoreConfig")
class SqliteStoreConfig(StoreConfig):
    type: Literal["SqliteStoreConfig"] = "SqliteStoreConfig"
    path: str


class StoreEnvelope(BaseModel):
    store: SubclassUnion[StoreConfig]


envelope = StoreEnvelope.model_validate_json(
    '{"store":{"type":"SqliteStoreConfig","name":"local","path":"./data.db"}}'
)
assert isinstance(envelope.store, SqliteStoreConfig)

Getting Started

Installing Swarmauri Base from pypi

To start developing with the Core Library, include it as a module in your Python project. Ensure you have Python 3.10 or later installed.

pip install swarmauri-base

Example Usage:

# Example of using the base class
from swarmauri_base.llms.LLMBase import LLMBase

class MyConcreteClass(LLMBase):
    
    pass

Features

agents

• AgentBase.py: Base class for agents.
• AgentConversationMixin.py: Mixin for conversation capabilities.
• AgentRetrieveMixin.py: Mixin for retrieval functionalities.
• AgentSystemContextMixin.py: Mixin for system context.
• AgentToolMixin.py: Mixin for tool integration.
• AgentVectorStoreMixin.py: Mixin for vector store functionalities.

chains

• ChainBase.py: Base class for chains.
• ChainContextBase.py: Base class for chain context.
• ChainStepBase.py: Base class for chain steps.
• PromptContextChainBase.py: Base class for prompt context chains.

chunkers

• ChunkerBase.py: Base class for chunkers.

conversations

• ConversationBase.py: Base class for conversations.
• ConversationSystemContextMixin.py: Mixin for system context in conversations.

dataconnectors

• DataConnectorBase.py: Base class for data connectors.

distances

• DistanceBase.py: Base class for distance calculations.
• VisionDistanceBase.py: Base class for vision distance calculations.

documents

• DocumentBase.py: Base class for documents.

document_stores

• DocumentStoreBase.py: Base class for document stores.
• DocumentStoreRetrieveBase.py: Base class for document retrieval.

embeddings

• EmbeddingBase.py: Base class for embeddings.
• VisionEmbeddingBase.py: Base class for vision embeddings.

factories

• FactoryBase.py: Base class for factories.

Factories can accept serialized component specs and hydrate concrete subclasses through SubclassUnion[...], avoiding brittle manual switches over type.

image_gens

• ImageGenBase.py: Base class for image generation.

llms

• LLMBase.py: Base class for large language models.

logging

• HandlerBase.py: Base class for logging handlers.
• LoggerBase.py: Base class for loggers.

measurements

• MeasurementAggregateMixin.py: Mixin for measurement aggregation.
• MeasurementBase.py: Base class for measurements.
• MeasurementCalculateMixin.py: Mixin for measurement calculations.
• MeasurementThresholdMixin.py: Mixin for measurement thresholds.

messages

• MessageBase.py: Base class for messages.

ocrs

• OCRBase.py: Base class for OCR functionalities.

parsers

• ParserBase.py: Base class for parsers.

pipelines

• PipelineBase.py: Base class for pipelines.

prompts

• PromptBase.py: Base class for prompts.
• PromptGeneratorBase.py: Base class for prompt generators.
• PromptMatrixBase.py: Base class for prompt matrices.

prompt_templates

• PromptTemplateBase.py: Base class for prompt templates.

schema_converters

• SchemaConverterBase.py: Base class for schema converters.

service_registries

• ServiceRegistryBase.py: Base class for service registries.

state

• StateBase.py: Base class for state management.

stt

• STTBase.py: Base class for speech-to-text functionalities.

swarms

• SwarmBase.py: Base class for swarm intelligence.

task_mgmt_strategies

• TaskMgmtStrategyBase.py: Base class for task management strategies.

toolkits

• ToolkitBase.py: Base class for toolkits.

tool_llms

• ToolLLMBase.py: Base class for tools for large language models.

tools

• ParameterBase.py: Base class for parameters.
• ToolBase.py: Base class for tools.

transports

• TransportBase.py: Base class for transports.

tts

• TTSBase.py: Base class for text-to-speech functionalities.

vectors

• VectorBase.py: Base class for vectors.

vector_stores

• VectorStoreBase.py: Base class for vector stores.
• VectorStoreCloudMixin.py: Mixin for cloud vector stores.
• VectorStorePersistentMixin.py: Mixin for persistent vector stores.
• VectorStoreRetrieveMixin.py: Mixin for vector store retrieval.
• VectorStoreSaveLoadMixin.py: Mixin for saving and loading vector stores.
• VisionVectorStoreBase.py: Base class for vision vector stores.

vlms

• VLMBase.py: Base class for visual language models.

Contributing

Contributions are welcome! If you'd like to add a new feature, fix a bug, or improve documentation, kindly go through the contributions guidelines first.