smallestai 2.2.0__py3-none-any.whl → 3.0.1__py3-none-any.whl

smallestai/atoms/exceptions.py

@@ -0,0 +1,216 @@
+# coding: utf-8
+
+"""
+    Agent Management API
+
+    API for managing agents, their templates, and call logs
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+from typing import Any, Optional
+from typing_extensions import Self
+
+class OpenApiException(Exception):
+    """The base exception class for all OpenAPIExceptions"""
+
+
+class ApiTypeError(OpenApiException, TypeError):
+    def __init__(self, msg, path_to_item=None, valid_classes=None,
+                 key_type=None) -> None:
+        """ Raises an exception for TypeErrors
+
+        Args:
+            msg (str): the exception message
+
+        Keyword Args:
+            path_to_item (list): a list of keys an indices to get to the
+                                 current_item
+                                 None if unset
+            valid_classes (tuple): the primitive classes that current item
+                                   should be an instance of
+                                   None if unset
+            key_type (bool): False if our value is a value in a dict
+                             True if it is a key in a dict
+                             False if our item is an item in a list
+                             None if unset
+        """
+        self.path_to_item = path_to_item
+        self.valid_classes = valid_classes
+        self.key_type = key_type
+        full_msg = msg
+        if path_to_item:
+            full_msg = "{0} at {1}".format(msg, render_path(path_to_item))
+        super(ApiTypeError, self).__init__(full_msg)
+
+
+class ApiValueError(OpenApiException, ValueError):
+    def __init__(self, msg, path_to_item=None) -> None:
+        """
+        Args:
+            msg (str): the exception message
+
+        Keyword Args:
+            path_to_item (list) the path to the exception in the
+                received_data dict. None if unset
+        """
+
+        self.path_to_item = path_to_item
+        full_msg = msg
+        if path_to_item:
+            full_msg = "{0} at {1}".format(msg, render_path(path_to_item))
+        super(ApiValueError, self).__init__(full_msg)
+
+
+class ApiAttributeError(OpenApiException, AttributeError):
+    def __init__(self, msg, path_to_item=None) -> None:
+        """
+        Raised when an attribute reference or assignment fails.
+
+        Args:
+            msg (str): the exception message
+
+        Keyword Args:
+            path_to_item (None/list) the path to the exception in the
+                received_data dict
+        """
+        self.path_to_item = path_to_item
+        full_msg = msg
+        if path_to_item:
+            full_msg = "{0} at {1}".format(msg, render_path(path_to_item))
+        super(ApiAttributeError, self).__init__(full_msg)
+
+
+class ApiKeyError(OpenApiException, KeyError):
+    def __init__(self, msg, path_to_item=None) -> None:
+        """
+        Args:
+            msg (str): the exception message
+
+        Keyword Args:
+            path_to_item (None/list) the path to the exception in the
+                received_data dict
+        """
+        self.path_to_item = path_to_item
+        full_msg = msg
+        if path_to_item:
+            full_msg = "{0} at {1}".format(msg, render_path(path_to_item))
+        super(ApiKeyError, self).__init__(full_msg)
+
+
+class ApiException(OpenApiException):
+
+    def __init__(
+        self, 
+        status=None, 
+        reason=None, 
+        http_resp=None,
+        *,
+        body: Optional[str] = None,
+        data: Optional[Any] = None,
+    ) -> None:
+        self.status = status
+        self.reason = reason
+        self.body = body
+        self.data = data
+        self.headers = None
+
+        if http_resp:
+            if self.status is None:
+                self.status = http_resp.status
+            if self.reason is None:
+                self.reason = http_resp.reason
+            if self.body is None:
+                try:
+                    self.body = http_resp.data.decode('utf-8')
+                except Exception:
+                    pass
+            self.headers = http_resp.getheaders()
+
+    @classmethod
+    def from_response(
+        cls, 
+        *, 
+        http_resp, 
+        body: Optional[str], 
+        data: Optional[Any],
+    ) -> Self:
+        if http_resp.status == 400:
+            raise BadRequestException(http_resp=http_resp, body=body, data=data)
+
+        if http_resp.status == 401:
+            raise UnauthorizedException(http_resp=http_resp, body=body, data=data)
+
+        if http_resp.status == 403:
+            raise ForbiddenException(http_resp=http_resp, body=body, data=data)
+
+        if http_resp.status == 404:
+            raise NotFoundException(http_resp=http_resp, body=body, data=data)
+
+        # Added new conditions for 409 and 422
+        if http_resp.status == 409:
+            raise ConflictException(http_resp=http_resp, body=body, data=data)
+
+        if http_resp.status == 422:
+            raise UnprocessableEntityException(http_resp=http_resp, body=body, data=data)
+
+        if 500 <= http_resp.status <= 599:
+            raise ServiceException(http_resp=http_resp, body=body, data=data)
+        raise ApiException(http_resp=http_resp, body=body, data=data)
+
+    def __str__(self):
+        """Custom error messages for exception"""
+        error_message = "({0})\n"\
+                        "Reason: {1}\n".format(self.status, self.reason)
+        if self.headers:
+            error_message += "HTTP response headers: {0}\n".format(
+                self.headers)
+
+        if self.data or self.body:
+            error_message += "HTTP response body: {0}\n".format(self.data or self.body)
+
+        return error_message
+
+
+class BadRequestException(ApiException):
+    pass
+
+
+class NotFoundException(ApiException):
+    pass
+
+
+class UnauthorizedException(ApiException):
+    pass
+
+
+class ForbiddenException(ApiException):
+    pass
+
+
+class ServiceException(ApiException):
+    pass
+
+
+class ConflictException(ApiException):
+    """Exception for HTTP 409 Conflict."""
+    pass
+
+
+class UnprocessableEntityException(ApiException):
+    """Exception for HTTP 422 Unprocessable Entity."""
+    pass
+
+
+def render_path(path_to_item):
+    """Returns a string representation of a path"""
+    result = ""
+    for pth in path_to_item:
+        if isinstance(pth, int):
+            result += "[{0}]".format(pth)
+        else:
+            result += "['{0}']".format(pth)
+    return result

smallestai/atoms/models/__init__.py

@@ -0,0 +1,72 @@
+# coding: utf-8
+
+# flake8: noqa
+"""
+    Agent Management API
+
+    API for managing agents, their templates, and call logs
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+# import models into model package
+from smallestai.atoms.models.agent_dto import AgentDTO
+from smallestai.atoms.models.agent_dto_language import AgentDTOLanguage
+from smallestai.atoms.models.agent_dto_synthesizer import AgentDTOSynthesizer
+from smallestai.atoms.models.agent_dto_synthesizer_voice_config import AgentDTOSynthesizerVoiceConfig
+from smallestai.atoms.models.api_response import ApiResponse
+from smallestai.atoms.models.bad_request_error_response import BadRequestErrorResponse
+from smallestai.atoms.models.create_agent_from_template200_response import CreateAgentFromTemplate200Response
+from smallestai.atoms.models.create_agent_from_template_request import CreateAgentFromTemplateRequest
+from smallestai.atoms.models.create_agent_request import CreateAgentRequest
+from smallestai.atoms.models.create_agent_request_language import CreateAgentRequestLanguage
+from smallestai.atoms.models.create_agent_request_language_synthesizer import CreateAgentRequestLanguageSynthesizer
+from smallestai.atoms.models.create_agent_request_language_synthesizer_voice_config import CreateAgentRequestLanguageSynthesizerVoiceConfig
+from smallestai.atoms.models.create_campaign201_response import CreateCampaign201Response
+from smallestai.atoms.models.create_campaign201_response_data import CreateCampaign201ResponseData
+from smallestai.atoms.models.create_campaign_request import CreateCampaignRequest
+from smallestai.atoms.models.create_knowledge_base201_response import CreateKnowledgeBase201Response
+from smallestai.atoms.models.create_knowledge_base_request import CreateKnowledgeBaseRequest
+from smallestai.atoms.models.delete_agent200_response import DeleteAgent200Response
+from smallestai.atoms.models.get_agent_by_id200_response import GetAgentById200Response
+from smallestai.atoms.models.get_agent_templates200_response import GetAgentTemplates200Response
+from smallestai.atoms.models.get_agent_templates200_response_data_inner import GetAgentTemplates200ResponseDataInner
+from smallestai.atoms.models.get_agents200_response import GetAgents200Response
+from smallestai.atoms.models.get_agents200_response_data import GetAgents200ResponseData
+from smallestai.atoms.models.get_campaign_by_id200_response import GetCampaignById200Response
+from smallestai.atoms.models.get_campaign_by_id200_response_data import GetCampaignById200ResponseData
+from smallestai.atoms.models.get_campaigns200_response import GetCampaigns200Response
+from smallestai.atoms.models.get_campaigns200_response_data_inner import GetCampaigns200ResponseDataInner
+from smallestai.atoms.models.get_campaigns200_response_data_inner_agent import GetCampaigns200ResponseDataInnerAgent
+from smallestai.atoms.models.get_campaigns200_response_data_inner_audience import GetCampaigns200ResponseDataInnerAudience
+from smallestai.atoms.models.get_campaigns_request import GetCampaignsRequest
+from smallestai.atoms.models.get_conversation_logs200_response import GetConversationLogs200Response
+from smallestai.atoms.models.get_conversation_logs200_response_data import GetConversationLogs200ResponseData
+from smallestai.atoms.models.get_current_user200_response import GetCurrentUser200Response
+from smallestai.atoms.models.get_current_user200_response_data import GetCurrentUser200ResponseData
+from smallestai.atoms.models.get_knowledge_base_by_id200_response import GetKnowledgeBaseById200Response
+from smallestai.atoms.models.get_knowledge_base_items200_response import GetKnowledgeBaseItems200Response
+from smallestai.atoms.models.get_knowledge_bases200_response import GetKnowledgeBases200Response
+from smallestai.atoms.models.get_organization200_response import GetOrganization200Response
+from smallestai.atoms.models.get_organization200_response_data import GetOrganization200ResponseData
+from smallestai.atoms.models.get_organization200_response_data_members_inner import GetOrganization200ResponseDataMembersInner
+from smallestai.atoms.models.get_organization200_response_data_subscription import GetOrganization200ResponseDataSubscription
+from smallestai.atoms.models.internal_server_error_response import InternalServerErrorResponse
+from smallestai.atoms.models.knowledge_base_dto import KnowledgeBaseDTO
+from smallestai.atoms.models.knowledge_base_item_dto import KnowledgeBaseItemDTO
+from smallestai.atoms.models.start_outbound_call200_response import StartOutboundCall200Response
+from smallestai.atoms.models.start_outbound_call200_response_data import StartOutboundCall200ResponseData
+from smallestai.atoms.models.start_outbound_call_request import StartOutboundCallRequest
+from smallestai.atoms.models.unauthorized_error_reponse import UnauthorizedErrorReponse
+from smallestai.atoms.models.update_agent200_response import UpdateAgent200Response
+from smallestai.atoms.models.update_agent_request import UpdateAgentRequest
+from smallestai.atoms.models.update_agent_request_language import UpdateAgentRequestLanguage
+from smallestai.atoms.models.update_agent_request_synthesizer import UpdateAgentRequestSynthesizer
+from smallestai.atoms.models.update_agent_request_synthesizer_voice_config import UpdateAgentRequestSynthesizerVoiceConfig
+from smallestai.atoms.models.update_agent_request_synthesizer_voice_config_one_of import UpdateAgentRequestSynthesizerVoiceConfigOneOf
+from smallestai.atoms.models.update_agent_request_synthesizer_voice_config_one_of1 import UpdateAgentRequestSynthesizerVoiceConfigOneOf1
+from smallestai.atoms.models.upload_text_to_knowledge_base_request import UploadTextToKnowledgeBaseRequest

smallestai/atoms/models/agent_dto.py

@@ -0,0 +1,130 @@
+# coding: utf-8
+
+"""
+    Agent Management API
+
+    API for managing agents, their templates, and call logs
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from smallestai.atoms.models.agent_dto_language import AgentDTOLanguage
+from smallestai.atoms.models.agent_dto_synthesizer import AgentDTOSynthesizer
+from typing import Optional, Set
+from typing_extensions import Self
+
+class AgentDTO(BaseModel):
+    """
+    AgentDTO
+    """ # noqa: E501
+    id: StrictStr = Field(description="The ID of the agent", alias="_id")
+    name: StrictStr = Field(description="The name of the agent")
+    description: Optional[StrictStr] = Field(default=None, description="The description of the agent")
+    organization: Optional[StrictStr] = Field(default=None, description="The organization ID of the agent")
+    workflow_id: Optional[StrictStr] = Field(default=None, description="The workflow ID of the agent", alias="workflowId")
+    created_by: Optional[StrictStr] = Field(default=None, description="The user ID of the user who created the agent", alias="createdBy")
+    global_knowledge_base_id: Optional[StrictStr] = Field(default=None, description="The global knowledge base ID of the agent", alias="globalKnowledgeBaseId")
+    language: Optional[AgentDTOLanguage] = None
+    synthesizer: Optional[AgentDTOSynthesizer] = None
+    slm_model: Optional[StrictStr] = Field(default=None, description="The LLM model to use for the agent. LLM model will be used to generate the response and take decisions based on the user's query.", alias="slmModel")
+    default_variables: Optional[Dict[str, Any]] = Field(default=None, description="The default variables to use for the agent. These variables will be used if no variables are provided when initiating a conversation with the agent.", alias="defaultVariables")
+    created_at: Optional[datetime] = Field(default=None, description="The date and time when the agent was created", alias="createdAt")
+    updated_at: Optional[datetime] = Field(default=None, description="The date and time when the agent was last updated", alias="updatedAt")
+    __properties: ClassVar[List[str]] = ["_id", "name", "description", "organization", "workflowId", "createdBy", "globalKnowledgeBaseId", "language", "synthesizer", "slmModel", "defaultVariables", "createdAt", "updatedAt"]
+
+    @field_validator('slm_model')
+    def slm_model_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['electron-v1', 'electron-v2', 'gpt-4o-mini']):
+            raise ValueError("must be one of enum values ('electron-v1', 'electron-v2', 'gpt-4o-mini')")
+        return value
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of AgentDTO from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of language
+        if self.language:
+            _dict['language'] = self.language.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of synthesizer
+        if self.synthesizer:
+            _dict['synthesizer'] = self.synthesizer.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentDTO from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "_id": obj.get("_id"),
+            "name": obj.get("name"),
+            "description": obj.get("description"),
+            "organization": obj.get("organization"),
+            "workflowId": obj.get("workflowId"),
+            "createdBy": obj.get("createdBy"),
+            "globalKnowledgeBaseId": obj.get("globalKnowledgeBaseId"),
+            "language": AgentDTOLanguage.from_dict(obj["language"]) if obj.get("language") is not None else None,
+            "synthesizer": AgentDTOSynthesizer.from_dict(obj["synthesizer"]) if obj.get("synthesizer") is not None else None,
+            "slmModel": obj.get("slmModel"),
+            "defaultVariables": obj.get("defaultVariables"),
+            "createdAt": obj.get("createdAt"),
+            "updatedAt": obj.get("updatedAt")
+        })
+        return _obj
+
+

smallestai/atoms/models/agent_dto_language.py

@@ -0,0 +1,91 @@
+# coding: utf-8
+
+"""
+    Agent Management API
+
+    API for managing agents, their templates, and call logs
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class AgentDTOLanguage(BaseModel):
+    """
+    The language configuration of the agent
+    """ # noqa: E501
+    enabled: Optional[StrictStr] = Field(default=None, description="The language of the agent")
+    switching: Optional[StrictBool] = Field(default=None, description="Whether the agent can switch between languages")
+    supported: Optional[List[StrictStr]] = Field(default=None, description="The supported languages of the agent")
+    __properties: ClassVar[List[str]] = ["enabled", "switching", "supported"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of AgentDTOLanguage from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentDTOLanguage from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "enabled": obj.get("enabled"),
+            "switching": obj.get("switching"),
+            "supported": obj.get("supported")
+        })
+        return _obj
+
+

smallestai/atoms/models/agent_dto_synthesizer.py

@@ -0,0 +1,99 @@
+# coding: utf-8
+
+"""
+    Agent Management API
+
+    API for managing agents, their templates, and call logs
+
+    The version of the OpenAPI document: 1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from smallestai.atoms.models.agent_dto_synthesizer_voice_config import AgentDTOSynthesizerVoiceConfig
+from typing import Optional, Set
+from typing_extensions import Self
+
+class AgentDTOSynthesizer(BaseModel):
+    """
+    The synthesizer (TTS) configuration of the agent
+    """ # noqa: E501
+    voice_config: Optional[AgentDTOSynthesizerVoiceConfig] = Field(default=None, alias="voiceConfig")
+    speed: Optional[Union[StrictFloat, StrictInt]] = 1.2
+    consistency: Optional[Union[StrictFloat, StrictInt]] = Field(default=0.5, description="The consistency of the synthesizer")
+    similarity: Optional[Union[StrictFloat, StrictInt]] = Field(default=0, description="The similarity of the synthesizer")
+    enhancement: Optional[Union[StrictFloat, StrictInt]] = Field(default=1, description="The enhancement of the synthesizer")
+    __properties: ClassVar[List[str]] = ["voiceConfig", "speed", "consistency", "similarity", "enhancement"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of AgentDTOSynthesizer from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of voice_config
+        if self.voice_config:
+            _dict['voiceConfig'] = self.voice_config.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentDTOSynthesizer from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "voiceConfig": AgentDTOSynthesizerVoiceConfig.from_dict(obj["voiceConfig"]) if obj.get("voiceConfig") is not None else None,
+            "speed": obj.get("speed") if obj.get("speed") is not None else 1.2,
+            "consistency": obj.get("consistency") if obj.get("consistency") is not None else 0.5,
+            "similarity": obj.get("similarity") if obj.get("similarity") is not None else 0,
+            "enhancement": obj.get("enhancement") if obj.get("enhancement") is not None else 1
+        })
+        return _obj
+
+