ai.assistant.model package

Submodules

ai.assistant.model.model_provider module

class ai.assistant.model.model_provider.ModelProvider(name: str, base_url: str | None = None, api_key: str | None = None)

Bases: object

Configuration for an AI model provider.

name

Unique name of the model provider.

Type:

str

base_url

Base URL for the provider’s API endpoint.

Type:

str | None

api_key

API key for authenticating requests.

Type:

str | None

name: str

base_url: str | None = None

api_key: str | None = None

__init__(name: str, base_url: str | None = None, api_key: str | None = None) → None

class ai.assistant.model.model_provider.ModelProviders

Bases: dict[str, ModelProvider]

Registry of ModelProvider instances, keyed by provider name.

__init__()

Initialize an empty ModelProviders registry.

add_dict(provider_dict: dict) → None

Create and add a ModelProvider from a dictionary.

Parameters:

provider_dict (dict) – Dictionary with keys matching ModelProvider fields.

add(provider: ModelProvider) → None

Add a ModelProvider to the registry.

Parameters:

provider (ModelProvider) – The provider to register.

Raises:

ValueError – If a provider with the same name already exists.

ai.assistant.model.model_provider_test module

ai.assistant.model.model_provider_test.test_adds_provider_correctly()

ai.assistant.model.model_provider_test.test_adds_provider_from_dict_correctly()

ai.assistant.model.model_provider_test.test_handles_duplicate_provider_names()

ai.assistant.model.model_provider_test.test_handles_empty_provider_dict()

Module contents