athena-intelligence 0.1.230__py3-none-any.whl → 0.1.242__py3-none-any.whl

athena/tools/sheets/raw_client.py

@@ -8,11 +8,15 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.http_response import AsyncHttpResponse, HttpResponse
 from ...core.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
+from ...core.serialization import convert_and_respect_annotation_metadata
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.cell_format import CellFormat
 from ...types.create_new_sheet_tab_response import CreateNewSheetTabResponse
 from ...types.get_table_response import GetTableResponse
+from ...types.sheet import Sheet
 from ...types.sheet_operation_response import SheetOperationResponse
 from ...types.table_row_data import TableRowData
+from .types.update_sheet_range_request_values_item_item import UpdateSheetRangeRequestValuesItemItem
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -503,10 +507,9 @@ class RawSheetsClient:
         asset_id: str,
         end_column: int,
         end_row: int,
+        formatting: CellFormat,
         start_column: int,
         start_row: int,
-        type: str,
-        value_json: str,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
@@ -524,18 +527,15 @@ class RawSheetsClient:
         end_row : int
             1-based ending row index
 
+        formatting : CellFormat
+            Cell format
+
         start_column : int
             1-based starting column index
 
         start_row : int
             1-based starting row index
 
-        type : str
-            Formatting type (e.g. backgroundColor, textFormat, borders, etc.)
-
-        value_json : str
-            JSON string for value param (e.g. '"#FF0000"' or '{"fontSize":12}')
-
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
 
@@ -554,11 +554,12 @@ class RawSheetsClient:
                 "asset_id": asset_id,
                 "end_column": end_column,
                 "end_row": end_row,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting, annotation=CellFormat, direction="write"
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "type": type,
-                "value_json": value_json,
             },
             headers={
                 "content-type": "application/json",
@@ -598,7 +599,8 @@ class RawSheetsClient:
         asset_id: str,
         start_column: int,
         start_row: int,
-        values: typing.Sequence[str],
+        values: typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+        formatting: typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]] = OMIT,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SheetOperationResponse]:
@@ -616,8 +618,11 @@ class RawSheetsClient:
         start_row : int
             1-based starting row index
 
-        values : typing.Sequence[str]
-            List of pipe-separated strings representing rows (e.g. ['A1|B1|C1', 'A2|B2|C2'])
+        values : typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]]
+            2D list of cells for each row
+
+        formatting : typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]]
+            Optional 2D list of cell formats matching the structure of values. Each row is a list of CellFormat objects for each cell in that row. Use None for cells without formatting. numberFormat is not required unless user explicity asked to change
 
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
@@ -635,10 +640,19 @@ class RawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]],
+                    direction="write",
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "values": values,
+                "values": convert_and_respect_annotation_metadata(
+                    object_=values,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+                    direction="write",
+                ),
             },
             headers={
                 "content-type": "application/json",
@@ -818,14 +832,7 @@ class RawSheetsClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     def create_tab(
-        self,
-        *,
-        asset_id: str,
-        sheet_id: int,
-        title: str,
-        active_sheet_id: typing.Optional[int] = OMIT,
-        tab_color: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, asset_id: str, sheet: Sheet, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[CreateNewSheetTabResponse]:
         """
         Create a new tab in an Athena spreadsheet.
@@ -835,17 +842,8 @@ class RawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        sheet_id : int
-            Sheet ID of the new tab
-
-        title : str
-            Title of the new tab
-
-        active_sheet_id : typing.Optional[int]
-            Currently active sheet ID
-
-        tab_color : typing.Optional[str]
-            Optional color of the new tab
+        sheet : Sheet
+            Sheet Specification
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -859,11 +857,8 @@ class RawSheetsClient:
             "api/v0/tools/sheets/tab/create",
             method="POST",
             json={
-                "active_sheet_id": active_sheet_id,
                 "asset_id": asset_id,
-                "sheet_id": sheet_id,
-                "tab_color": tab_color,
-                "title": title,
+                "sheet": convert_and_respect_annotation_metadata(object_=sheet, annotation=Sheet, direction="write"),
             },
             headers={
                 "content-type": "application/json",
@@ -1873,10 +1868,9 @@ class AsyncRawSheetsClient:
         asset_id: str,
         end_column: int,
         end_row: int,
+        formatting: CellFormat,
         start_column: int,
         start_row: int,
-        type: str,
-        value_json: str,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
@@ -1894,18 +1888,15 @@ class AsyncRawSheetsClient:
         end_row : int
             1-based ending row index
 
+        formatting : CellFormat
+            Cell format
+
         start_column : int
             1-based starting column index
 
         start_row : int
             1-based starting row index
 
-        type : str
-            Formatting type (e.g. backgroundColor, textFormat, borders, etc.)
-
-        value_json : str
-            JSON string for value param (e.g. '"#FF0000"' or '{"fontSize":12}')
-
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
 
@@ -1924,11 +1915,12 @@ class AsyncRawSheetsClient:
                 "asset_id": asset_id,
                 "end_column": end_column,
                 "end_row": end_row,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting, annotation=CellFormat, direction="write"
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "type": type,
-                "value_json": value_json,
             },
             headers={
                 "content-type": "application/json",
@@ -1968,7 +1960,8 @@ class AsyncRawSheetsClient:
         asset_id: str,
         start_column: int,
         start_row: int,
-        values: typing.Sequence[str],
+        values: typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+        formatting: typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]] = OMIT,
         sheet_id: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SheetOperationResponse]:
@@ -1986,8 +1979,11 @@ class AsyncRawSheetsClient:
         start_row : int
             1-based starting row index
 
-        values : typing.Sequence[str]
-            List of pipe-separated strings representing rows (e.g. ['A1|B1|C1', 'A2|B2|C2'])
+        values : typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]]
+            2D list of cells for each row
+
+        formatting : typing.Optional[typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]]]
+            Optional 2D list of cell formats matching the structure of values. Each row is a list of CellFormat objects for each cell in that row. Use None for cells without formatting. numberFormat is not required unless user explicity asked to change
 
         sheet_id : typing.Optional[int]
             Sheet ID (defaults to 1)
@@ -2005,10 +2001,19 @@ class AsyncRawSheetsClient:
             method="POST",
             json={
                 "asset_id": asset_id,
+                "formatting": convert_and_respect_annotation_metadata(
+                    object_=formatting,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[CellFormat]]],
+                    direction="write",
+                ),
                 "sheet_id": sheet_id,
                 "start_column": start_column,
                 "start_row": start_row,
-                "values": values,
+                "values": convert_and_respect_annotation_metadata(
+                    object_=values,
+                    annotation=typing.Sequence[typing.Sequence[typing.Optional[UpdateSheetRangeRequestValuesItemItem]]],
+                    direction="write",
+                ),
             },
             headers={
                 "content-type": "application/json",
@@ -2188,14 +2193,7 @@ class AsyncRawSheetsClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     async def create_tab(
-        self,
-        *,
-        asset_id: str,
-        sheet_id: int,
-        title: str,
-        active_sheet_id: typing.Optional[int] = OMIT,
-        tab_color: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, asset_id: str, sheet: Sheet, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[CreateNewSheetTabResponse]:
         """
         Create a new tab in an Athena spreadsheet.
@@ -2205,17 +2203,8 @@ class AsyncRawSheetsClient:
         asset_id : str
             The ID of the spreadsheet asset
 
-        sheet_id : int
-            Sheet ID of the new tab
-
-        title : str
-            Title of the new tab
-
-        active_sheet_id : typing.Optional[int]
-            Currently active sheet ID
-
-        tab_color : typing.Optional[str]
-            Optional color of the new tab
+        sheet : Sheet
+            Sheet Specification
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -2229,11 +2218,8 @@ class AsyncRawSheetsClient:
             "api/v0/tools/sheets/tab/create",
             method="POST",
             json={
-                "active_sheet_id": active_sheet_id,
                 "asset_id": asset_id,
-                "sheet_id": sheet_id,
-                "tab_color": tab_color,
-                "title": title,
+                "sheet": convert_and_respect_annotation_metadata(object_=sheet, annotation=Sheet, direction="write"),
             },
             headers={
                 "content-type": "application/json",

athena/tools/sheets/types/__init__.py

@@ -0,0 +1,36 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .update_sheet_range_request_values_item_item import UpdateSheetRangeRequestValuesItemItem
+_dynamic_imports: typing.Dict[str, str] = {
+    "UpdateSheetRangeRequestValuesItemItem": ".update_sheet_range_request_values_item_item"
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = ["UpdateSheetRangeRequestValuesItemItem"]

athena/tools/sheets/types/update_sheet_range_request_values_item_item.py

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

athena/tools/types/__init__.py

@@ -2,6 +2,33 @@
 
 # isort: skip_file
 
-from .tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
+_dynamic_imports: typing.Dict[str, str] = {"ToolsDataFrameRequestColumnsItem": ".tools_data_frame_request_columns_item"}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = ["ToolsDataFrameRequestColumnsItem"]

athena/types/__init__.py

@@ -2,57 +2,170 @@
 
 # isort: skip_file
 
-from .aop_async_execute_response_out import AopAsyncExecuteResponseOut
-from .aop_execute_request_in import AopExecuteRequestIn
-from .aop_execute_response_out import AopExecuteResponseOut
-from .asset_content_request_out import AssetContentRequestOut
-from .asset_node import AssetNode
-from .asset_screenshot_response_out import AssetScreenshotResponseOut
-from .chunk import Chunk
-from .chunk_content_item import ChunkContentItem, ChunkContentItem_ImageUrl, ChunkContentItem_Text
-from .chunk_result import ChunkResult
-from .chunk_result_chunk_id import ChunkResultChunkId
-from .content import Content
-from .conversation_asset_info import ConversationAssetInfo
-from .create_new_sheet_tab_response import CreateNewSheetTabResponse
-from .custom_agent_response import CustomAgentResponse
-from .data_frame_request_out import DataFrameRequestOut
-from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
-from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
-from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
-from .data_frame_unknown_format_error import DataFrameUnknownFormatError
-from .document_chunk import DocumentChunk
-from .drive_agent_response import DriveAgentResponse
-from .file_chunk_request_out import FileChunkRequestOut
-from .file_too_large_error import FileTooLargeError
-from .folder_response import FolderResponse
-from .general_agent_config import GeneralAgentConfig
-from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
-from .general_agent_request import GeneralAgentRequest
-from .general_agent_response import GeneralAgentResponse
-from .general_agent_response_message import GeneralAgentResponseMessage
-from .general_agent_response_message_kwargs import GeneralAgentResponseMessageKwargs
-from .get_table_response import GetTableResponse
-from .id import Id
-from .image_url_content import ImageUrlContent
-from .input_message import InputMessage
-from .input_message_content_item import (
-    InputMessageContentItem,
-    InputMessageContentItem_ImageUrl,
-    InputMessageContentItem_Text,
-)
-from .paginated_assets_out import PaginatedAssetsOut
-from .prompt_message import PromptMessage
-from .public_asset_out import PublicAssetOut
-from .research_agent_response import ResearchAgentResponse
-from .save_asset_request_out import SaveAssetRequestOut
-from .sheet_operation_response import SheetOperationResponse
-from .sql_agent_response import SqlAgentResponse
-from .structured_data_extractor_response import StructuredDataExtractorResponse
-from .table_row_data import TableRowData
-from .text_content import TextContent
-from .thread_status_response_out import ThreadStatusResponseOut
-from .type import Type
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .aop_async_execute_response_out import AopAsyncExecuteResponseOut
+    from .aop_execute_request_in import AopExecuteRequestIn
+    from .aop_execute_response_out import AopExecuteResponseOut
+    from .asset_content_request_out import AssetContentRequestOut
+    from .asset_node import AssetNode
+    from .asset_screenshot_response_out import AssetScreenshotResponseOut
+    from .backgroundcolor import Backgroundcolor
+    from .border_model import BorderModel
+    from .border_style import BorderStyle
+    from .borders_model import BordersModel
+    from .cell_format import CellFormat
+    from .cell_format_horizontal_alignment import CellFormatHorizontalAlignment
+    from .cell_format_vertical_alignment import CellFormatVerticalAlignment
+    from .chunk import Chunk
+    from .chunk_content_item import ChunkContentItem, ChunkContentItem_ImageUrl, ChunkContentItem_Text
+    from .chunk_result import ChunkResult
+    from .chunk_result_chunk_id import ChunkResultChunkId
+    from .color import Color
+    from .content import Content
+    from .conversation_asset_info import ConversationAssetInfo
+    from .conversation_message import ConversationMessage
+    from .conversation_result import ConversationResult
+    from .create_new_sheet_tab_response import CreateNewSheetTabResponse
+    from .custom_agent_response import CustomAgentResponse
+    from .data_frame_request_out import DataFrameRequestOut
+    from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
+    from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
+    from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
+    from .data_frame_unknown_format_error import DataFrameUnknownFormatError
+    from .document_chunk import DocumentChunk
+    from .drive_agent_response import DriveAgentResponse
+    from .file_chunk_request_out import FileChunkRequestOut
+    from .file_too_large_error import FileTooLargeError
+    from .folder_response import FolderResponse
+    from .general_agent_config import GeneralAgentConfig
+    from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
+    from .general_agent_request import GeneralAgentRequest
+    from .general_agent_response import GeneralAgentResponse
+    from .general_agent_response_message import GeneralAgentResponseMessage
+    from .general_agent_response_message_kwargs import GeneralAgentResponseMessageKwargs
+    from .get_table_response import GetTableResponse
+    from .id import Id
+    from .image_url_content import ImageUrlContent
+    from .input_message import InputMessage
+    from .input_message_content_item import (
+        InputMessageContentItem,
+        InputMessageContentItem_ImageUrl,
+        InputMessageContentItem_Text,
+    )
+    from .number_format_model import NumberFormatModel
+    from .number_format_type import NumberFormatType
+    from .paginated_assets_out import PaginatedAssetsOut
+    from .prompt_message import PromptMessage
+    from .public_asset_out import PublicAssetOut
+    from .research_agent_response import ResearchAgentResponse
+    from .save_asset_request_out import SaveAssetRequestOut
+    from .sheet import Sheet
+    from .sheet_operation_response import SheetOperationResponse
+    from .sql_agent_response import SqlAgentResponse
+    from .structured_data_extractor_response import StructuredDataExtractorResponse
+    from .table_row_data import TableRowData
+    from .text_content import TextContent
+    from .text_format_model import TextFormatModel
+    from .textrotation import Textrotation
+    from .theme_color import ThemeColor
+    from .thread_status_response_out import ThreadStatusResponseOut
+    from .type import Type
+    from .wrap_strategy import WrapStrategy
+_dynamic_imports: typing.Dict[str, str] = {
+    "AopAsyncExecuteResponseOut": ".aop_async_execute_response_out",
+    "AopExecuteRequestIn": ".aop_execute_request_in",
+    "AopExecuteResponseOut": ".aop_execute_response_out",
+    "AssetContentRequestOut": ".asset_content_request_out",
+    "AssetNode": ".asset_node",
+    "AssetScreenshotResponseOut": ".asset_screenshot_response_out",
+    "Backgroundcolor": ".backgroundcolor",
+    "BorderModel": ".border_model",
+    "BorderStyle": ".border_style",
+    "BordersModel": ".borders_model",
+    "CellFormat": ".cell_format",
+    "CellFormatHorizontalAlignment": ".cell_format_horizontal_alignment",
+    "CellFormatVerticalAlignment": ".cell_format_vertical_alignment",
+    "Chunk": ".chunk",
+    "ChunkContentItem": ".chunk_content_item",
+    "ChunkContentItem_ImageUrl": ".chunk_content_item",
+    "ChunkContentItem_Text": ".chunk_content_item",
+    "ChunkResult": ".chunk_result",
+    "ChunkResultChunkId": ".chunk_result_chunk_id",
+    "Color": ".color",
+    "Content": ".content",
+    "ConversationAssetInfo": ".conversation_asset_info",
+    "ConversationMessage": ".conversation_message",
+    "ConversationResult": ".conversation_result",
+    "CreateNewSheetTabResponse": ".create_new_sheet_tab_response",
+    "CustomAgentResponse": ".custom_agent_response",
+    "DataFrameRequestOut": ".data_frame_request_out",
+    "DataFrameRequestOutColumnsItem": ".data_frame_request_out_columns_item",
+    "DataFrameRequestOutDataItemItem": ".data_frame_request_out_data_item_item",
+    "DataFrameRequestOutIndexItem": ".data_frame_request_out_index_item",
+    "DataFrameUnknownFormatError": ".data_frame_unknown_format_error",
+    "DocumentChunk": ".document_chunk",
+    "DriveAgentResponse": ".drive_agent_response",
+    "FileChunkRequestOut": ".file_chunk_request_out",
+    "FileTooLargeError": ".file_too_large_error",
+    "FolderResponse": ".folder_response",
+    "GeneralAgentConfig": ".general_agent_config",
+    "GeneralAgentConfigEnabledToolsItem": ".general_agent_config_enabled_tools_item",
+    "GeneralAgentRequest": ".general_agent_request",
+    "GeneralAgentResponse": ".general_agent_response",
+    "GeneralAgentResponseMessage": ".general_agent_response_message",
+    "GeneralAgentResponseMessageKwargs": ".general_agent_response_message_kwargs",
+    "GetTableResponse": ".get_table_response",
+    "Id": ".id",
+    "ImageUrlContent": ".image_url_content",
+    "InputMessage": ".input_message",
+    "InputMessageContentItem": ".input_message_content_item",
+    "InputMessageContentItem_ImageUrl": ".input_message_content_item",
+    "InputMessageContentItem_Text": ".input_message_content_item",
+    "NumberFormatModel": ".number_format_model",
+    "NumberFormatType": ".number_format_type",
+    "PaginatedAssetsOut": ".paginated_assets_out",
+    "PromptMessage": ".prompt_message",
+    "PublicAssetOut": ".public_asset_out",
+    "ResearchAgentResponse": ".research_agent_response",
+    "SaveAssetRequestOut": ".save_asset_request_out",
+    "Sheet": ".sheet",
+    "SheetOperationResponse": ".sheet_operation_response",
+    "SqlAgentResponse": ".sql_agent_response",
+    "StructuredDataExtractorResponse": ".structured_data_extractor_response",
+    "TableRowData": ".table_row_data",
+    "TextContent": ".text_content",
+    "TextFormatModel": ".text_format_model",
+    "Textrotation": ".textrotation",
+    "ThemeColor": ".theme_color",
+    "ThreadStatusResponseOut": ".thread_status_response_out",
+    "Type": ".type",
+    "WrapStrategy": ".wrap_strategy",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = [
     "AopAsyncExecuteResponseOut",
@@ -61,14 +174,24 @@ __all__ = [
     "AssetContentRequestOut",
     "AssetNode",
     "AssetScreenshotResponseOut",
+    "Backgroundcolor",
+    "BorderModel",
+    "BorderStyle",
+    "BordersModel",
+    "CellFormat",
+    "CellFormatHorizontalAlignment",
+    "CellFormatVerticalAlignment",
     "Chunk",
     "ChunkContentItem",
     "ChunkContentItem_ImageUrl",
     "ChunkContentItem_Text",
     "ChunkResult",
     "ChunkResultChunkId",
+    "Color",
     "Content",
     "ConversationAssetInfo",
+    "ConversationMessage",
+    "ConversationResult",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",
@@ -94,16 +217,23 @@ __all__ = [
     "InputMessageContentItem",
     "InputMessageContentItem_ImageUrl",
     "InputMessageContentItem_Text",
+    "NumberFormatModel",
+    "NumberFormatType",
     "PaginatedAssetsOut",
     "PromptMessage",
     "PublicAssetOut",
     "ResearchAgentResponse",
     "SaveAssetRequestOut",
+    "Sheet",
     "SheetOperationResponse",
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
     "TableRowData",
     "TextContent",
+    "TextFormatModel",
+    "Textrotation",
+    "ThemeColor",
     "ThreadStatusResponseOut",
     "Type",
+    "WrapStrategy",
 ]

athena/types/aop_execute_response_out.py

@@ -4,6 +4,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .conversation_result import ConversationResult
 
 
 class AopExecuteResponseOut(UniversalBaseModel):
@@ -31,7 +32,7 @@ class AopExecuteResponseOut(UniversalBaseModel):
     Base prompt of the AOP before user inputs were added
     """
 
-    conversation: typing.Optional[typing.Optional[typing.Any]] = pydantic.Field(default=None)
+    conversation: typing.Optional[ConversationResult] = pydantic.Field(default=None)
     """
     The conversation result from the AOP execution
     """

athena/types/backgroundcolor.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .theme_color import ThemeColor
+
+Backgroundcolor = typing.Union[str, ThemeColor]