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

athena/types/agent_chat_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 AgentChatRequest(pydantic.BaseModel):
-    user_message: str
-    workspace_id: str
-    sdataset_ids: typing.List[str]
-    model: 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/agent_chat_request_wait.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 AgentChatRequestWait(pydantic.BaseModel):
-    seconds: int
-
-    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/ai_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 AiMessage(pydantic.BaseModel):
-    """
-    A Message from an AI.
-    """
-
-    content: Content
-    additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None
-    example: typing.Optional[bool] = 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/ai_message_content_item.py

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

athena/types/api_message_submitted.py

@@ -1,36 +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 ApiMessageSubmitted(pydantic.BaseModel):
-    content: str
-    conversation_id: typing.Optional[str] = None
-    sdataset_ids: typing.Optional[typing.List[str]] = None
-    tools: typing.Optional[typing.List[str]] = None
-    model: typing.Optional[str] = None
-    documents: typing.Optional[typing.List[str]] = None
-    config: typing.Optional[typing.Dict[str, typing.Any]] = None
-    conversation_name: typing.Optional[str] = None
-    additional_context: 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/append_final_msg_tokens_request.py

@@ -1,29 +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 AppendFinalMsgTokensRequest(pydantic.BaseModel):
-    new_tokens: typing.List[str]
-    msg_id: 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/append_msg_tokens_request.py

@@ -1,29 +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 AppendMsgTokensRequest(pydantic.BaseModel):
-    new_tokens: typing.List[str]
-    msg_id: 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/attachment.py

@@ -1,32 +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 Attachment(pydantic.BaseModel):
-    content: typing.Optional[str] = None
-    file_name: typing.Optional[str] = None
-    content_type: typing.Optional[str] = None
-    size: typing.Optional[int] = None
-    disposition: 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/audio_request.py

@@ -1,29 +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 AudioRequest(pydantic.BaseModel):
-    text: str
-    turbo: typing.Optional[bool] = 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/base_message.py

@@ -1,37 +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 BaseMessage(pydantic.BaseModel):
-    """
-    The base abstract Message class.
-
-    Messages are the inputs and outputs of ChatModels.
-    """
-
-    content: Content
-    additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = None
-    type: 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/base_message_content_item.py

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

athena/types/batch_response_metadata.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 BatchResponseMetadata(pydantic.BaseModel):
-    """
-    Represents response metadata used for batches of input/output LangServe
-    responses.
-    """
-
-    run_ids: typing.List[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/callback_event.py

@@ -1,142 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import typing
-
-from .on_chain_end import OnChainEnd
-from .on_chain_error import OnChainError
-from .on_chain_start import OnChainStart
-from .on_chat_model_start import OnChatModelStart
-from .on_llm_end import OnLlmEnd
-from .on_llm_start import OnLlmStart
-from .on_retriever_end import OnRetrieverEnd
-from .on_retriever_error import OnRetrieverError
-from .on_retriever_start import OnRetrieverStart
-from .on_tool_end import OnToolEnd
-from .on_tool_error import OnToolError
-from .on_tool_start import OnToolStart
-
-
-class CallbackEvent_OnChainStart(OnChainStart):
-    type: typing.Literal["on_chain_start"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnChainEnd(OnChainEnd):
-    type: typing.Literal["on_chain_end"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnChainError(OnChainError):
-    type: typing.Literal["on_chain_error"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnChatModelStart(OnChatModelStart):
-    type: typing.Literal["on_chat_model_start"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnLlmStart(OnLlmStart):
-    type: typing.Literal["on_llm_start"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnLlmEnd(OnLlmEnd):
-    type: typing.Literal["on_llm_end"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnToolStart(OnToolStart):
-    type: typing.Literal["on_tool_start"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnToolEnd(OnToolEnd):
-    type: typing.Literal["on_tool_end"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnToolError(OnToolError):
-    type: typing.Literal["on_tool_error"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnRetrieverStart(OnRetrieverStart):
-    type: typing.Literal["on_retriever_start"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnRetrieverEnd(OnRetrieverEnd):
-    type: typing.Literal["on_retriever_end"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class CallbackEvent_OnRetrieverError(OnRetrieverError):
-    type: typing.Literal["on_retriever_error"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-CallbackEvent = typing.Union[
-    CallbackEvent_OnChainStart,
-    CallbackEvent_OnChainEnd,
-    CallbackEvent_OnChainError,
-    CallbackEvent_OnChatModelStart,
-    CallbackEvent_OnLlmStart,
-    CallbackEvent_OnLlmEnd,
-    CallbackEvent_OnToolStart,
-    CallbackEvent_OnToolEnd,
-    CallbackEvent_OnToolError,
-    CallbackEvent_OnRetrieverStart,
-    CallbackEvent_OnRetrieverEnd,
-    CallbackEvent_OnRetrieverError,
-]

athena/types/callbacks.py

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

athena/types/chart_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 ChartRequest(pydantic.BaseModel):
-    question: str
-    workspace_id: str
-    sdataset_ids: typing.List[str]
-    model: 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/chat_fireworks_input.py

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

athena/types/chat_fireworks_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 ChatFireworksInputItem_Ai(AiMessage):
-    type: typing.Literal["ai"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksInputItem_Human(HumanMessage):
-    type: typing.Literal["human"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksInputItem_Chat(ChatMessage):
-    type: typing.Literal["chat"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksInputItem_System(SystemMessage):
-    type: typing.Literal["system"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksInputItem_Function(FunctionMessage):
-    type: typing.Literal["function"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksInputItem_Tool(ToolMessage):
-    type: typing.Literal["tool"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-ChatFireworksInputItem = typing.Union[
-    ChatFireworksInputItem_Ai,
-    ChatFireworksInputItem_Human,
-    ChatFireworksInputItem_Chat,
-    ChatFireworksInputItem_System,
-    ChatFireworksInputItem_Function,
-    ChatFireworksInputItem_Tool,
-]

athena/types/chat_fireworks_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 ChatFireworksOutput_Ai(AiMessage):
-    type: typing.Literal["ai"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksOutput_Human(HumanMessage):
-    type: typing.Literal["human"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksOutput_Chat(ChatMessage):
-    type: typing.Literal["chat"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksOutput_System(SystemMessage):
-    type: typing.Literal["system"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksOutput_Function(FunctionMessage):
-    type: typing.Literal["function"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class ChatFireworksOutput_Tool(ToolMessage):
-    type: typing.Literal["tool"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-ChatFireworksOutput = typing.Union[
-    ChatFireworksOutput_Ai,
-    ChatFireworksOutput_Human,
-    ChatFireworksOutput_Chat,
-    ChatFireworksOutput_System,
-    ChatFireworksOutput_Function,
-    ChatFireworksOutput_Tool,
-]