athena-intelligence 0.1.93__py3-none-any.whl → 0.1.95__py3-none-any.whl

athena/types/filter_operator.py

@@ -1,73 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class FilterOperator(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    EQUALS = "equals"
-    NOT_EQUALS = "notEquals"
-    CONTAINS = "contains"
-    NOT_CONTAINS = "notContains"
-    STARTS_WITH = "startsWith"
-    ENDS_WITH = "endsWith"
-    GT = "gt"
-    GTE = "gte"
-    LT = "lt"
-    LTE = "lte"
-    SET = "set"
-    NOT_SET = "notSet"
-    IN = "in"
-    NOT_IN = "notIn"
-
-    def visit(
-        self,
-        equals: typing.Callable[[], T_Result],
-        not_equals: typing.Callable[[], T_Result],
-        contains: typing.Callable[[], T_Result],
-        not_contains: typing.Callable[[], T_Result],
-        starts_with: typing.Callable[[], T_Result],
-        ends_with: typing.Callable[[], T_Result],
-        gt: typing.Callable[[], T_Result],
-        gte: typing.Callable[[], T_Result],
-        lt: typing.Callable[[], T_Result],
-        lte: typing.Callable[[], T_Result],
-        set_: typing.Callable[[], T_Result],
-        not_set: typing.Callable[[], T_Result],
-        in_: typing.Callable[[], T_Result],
-        not_in: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is FilterOperator.EQUALS:
-            return equals()
-        if self is FilterOperator.NOT_EQUALS:
-            return not_equals()
-        if self is FilterOperator.CONTAINS:
-            return contains()
-        if self is FilterOperator.NOT_CONTAINS:
-            return not_contains()
-        if self is FilterOperator.STARTS_WITH:
-            return starts_with()
-        if self is FilterOperator.ENDS_WITH:
-            return ends_with()
-        if self is FilterOperator.GT:
-            return gt()
-        if self is FilterOperator.GTE:
-            return gte()
-        if self is FilterOperator.LT:
-            return lt()
-        if self is FilterOperator.LTE:
-            return lte()
-        if self is FilterOperator.SET:
-            return set_()
-        if self is FilterOperator.NOT_SET:
-            return not_set()
-        if self is FilterOperator.IN:
-            return in_()
-        if self is FilterOperator.NOT_IN:
-            return not_in()

athena/types/firecrawl_scrape_url_data_reponse_dto.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .firecrawl_scrape_url_metadata import FirecrawlScrapeUrlMetadata
-
-
-class FirecrawlScrapeUrlDataReponseDto(pydantic_v1.BaseModel):
-    content: str
-    markdown: str
-    metadata: FirecrawlScrapeUrlMetadata
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/firecrawl_scrape_url_metadata.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class FirecrawlScrapeUrlMetadata(pydantic_v1.BaseModel):
-    title: typing.Optional[str] = None
-    description: typing.Optional[str] = None
-    language: typing.Optional[str] = None
-    source_url: typing.Optional[str] = pydantic_v1.Field(alias="sourceURL", default=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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-        populate_by_name = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/get_datasets_response.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .dataset import Dataset
-
-
-class GetDatasetsResponse(pydantic_v1.BaseModel):
-    datasets: typing.List[Dataset]
-    total: int
-    page: int
-    page_size: int
-    pages: 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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/get_snippet_out.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class GetSnippetOut(pydantic_v1.BaseModel):
-    response: typing.Dict[str, typing.Any]
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/get_snippets_response.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .snippet import Snippet
-
-
-class GetSnippetsResponse(pydantic_v1.BaseModel):
-    snippets: typing.List[Snippet]
-    total: int
-    page: int
-    page_size: int
-    pages: 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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/langchain_documents_request_out.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class LangchainDocumentsRequestOut(pydantic_v1.BaseModel):
-    output: typing.Dict[str, typing.Any]
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/llm_model.py

@@ -1,141 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class LlmModel(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    GPT_35_TURBO = "gpt-3.5-turbo"
-    GPT_4_TURBO = "gpt-4-turbo"
-    GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
-    GPT_4_O_MINI = "gpt-4o-mini"
-    GPT_4_O = "gpt-4o"
-    GPT_4_O_20240806 = "gpt-4o-2024-08-06"
-    GPT_4 = "gpt-4"
-    MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
-    MISTRAL_LARGE_0224 = "mistral-large-0224"
-    MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
-    LLAMA_V_38_B_INSTRUCT = "llama-v3-8b-instruct"
-    LLAMA_V_370_B_INSTRUCT = "llama-v3-70b-instruct"
-    LLAMA_V_3_P_1405_B_INSTRUCT = "llama-v3p1-405b-instruct"
-    CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
-    CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
-    CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
-    CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
-    GROQ_MIXTRAL_8_X_7_B_32768 = "groq-mixtral-8x7b-32768"
-    GROQ_LLAMA_38_B_8192 = "groq-llama3-8b-8192"
-    GROQ_LLAMA_370_B_8192 = "groq-llama3-70b-8192"
-    GROQ_GEMMA_7_B_IT = "groq-gemma-7b-it"
-    GROQ_LLAMA_31405_B_REASONING = "groq-llama-3.1-405b-reasoning"
-    GROQ_LLAMA_3170_B_VERSATILE = "groq-llama-3.1-70b-versatile"
-    GROQ_LLAMA_318_B_INSTANT = "groq-llama-3.1-8b-instant"
-    GROQ_LLAMA_3_GROQ_70_B_8192_TOOL_USE_PREVIEW = "groq-llama3-groq-70b-8192-tool-use-preview"
-    GROQ_LLAMA_3_GROQ_8_B_8192_TOOL_USE_PREVIEW = "groq-llama3-groq-8b-8192-tool-use-preview"
-    DATABRICKS_DBRX = "databricks-dbrx"
-    GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
-    GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
-    GOOGLE_GEMINI_15_FLASH_001 = "google-gemini-1.5-flash-001"
-    GOOGLE_GEMINI_15_PRO_001 = "google-gemini-1.5-pro-001"
-
-    def visit(
-        self,
-        gpt_35_turbo: typing.Callable[[], T_Result],
-        gpt_4_turbo: typing.Callable[[], T_Result],
-        gpt_4_turbo_preview: typing.Callable[[], T_Result],
-        gpt_4_o_mini: typing.Callable[[], T_Result],
-        gpt_4_o: typing.Callable[[], T_Result],
-        gpt_4_o_20240806: typing.Callable[[], T_Result],
-        gpt_4: typing.Callable[[], T_Result],
-        mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
-        mistral_large_0224: typing.Callable[[], T_Result],
-        mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
-        llama_v_38_b_instruct: typing.Callable[[], T_Result],
-        llama_v_370_b_instruct: typing.Callable[[], T_Result],
-        llama_v_3_p_1405_b_instruct: typing.Callable[[], T_Result],
-        claude_35_sonnet_20240620: typing.Callable[[], T_Result],
-        claude_3_opus_20240229: typing.Callable[[], T_Result],
-        claude_3_sonnet_20240229: typing.Callable[[], T_Result],
-        claude_3_haiku_20240307: typing.Callable[[], T_Result],
-        groq_mixtral_8_x_7_b_32768: typing.Callable[[], T_Result],
-        groq_llama_38_b_8192: typing.Callable[[], T_Result],
-        groq_llama_370_b_8192: typing.Callable[[], T_Result],
-        groq_gemma_7_b_it: typing.Callable[[], T_Result],
-        groq_llama_31405_b_reasoning: typing.Callable[[], T_Result],
-        groq_llama_3170_b_versatile: typing.Callable[[], T_Result],
-        groq_llama_318_b_instant: typing.Callable[[], T_Result],
-        groq_llama_3_groq_70_b_8192_tool_use_preview: typing.Callable[[], T_Result],
-        groq_llama_3_groq_8_b_8192_tool_use_preview: typing.Callable[[], T_Result],
-        databricks_dbrx: typing.Callable[[], T_Result],
-        google_gemini_10_pro_latest: typing.Callable[[], T_Result],
-        google_gemini_15_pro_latest: typing.Callable[[], T_Result],
-        google_gemini_15_flash_001: typing.Callable[[], T_Result],
-        google_gemini_15_pro_001: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is LlmModel.GPT_35_TURBO:
-            return gpt_35_turbo()
-        if self is LlmModel.GPT_4_TURBO:
-            return gpt_4_turbo()
-        if self is LlmModel.GPT_4_TURBO_PREVIEW:
-            return gpt_4_turbo_preview()
-        if self is LlmModel.GPT_4_O_MINI:
-            return gpt_4_o_mini()
-        if self is LlmModel.GPT_4_O:
-            return gpt_4_o()
-        if self is LlmModel.GPT_4_O_20240806:
-            return gpt_4_o_20240806()
-        if self is LlmModel.GPT_4:
-            return gpt_4()
-        if self is LlmModel.MIXTRAL_SMALL_8_X_7_B_0211:
-            return mixtral_small_8_x_7_b_0211()
-        if self is LlmModel.MISTRAL_LARGE_0224:
-            return mistral_large_0224()
-        if self is LlmModel.MIXTRAL_8_X_22_B_INSTRUCT:
-            return mixtral_8_x_22_b_instruct()
-        if self is LlmModel.LLAMA_V_38_B_INSTRUCT:
-            return llama_v_38_b_instruct()
-        if self is LlmModel.LLAMA_V_370_B_INSTRUCT:
-            return llama_v_370_b_instruct()
-        if self is LlmModel.LLAMA_V_3_P_1405_B_INSTRUCT:
-            return llama_v_3_p_1405_b_instruct()
-        if self is LlmModel.CLAUDE_35_SONNET_20240620:
-            return claude_35_sonnet_20240620()
-        if self is LlmModel.CLAUDE_3_OPUS_20240229:
-            return claude_3_opus_20240229()
-        if self is LlmModel.CLAUDE_3_SONNET_20240229:
-            return claude_3_sonnet_20240229()
-        if self is LlmModel.CLAUDE_3_HAIKU_20240307:
-            return claude_3_haiku_20240307()
-        if self is LlmModel.GROQ_MIXTRAL_8_X_7_B_32768:
-            return groq_mixtral_8_x_7_b_32768()
-        if self is LlmModel.GROQ_LLAMA_38_B_8192:
-            return groq_llama_38_b_8192()
-        if self is LlmModel.GROQ_LLAMA_370_B_8192:
-            return groq_llama_370_b_8192()
-        if self is LlmModel.GROQ_GEMMA_7_B_IT:
-            return groq_gemma_7_b_it()
-        if self is LlmModel.GROQ_LLAMA_31405_B_REASONING:
-            return groq_llama_31405_b_reasoning()
-        if self is LlmModel.GROQ_LLAMA_3170_B_VERSATILE:
-            return groq_llama_3170_b_versatile()
-        if self is LlmModel.GROQ_LLAMA_318_B_INSTANT:
-            return groq_llama_318_b_instant()
-        if self is LlmModel.GROQ_LLAMA_3_GROQ_70_B_8192_TOOL_USE_PREVIEW:
-            return groq_llama_3_groq_70_b_8192_tool_use_preview()
-        if self is LlmModel.GROQ_LLAMA_3_GROQ_8_B_8192_TOOL_USE_PREVIEW:
-            return groq_llama_3_groq_8_b_8192_tool_use_preview()
-        if self is LlmModel.DATABRICKS_DBRX:
-            return databricks_dbrx()
-        if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:
-            return google_gemini_10_pro_latest()
-        if self is LlmModel.GOOGLE_GEMINI_15_PRO_LATEST:
-            return google_gemini_15_pro_latest()
-        if self is LlmModel.GOOGLE_GEMINI_15_FLASH_001:
-            return google_gemini_15_flash_001()
-        if self is LlmModel.GOOGLE_GEMINI_15_PRO_001:
-            return google_gemini_15_pro_001()

athena/types/map_reduce_chain_out.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .document import Document
-
-
-class MapReduceChainOut(pydantic_v1.BaseModel):
-    input: str
-    input_documents: typing.List[Document]
-    output_text: 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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/message_out.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class MessageOut(pydantic_v1.BaseModel):
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/message_out_dto.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 ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .status_enum import StatusEnum
-
-
-class MessageOutDto(pydantic_v1.BaseModel):
-    id: str
-    conversation_id: str
-    logs: typing.Optional[str] = None
-    content: typing.Optional[str] = None
-    created_at: str
-    status: StatusEnum
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/model.py

@@ -1,93 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class Model(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    GPT_35_TURBO = "gpt-3.5-turbo"
-    GPT_4_TURBO = "gpt-4-turbo"
-    GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
-    GPT_4 = "gpt-4"
-    GPT_4_O = "gpt-4o"
-    GPT_4_O_MINI = "gpt-4o-mini"
-    GPT_4_O_20240806 = "gpt-4o-2024-08-06"
-    MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
-    MISTRAL_LARGE_0224 = "mistral-large-0224"
-    MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
-    LLAMA_V_370_B_INSTRUCT = "llama-v3-70b-instruct"
-    LLAMA_V_38_B_INSTRUCT = "llama-v3-8b-instruct"
-    CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
-    CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
-    CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
-    CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
-    GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
-    GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
-    DATABRICKS_DBRX = "databricks-dbrx"
-
-    def visit(
-        self,
-        gpt_35_turbo: typing.Callable[[], T_Result],
-        gpt_4_turbo: typing.Callable[[], T_Result],
-        gpt_4_turbo_preview: typing.Callable[[], T_Result],
-        gpt_4: typing.Callable[[], T_Result],
-        gpt_4_o: typing.Callable[[], T_Result],
-        gpt_4_o_mini: typing.Callable[[], T_Result],
-        gpt_4_o_20240806: typing.Callable[[], T_Result],
-        mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
-        mistral_large_0224: typing.Callable[[], T_Result],
-        mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
-        llama_v_370_b_instruct: typing.Callable[[], T_Result],
-        llama_v_38_b_instruct: typing.Callable[[], T_Result],
-        claude_3_opus_20240229: typing.Callable[[], T_Result],
-        claude_3_sonnet_20240229: typing.Callable[[], T_Result],
-        claude_3_haiku_20240307: typing.Callable[[], T_Result],
-        claude_35_sonnet_20240620: typing.Callable[[], T_Result],
-        google_gemini_10_pro_latest: typing.Callable[[], T_Result],
-        google_gemini_15_pro_latest: typing.Callable[[], T_Result],
-        databricks_dbrx: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is Model.GPT_35_TURBO:
-            return gpt_35_turbo()
-        if self is Model.GPT_4_TURBO:
-            return gpt_4_turbo()
-        if self is Model.GPT_4_TURBO_PREVIEW:
-            return gpt_4_turbo_preview()
-        if self is Model.GPT_4:
-            return gpt_4()
-        if self is Model.GPT_4_O:
-            return gpt_4_o()
-        if self is Model.GPT_4_O_MINI:
-            return gpt_4_o_mini()
-        if self is Model.GPT_4_O_20240806:
-            return gpt_4_o_20240806()
-        if self is Model.MIXTRAL_SMALL_8_X_7_B_0211:
-            return mixtral_small_8_x_7_b_0211()
-        if self is Model.MISTRAL_LARGE_0224:
-            return mistral_large_0224()
-        if self is Model.MIXTRAL_8_X_22_B_INSTRUCT:
-            return mixtral_8_x_22_b_instruct()
-        if self is Model.LLAMA_V_370_B_INSTRUCT:
-            return llama_v_370_b_instruct()
-        if self is Model.LLAMA_V_38_B_INSTRUCT:
-            return llama_v_38_b_instruct()
-        if self is Model.CLAUDE_3_OPUS_20240229:
-            return claude_3_opus_20240229()
-        if self is Model.CLAUDE_3_SONNET_20240229:
-            return claude_3_sonnet_20240229()
-        if self is Model.CLAUDE_3_HAIKU_20240307:
-            return claude_3_haiku_20240307()
-        if self is Model.CLAUDE_35_SONNET_20240620:
-            return claude_35_sonnet_20240620()
-        if self is Model.GOOGLE_GEMINI_10_PRO_LATEST:
-            return google_gemini_10_pro_latest()
-        if self is Model.GOOGLE_GEMINI_15_PRO_LATEST:
-            return google_gemini_15_pro_latest()
-        if self is Model.DATABRICKS_DBRX:
-            return databricks_dbrx()

athena/types/publish_formats.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
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class PublishFormats(pydantic_v1.BaseModel):
-    markdown: bool
-    pdf: bool
-    docx: bool
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/query_model.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 ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .filter_model import FilterModel
-from .time_dimension_model import TimeDimensionModel
-
-
-class QueryModel(pydantic_v1.BaseModel):
-    measures: typing.Optional[typing.List[str]] = None
-    time_dimensions: typing.Optional[typing.List[TimeDimensionModel]] = pydantic_v1.Field(
-        alias="timeDimensions", default=None
-    )
-    dimensions: typing.Optional[typing.List[str]] = None
-    filters: typing.Optional[typing.List[FilterModel]] = None
-    order: typing.Optional[typing.Dict[str, typing.Any]] = None
-    limit: typing.Optional[int] = 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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-        populate_by_name = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}