Skip to content

Commit 222fa85

Browse files
EvanWang2EvanWang2
committed
Update SDK: Generate v2 client with new annotation types
- Generated v2 client using openapi-generator-cli - Added support for AnnotationLabeledText, AnnotationRubricCriteria, and AnnotationRubricRating - Updated annotation.py with new type discrimination logic - Updated pre-commit configuration for Python 3.9 compatibility - All 103 files updated with latest Scale API models and endpoints
1 parent f5dba5b commit 222fa85

File tree

103 files changed

+6534
-812
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

103 files changed

+6534
-812
lines changed

‎.openapi-generator/FILES‎

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,17 @@ scaleapi/api_client/v2/models/annotation_file_properties.py
1818
scaleapi/api_client/v2/models/annotation_file_properties_value.py
1919
scaleapi/api_client/v2/models/annotation_integer.py
2020
scaleapi/api_client/v2/models/annotation_integer_properties.py
21+
scaleapi/api_client/v2/models/annotation_labeled_text.py
22+
scaleapi/api_client/v2/models/annotation_labeled_text_properties.py
2123
scaleapi/api_client/v2/models/annotation_metadata.py
2224
scaleapi/api_client/v2/models/annotation_ranked_choices.py
2325
scaleapi/api_client/v2/models/annotation_ranked_choices_properties.py
2426
scaleapi/api_client/v2/models/annotation_ranked_groups.py
2527
scaleapi/api_client/v2/models/annotation_ranked_groups_properties.py
28+
scaleapi/api_client/v2/models/annotation_rubric_criteria.py
29+
scaleapi/api_client/v2/models/annotation_rubric_criteria_properties.py
30+
scaleapi/api_client/v2/models/annotation_rubric_rating.py
31+
scaleapi/api_client/v2/models/annotation_rubric_rating_properties.py
2632
scaleapi/api_client/v2/models/annotation_text.py
2733
scaleapi/api_client/v2/models/annotation_text_properties.py
2834
scaleapi/api_client/v2/models/annotation_type.py
@@ -33,10 +39,16 @@ scaleapi/api_client/v2/models/base_annotation.py
3339
scaleapi/api_client/v2/models/base_chunk.py
3440
scaleapi/api_client/v2/models/basic_file.py
3541
scaleapi/api_client/v2/models/batch.py
42+
scaleapi/api_client/v2/models/batch_status.py
43+
scaleapi/api_client/v2/models/cancel_batch200_response.py
3644
scaleapi/api_client/v2/models/chunk.py
3745
scaleapi/api_client/v2/models/chunk_text.py
3846
scaleapi/api_client/v2/models/chunk_text_properties.py
3947
scaleapi/api_client/v2/models/content_and_url.py
48+
scaleapi/api_client/v2/models/create_batch_request.py
49+
scaleapi/api_client/v2/models/create_batch_request_project.py
50+
scaleapi/api_client/v2/models/criterion_definition.py
51+
scaleapi/api_client/v2/models/criterion_evaluation.py
4052
scaleapi/api_client/v2/models/delivery.py
4153
scaleapi/api_client/v2/models/detailed_file.py
4254
scaleapi/api_client/v2/models/error_detail.py
@@ -45,24 +57,37 @@ scaleapi/api_client/v2/models/expandable.py
4557
scaleapi/api_client/v2/models/expandable_annotation.py
4658
scaleapi/api_client/v2/models/expandable_batch.py
4759
scaleapi/api_client/v2/models/expandable_delivery.py
60+
scaleapi/api_client/v2/models/expandable_enum_batch.py
4861
scaleapi/api_client/v2/models/expandable_enum_deliveries.py
4962
scaleapi/api_client/v2/models/expandable_enum_delivery.py
5063
scaleapi/api_client/v2/models/expandable_enum_task.py
5164
scaleapi/api_client/v2/models/expandable_project.py
65+
scaleapi/api_client/v2/models/finalize_batch_request.py
66+
scaleapi/api_client/v2/models/gen_ai_project_type.py
67+
scaleapi/api_client/v2/models/get_batch500_response.py
68+
scaleapi/api_client/v2/models/get_batches_response.py
5269
scaleapi/api_client/v2/models/get_delivered_tasks_response.py
5370
scaleapi/api_client/v2/models/get_deliveries_response.py
54-
scaleapi/api_client/v2/models/get_task500_response.py
71+
scaleapi/api_client/v2/models/get_projects_response.py
5572
scaleapi/api_client/v2/models/get_tasks_response.py
5673
scaleapi/api_client/v2/models/image_file.py
74+
scaleapi/api_client/v2/models/labeled_text_value.py
5775
scaleapi/api_client/v2/models/message.py
5876
scaleapi/api_client/v2/models/message_content.py
5977
scaleapi/api_client/v2/models/message_role.py
6078
scaleapi/api_client/v2/models/model_parameters.py
6179
scaleapi/api_client/v2/models/option.py
80+
scaleapi/api_client/v2/models/pause_batch200_response.py
6281
scaleapi/api_client/v2/models/project.py
6382
scaleapi/api_client/v2/models/reasoning.py
6483
scaleapi/api_client/v2/models/reference_text.py
84+
scaleapi/api_client/v2/models/resume_batch200_response.py
85+
scaleapi/api_client/v2/models/rubric.py
86+
scaleapi/api_client/v2/models/rubric_criteria_value.py
87+
scaleapi/api_client/v2/models/rubric_evaluation.py
88+
scaleapi/api_client/v2/models/rubric_rating_value.py
6589
scaleapi/api_client/v2/models/sensitive_content_report.py
90+
scaleapi/api_client/v2/models/set_batch_metadata_request.py
6691
scaleapi/api_client/v2/models/submission.py
6792
scaleapi/api_client/v2/models/task.py
6893
scaleapi/api_client/v2/models/task_status.py

‎.pre-commit-config.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,4 +45,4 @@ repos:
4545
- "python_dateutil>=2.8.2"
4646
- "pydantic>=2"
4747
- "typing-extensions>=4.7.1"
48-
language_version: python3.8
48+
language_version: python3.9

‎scaleapi/api_client/v2/__init__.py‎

Lines changed: 56 additions & 81 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,14 @@
33
# flake8: noqa
44

55
"""
6-
GenAI API Spec
6+
GenAI API Spec
77
8-
Data Engine: Generative AI API Specification
8+
Data Engine: Generative AI API Specification
99
10-
The version of the OpenAPI document: 0.0.1
11-
Generated by OpenAPI Generator (https://openapi-generator.tech)
10+
The version of the OpenAPI document: 0.0.1
11+
Generated by OpenAPI Generator (https://openapi-generator.tech)
1212
13-
Do not edit the class manually.
13+
Do not edit the class manually.
1414
""" # noqa: E501
1515

1616

@@ -33,63 +33,47 @@
3333
# import models into sdk package
3434
from scaleapi.api_client.v2.models.annotation import Annotation
3535
from scaleapi.api_client.v2.models.annotation_boolean import AnnotationBoolean
36-
from scaleapi.api_client.v2.models.annotation_boolean_properties import (
37-
AnnotationBooleanProperties,
38-
)
36+
from scaleapi.api_client.v2.models.annotation_boolean_properties import AnnotationBooleanProperties
3937
from scaleapi.api_client.v2.models.annotation_category import AnnotationCategory
40-
from scaleapi.api_client.v2.models.annotation_category_multiple import (
41-
AnnotationCategoryMultiple,
42-
)
43-
from scaleapi.api_client.v2.models.annotation_category_multiple_properties import (
44-
AnnotationCategoryMultipleProperties,
45-
)
46-
from scaleapi.api_client.v2.models.annotation_category_properties import (
47-
AnnotationCategoryProperties,
48-
)
38+
from scaleapi.api_client.v2.models.annotation_category_multiple import AnnotationCategoryMultiple
39+
from scaleapi.api_client.v2.models.annotation_category_multiple_properties import AnnotationCategoryMultipleProperties
40+
from scaleapi.api_client.v2.models.annotation_category_properties import AnnotationCategoryProperties
4941
from scaleapi.api_client.v2.models.annotation_file import AnnotationFile
50-
from scaleapi.api_client.v2.models.annotation_file_properties import (
51-
AnnotationFileProperties,
52-
)
53-
from scaleapi.api_client.v2.models.annotation_file_properties_value import (
54-
AnnotationFilePropertiesValue,
55-
)
42+
from scaleapi.api_client.v2.models.annotation_file_properties import AnnotationFileProperties
43+
from scaleapi.api_client.v2.models.annotation_file_properties_value import AnnotationFilePropertiesValue
5644
from scaleapi.api_client.v2.models.annotation_integer import AnnotationInteger
57-
from scaleapi.api_client.v2.models.annotation_integer_properties import (
58-
AnnotationIntegerProperties,
59-
)
45+
from scaleapi.api_client.v2.models.annotation_integer_properties import AnnotationIntegerProperties
46+
from scaleapi.api_client.v2.models.annotation_labeled_text import AnnotationLabeledText
47+
from scaleapi.api_client.v2.models.annotation_labeled_text_properties import AnnotationLabeledTextProperties
6048
from scaleapi.api_client.v2.models.annotation_metadata import AnnotationMetadata
61-
from scaleapi.api_client.v2.models.annotation_ranked_choices import (
62-
AnnotationRankedChoices,
63-
)
64-
from scaleapi.api_client.v2.models.annotation_ranked_choices_properties import (
65-
AnnotationRankedChoicesProperties,
66-
)
67-
from scaleapi.api_client.v2.models.annotation_ranked_groups import (
68-
AnnotationRankedGroups,
69-
)
70-
from scaleapi.api_client.v2.models.annotation_ranked_groups_properties import (
71-
AnnotationRankedGroupsProperties,
72-
)
49+
from scaleapi.api_client.v2.models.annotation_ranked_choices import AnnotationRankedChoices
50+
from scaleapi.api_client.v2.models.annotation_ranked_choices_properties import AnnotationRankedChoicesProperties
51+
from scaleapi.api_client.v2.models.annotation_ranked_groups import AnnotationRankedGroups
52+
from scaleapi.api_client.v2.models.annotation_ranked_groups_properties import AnnotationRankedGroupsProperties
53+
from scaleapi.api_client.v2.models.annotation_rubric_criteria import AnnotationRubricCriteria
54+
from scaleapi.api_client.v2.models.annotation_rubric_criteria_properties import AnnotationRubricCriteriaProperties
55+
from scaleapi.api_client.v2.models.annotation_rubric_rating import AnnotationRubricRating
56+
from scaleapi.api_client.v2.models.annotation_rubric_rating_properties import AnnotationRubricRatingProperties
7357
from scaleapi.api_client.v2.models.annotation_text import AnnotationText
74-
from scaleapi.api_client.v2.models.annotation_text_properties import (
75-
AnnotationTextProperties,
76-
)
58+
from scaleapi.api_client.v2.models.annotation_text_properties import AnnotationTextProperties
7759
from scaleapi.api_client.v2.models.annotation_type import AnnotationType
78-
from scaleapi.api_client.v2.models.annotation_workspace_container import (
79-
AnnotationWorkspaceContainer,
80-
)
81-
from scaleapi.api_client.v2.models.annotation_workspace_container_properties import (
82-
AnnotationWorkspaceContainerProperties,
83-
)
60+
from scaleapi.api_client.v2.models.annotation_workspace_container import AnnotationWorkspaceContainer
61+
from scaleapi.api_client.v2.models.annotation_workspace_container_properties import AnnotationWorkspaceContainerProperties
8462
from scaleapi.api_client.v2.models.audio_file import AudioFile
8563
from scaleapi.api_client.v2.models.base_annotation import BaseAnnotation
8664
from scaleapi.api_client.v2.models.base_chunk import BaseChunk
8765
from scaleapi.api_client.v2.models.basic_file import BasicFile
8866
from scaleapi.api_client.v2.models.batch import Batch
67+
from scaleapi.api_client.v2.models.batch_status import BatchStatus
68+
from scaleapi.api_client.v2.models.cancel_batch200_response import CancelBatch200Response
8969
from scaleapi.api_client.v2.models.chunk import Chunk
9070
from scaleapi.api_client.v2.models.chunk_text import ChunkText
9171
from scaleapi.api_client.v2.models.chunk_text_properties import ChunkTextProperties
9272
from scaleapi.api_client.v2.models.content_and_url import ContentAndUrl
73+
from scaleapi.api_client.v2.models.create_batch_request import CreateBatchRequest
74+
from scaleapi.api_client.v2.models.create_batch_request_project import CreateBatchRequestProject
75+
from scaleapi.api_client.v2.models.criterion_definition import CriterionDefinition
76+
from scaleapi.api_client.v2.models.criterion_evaluation import CriterionEvaluation
9377
from scaleapi.api_client.v2.models.delivery import Delivery
9478
from scaleapi.api_client.v2.models.detailed_file import DetailedFile
9579
from scaleapi.api_client.v2.models.error_detail import ErrorDetail
@@ -98,56 +82,47 @@
9882
from scaleapi.api_client.v2.models.expandable_annotation import ExpandableAnnotation
9983
from scaleapi.api_client.v2.models.expandable_batch import ExpandableBatch
10084
from scaleapi.api_client.v2.models.expandable_delivery import ExpandableDelivery
101-
from scaleapi.api_client.v2.models.expandable_enum_deliveries import (
102-
ExpandableEnumDeliveries,
103-
)
104-
from scaleapi.api_client.v2.models.expandable_enum_delivery import (
105-
ExpandableEnumDelivery,
106-
)
85+
from scaleapi.api_client.v2.models.expandable_enum_batch import ExpandableEnumBatch
86+
from scaleapi.api_client.v2.models.expandable_enum_deliveries import ExpandableEnumDeliveries
87+
from scaleapi.api_client.v2.models.expandable_enum_delivery import ExpandableEnumDelivery
10788
from scaleapi.api_client.v2.models.expandable_enum_task import ExpandableEnumTask
10889
from scaleapi.api_client.v2.models.expandable_project import ExpandableProject
109-
from scaleapi.api_client.v2.models.get_delivered_tasks_response import (
110-
GetDeliveredTasksResponse,
111-
)
90+
from scaleapi.api_client.v2.models.finalize_batch_request import FinalizeBatchRequest
91+
from scaleapi.api_client.v2.models.gen_ai_project_type import GenAIProjectType
92+
from scaleapi.api_client.v2.models.get_batch500_response import GetBatch500Response
93+
from scaleapi.api_client.v2.models.get_batches_response import GetBatchesResponse
94+
from scaleapi.api_client.v2.models.get_delivered_tasks_response import GetDeliveredTasksResponse
11295
from scaleapi.api_client.v2.models.get_deliveries_response import GetDeliveriesResponse
113-
from scaleapi.api_client.v2.models.get_task500_response import GetTask500Response
96+
from scaleapi.api_client.v2.models.get_projects_response import GetProjectsResponse
11497
from scaleapi.api_client.v2.models.get_tasks_response import GetTasksResponse
11598
from scaleapi.api_client.v2.models.image_file import ImageFile
99+
from scaleapi.api_client.v2.models.labeled_text_value import LabeledTextValue
116100
from scaleapi.api_client.v2.models.message import Message
117101
from scaleapi.api_client.v2.models.message_content import MessageContent
118102
from scaleapi.api_client.v2.models.message_role import MessageRole
119103
from scaleapi.api_client.v2.models.model_parameters import ModelParameters
120104
from scaleapi.api_client.v2.models.option import Option
105+
from scaleapi.api_client.v2.models.pause_batch200_response import PauseBatch200Response
121106
from scaleapi.api_client.v2.models.project import Project
122107
from scaleapi.api_client.v2.models.reasoning import Reasoning
123108
from scaleapi.api_client.v2.models.reference_text import ReferenceText
124-
from scaleapi.api_client.v2.models.sensitive_content_report import (
125-
SensitiveContentReport,
126-
)
109+
from scaleapi.api_client.v2.models.resume_batch200_response import ResumeBatch200Response
110+
from scaleapi.api_client.v2.models.rubric import Rubric
111+
from scaleapi.api_client.v2.models.rubric_criteria_value import RubricCriteriaValue
112+
from scaleapi.api_client.v2.models.rubric_evaluation import RubricEvaluation
113+
from scaleapi.api_client.v2.models.rubric_rating_value import RubricRatingValue
114+
from scaleapi.api_client.v2.models.sensitive_content_report import SensitiveContentReport
115+
from scaleapi.api_client.v2.models.set_batch_metadata_request import SetBatchMetadataRequest
127116
from scaleapi.api_client.v2.models.submission import Submission
128117
from scaleapi.api_client.v2.models.task import Task
129118
from scaleapi.api_client.v2.models.task_status import TaskStatus
130119
from scaleapi.api_client.v2.models.thread import Thread
131120
from scaleapi.api_client.v2.models.turn import Turn
132-
from scaleapi.api_client.v2.models.v1_task_task_id_get200_response import (
133-
V1TaskTaskIdGet200Response,
134-
)
135-
from scaleapi.api_client.v2.models.v1_task_task_id_get200_response_response import (
136-
V1TaskTaskIdGet200ResponseResponse,
137-
)
138-
from scaleapi.api_client.v2.models.workspace_container_config import (
139-
WorkspaceContainerConfig,
140-
)
141-
from scaleapi.api_client.v2.models.workspace_container_value import (
142-
WorkspaceContainerValue,
143-
)
144-
from scaleapi.api_client.v2.models.workspace_execution_data import (
145-
WorkspaceExecutionData,
146-
)
147-
from scaleapi.api_client.v2.models.workspace_execution_data_result import (
148-
WorkspaceExecutionDataResult,
149-
)
150-
from scaleapi.api_client.v2.models.workspace_execution_data_result_status import (
151-
WorkspaceExecutionDataResultStatus,
152-
)
121+
from scaleapi.api_client.v2.models.v1_task_task_id_get200_response import V1TaskTaskIdGet200Response
122+
from scaleapi.api_client.v2.models.v1_task_task_id_get200_response_response import V1TaskTaskIdGet200ResponseResponse
123+
from scaleapi.api_client.v2.models.workspace_container_config import WorkspaceContainerConfig
124+
from scaleapi.api_client.v2.models.workspace_container_value import WorkspaceContainerValue
125+
from scaleapi.api_client.v2.models.workspace_execution_data import WorkspaceExecutionData
126+
from scaleapi.api_client.v2.models.workspace_execution_data_result import WorkspaceExecutionDataResult
127+
from scaleapi.api_client.v2.models.workspace_execution_data_result_status import WorkspaceExecutionDataResultStatus
153128
from scaleapi.api_client.v2.models.workspace_file import WorkspaceFile

‎scaleapi/api_client/v2/api/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,4 @@
22

33
# import apis into api package
44
from scaleapi.api_client.v2.api.v2_api import V2Api
5+

0 commit comments

Comments
 (0)