athena-intelligence 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

athena/types/chat_generation.py

@@ -1,35 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .base_message import BaseMessage
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ChatGeneration(pydantic.BaseModel):
-    """
-    A single chat generation output.
-    """
-
-    text: typing.Optional[str] = None
-    generation_info: typing.Optional[typing.Dict[str, typing.Any]] = None
-    message: BaseMessage
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/chat_message.py

@@ -1,35 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .content import Content
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ChatMessage(pydantic.BaseModel):
-    """
-    A Message that can be assigned an arbitrary speaker (i.e. role).
-    """
-
-    content: Content
-    additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None
-    role: str
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/chat_message_content_item.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-ChatMessageContentItem = typing.Union[str, typing.Dict[str, typing.Any]]

athena/types/chat_open_ai_input.py

@@ -1,9 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-from .chat_open_ai_input_item import ChatOpenAiInputItem
-from .chat_prompt_value_concrete import ChatPromptValueConcrete
-from .string_prompt_value import StringPromptValue
-
-ChatOpenAiInput = typing.Union[str, StringPromptValue, ChatPromptValueConcrete, typing.List[ChatOpenAiInputItem]]

athena/types/chat_open_ai_input_item.py

@@ -1,76 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import typing
-
-from .ai_message import AiMessage
-from .chat_message import ChatMessage
-from .function_message import FunctionMessage
-from .human_message import HumanMessage
-from .system_message import SystemMessage
-from .tool_message import ToolMessage
-
-
-class ChatOpenAiInputItem_Ai(AiMessage):
-    type: typing.Literal["ai"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiInputItem_Human(HumanMessage):
-    type: typing.Literal["human"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiInputItem_Chat(ChatMessage):
-    type: typing.Literal["chat"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiInputItem_System(SystemMessage):
-    type: typing.Literal["system"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiInputItem_Function(FunctionMessage):
-    type: typing.Literal["function"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiInputItem_Tool(ToolMessage):
-    type: typing.Literal["tool"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-ChatOpenAiInputItem = typing.Union[
-    ChatOpenAiInputItem_Ai,
-    ChatOpenAiInputItem_Human,
-    ChatOpenAiInputItem_Chat,
-    ChatOpenAiInputItem_System,
-    ChatOpenAiInputItem_Function,
-    ChatOpenAiInputItem_Tool,
-]

athena/types/chat_open_ai_output.py

@@ -1,76 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import typing
-
-from .ai_message import AiMessage
-from .chat_message import ChatMessage
-from .function_message import FunctionMessage
-from .human_message import HumanMessage
-from .system_message import SystemMessage
-from .tool_message import ToolMessage
-
-
-class ChatOpenAiOutput_Ai(AiMessage):
-    type: typing.Literal["ai"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiOutput_Human(HumanMessage):
-    type: typing.Literal["human"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiOutput_Chat(ChatMessage):
-    type: typing.Literal["chat"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiOutput_System(SystemMessage):
-    type: typing.Literal["system"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiOutput_Function(FunctionMessage):
-    type: typing.Literal["function"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatOpenAiOutput_Tool(ToolMessage):
-    type: typing.Literal["tool"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-ChatOpenAiOutput = typing.Union[
-    ChatOpenAiOutput_Ai,
-    ChatOpenAiOutput_Human,
-    ChatOpenAiOutput_Chat,
-    ChatOpenAiOutput_System,
-    ChatOpenAiOutput_Function,
-    ChatOpenAiOutput_Tool,
-]

athena/types/chat_prompt_value_concrete.py

@@ -1,35 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .chat_prompt_value_concrete_messages_item import ChatPromptValueConcreteMessagesItem
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ChatPromptValueConcrete(pydantic.BaseModel):
-    """
-    Chat prompt value which explicitly lists out the message types it accepts.
-    For use in external schemas.
-    """
-
-    messages: typing.List[ChatPromptValueConcreteMessagesItem]
-    type: typing.Optional[typing.Literal["ChatPromptValueConcrete"]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/chat_prompt_value_concrete_messages_item.py

@@ -1,76 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import typing
-
-from .ai_message import AiMessage
-from .chat_message import ChatMessage
-from .function_message import FunctionMessage
-from .human_message import HumanMessage
-from .system_message import SystemMessage
-from .tool_message import ToolMessage
-
-
-class ChatPromptValueConcreteMessagesItem_Ai(AiMessage):
-    type: typing.Literal["ai"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatPromptValueConcreteMessagesItem_Human(HumanMessage):
-    type: typing.Literal["human"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatPromptValueConcreteMessagesItem_Chat(ChatMessage):
-    type: typing.Literal["chat"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatPromptValueConcreteMessagesItem_System(SystemMessage):
-    type: typing.Literal["system"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatPromptValueConcreteMessagesItem_Function(FunctionMessage):
-    type: typing.Literal["function"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatPromptValueConcreteMessagesItem_Tool(ToolMessage):
-    type: typing.Literal["tool"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-ChatPromptValueConcreteMessagesItem = typing.Union[
-    ChatPromptValueConcreteMessagesItem_Ai,
-    ChatPromptValueConcreteMessagesItem_Human,
-    ChatPromptValueConcreteMessagesItem_Chat,
-    ChatPromptValueConcreteMessagesItem_System,
-    ChatPromptValueConcreteMessagesItem_Function,
-    ChatPromptValueConcreteMessagesItem_Tool,
-]

athena/types/content.py

@@ -1,7 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-from .tool_message_content_item import ToolMessageContentItem
-
-Content = typing.Union[str, typing.List[ToolMessageContentItem]]

athena/types/create_room_request.py

@@ -1,28 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class CreateRoomRequest(pydantic.BaseModel):
-    room_id: str
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/delete_room_request.py

@@ -1,28 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class DeleteRoomRequest(pydantic.BaseModel):
-    room_id: str
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/document.py

@@ -1,34 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class Document(pydantic.BaseModel):
-    """
-    Class for storing a piece of text and associated metadata.
-    """
-
-    page_content: str
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
-    type: typing.Optional[typing.Literal["Document"]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/document_webhook_payload.py

@@ -1,33 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class DocumentWebhookPayload(pydantic.BaseModel):
-    app_name: str = pydantic.Field(alias="appName")
-    name: str
-    time: str
-    tiptap_json: typing.Dict[str, typing.Any] = pydantic.Field(alias="tiptapJson")
-    clients_count: int = pydantic.Field(alias="clientsCount")
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/email_data.py

@@ -1,40 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .attachment import Attachment
-from .envelope import Envelope
-from .headers import Headers
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class EmailData(pydantic.BaseModel):
-    """
-    Email model.
-    """
-
-    headers: typing.Optional[Headers] = None
-    envelope: typing.Optional[Envelope] = None
-    plain: typing.Optional[str] = None
-    html: typing.Optional[str] = None
-    reply_plain: typing.Optional[str] = None
-    attachments: typing.Optional[typing.List[Attachment]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/envelope.py

@@ -1,38 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .spf import Spf
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class Envelope(pydantic.BaseModel):
-    to: typing.Optional[str] = None
-    from_: typing.Optional[str] = pydantic.Field(alias="from", default=None)
-    helo_domain: typing.Optional[str] = None
-    remote_ip: typing.Optional[str] = None
-    recipients: typing.Optional[typing.List[str]] = None
-    tls: typing.Optional[bool] = None
-    tls_cipher: typing.Optional[str] = None
-    md_5: typing.Optional[str] = pydantic.Field(alias="md5", default=None)
-    spf: typing.Optional[Spf] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/error.py

@@ -1,34 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class Error(pydantic.BaseModel):
-    """
-    Error object that is modeled after an HTTP error format.
-    """
-
-    status_code: int
-    message: str
-    type: typing.Optional[typing.Literal["error"]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/forecast_request.py

@@ -1,31 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ForecastRequest(pydantic.BaseModel):
-    historical: typing.List[typing.Dict[str, typing.Any]]
-    forecast_horizon: int
-    confidence_level: typing.Optional[typing.List[int]] = None
-    freq: typing.Optional[str] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/function_message.py

@@ -1,35 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .content import Content
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class FunctionMessage(pydantic.BaseModel):
-    """
-    A Message for passing the result of executing a function back to a model.
-    """
-
-    content: Content
-    additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None
-    name: str
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/function_message_content_item.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-FunctionMessageContentItem = typing.Union[str, typing.Dict[str, typing.Any]]