"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from .basesdk import BaseSDK
from .sdkconfiguration import SDKConfiguration
from mistralai.conversations import Conversations
from mistralai.libraries import Libraries
from mistralai.mistral_agents import MistralAgents
from typing import Optional


class Beta(BaseSDK):
    conversations: Conversations
    r"""(beta) Conversations API"""
    agents: MistralAgents
    r"""(beta) Agents API"""
    libraries: Libraries
    r"""(beta) Libraries API to create and manage libraries - index your documents to enhance agent capabilities."""

    def __init__(
        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
    ) -> None:
        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
        self.sdk_configuration = sdk_config
        self._init_sdks()

    def _init_sdks(self):
        self.conversations = Conversations(
            self.sdk_configuration, parent_ref=self.parent_ref
        )
        self.agents = MistralAgents(self.sdk_configuration, parent_ref=self.parent_ref)
        self.libraries = Libraries(self.sdk_configuration, parent_ref=self.parent_ref)
