everyrow 0.1.9__py3-none-any.whl → 0.2.0__py3-none-any.whl

everyrow/generated/api/{default/get_task_status_endpoint_tasks_task_id_status_get.py → tasks/get_task_status_tasks_task_id_status_get.py}

@@ -7,6 +7,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
+from ...models.error_response import ErrorResponse
 from ...models.http_validation_error import HTTPValidationError
 from ...models.task_status_response import TaskStatusResponse
 from ...types import Response
@@ -27,12 +28,17 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> HTTPValidationError | TaskStatusResponse | None:
+) -> ErrorResponse | HTTPValidationError | TaskStatusResponse | None:
     if response.status_code == 200:
         response_200 = TaskStatusResponse.from_dict(response.json())
 
         return response_200
 
+    if response.status_code == 404:
+        response_404 = ErrorResponse.from_dict(response.json())
+
+        return response_404
+
     if response.status_code == 422:
         response_422 = HTTPValidationError.from_dict(response.json())
 
@@ -46,7 +52,7 @@ def _parse_response(
 
 def _build_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> Response[HTTPValidationError | TaskStatusResponse]:
+) -> Response[ErrorResponse | HTTPValidationError | TaskStatusResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -59,16 +65,10 @@ def sync_detailed(
     task_id: UUID,
     *,
     client: AuthenticatedClient,
-) -> Response[HTTPValidationError | TaskStatusResponse]:
-    """Get Task Status Endpoint
+) -> Response[ErrorResponse | HTTPValidationError | TaskStatusResponse]:
+    """Get task details
 
-     Get the status and artifact_id of a task by its ID.
-
-    Returns:
-        TaskStatusResponse containing the task's current status and artifact_id (if available).
-
-    Raises:
-        HTTPException: 404 if the task is not found.
+     Get the status and metadata of a task by its ID.
 
     Args:
         task_id (UUID):
@@ -78,7 +78,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[HTTPValidationError | TaskStatusResponse]
+        Response[ErrorResponse | HTTPValidationError | TaskStatusResponse]
     """
 
     kwargs = _get_kwargs(
@@ -96,16 +96,10 @@ def sync(
     task_id: UUID,
     *,
     client: AuthenticatedClient,
-) -> HTTPValidationError | TaskStatusResponse | None:
-    """Get Task Status Endpoint
-
-     Get the status and artifact_id of a task by its ID.
-
-    Returns:
-        TaskStatusResponse containing the task's current status and artifact_id (if available).
+) -> ErrorResponse | HTTPValidationError | TaskStatusResponse | None:
+    """Get task details
 
-    Raises:
-        HTTPException: 404 if the task is not found.
+     Get the status and metadata of a task by its ID.
 
     Args:
         task_id (UUID):
@@ -115,7 +109,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        HTTPValidationError | TaskStatusResponse
+        ErrorResponse | HTTPValidationError | TaskStatusResponse
     """
 
     return sync_detailed(
@@ -128,16 +122,10 @@ async def asyncio_detailed(
     task_id: UUID,
     *,
     client: AuthenticatedClient,
-) -> Response[HTTPValidationError | TaskStatusResponse]:
-    """Get Task Status Endpoint
-
-     Get the status and artifact_id of a task by its ID.
+) -> Response[ErrorResponse | HTTPValidationError | TaskStatusResponse]:
+    """Get task details
 
-    Returns:
-        TaskStatusResponse containing the task's current status and artifact_id (if available).
-
-    Raises:
-        HTTPException: 404 if the task is not found.
+     Get the status and metadata of a task by its ID.
 
     Args:
         task_id (UUID):
@@ -147,7 +135,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[HTTPValidationError | TaskStatusResponse]
+        Response[ErrorResponse | HTTPValidationError | TaskStatusResponse]
     """
 
     kwargs = _get_kwargs(
@@ -163,16 +151,10 @@ async def asyncio(
     task_id: UUID,
     *,
     client: AuthenticatedClient,
-) -> HTTPValidationError | TaskStatusResponse | None:
-    """Get Task Status Endpoint
+) -> ErrorResponse | HTTPValidationError | TaskStatusResponse | None:
+    """Get task details
 
-     Get the status and artifact_id of a task by its ID.
-
-    Returns:
-        TaskStatusResponse containing the task's current status and artifact_id (if available).
-
-    Raises:
-        HTTPException: 404 if the task is not found.
+     Get the status and metadata of a task by its ID.
 
     Args:
         task_id (UUID):
@@ -182,7 +164,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        HTTPValidationError | TaskStatusResponse
+        ErrorResponse | HTTPValidationError | TaskStatusResponse
     """
 
     return (

everyrow/generated/models/__init__.py

@@ -1,281 +1,97 @@
 """Contains all the data models used in inputs/outputs"""
 
-from .agent_improvement_instruction import AgentImprovementInstruction
-from .agent_query_params import AgentQueryParams
-from .agent_query_params_system_prompt_kind_type_0 import AgentQueryParamsSystemPromptKindType0
-from .agent_task_args import AgentTaskArgs
-from .agent_task_args_processing_mode import AgentTaskArgsProcessingMode
-from .allowed_suggestions import AllowedSuggestions
-from .api_key_info import APIKeyInfo
-from .artifact_changed_payload import ArtifactChangedPayload
-from .artifact_group_record import ArtifactGroupRecord
-from .artifact_group_record_analysis_type_0 import ArtifactGroupRecordAnalysisType0
-from .artifact_group_record_metadata_type_0 import ArtifactGroupRecordMetadataType0
-from .artifact_group_record_trace_mapping_type_0 import ArtifactGroupRecordTraceMappingType0
-from .artifact_status import ArtifactStatus
-from .auto_cohort_conversation_message import AutoCohortConversationMessage
-from .aux_data import AuxData
-from .aux_data_source_bank import AuxDataSourceBank
-from .chat_completion_message_tool_call import ChatCompletionMessageToolCall
-from .chat_message_metadata import ChatMessageMetadata
-from .concatenate_query_params import ConcatenateQueryParams
-from .concatenate_request import ConcatenateRequest
-from .continue_reason import ContinueReason
-from .continue_task_request import ContinueTaskRequest
-from .controller_improvement_round import ControllerImprovementRound
-from .conversation_changed_payload import ConversationChangedPayload
-from .copy_artifacts_request import CopyArtifactsRequest
-from .copy_artifacts_response import CopyArtifactsResponse
-from .copy_workflow_request import CopyWorkflowRequest
-from .copy_workflow_response import CopyWorkflowResponse
-from .create_api_key_request import CreateAPIKeyRequest
-from .create_api_key_response import CreateAPIKeyResponse
-from .create_group_query_params import CreateGroupQueryParams
-from .create_group_request import CreateGroupRequest
-from .create_query_params import CreateQueryParams
-from .create_request import CreateRequest
-from .create_session_request import CreateSessionRequest
-from .create_session_response import CreateSessionResponse
-from .create_workflow_from_artifact_request import CreateWorkflowFromArtifactRequest
-from .create_workflow_from_artifact_response import CreateWorkflowFromArtifactResponse
-from .data_frame_method import DataFrameMethod
-from .date_cutoffs import DateCutoffs
-from .dedupe_public_params import DedupePublicParams
-from .dedupe_request_params import DedupeRequestParams
-from .deep_merge_public_params import DeepMergePublicParams
-from .deep_merge_request import DeepMergeRequest
-from .deep_rank_public_params import DeepRankPublicParams
-from .deep_rank_request import DeepRankRequest
-from .deep_screen_public_params import DeepScreenPublicParams
-from .deep_screen_request import DeepScreenRequest
-from .derive_expression import DeriveExpression
-from .derive_query_params import DeriveQueryParams
-from .derive_request import DeriveRequest
-from .document_query_tool import DocumentQueryTool
-from .drop_columns_query_params import DropColumnsQueryParams
-from .drop_columns_request import DropColumnsRequest
-from .event_type import EventType
-from .execution_metadata import ExecutionMetadata
-from .export_request import ExportRequest
-from .export_request_token_data import ExportRequestTokenData
-from .export_to_google_sheets_export_post_response_export_to_google_sheets_export_post import (
-    ExportToGoogleSheetsExportPostResponseExportToGoogleSheetsExportPost,
-)
-from .filter_query_params import FilterQueryParams
-from .filter_request import FilterRequest
-from .flatten_query_params import FlattenQueryParams
-from .flatten_request import FlattenRequest
-from .generate_feedback_request import GenerateFeedbackRequest
-from .group_by_query_params import GroupByQueryParams
-from .group_by_request import GroupByRequest
-from .healthz_healthz_get_response_healthz_healthz_get import HealthzHealthzGetResponseHealthzHealthzGet
+from .agent_map_operation import AgentMapOperation
+from .agent_map_operation_input_type_1_item import AgentMapOperationInputType1Item
+from .agent_map_operation_input_type_2 import AgentMapOperationInputType2
+from .agent_map_operation_response_schema_type_0 import AgentMapOperationResponseSchemaType0
+from .billing_response import BillingResponse
+from .create_artifact_request import CreateArtifactRequest
+from .create_artifact_request_data_type_0_item import CreateArtifactRequestDataType0Item
+from .create_artifact_request_data_type_1 import CreateArtifactRequestDataType1
+from .create_artifact_response import CreateArtifactResponse
+from .create_session import CreateSession
+from .dedupe_operation import DedupeOperation
+from .dedupe_operation_input_type_1_item import DedupeOperationInputType1Item
+from .dedupe_operation_input_type_2 import DedupeOperationInputType2
+from .error_response import ErrorResponse
+from .error_response_details_type_0 import ErrorResponseDetailsType0
 from .http_validation_error import HTTPValidationError
-from .image_chat_content_part import ImageChatContentPart
-from .image_chat_content_part_image_url import ImageChatContentPartImageUrl
-from .import_from_google_sheets_import_post_response_import_from_google_sheets_import_post import (
-    ImportFromGoogleSheetsImportPostResponseImportFromGoogleSheetsImportPost,
-)
-from .import_request import ImportRequest
-from .import_request_token_data import ImportRequestTokenData
 from .insufficient_balance_error import InsufficientBalanceError
-from .join_query_params import JoinQueryParams
-from .join_request import JoinRequest
-from .llm_enum import LLMEnum
-from .map_agent_request_params import MapAgentRequestParams
-from .map_multi_agent_request_params import MapMultiAgentRequestParams
-from .message_created_payload import MessageCreatedPayload
-from .multi_agent_effort_level import MultiAgentEffortLevel
-from .multi_agent_query_params import MultiAgentQueryParams
-from .multi_modal_chat_message import MultiModalChatMessage
-from .multi_modal_chat_message_role import MultiModalChatMessageRole
-from .preview_metadata import PreviewMetadata
-from .processing_mode import ProcessingMode
-from .progress_status import ProgressStatus
-from .queue_stats import QueueStats
-from .reduce_agent_request_params import ReduceAgentRequestParams
-from .reduce_multi_agent_request_params import ReduceMultiAgentRequestParams
-from .resource_estimation_response import ResourceEstimationResponse
-from .response_schema_type import ResponseSchemaType
-from .revoke_api_key_response import RevokeAPIKeyResponse
-from .rollback_to_message_request import RollbackToMessageRequest
-from .rollback_to_message_response import RollbackToMessageResponse
-from .session_changed_payload import SessionChangedPayload
-from .simple_chat_message import SimpleChatMessage
-from .simple_chat_message_role import SimpleChatMessageRole
-from .simple_chat_message_with_tool_calls import SimpleChatMessageWithToolCalls
-from .source_database_entry import SourceDatabaseEntry
-from .standalone_artifact_record import StandaloneArtifactRecord
-from .standalone_artifact_record_analysis_type_0 import StandaloneArtifactRecordAnalysisType0
-from .standalone_artifact_record_metadata_type_0 import StandaloneArtifactRecordMetadataType0
-from .standalone_artifact_record_trace_mapping_type_0 import StandaloneArtifactRecordTraceMappingType0
-from .status_count import StatusCount
-from .status_count_status import StatusCountStatus
-from .submit_chat_task_body import SubmitChatTaskBody
-from .submit_chat_task_body_selected_task_type_type_0 import SubmitChatTaskBodySelectedTaskTypeType0
-from .submit_task_body import SubmitTaskBody
-from .task_changed_payload import TaskChangedPayload
-from .task_effort import TaskEffort
-from .task_id_request import TaskIdRequest
-from .task_insert import TaskInsert
-from .task_insert_query_params import TaskInsertQueryParams
-from .task_metadata import TaskMetadata
-from .task_metadata_cols_to_rename_type_0 import TaskMetadataColsToRenameType0
-from .task_response import TaskResponse
+from .llm_enum_public import LLMEnumPublic
+from .merge_operation import MergeOperation
+from .merge_operation_left_input_type_1_item import MergeOperationLeftInputType1Item
+from .merge_operation_left_input_type_2 import MergeOperationLeftInputType2
+from .merge_operation_right_input_type_1_item import MergeOperationRightInputType1Item
+from .merge_operation_right_input_type_2 import MergeOperationRightInputType2
+from .merge_operation_use_web_search_type_0 import MergeOperationUseWebSearchType0
+from .operation_response import OperationResponse
+from .public_effort_level import PublicEffortLevel
+from .public_task_type import PublicTaskType
+from .rank_operation import RankOperation
+from .rank_operation_input_type_1_item import RankOperationInputType1Item
+from .rank_operation_input_type_2 import RankOperationInputType2
+from .rank_operation_response_schema_type_0 import RankOperationResponseSchemaType0
+from .screen_operation import ScreenOperation
+from .screen_operation_input_type_1_item import ScreenOperationInputType1Item
+from .screen_operation_input_type_2 import ScreenOperationInputType2
+from .screen_operation_response_schema_type_0 import ScreenOperationResponseSchemaType0
+from .session_response import SessionResponse
+from .single_agent_operation import SingleAgentOperation
+from .single_agent_operation_input_type_1_item import SingleAgentOperationInputType1Item
+from .single_agent_operation_input_type_2 import SingleAgentOperationInputType2
+from .single_agent_operation_response_schema_type_0 import SingleAgentOperationResponseSchemaType0
+from .task_result_response import TaskResultResponse
+from .task_result_response_data_type_0_item import TaskResultResponseDataType0Item
+from .task_result_response_data_type_1 import TaskResultResponseDataType1
 from .task_status import TaskStatus
 from .task_status_response import TaskStatusResponse
-from .task_type import TaskType
-from .text_chat_content_part import TextChatContentPart
-from .tool_response_message import ToolResponseMessage
-from .toolkit_constants import ToolkitConstants
-from .trace_changed_payload import TraceChangedPayload
-from .trace_info import TraceInfo
-from .trigger_workflow_execution_request import TriggerWorkflowExecutionRequest
-from .trigger_workflow_execution_request_task_params import TriggerWorkflowExecutionRequestTaskParams
-from .trigger_workflow_execution_request_task_params_additional_property import (
-    TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty,
-)
-from .trigger_workflow_execution_response import TriggerWorkflowExecutionResponse
-from .upload_csv_payload import UploadCsvPayload
-from .upload_csv_query_params import UploadCsvQueryParams
-from .usage_response import UsageResponse
 from .validation_error import ValidationError
-from .whoami_whoami_get_response_whoami_whoami_get import WhoamiWhoamiGetResponseWhoamiWhoamiGet
-from .workflow_leaf_node_input import WorkflowLeafNodeInput
 
 __all__ = (
-    "AgentImprovementInstruction",
-    "AgentQueryParams",
-    "AgentQueryParamsSystemPromptKindType0",
-    "AgentTaskArgs",
-    "AgentTaskArgsProcessingMode",
-    "AllowedSuggestions",
-    "APIKeyInfo",
-    "ArtifactChangedPayload",
-    "ArtifactGroupRecord",
-    "ArtifactGroupRecordAnalysisType0",
-    "ArtifactGroupRecordMetadataType0",
-    "ArtifactGroupRecordTraceMappingType0",
-    "ArtifactStatus",
-    "AutoCohortConversationMessage",
-    "AuxData",
-    "AuxDataSourceBank",
-    "ChatCompletionMessageToolCall",
-    "ChatMessageMetadata",
-    "ConcatenateQueryParams",
-    "ConcatenateRequest",
-    "ContinueReason",
-    "ContinueTaskRequest",
-    "ControllerImprovementRound",
-    "ConversationChangedPayload",
-    "CopyArtifactsRequest",
-    "CopyArtifactsResponse",
-    "CopyWorkflowRequest",
-    "CopyWorkflowResponse",
-    "CreateAPIKeyRequest",
-    "CreateAPIKeyResponse",
-    "CreateGroupQueryParams",
-    "CreateGroupRequest",
-    "CreateQueryParams",
-    "CreateRequest",
-    "CreateSessionRequest",
-    "CreateSessionResponse",
-    "CreateWorkflowFromArtifactRequest",
-    "CreateWorkflowFromArtifactResponse",
-    "DataFrameMethod",
-    "DateCutoffs",
-    "DedupePublicParams",
-    "DedupeRequestParams",
-    "DeepMergePublicParams",
-    "DeepMergeRequest",
-    "DeepRankPublicParams",
-    "DeepRankRequest",
-    "DeepScreenPublicParams",
-    "DeepScreenRequest",
-    "DeriveExpression",
-    "DeriveQueryParams",
-    "DeriveRequest",
-    "DocumentQueryTool",
-    "DropColumnsQueryParams",
-    "DropColumnsRequest",
-    "EventType",
-    "ExecutionMetadata",
-    "ExportRequest",
-    "ExportRequestTokenData",
-    "ExportToGoogleSheetsExportPostResponseExportToGoogleSheetsExportPost",
-    "FilterQueryParams",
-    "FilterRequest",
-    "FlattenQueryParams",
-    "FlattenRequest",
-    "GenerateFeedbackRequest",
-    "GroupByQueryParams",
-    "GroupByRequest",
-    "HealthzHealthzGetResponseHealthzHealthzGet",
+    "AgentMapOperation",
+    "AgentMapOperationInputType1Item",
+    "AgentMapOperationInputType2",
+    "AgentMapOperationResponseSchemaType0",
+    "BillingResponse",
+    "CreateArtifactRequest",
+    "CreateArtifactRequestDataType0Item",
+    "CreateArtifactRequestDataType1",
+    "CreateArtifactResponse",
+    "CreateSession",
+    "DedupeOperation",
+    "DedupeOperationInputType1Item",
+    "DedupeOperationInputType2",
+    "ErrorResponse",
+    "ErrorResponseDetailsType0",
     "HTTPValidationError",
-    "ImageChatContentPart",
-    "ImageChatContentPartImageUrl",
-    "ImportFromGoogleSheetsImportPostResponseImportFromGoogleSheetsImportPost",
-    "ImportRequest",
-    "ImportRequestTokenData",
     "InsufficientBalanceError",
-    "JoinQueryParams",
-    "JoinRequest",
-    "LLMEnum",
-    "MapAgentRequestParams",
-    "MapMultiAgentRequestParams",
-    "MessageCreatedPayload",
-    "MultiAgentEffortLevel",
-    "MultiAgentQueryParams",
-    "MultiModalChatMessage",
-    "MultiModalChatMessageRole",
-    "PreviewMetadata",
-    "ProcessingMode",
-    "ProgressStatus",
-    "QueueStats",
-    "ReduceAgentRequestParams",
-    "ReduceMultiAgentRequestParams",
-    "ResourceEstimationResponse",
-    "ResponseSchemaType",
-    "RevokeAPIKeyResponse",
-    "RollbackToMessageRequest",
-    "RollbackToMessageResponse",
-    "SessionChangedPayload",
-    "SimpleChatMessage",
-    "SimpleChatMessageRole",
-    "SimpleChatMessageWithToolCalls",
-    "SourceDatabaseEntry",
-    "StandaloneArtifactRecord",
-    "StandaloneArtifactRecordAnalysisType0",
-    "StandaloneArtifactRecordMetadataType0",
-    "StandaloneArtifactRecordTraceMappingType0",
-    "StatusCount",
-    "StatusCountStatus",
-    "SubmitChatTaskBody",
-    "SubmitChatTaskBodySelectedTaskTypeType0",
-    "SubmitTaskBody",
-    "TaskChangedPayload",
-    "TaskEffort",
-    "TaskIdRequest",
-    "TaskInsert",
-    "TaskInsertQueryParams",
-    "TaskMetadata",
-    "TaskMetadataColsToRenameType0",
-    "TaskResponse",
+    "LLMEnumPublic",
+    "MergeOperation",
+    "MergeOperationLeftInputType1Item",
+    "MergeOperationLeftInputType2",
+    "MergeOperationRightInputType1Item",
+    "MergeOperationRightInputType2",
+    "MergeOperationUseWebSearchType0",
+    "OperationResponse",
+    "PublicEffortLevel",
+    "PublicTaskType",
+    "RankOperation",
+    "RankOperationInputType1Item",
+    "RankOperationInputType2",
+    "RankOperationResponseSchemaType0",
+    "ScreenOperation",
+    "ScreenOperationInputType1Item",
+    "ScreenOperationInputType2",
+    "ScreenOperationResponseSchemaType0",
+    "SessionResponse",
+    "SingleAgentOperation",
+    "SingleAgentOperationInputType1Item",
+    "SingleAgentOperationInputType2",
+    "SingleAgentOperationResponseSchemaType0",
+    "TaskResultResponse",
+    "TaskResultResponseDataType0Item",
+    "TaskResultResponseDataType1",
     "TaskStatus",
     "TaskStatusResponse",
-    "TaskType",
-    "TextChatContentPart",
-    "ToolkitConstants",
-    "ToolResponseMessage",
-    "TraceChangedPayload",
-    "TraceInfo",
-    "TriggerWorkflowExecutionRequest",
-    "TriggerWorkflowExecutionRequestTaskParams",
-    "TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty",
-    "TriggerWorkflowExecutionResponse",
-    "UploadCsvPayload",
-    "UploadCsvQueryParams",
-    "UsageResponse",
     "ValidationError",
-    "WhoamiWhoamiGetResponseWhoamiWhoamiGet",
-    "WorkflowLeafNodeInput",
 )