diff --git a/openapi-client-config.yaml b/openapi-client-config.yaml index c4834eef2..becde9a43 100644 --- a/openapi-client-config.yaml +++ b/openapi-client-config.yaml @@ -1,3 +1,5 @@ post_hooks: + # Fix missing status codes in generated API files + - 'python ../../../scripts/patch_status_codes.py "../../../src/galileo/resources/api"' # Commenting this out for now as it's not working # - 'python ../../../scripts/patch_http_validation_error.py "../../../src/galileo/resources/models/http_validation_error.py"' diff --git a/openapi.yaml b/openapi.yaml index 17a19c0ee..d787e5e38 100644 --- a/openapi.yaml +++ b/openapi.yaml @@ -1782,6 +1782,74 @@ components: - end_bucket_time title: BucketedMetrics type: object + BulkDeleteDatasetsRequest: + description: Request to delete multiple datasets. + properties: + dataset_ids: + items: + format: uuid4 + type: string + maxItems: 100 + minItems: 1 + title: Dataset Ids + type: array + required: + - dataset_ids + title: BulkDeleteDatasetsRequest + type: object + BulkDeleteDatasetsResponse: + description: Response from bulk deletion operation. + properties: + deleted_count: + title: Deleted Count + type: integer + failed_deletions: + items: + $ref: '#/components/schemas/api__schemas__content__dataset__BulkDeleteFailure' + title: Failed Deletions + type: array + message: + title: Message + type: string + required: + - deleted_count + - message + title: BulkDeleteDatasetsResponse + type: object + BulkDeletePromptTemplatesRequest: + description: Request to delete multiple prompt templates. + properties: + template_ids: + items: + format: uuid4 + type: string + maxItems: 100 + minItems: 1 + title: Template Ids + type: array + required: + - template_ids + title: BulkDeletePromptTemplatesRequest + type: object + BulkDeletePromptTemplatesResponse: + description: Response from bulk deletion operation. + properties: + deleted_count: + title: Deleted Count + type: integer + failed_deletions: + items: + $ref: '#/components/schemas/api__schemas__content__prompt__BulkDeleteFailure' + title: Failed Deletions + type: array + message: + title: Message + type: string + required: + - deleted_count + - message + title: BulkDeletePromptTemplatesResponse + type: object ChainPollTemplate: description: 'Template for a chainpoll metric prompt, @@ -2392,6 +2460,11 @@ components: type: string type: array title: Labels + luna_model: + anyOf: + - type: string + - type: 'null' + title: Luna Model metric_critique_configuration: anyOf: - $ref: '#/components/schemas/MetricCritiqueJobConfiguration' @@ -2696,6 +2769,11 @@ components: link: title: Link type: string + luna_model: + anyOf: + - type: string + - type: 'null' + title: Luna Model message: title: Message type: string @@ -3109,6 +3187,11 @@ components: type: string title: Tags type: array + user_prompt: + anyOf: + - type: string + - type: 'null' + title: User Prompt required: - name - scorer_type @@ -7524,6 +7607,15 @@ components: - message title: DeletePromptResponse type: object + DeleteRunResponse: + properties: + message: + title: Message + type: string + required: + - message + title: DeleteRunResponse + type: object DeleteScorerResponse: properties: message: @@ -11014,27 +11106,22 @@ components: FeedbackRatingInfo: properties: explanation: + anyOf: + - type: string + - type: 'null' title: Explanation - type: string feedback_type: - enum: - - like_dislike - - star - - score - - tags - - text - title: Feedback Type - type: string + $ref: '#/components/schemas/FeedbackType' value: anyOf: - type: boolean - type: integer + - type: string - items: maxLength: 255 minLength: 1 type: string type: array - - type: string title: Value required: - feedback_type @@ -11042,6 +11129,15 @@ components: - explanation title: FeedbackRatingInfo type: object + FeedbackType: + enum: + - like_dislike + - star + - score + - tags + - text + title: FeedbackType + type: string FewShotExample: description: Few-shot example for a chainpoll metric prompt. properties: @@ -12969,12 +13065,12 @@ components: examples: - columns: - applicable_types: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -12989,12 +13085,12 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13009,12 +13105,12 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13029,12 +13125,12 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13049,11 +13145,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13068,11 +13164,11 @@ components: multi_valued: true sortable: false - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13088,12 +13184,12 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13108,11 +13204,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13127,11 +13223,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13146,12 +13242,12 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13166,11 +13262,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13186,11 +13282,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13206,11 +13302,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13226,11 +13322,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13245,11 +13341,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13264,11 +13360,11 @@ components: multi_valued: false sortable: true - applicable_types: - - trace - - workflow - - agent - retriever + - agent - tool + - workflow + - trace - llm category: standard complex: false @@ -13283,18 +13379,18 @@ components: multi_valued: false sortable: true - allowed_values: - - trace - - workflow - - agent - session - retriever + - agent - tool + - workflow + - trace - llm applicable_types: - - workflow - - agent - retriever + - agent - tool + - workflow - llm category: standard complex: false @@ -13309,10 +13405,10 @@ components: multi_valued: false sortable: true - applicable_types: - - workflow - - agent - retriever + - agent - tool + - workflow - llm category: standard complex: false @@ -13328,10 +13424,10 @@ components: multi_valued: false sortable: true - applicable_types: - - workflow - - agent - retriever + - agent - tool + - workflow - llm category: standard complex: false @@ -13346,10 +13442,10 @@ components: multi_valued: false sortable: true - applicable_types: - - workflow - - agent - retriever + - agent - tool + - workflow - llm category: standard complex: false @@ -13364,13 +13460,13 @@ components: multi_valued: false sortable: true - allowed_values: + - classifier + - default + - router - reflection - planner - supervisor - react - - router - - classifier - - default - judge applicable_types: - agent @@ -13543,6 +13639,67 @@ components: - value title: LogRecordsDateFilter type: object + LogRecordsDeleteRequest: + example: + filters: + - case_sensitive: true + name: input + operator: eq + type: text + value: example input + log_stream_id: 74aec44e-ec21-4c9f-a3e2-b2ab2b81b4db + properties: + experiment_id: + anyOf: + - format: uuid4 + type: string + - type: 'null' + description: Experiment id associated with the traces. + title: Experiment Id + filters: + items: + discriminator: + mapping: + boolean: '#/components/schemas/LogRecordsBooleanFilter' + date: '#/components/schemas/LogRecordsDateFilter' + id: '#/components/schemas/LogRecordsIDFilter' + number: '#/components/schemas/LogRecordsNumberFilter' + text: '#/components/schemas/LogRecordsTextFilter' + propertyName: type + oneOf: + - $ref: '#/components/schemas/LogRecordsIDFilter' + - $ref: '#/components/schemas/LogRecordsDateFilter' + - $ref: '#/components/schemas/LogRecordsNumberFilter' + - $ref: '#/components/schemas/LogRecordsBooleanFilter' + - $ref: '#/components/schemas/LogRecordsTextFilter' + title: Filters + type: array + log_stream_id: + anyOf: + - format: uuid4 + type: string + - type: 'null' + description: Log stream id associated with the traces. + title: Log Stream Id + metrics_testing_id: + anyOf: + - format: uuid4 + type: string + - type: 'null' + description: Metrics testing id associated with the traces. + title: Metrics Testing Id + title: LogRecordsDeleteRequest + type: object + LogRecordsDeleteResponse: + properties: + message: + description: Message + title: Message + type: string + required: + - message + title: LogRecordsDeleteResponse + type: object LogRecordsExportRequest: description: Request schema for exporting log records (sessions, traces, spans). examples: @@ -13888,7 +14045,7 @@ components: num_rows: 2 paginated: true records: - - created_at: '2025-09-04T23:59:25.923814Z' + - created_at: '2025-09-24T18:07:03.707386Z' dataset_metadata: {} feedback_rating_info: {} id: 1a0939d1-8b43-4fe3-a91c-196e2d9847e3 @@ -13905,7 +14062,7 @@ components: trace_id: 1a0939d1-8b43-4fe3-a91c-196e2d9847e3 type: trace user_metadata: {} - - created_at: '2025-09-04T23:59:25.924048Z' + - created_at: '2025-09-24T18:07:03.707533Z' dataset_metadata: {} id: 1edc2401-24b4-448c-a392-b36932293061 input: @@ -14171,14 +14328,14 @@ components: - log_stream_id: 00000000-0000-0000-0000-000000000000 parent_id: 11000011-0000-0000-0000-110000110000 spans: - - created_at: '2025-09-04T23:59:25.885822Z' + - created_at: '2025-09-24T18:07:03.679788Z' dataset_metadata: {} input: who is a smart LLM? metrics: {} name: '' output: I am! spans: - - created_at: '2025-09-04T23:59:25.861579Z' + - created_at: '2025-09-24T18:07:03.663157Z' dataset_metadata: {} id: 22222222-2222-4222-a222-222222222222 input: @@ -14533,14 +14690,14 @@ components: - log_stream_id: 00000000-0000-0000-0000-000000000000 session_id: 00000000-0000-0000-0000-000000000000 traces: - - created_at: '2025-09-04T23:59:25.851577Z' + - created_at: '2025-09-24T18:07:03.655609Z' dataset_metadata: {} input: who is a smart LLM? metrics: {} name: '' output: I am! spans: - - created_at: '2025-09-04T23:59:25.851522Z' + - created_at: '2025-09-24T18:07:03.655559Z' dataset_metadata: {} input: - content: 'Question: who is a smart LLM?' @@ -14558,14 +14715,14 @@ components: user_metadata: {} - experiment_id: 00000000-0000-0000-0000-000000000000 traces: - - created_at: '2025-09-04T23:59:25.852372Z' + - created_at: '2025-09-24T18:07:03.656367Z' dataset_metadata: {} input: who is a smart LLM? metrics: {} name: '' output: I am! spans: - - created_at: '2025-09-04T23:59:25.852327Z' + - created_at: '2025-09-24T18:07:03.656296Z' dataset_metadata: {} id: 11111111-1111-4111-a111-111111111111 input: @@ -15559,6 +15716,7 @@ components: - discrete - freeform - percentage + - multilabel title: OutputTypeEnum type: string OverrideAction: @@ -16650,7 +16808,7 @@ components: title: Max Tokens type: integer model_alias: - default: gpt-5-mini + default: GPT-4o title: Model Alias type: string n: @@ -17912,27 +18070,46 @@ components: - scorers title: RunScorerSettingsResponse type: object - RunTagDB: + RunTagCreateRequest: properties: - created_at: - format: date-time - title: Created At - type: string - created_by: - format: uuid4 - title: Created By - type: string - id: - format: uuid4 - title: Id - type: string key: maxLength: 256 title: Key type: string - project_id: - format: uuid4 - title: Project Id + tag_type: + title: Tag Type + type: string + value: + maxLength: 256 + title: Value + type: string + required: + - key + - value + - tag_type + title: RunTagCreateRequest + type: object + RunTagDB: + properties: + created_at: + format: date-time + title: Created At + type: string + created_by: + format: uuid4 + title: Created By + type: string + id: + format: uuid4 + title: Id + type: string + key: + maxLength: 256 + title: Key + type: string + project_id: + format: uuid4 + title: Project Id type: string run_id: format: uuid4 @@ -18274,10 +18451,6 @@ components: type: object ScorerResponse: properties: - chain_poll_template: - anyOf: - - $ref: '#/components/schemas/ChainPollTemplate' - - type: 'null' created_at: anyOf: - format: date-time @@ -18334,11 +18507,6 @@ components: anyOf: - $ref: '#/components/schemas/BaseScorerVersionDB' - type: 'null' - metric_name: - anyOf: - - type: string - - type: 'null' - title: Metric Name model_type: anyOf: - $ref: '#/components/schemas/ModelType' @@ -18359,6 +18527,11 @@ components: type: string - type: 'null' title: Updated At + user_prompt: + anyOf: + - type: string + - type: 'null' + title: User Prompt required: - id - name @@ -20134,6 +20307,11 @@ components: type: array - type: 'null' title: Tags + user_prompt: + anyOf: + - type: string + - type: 'null' + title: User Prompt title: UpdateScorerRequest type: object UpsertDatasetContentRequest: @@ -20608,6 +20786,40 @@ components: type: object title: WorkflowSpan type: object + api__schemas__content__dataset__BulkDeleteFailure: + description: Details about a failed deletion. + properties: + dataset_id: + format: uuid4 + title: Dataset Id + type: string + dataset_name: + title: Dataset Name + type: string + reason: + title: Reason + type: string + required: + - dataset_id + - dataset_name + - reason + title: BulkDeleteFailure + type: object + api__schemas__content__prompt__BulkDeleteFailure: + description: Details about a failed deletion. + properties: + reason: + title: Reason + type: string + template_id: + format: uuid4 + title: Template Id + type: string + required: + - template_id + - reason + title: BulkDeleteFailure + type: object api__schemas__project__GetProjectsPaginatedResponse: properties: limit: @@ -20965,6 +21177,45 @@ paths: summary: Create Dataset tags: - datasets + /datasets/bulk_delete: + delete: + description: "Delete multiple datasets in bulk.\n\nThis endpoint allows efficient\ + \ deletion of multiple datasets at once.\nIt validates permissions for each\ + \ dataset in the service and provides detailed feedback about\nsuccessful\ + \ and failed deletions for each dataset.\n\nParameters\n----------\ndelete_request\ + \ : BulkDeleteDatasetsRequest\n Request containing list of dataset IDs\ + \ to delete (max 100)\nctx : Context\n Request context including authentication\ + \ information\n\nReturns\n-------\nBulkDeleteDatasetsResponse\n Details\ + \ about the bulk deletion operation including:\n - Number of successfully\ + \ deleted datasets\n - List of failed deletions with reasons\n - Summary\ + \ message" + operationId: bulk_delete_datasets_datasets_bulk_delete_delete + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/BulkDeleteDatasetsRequest' + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/BulkDeleteDatasetsResponse' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Bulk Delete Datasets + tags: + - datasets /datasets/extend: post: description: Extends the dataset content @@ -22790,6 +23041,242 @@ paths: summary: Get Experiment Metrics tags: - experiment + /projects/{project_id}/experiments/{experiment_id}/tags: + get: + description: Gets tags for a given project_id/experiment_id. + operationId: get_experiment_tags_projects__project_id__experiments__experiment_id__tags_get + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + - in: path + name: experiment_id + required: true + schema: + format: uuid4 + title: Experiment Id + type: string + responses: + '200': + content: + application/json: + schema: + items: + $ref: '#/components/schemas/RunTagDB' + title: Response Get Experiment Tags Projects Project Id Experiments Experiment + Id Tags Get + type: array + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Get Experiment Tags + tags: + - experiment_tags + post: + description: Sets a tag for an experiment. + operationId: set_tag_for_experiment_projects__project_id__experiments__experiment_id__tags_post + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + - in: path + name: experiment_id + required: true + schema: + format: uuid4 + title: Experiment Id + type: string + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/RunTagCreateRequest' + examples: + - key: Vector DB + tag_type: rag + value: Pinecone + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/RunTagDB' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Set Tag For Experiment + tags: + - experiment_tags + /projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}: + delete: + operationId: delete_experiment_tag_projects__project_id__experiments__experiment_id__tags__tag_id__delete + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + - in: path + name: experiment_id + required: true + schema: + format: uuid4 + title: Experiment Id + type: string + - in: path + name: tag_id + required: true + schema: + format: uuid4 + title: Tag Id + type: string + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/DeleteRunResponse' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Delete Experiment Tag + tags: + - experiment_tags + get: + description: Gets a tag for a given project_id/experiment_id. + operationId: get_experiment_tag_projects__project_id__experiments__experiment_id__tags__tag_id__get + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + - in: path + name: experiment_id + required: true + schema: + format: uuid4 + title: Experiment Id + type: string + - in: path + name: tag_id + required: true + schema: + format: uuid4 + title: Tag Id + type: string + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/RunTagDB' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Get Experiment Tag + tags: + - experiment_tags + put: + description: Sets or updates a tag for an experiment. + operationId: update_tag_for_experiment_projects__project_id__experiments__experiment_id__tags__tag_id__put + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + - in: path + name: experiment_id + required: true + schema: + format: uuid4 + title: Experiment Id + type: string + - in: path + name: tag_id + required: true + schema: + format: uuid4 + title: Tag Id + type: string + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/RunTagCreateRequest' + examples: + - key: Vector DB + tag_type: rag + value: Pinecone + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/RunTagDB' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Update Tag For Experiment + tags: + - experiment_tags /projects/{project_id}/export_records: post: operationId: export_records_projects__project_id__export_records_post @@ -23837,6 +24324,44 @@ paths: summary: Sessions Available Columns tags: - trace + /projects/{project_id}/sessions/delete: + post: + description: Delete all session records that match the provided filters. + operationId: delete_sessions_projects__project_id__sessions_delete_post + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/LogRecordsDeleteRequest' + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/LogRecordsDeleteResponse' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Delete Sessions + tags: + - trace /projects/{project_id}/sessions/search: post: operationId: query_sessions_projects__project_id__sessions_search_post @@ -23986,6 +24511,44 @@ paths: summary: Spans Available Columns tags: - trace + /projects/{project_id}/spans/delete: + post: + description: Delete all span records that match the provided filters. + operationId: delete_spans_projects__project_id__spans_delete_post + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/LogRecordsDeleteRequest' + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/LogRecordsDeleteResponse' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Delete Spans + tags: + - trace /projects/{project_id}/spans/search: post: operationId: query_spans_projects__project_id__spans_search_post @@ -24776,6 +25339,44 @@ paths: summary: Traces Available Columns tags: - trace + /projects/{project_id}/traces/delete: + post: + description: Delete all trace records that match the provided filters. + operationId: delete_traces_projects__project_id__traces_delete_post + parameters: + - in: path + name: project_id + required: true + schema: + format: uuid4 + title: Project Id + type: string + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/LogRecordsDeleteRequest' + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/LogRecordsDeleteResponse' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Delete Traces + tags: + - trace /projects/{project_id}/traces/search: post: operationId: query_traces_projects__project_id__traces_search_post @@ -25946,6 +26547,45 @@ paths: summary: Create Global Prompt Template tags: - prompts + /templates/bulk_delete: + delete: + description: "Delete multiple global prompt templates in bulk.\n\nThis endpoint\ + \ allows efficient deletion of multiple global prompt templates at once.\n\ + It validates permissions for each template in the service and provides detailed\ + \ feedback about\nsuccessful and failed deletions for each template.\n\nParameters\n\ + ----------\ndelete_request : BulkDeletePromptTemplatesRequest\n Request\ + \ containing list of template IDs to delete (max 100)\nctx : Context\n \ + \ Request context including authentication information\n\nReturns\n-------\n\ + BulkDeletePromptTemplatesResponse\n Details about the bulk deletion operation\ + \ including:\n - Number of successfully deleted templates\n - List of\ + \ failed deletions with reasons\n - Summary message" + operationId: bulk_delete_global_templates_templates_bulk_delete_delete + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/BulkDeletePromptTemplatesRequest' + required: true + responses: + '200': + content: + application/json: + schema: + $ref: '#/components/schemas/BulkDeletePromptTemplatesResponse' + description: Successful Response + '422': + content: + application/json: + schema: + $ref: '#/components/schemas/HTTPValidationError' + description: Validation Error + security: + - APIKeyHeader: [] + - OAuth2PasswordBearer: [] + - HTTPBasic: [] + summary: Bulk Delete Global Templates + tags: + - prompts /templates/query: post: description: "Query prompt templates the user has access to.\n\nParameters\n\ diff --git a/scripts/patch_status_codes.py b/scripts/patch_status_codes.py new file mode 100755 index 000000000..db5d21027 --- /dev/null +++ b/scripts/patch_status_codes.py @@ -0,0 +1,97 @@ +#!/usr/bin/env python3 +""" +Patch generated API files to fix missing status codes. + +Workflow +-------- +1. Find all API endpoint files in the generated directory. +2. For each file, locate lines with 'if response.status_code == :' +3. Replace with proper status codes by extracting from response variable names. +4. Write the file back only if changes were made. + +Run this script as a post-generation hook for openapi-python-client. + +Exit codes +0 - patches applied successfully +1 - I/O or other error +""" + +from __future__ import annotations + +import re +import sys +from pathlib import Path + + +def fix_status_codes_in_file(file_path: Path) -> bool: + """Fix missing status codes in a single file. Returns True if changes were made.""" + try: + content = file_path.read_text(encoding="utf-8") + original_content = content + + # Simple approach: find each broken status code check and fix it by looking for context + lines = content.split("\n") + modified = False + + for i, line in enumerate(lines): + if "if response.status_code == :" in line: + # Look for status code in the next few lines + status_code = None + for j in range(i + 1, min(i + 10, len(lines))): + # Check for patterns like 'response_200', '_parse_response_200', etc. + matches = re.findall(r"(?:response_|_parse_response_|_)(\d{3})", lines[j]) + if matches: + status_code = matches[0] + break + + if status_code: + lines[i] = line.replace( + "if response.status_code == :", f"if response.status_code == {status_code}:" + ) + modified = True + + if modified: + content = "\n".join(lines) + + if content != original_content: + file_path.write_text(content, encoding="utf-8") + return True + + return False + + except Exception as e: + print(f"Error processing {file_path}: {e}", file=sys.stderr) + return False + + +def fix_status_codes_in_directory(directory: Path) -> int: + """Fix status codes in all Python files in directory. Returns number of files patched.""" + if not directory.exists(): + print(f"Directory {directory} not found", file=sys.stderr) + return 0 + + patched_count = 0 + python_files = list(directory.rglob("*.py")) + + for file_path in python_files: + if fix_status_codes_in_file(file_path): + print(f"Patched {file_path}") + patched_count += 1 + + return patched_count + + +def main() -> int: + if len(sys.argv) != 2: + print(f"Usage: {Path(sys.argv[0]).name} ", file=sys.stderr) + return 1 + + directory = Path(sys.argv[1]) + patched_count = fix_status_codes_in_directory(directory) + + print(f"Patched {patched_count} files") + return 0 + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/src/galileo/resources/api/auth/login_api_key_login_api_key_post.py b/src/galileo/resources/api/auth/login_api_key_login_api_key_post.py index 569de39ca..d2de54e80 100644 --- a/src/galileo/resources/api/auth/login_api_key_login_api_key_post.py +++ b/src/galileo/resources/api/auth/login_api_key_login_api_key_post.py @@ -31,17 +31,14 @@ def _get_kwargs(*, body: ApiKeyLoginRequest) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Token]]: if response.status_code == 200: - response_200 = Token.from_dict(response.json()) + return Token.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, Token]]: diff --git a/src/galileo/resources/api/auth/login_email_login_post.py b/src/galileo/resources/api/auth/login_email_login_post.py index 180480a90..881357f3f 100644 --- a/src/galileo/resources/api/auth/login_email_login_post.py +++ b/src/galileo/resources/api/auth/login_email_login_post.py @@ -31,17 +31,14 @@ def _get_kwargs(*, body: BodyLoginEmailLoginPost) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Token]]: if response.status_code == 200: - response_200 = Token.from_dict(response.json()) + return Token.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, Token]]: diff --git a/src/galileo/resources/api/data/autogen_llm_scorer_scorers_llm_autogen_post.py b/src/galileo/resources/api/data/autogen_llm_scorer_scorers_llm_autogen_post.py index cc9c0f608..ec148076d 100644 --- a/src/galileo/resources/api/data/autogen_llm_scorer_scorers_llm_autogen_post.py +++ b/src/galileo/resources/api/data/autogen_llm_scorer_scorers_llm_autogen_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GenerationResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = GenerationResponse.from_dict(response.json()) + return GenerationResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/create_code_scorer_version_scorers_scorer_id_version_code_post.py b/src/galileo/resources/api/data/create_code_scorer_version_scorers_scorer_id_version_code_post.py index 3f919ab0e..f15ac3339 100644 --- a/src/galileo/resources/api/data/create_code_scorer_version_scorers_scorer_id_version_code_post.py +++ b/src/galileo/resources/api/data/create_code_scorer_version_scorers_scorer_id_version_code_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BaseScorerVersionResponse.from_dict(response.json()) + return BaseScorerVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/create_llm_scorer_version_scorers_scorer_id_version_llm_post.py b/src/galileo/resources/api/data/create_llm_scorer_version_scorers_scorer_id_version_llm_post.py index 3c54dc1a4..0b14551cb 100644 --- a/src/galileo/resources/api/data/create_llm_scorer_version_scorers_scorer_id_version_llm_post.py +++ b/src/galileo/resources/api/data/create_llm_scorer_version_scorers_scorer_id_version_llm_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BaseScorerVersionResponse.from_dict(response.json()) + return BaseScorerVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/create_luna_scorer_version_scorers_scorer_id_version_luna_post.py b/src/galileo/resources/api/data/create_luna_scorer_version_scorers_scorer_id_version_luna_post.py index 2e668f0b9..c45e268c3 100644 --- a/src/galileo/resources/api/data/create_luna_scorer_version_scorers_scorer_id_version_luna_post.py +++ b/src/galileo/resources/api/data/create_luna_scorer_version_scorers_scorer_id_version_luna_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BaseScorerVersionResponse.from_dict(response.json()) + return BaseScorerVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/create_preset_scorer_version_scorers_scorer_id_version_preset_post.py b/src/galileo/resources/api/data/create_preset_scorer_version_scorers_scorer_id_version_preset_post.py index 9b2c1bf6c..5ef89ba18 100644 --- a/src/galileo/resources/api/data/create_preset_scorer_version_scorers_scorer_id_version_preset_post.py +++ b/src/galileo/resources/api/data/create_preset_scorer_version_scorers_scorer_id_version_preset_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BaseScorerVersionResponse.from_dict(response.json()) + return BaseScorerVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/create_scorers_post.py b/src/galileo/resources/api/data/create_scorers_post.py index abded7ab7..7bc84c7ce 100644 --- a/src/galileo/resources/api/data/create_scorers_post.py +++ b/src/galileo/resources/api/data/create_scorers_post.py @@ -33,17 +33,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: if response.status_code == 200: - response_200 = ScorerResponse.from_dict(response.json()) + return ScorerResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/delete_scorer_scorers_scorer_id_delete.py b/src/galileo/resources/api/data/delete_scorer_scorers_scorer_id_delete.py index 7883e125b..84a7a73ec 100644 --- a/src/galileo/resources/api/data/delete_scorer_scorers_scorer_id_delete.py +++ b/src/galileo/resources/api/data/delete_scorer_scorers_scorer_id_delete.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = DeleteScorerResponse.from_dict(response.json()) + return DeleteScorerResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/get_scorer_scorers_scorer_id_get.py b/src/galileo/resources/api/data/get_scorer_scorers_scorer_id_get.py index 1cd47c668..ef8fd9177 100644 --- a/src/galileo/resources/api/data/get_scorer_scorers_scorer_id_get.py +++ b/src/galileo/resources/api/data/get_scorer_scorers_scorer_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: if response.status_code == 200: - response_200 = ScorerResponse.from_dict(response.json()) + return ScorerResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/get_scorer_version_code_scorers_scorer_id_version_code_get.py b/src/galileo/resources/api/data/get_scorer_version_code_scorers_scorer_id_version_code_get.py index cbe892494..200224cc1 100644 --- a/src/galileo/resources/api/data/get_scorer_version_code_scorers_scorer_id_version_code_get.py +++ b/src/galileo/resources/api/data/get_scorer_version_code_scorers_scorer_id_version_code_get.py @@ -18,10 +18,7 @@ def _get_kwargs(scorer_id: str, *, version: Union[None, Unset, int] = UNSET) -> params: dict[str, Any] = {} json_version: Union[None, Unset, int] - if isinstance(version, Unset): - json_version = UNSET - else: - json_version = version + json_version = UNSET if isinstance(version, Unset) else version params["version"] = json_version params = {k: v for k, v in params.items() if v is not UNSET and v is not None} @@ -41,16 +38,13 @@ def _get_kwargs(scorer_id: str, *, version: Union[None, Unset, int] = UNSET) -> def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/data/get_scorer_version_or_latest_scorers_scorer_id_version_get.py b/src/galileo/resources/api/data/get_scorer_version_or_latest_scorers_scorer_id_version_get.py index 7d422c625..7d55ff98f 100644 --- a/src/galileo/resources/api/data/get_scorer_version_or_latest_scorers_scorer_id_version_get.py +++ b/src/galileo/resources/api/data/get_scorer_version_or_latest_scorers_scorer_id_version_get.py @@ -39,17 +39,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BaseScorerVersionResponse.from_dict(response.json()) + return BaseScorerVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/list_all_versions_for_scorer_scorers_scorer_id_versions_get.py b/src/galileo/resources/api/data/list_all_versions_for_scorer_scorers_scorer_id_versions_get.py index 86bcbe56f..bf493065a 100644 --- a/src/galileo/resources/api/data/list_all_versions_for_scorer_scorers_scorer_id_versions_get.py +++ b/src/galileo/resources/api/data/list_all_versions_for_scorer_scorers_scorer_id_versions_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: if response.status_code == 200: - response_200 = ListScorerVersionsResponse.from_dict(response.json()) + return ListScorerVersionsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/list_projects_for_scorer_route_scorers_scorer_id_projects_get.py b/src/galileo/resources/api/data/list_projects_for_scorer_route_scorers_scorer_id_projects_get.py index 05abc7629..6560e3785 100644 --- a/src/galileo/resources/api/data/list_projects_for_scorer_route_scorers_scorer_id_projects_get.py +++ b/src/galileo/resources/api/data/list_projects_for_scorer_route_scorers_scorer_id_projects_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: if response.status_code == 200: - response_200 = GetProjectsPaginatedResponseV2.from_dict(response.json()) + return GetProjectsPaginatedResponseV2.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/list_projects_for_scorer_version_route_scorers_versions_scorer_version_id_projects_get.py b/src/galileo/resources/api/data/list_projects_for_scorer_version_route_scorers_versions_scorer_version_id_projects_get.py index 96726e16f..7e5662a61 100644 --- a/src/galileo/resources/api/data/list_projects_for_scorer_version_route_scorers_versions_scorer_version_id_projects_get.py +++ b/src/galileo/resources/api/data/list_projects_for_scorer_version_route_scorers_versions_scorer_version_id_projects_get.py @@ -45,17 +45,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: if response.status_code == 200: - response_200 = GetProjectsPaginatedResponseV2.from_dict(response.json()) + return GetProjectsPaginatedResponseV2.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/list_scorers_with_filters_scorers_list_post.py b/src/galileo/resources/api/data/list_scorers_with_filters_scorers_list_post.py index 54f6e8154..92087a500 100644 --- a/src/galileo/resources/api/data/list_scorers_with_filters_scorers_list_post.py +++ b/src/galileo/resources/api/data/list_scorers_with_filters_scorers_list_post.py @@ -48,17 +48,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: if response.status_code == 200: - response_200 = ListScorersResponse.from_dict(response.json()) + return ListScorersResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/list_tags_scorers_tags_get.py b/src/galileo/resources/api/data/list_tags_scorers_tags_get.py index 5815fc781..2594edb7d 100644 --- a/src/galileo/resources/api/data/list_tags_scorers_tags_get.py +++ b/src/galileo/resources/api/data/list_tags_scorers_tags_get.py @@ -24,13 +24,11 @@ def _get_kwargs() -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list[str]]: if response.status_code == 200: - response_200 = cast(list[str], response.json()) + return cast(list[str], response.json()) - return response_200 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[list[str]]: diff --git a/src/galileo/resources/api/data/manual_llm_validate_scorers_llm_validate_post.py b/src/galileo/resources/api/data/manual_llm_validate_scorers_llm_validate_post.py index e2f714099..91be3cfb9 100644 --- a/src/galileo/resources/api/data/manual_llm_validate_scorers_llm_validate_post.py +++ b/src/galileo/resources/api/data/manual_llm_validate_scorers_llm_validate_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = GeneratedScorerValidationResponse.from_dict(response.json()) + return GeneratedScorerValidationResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/restore_scorer_version_scorers_scorer_id_versions_version_number_restore_post.py b/src/galileo/resources/api/data/restore_scorer_version_scorers_scorer_id_versions_version_number_restore_post.py index de799821b..fbef921b6 100644 --- a/src/galileo/resources/api/data/restore_scorer_version_scorers_scorer_id_versions_version_number_restore_post.py +++ b/src/galileo/resources/api/data/restore_scorer_version_scorers_scorer_id_versions_version_number_restore_post.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BaseScorerVersionResponse.from_dict(response.json()) + return BaseScorerVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/update_scorers_scorer_id_patch.py b/src/galileo/resources/api/data/update_scorers_scorer_id_patch.py index 5bdf27f76..d931ab3b2 100644 --- a/src/galileo/resources/api/data/update_scorers_scorer_id_patch.py +++ b/src/galileo/resources/api/data/update_scorers_scorer_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: if response.status_code == 200: - response_200 = ScorerResponse.from_dict(response.json()) + return ScorerResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/data/validate_llm_scorer_log_record_scorers_llm_validate_log_record_post.py b/src/galileo/resources/api/data/validate_llm_scorer_log_record_scorers_llm_validate_log_record_post.py index 2f10fea8a..e98bc3f02 100644 --- a/src/galileo/resources/api/data/validate_llm_scorer_log_record_scorers_llm_validate_log_record_post.py +++ b/src/galileo/resources/api/data/validate_llm_scorer_log_record_scorers_llm_validate_log_record_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: if response.status_code == 200: - response_200 = ValidateLLMScorerLogRecordResponse.from_dict(response.json()) + return ValidateLLMScorerLogRecordResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/bulk_delete_datasets_datasets_bulk_delete_delete.py b/src/galileo/resources/api/datasets/bulk_delete_datasets_datasets_bulk_delete_delete.py index a78f96a13..ed711657f 100644 --- a/src/galileo/resources/api/datasets/bulk_delete_datasets_datasets_bulk_delete_delete.py +++ b/src/galileo/resources/api/datasets/bulk_delete_datasets_datasets_bulk_delete_delete.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -26,6 +27,8 @@ def _get_kwargs(*, body: BulkDeleteDatasetsRequest) -> dict[str, Any]: headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs @@ -34,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BulkDeleteDatasetsResponse.from_dict(response.json()) + return BulkDeleteDatasetsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/create_dataset_datasets_post.py b/src/galileo/resources/api/datasets/create_dataset_datasets_post.py index 88f6c6da3..ad73a929d 100644 --- a/src/galileo/resources/api/datasets/create_dataset_datasets_post.py +++ b/src/galileo/resources/api/datasets/create_dataset_datasets_post.py @@ -52,17 +52,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetDB.from_dict(response.json()) + return DatasetDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetDB, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/create_group_dataset_collaborators_datasets_dataset_id_groups_post.py b/src/galileo/resources/api/datasets/create_group_dataset_collaborators_datasets_dataset_id_groups_post.py index 07d946b6e..dc2112c8d 100644 --- a/src/galileo/resources/api/datasets/create_group_dataset_collaborators_datasets_dataset_id_groups_post.py +++ b/src/galileo/resources/api/datasets/create_group_dataset_collaborators_datasets_dataset_id_groups_post.py @@ -49,13 +49,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/create_user_dataset_collaborators_datasets_dataset_id_users_post.py b/src/galileo/resources/api/datasets/create_user_dataset_collaborators_datasets_dataset_id_users_post.py index c6e1d792a..d1870fcd9 100644 --- a/src/galileo/resources/api/datasets/create_user_dataset_collaborators_datasets_dataset_id_users_post.py +++ b/src/galileo/resources/api/datasets/create_user_dataset_collaborators_datasets_dataset_id_users_post.py @@ -49,13 +49,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/delete_dataset_datasets_dataset_id_delete.py b/src/galileo/resources/api/datasets/delete_dataset_datasets_dataset_id_delete.py index b3c41d8ed..5afff6254 100644 --- a/src/galileo/resources/api/datasets/delete_dataset_datasets_dataset_id_delete.py +++ b/src/galileo/resources/api/datasets/delete_dataset_datasets_dataset_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(dataset_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/delete_group_dataset_collaborator_datasets_dataset_id_groups_group_id_delete.py b/src/galileo/resources/api/datasets/delete_group_dataset_collaborator_datasets_dataset_id_groups_group_id_delete.py index 614665132..c14b71f10 100644 --- a/src/galileo/resources/api/datasets/delete_group_dataset_collaborator_datasets_dataset_id_groups_group_id_delete.py +++ b/src/galileo/resources/api/datasets/delete_group_dataset_collaborator_datasets_dataset_id_groups_group_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(dataset_id: str, group_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/delete_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_delete.py b/src/galileo/resources/api/datasets/delete_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_delete.py index f2d816858..34c7af68e 100644 --- a/src/galileo/resources/api/datasets/delete_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_delete.py +++ b/src/galileo/resources/api/datasets/delete_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(project_id: str, dataset_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/delete_user_dataset_collaborator_datasets_dataset_id_users_user_id_delete.py b/src/galileo/resources/api/datasets/delete_user_dataset_collaborator_datasets_dataset_id_users_user_id_delete.py index 2c45f771a..5d61855ab 100644 --- a/src/galileo/resources/api/datasets/delete_user_dataset_collaborator_datasets_dataset_id_users_user_id_delete.py +++ b/src/galileo/resources/api/datasets/delete_user_dataset_collaborator_datasets_dataset_id_users_user_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(dataset_id: str, user_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/download_dataset_datasets_dataset_id_download_get.py b/src/galileo/resources/api/datasets/download_dataset_datasets_dataset_id_download_get.py index e7c98b3b5..a6695e316 100644 --- a/src/galileo/resources/api/datasets/download_dataset_datasets_dataset_id_download_get.py +++ b/src/galileo/resources/api/datasets/download_dataset_datasets_dataset_id_download_get.py @@ -29,16 +29,13 @@ def _get_kwargs(dataset_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/download_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_get.py b/src/galileo/resources/api/datasets/download_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_get.py index 51c314933..ea585724c 100644 --- a/src/galileo/resources/api/datasets/download_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_get.py +++ b/src/galileo/resources/api/datasets/download_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_get.py @@ -29,16 +29,13 @@ def _get_kwargs(project_id: str, dataset_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = cast(Any, None) - return response_200 + return cast(Any, None) if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/extend_dataset_content_datasets_extend_post.py b/src/galileo/resources/api/datasets/extend_dataset_content_datasets_extend_post.py index ab57c139e..48dde46e1 100644 --- a/src/galileo/resources/api/datasets/extend_dataset_content_datasets_extend_post.py +++ b/src/galileo/resources/api/datasets/extend_dataset_content_datasets_extend_post.py @@ -33,17 +33,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: if response.status_code == 200: - response_200 = SyntheticDatasetExtensionResponse.from_dict(response.json()) + return SyntheticDatasetExtensionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/get_dataset_content_datasets_dataset_id_content_get.py b/src/galileo/resources/api/datasets/get_dataset_content_datasets_dataset_id_content_get.py index 5000708bc..db3f2c51c 100644 --- a/src/galileo/resources/api/datasets/get_dataset_content_datasets_dataset_id_content_get.py +++ b/src/galileo/resources/api/datasets/get_dataset_content_datasets_dataset_id_content_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DatasetContent, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetContent.from_dict(response.json()) + return DatasetContent.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/get_dataset_datasets_dataset_id_get.py b/src/galileo/resources/api/datasets/get_dataset_datasets_dataset_id_get.py index 893fa17ab..7bf7a6a6d 100644 --- a/src/galileo/resources/api/datasets/get_dataset_datasets_dataset_id_get.py +++ b/src/galileo/resources/api/datasets/get_dataset_datasets_dataset_id_get.py @@ -30,17 +30,14 @@ def _get_kwargs(dataset_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetDB.from_dict(response.json()) + return DatasetDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetDB, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/get_dataset_synthetic_extend_status_datasets_extend_dataset_id_get.py b/src/galileo/resources/api/datasets/get_dataset_synthetic_extend_status_datasets_extend_dataset_id_get.py index ab5b4a00c..645da0850 100644 --- a/src/galileo/resources/api/datasets/get_dataset_synthetic_extend_status_datasets_extend_dataset_id_get.py +++ b/src/galileo/resources/api/datasets/get_dataset_synthetic_extend_status_datasets_extend_dataset_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, JobProgress]]: if response.status_code == 200: - response_200 = JobProgress.from_dict(response.json()) + return JobProgress.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/get_dataset_version_content_datasets_dataset_id_versions_version_index_content_get.py b/src/galileo/resources/api/datasets/get_dataset_version_content_datasets_dataset_id_versions_version_index_content_get.py index ac2e7edb6..845f76097 100644 --- a/src/galileo/resources/api/datasets/get_dataset_version_content_datasets_dataset_id_versions_version_index_content_get.py +++ b/src/galileo/resources/api/datasets/get_dataset_version_content_datasets_dataset_id_versions_version_index_content_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DatasetContent, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetContent.from_dict(response.json()) + return DatasetContent.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/list_dataset_projects_datasets_dataset_id_projects_get.py b/src/galileo/resources/api/datasets/list_dataset_projects_datasets_dataset_id_projects_get.py index 1c30cb800..e6582e9c5 100644 --- a/src/galileo/resources/api/datasets/list_dataset_projects_datasets_dataset_id_projects_get.py +++ b/src/galileo/resources/api/datasets/list_dataset_projects_datasets_dataset_id_projects_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: if response.status_code == 200: - response_200 = ListDatasetProjectsResponse.from_dict(response.json()) + return ListDatasetProjectsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/list_datasets_datasets_get.py b/src/galileo/resources/api/datasets/list_datasets_datasets_get.py index 83aa0ada6..9df57d02d 100644 --- a/src/galileo/resources/api/datasets/list_datasets_datasets_get.py +++ b/src/galileo/resources/api/datasets/list_datasets_datasets_get.py @@ -56,17 +56,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: if response.status_code == 200: - response_200 = ListDatasetResponse.from_dict(response.json()) + return ListDatasetResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/list_group_dataset_collaborators_datasets_dataset_id_groups_get.py b/src/galileo/resources/api/datasets/list_group_dataset_collaborators_datasets_dataset_id_groups_get.py index 9429f5688..6a629addf 100644 --- a/src/galileo/resources/api/datasets/list_group_dataset_collaborators_datasets_dataset_id_groups_get.py +++ b/src/galileo/resources/api/datasets/list_group_dataset_collaborators_datasets_dataset_id_groups_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: if response.status_code == 200: - response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) + return ListGroupCollaboratorsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/list_prompt_datasets_projects_project_id_prompt_datasets_get.py b/src/galileo/resources/api/datasets/list_prompt_datasets_projects_project_id_prompt_datasets_get.py index d48bb3f80..e0394a2e6 100644 --- a/src/galileo/resources/api/datasets/list_prompt_datasets_projects_project_id_prompt_datasets_get.py +++ b/src/galileo/resources/api/datasets/list_prompt_datasets_projects_project_id_prompt_datasets_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: if response.status_code == 200: - response_200 = ListPromptDatasetResponse.from_dict(response.json()) + return ListPromptDatasetResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/list_user_dataset_collaborators_datasets_dataset_id_users_get.py b/src/galileo/resources/api/datasets/list_user_dataset_collaborators_datasets_dataset_id_users_get.py index aee566ca8..35e378c87 100644 --- a/src/galileo/resources/api/datasets/list_user_dataset_collaborators_datasets_dataset_id_users_get.py +++ b/src/galileo/resources/api/datasets/list_user_dataset_collaborators_datasets_dataset_id_users_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: if response.status_code == 200: - response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) + return ListUserCollaboratorsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/preview_dataset_datasets_dataset_id_preview_post.py b/src/galileo/resources/api/datasets/preview_dataset_datasets_dataset_id_preview_post.py index 61ff097b5..4248eba90 100644 --- a/src/galileo/resources/api/datasets/preview_dataset_datasets_dataset_id_preview_post.py +++ b/src/galileo/resources/api/datasets/preview_dataset_datasets_dataset_id_preview_post.py @@ -52,17 +52,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DatasetContent, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetContent.from_dict(response.json()) + return DatasetContent.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/query_dataset_content_datasets_dataset_id_content_query_post.py b/src/galileo/resources/api/datasets/query_dataset_content_datasets_dataset_id_content_query_post.py index 477fb606d..70874482b 100644 --- a/src/galileo/resources/api/datasets/query_dataset_content_datasets_dataset_id_content_query_post.py +++ b/src/galileo/resources/api/datasets/query_dataset_content_datasets_dataset_id_content_query_post.py @@ -48,17 +48,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DatasetContent, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetContent.from_dict(response.json()) + return DatasetContent.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/query_dataset_versions_datasets_dataset_id_versions_query_post.py b/src/galileo/resources/api/datasets/query_dataset_versions_datasets_dataset_id_versions_query_post.py index 82ef9c2ba..e4773fc88 100644 --- a/src/galileo/resources/api/datasets/query_dataset_versions_datasets_dataset_id_versions_query_post.py +++ b/src/galileo/resources/api/datasets/query_dataset_versions_datasets_dataset_id_versions_query_post.py @@ -52,17 +52,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: if response.status_code == 200: - response_200 = ListDatasetVersionResponse.from_dict(response.json()) + return ListDatasetVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/query_datasets_datasets_query_post.py b/src/galileo/resources/api/datasets/query_datasets_datasets_query_post.py index b1b083fe5..9fdf5c721 100644 --- a/src/galileo/resources/api/datasets/query_datasets_datasets_query_post.py +++ b/src/galileo/resources/api/datasets/query_datasets_datasets_query_post.py @@ -62,17 +62,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: if response.status_code == 200: - response_200 = ListDatasetResponse.from_dict(response.json()) + return ListDatasetResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/update_dataset_content_datasets_dataset_id_content_patch.py b/src/galileo/resources/api/datasets/update_dataset_content_datasets_dataset_id_content_patch.py index cb8dfcd79..332bd1410 100644 --- a/src/galileo/resources/api/datasets/update_dataset_content_datasets_dataset_id_content_patch.py +++ b/src/galileo/resources/api/datasets/update_dataset_content_datasets_dataset_id_content_patch.py @@ -38,16 +38,13 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/update_dataset_datasets_dataset_id_patch.py b/src/galileo/resources/api/datasets/update_dataset_datasets_dataset_id_patch.py index 77a20b11c..e73b639de 100644 --- a/src/galileo/resources/api/datasets/update_dataset_datasets_dataset_id_patch.py +++ b/src/galileo/resources/api/datasets/update_dataset_datasets_dataset_id_patch.py @@ -35,17 +35,14 @@ def _get_kwargs(dataset_id: str, *, body: UpdateDatasetRequest) -> dict[str, Any def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetDB.from_dict(response.json()) + return DatasetDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetDB, HTTPValidationError]]: diff --git a/src/galileo/resources/api/datasets/update_dataset_version_datasets_dataset_id_versions_version_index_patch.py b/src/galileo/resources/api/datasets/update_dataset_version_datasets_dataset_id_versions_version_index_patch.py index e1f0cea97..7087e1159 100644 --- a/src/galileo/resources/api/datasets/update_dataset_version_datasets_dataset_id_versions_version_index_patch.py +++ b/src/galileo/resources/api/datasets/update_dataset_version_datasets_dataset_id_versions_version_index_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: if response.status_code == 200: - response_200 = DatasetVersionDB.from_dict(response.json()) + return DatasetVersionDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/update_group_dataset_collaborator_datasets_dataset_id_groups_group_id_patch.py b/src/galileo/resources/api/datasets/update_group_dataset_collaborator_datasets_dataset_id_groups_group_id_patch.py index 16ffb8be3..b4ea5ed9a 100644 --- a/src/galileo/resources/api/datasets/update_group_dataset_collaborator_datasets_dataset_id_groups_group_id_patch.py +++ b/src/galileo/resources/api/datasets/update_group_dataset_collaborator_datasets_dataset_id_groups_group_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: if response.status_code == 200: - response_200 = GroupCollaborator.from_dict(response.json()) + return GroupCollaborator.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py b/src/galileo/resources/api/datasets/update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py index ca3a1f4ef..918338b5d 100644 --- a/src/galileo/resources/api/datasets/update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py +++ b/src/galileo/resources/api/datasets/update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py @@ -32,17 +32,11 @@ def _get_kwargs( params: dict[str, Any] = {} json_file_name: Union[None, Unset, str] - if isinstance(file_name, Unset): - json_file_name = UNSET - else: - json_file_name = file_name + json_file_name = UNSET if isinstance(file_name, Unset) else file_name params["file_name"] = json_file_name json_num_rows: Union[None, Unset, int] - if isinstance(num_rows, Unset): - json_num_rows = UNSET - else: - json_num_rows = num_rows + json_num_rows = UNSET if isinstance(num_rows, Unset) else num_rows params["num_rows"] = json_num_rows json_format_: Union[Unset, str] = UNSET @@ -74,17 +68,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: if response.status_code == 200: - response_200 = PromptDatasetDB.from_dict(response.json()) + return PromptDatasetDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/update_user_dataset_collaborator_datasets_dataset_id_users_user_id_patch.py b/src/galileo/resources/api/datasets/update_user_dataset_collaborator_datasets_dataset_id_users_user_id_patch.py index b99b66ea9..b10ffde13 100644 --- a/src/galileo/resources/api/datasets/update_user_dataset_collaborator_datasets_dataset_id_users_user_id_patch.py +++ b/src/galileo/resources/api/datasets/update_user_dataset_collaborator_datasets_dataset_id_users_user_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: if response.status_code == 200: - response_200 = UserCollaborator.from_dict(response.json()) + return UserCollaborator.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post.py b/src/galileo/resources/api/datasets/upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post.py index 1828934ee..b1c9a01a1 100644 --- a/src/galileo/resources/api/datasets/upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post.py +++ b/src/galileo/resources/api/datasets/upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post.py @@ -57,17 +57,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: if response.status_code == 200: - response_200 = PromptDatasetDB.from_dict(response.json()) + return PromptDatasetDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/datasets/upsert_dataset_content_datasets_dataset_id_content_put.py b/src/galileo/resources/api/datasets/upsert_dataset_content_datasets_dataset_id_content_put.py index 8dfe90f3b..3693c8593 100644 --- a/src/galileo/resources/api/datasets/upsert_dataset_content_datasets_dataset_id_content_put.py +++ b/src/galileo/resources/api/datasets/upsert_dataset_content_datasets_dataset_id_content_put.py @@ -39,16 +39,13 @@ def _get_kwargs(dataset_id: str, *, body: Union["RollbackRequest", "UpsertDatase def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/experiment/create_experiment_projects_project_id_experiments_post.py b/src/galileo/resources/api/experiment/create_experiment_projects_project_id_experiments_post.py index a699b3363..57cc5a009 100644 --- a/src/galileo/resources/api/experiment/create_experiment_projects_project_id_experiments_post.py +++ b/src/galileo/resources/api/experiment/create_experiment_projects_project_id_experiments_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExperimentResponse.from_dict(response.json()) + return ExperimentResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment/delete_experiment_projects_project_id_experiments_experiment_id_delete.py b/src/galileo/resources/api/experiment/delete_experiment_projects_project_id_experiments_experiment_id_delete.py index 334d0123f..7f08c229b 100644 --- a/src/galileo/resources/api/experiment/delete_experiment_projects_project_id_experiments_experiment_id_delete.py +++ b/src/galileo/resources/api/experiment/delete_experiment_projects_project_id_experiments_experiment_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(project_id: str, experiment_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 204: - response_204 = cast(Any, None) - return response_204 + return cast(Any, None) if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/experiment/experiments_available_columns_projects_project_id_experiments_available_columns_post.py b/src/galileo/resources/api/experiment/experiments_available_columns_projects_project_id_experiments_available_columns_post.py index fc9e703cb..0beb2a036 100644 --- a/src/galileo/resources/api/experiment/experiments_available_columns_projects_project_id_experiments_available_columns_post.py +++ b/src/galileo/resources/api/experiment/experiments_available_columns_projects_project_id_experiments_available_columns_post.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExperimentsAvailableColumnsResponse.from_dict(response.json()) + return ExperimentsAvailableColumnsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment/get_experiment_metrics_projects_project_id_experiments_experiment_id_metrics_post.py b/src/galileo/resources/api/experiment/get_experiment_metrics_projects_project_id_experiments_experiment_id_metrics_post.py index 5b32db2ed..4aebef1a7 100644 --- a/src/galileo/resources/api/experiment/get_experiment_metrics_projects_project_id_experiments_experiment_id_metrics_post.py +++ b/src/galileo/resources/api/experiment/get_experiment_metrics_projects_project_id_experiments_experiment_id_metrics_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExperimentMetricsResponse.from_dict(response.json()) + return ExperimentMetricsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment/get_experiment_projects_project_id_experiments_experiment_id_get.py b/src/galileo/resources/api/experiment/get_experiment_projects_project_id_experiments_experiment_id_get.py index 704b72452..8154c926d 100644 --- a/src/galileo/resources/api/experiment/get_experiment_projects_project_id_experiments_experiment_id_get.py +++ b/src/galileo/resources/api/experiment/get_experiment_projects_project_id_experiments_experiment_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExperimentResponse.from_dict(response.json()) + return ExperimentResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment/get_experiments_metrics_projects_project_id_experiments_metrics_post.py b/src/galileo/resources/api/experiment/get_experiments_metrics_projects_project_id_experiments_metrics_post.py index f8d48d818..fba2bd92d 100644 --- a/src/galileo/resources/api/experiment/get_experiments_metrics_projects_project_id_experiments_metrics_post.py +++ b/src/galileo/resources/api/experiment/get_experiments_metrics_projects_project_id_experiments_metrics_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExperimentMetricsResponse.from_dict(response.json()) + return ExperimentMetricsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment/list_experiments_projects_project_id_experiments_get.py b/src/galileo/resources/api/experiment/list_experiments_projects_project_id_experiments_get.py index a1ff166d7..773a90fc0 100644 --- a/src/galileo/resources/api/experiment/list_experiments_projects_project_id_experiments_get.py +++ b/src/galileo/resources/api/experiment/list_experiments_projects_project_id_experiments_get.py @@ -41,13 +41,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment/update_experiment_projects_project_id_experiments_experiment_id_put.py b/src/galileo/resources/api/experiment/update_experiment_projects_project_id_experiments_experiment_id_put.py index 43a8736a4..33d8cec89 100644 --- a/src/galileo/resources/api/experiment/update_experiment_projects_project_id_experiments_experiment_id_put.py +++ b/src/galileo/resources/api/experiment/update_experiment_projects_project_id_experiments_experiment_id_put.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExperimentResponse.from_dict(response.json()) + return ExperimentResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment_tags/__init__.py b/src/galileo/resources/api/experiment_tags/__init__.py new file mode 100644 index 000000000..2d7c0b23d --- /dev/null +++ b/src/galileo/resources/api/experiment_tags/__init__.py @@ -0,0 +1 @@ +"""Contains endpoint functions for accessing the API""" diff --git a/src/galileo/resources/api/experiment_tags/delete_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_delete.py b/src/galileo/resources/api/experiment_tags/delete_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_delete.py index 9a024b141..9fbcbedde 100644 --- a/src/galileo/resources/api/experiment_tags/delete_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_delete.py +++ b/src/galileo/resources/api/experiment_tags/delete_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_delete.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -13,12 +14,17 @@ def _get_kwargs(project_id: str, experiment_id: str, tag_id: str) -> dict[str, Any]: + headers: dict[str, Any] = {} + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, "path": f"/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}", } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + + _kwargs["content_headers"] = headers return _kwargs @@ -26,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = DeleteRunResponse.from_dict(response.json()) + return DeleteRunResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment_tags/get_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_get.py b/src/galileo/resources/api/experiment_tags/get_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_get.py index 7a1c55683..076fe3f5b 100644 --- a/src/galileo/resources/api/experiment_tags/get_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_get.py +++ b/src/galileo/resources/api/experiment_tags/get_experiment_tag_projects_project_id_experiments_experiment_id_tags_tag_id_get.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -13,28 +14,30 @@ def _get_kwargs(project_id: str, experiment_id: str, tag_id: str) -> dict[str, Any]: + headers: dict[str, Any] = {} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, "path": f"/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}", } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + + _kwargs["content_headers"] = headers return _kwargs def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunTagDB]]: if response.status_code == 200: - response_200 = RunTagDB.from_dict(response.json()) + return RunTagDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RunTagDB]]: diff --git a/src/galileo/resources/api/experiment_tags/get_experiment_tags_projects_project_id_experiments_experiment_id_tags_get.py b/src/galileo/resources/api/experiment_tags/get_experiment_tags_projects_project_id_experiments_experiment_id_tags_get.py index 2946a3bbf..67ad88114 100644 --- a/src/galileo/resources/api/experiment_tags/get_experiment_tags_projects_project_id_experiments_experiment_id_tags_get.py +++ b/src/galileo/resources/api/experiment_tags/get_experiment_tags_projects_project_id_experiments_experiment_id_tags_get.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -13,12 +14,17 @@ def _get_kwargs(project_id: str, experiment_id: str) -> dict[str, Any]: + headers: dict[str, Any] = {} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, "path": f"/projects/{project_id}/experiments/{experiment_id}/tags", } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + + _kwargs["content_headers"] = headers return _kwargs @@ -35,13 +41,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/experiment_tags/set_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_post.py b/src/galileo/resources/api/experiment_tags/set_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_post.py index 9bf08fa49..b511f86d3 100644 --- a/src/galileo/resources/api/experiment_tags/set_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_post.py +++ b/src/galileo/resources/api/experiment_tags/set_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_post.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -26,23 +27,22 @@ def _get_kwargs(project_id: str, experiment_id: str, *, body: RunTagCreateReques headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunTagDB]]: if response.status_code == 200: - response_200 = RunTagDB.from_dict(response.json()) + return RunTagDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RunTagDB]]: diff --git a/src/galileo/resources/api/experiment_tags/update_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_tag_id_put.py b/src/galileo/resources/api/experiment_tags/update_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_tag_id_put.py index d7cd406ed..1879c3d58 100644 --- a/src/galileo/resources/api/experiment_tags/update_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_tag_id_put.py +++ b/src/galileo/resources/api/experiment_tags/update_tag_for_experiment_projects_project_id_experiments_experiment_id_tags_tag_id_put.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -26,23 +27,22 @@ def _get_kwargs(project_id: str, experiment_id: str, tag_id: str, *, body: RunTa headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunTagDB]]: if response.status_code == 200: - response_200 = RunTagDB.from_dict(response.json()) + return RunTagDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RunTagDB]]: diff --git a/src/galileo/resources/api/health/healthcheck_healthcheck_get.py b/src/galileo/resources/api/health/healthcheck_healthcheck_get.py index 2045b14aa..59d64e220 100644 --- a/src/galileo/resources/api/health/healthcheck_healthcheck_get.py +++ b/src/galileo/resources/api/health/healthcheck_healthcheck_get.py @@ -25,13 +25,11 @@ def _get_kwargs() -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[HealthcheckResponse]: if response.status_code == 200: - response_200 = HealthcheckResponse.from_dict(response.json()) + return HealthcheckResponse.from_dict(response.json()) - return response_200 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[HealthcheckResponse]: diff --git a/src/galileo/resources/api/jobs/create_job_jobs_post.py b/src/galileo/resources/api/jobs/create_job_jobs_post.py index 5b7bd2f5e..df317fbf6 100644 --- a/src/galileo/resources/api/jobs/create_job_jobs_post.py +++ b/src/galileo/resources/api/jobs/create_job_jobs_post.py @@ -33,17 +33,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = CreateJobResponse.from_dict(response.json()) + return CreateJobResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/jobs/get_job_jobs_job_id_get.py b/src/galileo/resources/api/jobs/get_job_jobs_job_id_get.py index eea80f35c..03d2e6d60 100644 --- a/src/galileo/resources/api/jobs/get_job_jobs_job_id_get.py +++ b/src/galileo/resources/api/jobs/get_job_jobs_job_id_get.py @@ -26,17 +26,14 @@ def _get_kwargs(job_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, JobDB]]: if response.status_code == 200: - response_200 = JobDB.from_dict(response.json()) + return JobDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, JobDB]]: diff --git a/src/galileo/resources/api/jobs/get_jobs_for_project_run_projects_project_id_runs_run_id_jobs_get.py b/src/galileo/resources/api/jobs/get_jobs_for_project_run_projects_project_id_runs_run_id_jobs_get.py index 6ba16320d..7f81f60e9 100644 --- a/src/galileo/resources/api/jobs/get_jobs_for_project_run_projects_project_id_runs_run_id_jobs_get.py +++ b/src/galileo/resources/api/jobs/get_jobs_for_project_run_projects_project_id_runs_run_id_jobs_get.py @@ -19,10 +19,7 @@ def _get_kwargs(project_id: str, run_id: str, *, status: Union[None, Unset, str] params: dict[str, Any] = {} json_status: Union[None, Unset, str] - if isinstance(status, Unset): - json_status = UNSET - else: - json_status = status + json_status = UNSET if isinstance(status, Unset) else status params["status"] = json_status params = {k: v for k, v in params.items() if v is not UNSET and v is not None} @@ -53,13 +50,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/jobs/get_latest_job_for_project_run_projects_project_id_runs_run_id_jobs_latest_get.py b/src/galileo/resources/api/jobs/get_latest_job_for_project_run_projects_project_id_runs_run_id_jobs_latest_get.py index f634f1a45..8d9198e3f 100644 --- a/src/galileo/resources/api/jobs/get_latest_job_for_project_run_projects_project_id_runs_run_id_jobs_latest_get.py +++ b/src/galileo/resources/api/jobs/get_latest_job_for_project_run_projects_project_id_runs_run_id_jobs_latest_get.py @@ -6,34 +6,44 @@ from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient - +from ...types import Response, UNSET from ... import errors + from ...models.http_validation_error import HTTPValidationError from ...models.job_db import JobDB -from ...types import Response +from typing import cast +from typing import cast, Union + + +def _get_kwargs( + project_id: str, + run_id: str, -def _get_kwargs(project_id: str, run_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/runs/{run_id}/jobs/latest", + "path": "/projects/{project_id}/runs/{run_id}/jobs/latest".format(project_id=project_id,run_id=run_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, Union["JobDB", None]]]: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['JobDB', None]]]: if response.status_code == 200: - - def _parse_response_200(data: object) -> Union["JobDB", None]: + def _parse_response_200(data: object) -> Union['JobDB', None]: if data is None: return data try: @@ -41,10 +51,12 @@ def _parse_response_200(data: object) -> Union["JobDB", None]: raise TypeError() response_200_type_0 = JobDB.from_dict(data) + + return response_200_type_0 - except: # noqa: E722 + except: # noqa: E722 pass - return cast(Union["JobDB", None], data) + return cast(Union['JobDB', None], data) response_200 = _parse_response_200(response.json()) @@ -52,6 +64,8 @@ def _parse_response_200(data: object) -> Union["JobDB", None]: if response.status_code == 422: response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -59,9 +73,7 @@ def _parse_response_200(data: object) -> Union["JobDB", None]: return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, Union["JobDB", None]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, Union['JobDB', None]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -71,9 +83,13 @@ def _build_response( def sync_detailed( - project_id: str, run_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, Union["JobDB", None]]]: - """Get Latest Job For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, Union['JobDB', None]]]: + """ Get Latest Job For Project Run Returns the most recently updated job for a run. @@ -87,19 +103,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, Union['JobDB', None]]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, run_id: str, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, Union["JobDB", None]]]: - """Get Latest Job For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, Union['JobDB', None]]]: + """ Get Latest Job For Project Run Returns the most recently updated job for a run. @@ -113,15 +139,24 @@ def sync( Returns: Union[HTTPValidationError, Union['JobDB', None]] - """ + """ - return sync_detailed(project_id=project_id, run_id=run_id, client=client).parsed + return sync_detailed( + project_id=project_id, +run_id=run_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, run_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, Union["JobDB", None]]]: - """Get Latest Job For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, Union['JobDB', None]]]: + """ Get Latest Job For Project Run Returns the most recently updated job for a run. @@ -135,19 +170,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Union['JobDB', None]]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - project_id: str, run_id: str, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, Union["JobDB", None]]]: - """Get Latest Job For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, Union['JobDB', None]]]: + """ Get Latest Job For Project Run Returns the most recently updated job for a run. @@ -161,6 +206,12 @@ async def asyncio( Returns: Union[HTTPValidationError, Union['JobDB', None]] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +run_id=run_id, +client=client, - return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client)).parsed + )).parsed diff --git a/src/galileo/resources/api/log_stream/create_log_stream_projects_project_id_log_streams_post.py b/src/galileo/resources/api/log_stream/create_log_stream_projects_project_id_log_streams_post.py index 8bc0b7f7e..953887958 100644 --- a/src/galileo/resources/api/log_stream/create_log_stream_projects_project_id_log_streams_post.py +++ b/src/galileo/resources/api/log_stream/create_log_stream_projects_project_id_log_streams_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: if response.status_code == 200: - response_200 = LogStreamResponse.from_dict(response.json()) + return LogStreamResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/log_stream/delete_log_stream_projects_project_id_log_streams_log_stream_id_delete.py b/src/galileo/resources/api/log_stream/delete_log_stream_projects_project_id_log_streams_log_stream_id_delete.py index dec565659..6a566ceec 100644 --- a/src/galileo/resources/api/log_stream/delete_log_stream_projects_project_id_log_streams_log_stream_id_delete.py +++ b/src/galileo/resources/api/log_stream/delete_log_stream_projects_project_id_log_streams_log_stream_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(project_id: str, log_stream_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 204: - response_204 = cast(Any, None) - return response_204 + return cast(Any, None) if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/log_stream/get_log_stream_projects_project_id_log_streams_log_stream_id_get.py b/src/galileo/resources/api/log_stream/get_log_stream_projects_project_id_log_streams_log_stream_id_get.py index 010d432b2..83fbfa193 100644 --- a/src/galileo/resources/api/log_stream/get_log_stream_projects_project_id_log_streams_log_stream_id_get.py +++ b/src/galileo/resources/api/log_stream/get_log_stream_projects_project_id_log_streams_log_stream_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: if response.status_code == 200: - response_200 = LogStreamResponse.from_dict(response.json()) + return LogStreamResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/log_stream/list_log_streams_projects_project_id_log_streams_get.py b/src/galileo/resources/api/log_stream/list_log_streams_projects_project_id_log_streams_get.py index 01be8c100..68423a939 100644 --- a/src/galileo/resources/api/log_stream/list_log_streams_projects_project_id_log_streams_get.py +++ b/src/galileo/resources/api/log_stream/list_log_streams_projects_project_id_log_streams_get.py @@ -41,13 +41,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/log_stream/update_log_stream_projects_project_id_log_streams_log_stream_id_put.py b/src/galileo/resources/api/log_stream/update_log_stream_projects_project_id_log_streams_log_stream_id_put.py index 73d389d98..0a763f915 100644 --- a/src/galileo/resources/api/log_stream/update_log_stream_projects_project_id_log_streams_log_stream_id_put.py +++ b/src/galileo/resources/api/log_stream/update_log_stream_projects_project_id_log_streams_log_stream_id_put.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: if response.status_code == 200: - response_200 = LogStreamResponse.from_dict(response.json()) + return LogStreamResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/create_group_project_collaborators_projects_project_id_groups_post.py b/src/galileo/resources/api/projects/create_group_project_collaborators_projects_project_id_groups_post.py index 58d911e73..9794851cb 100644 --- a/src/galileo/resources/api/projects/create_group_project_collaborators_projects_project_id_groups_post.py +++ b/src/galileo/resources/api/projects/create_group_project_collaborators_projects_project_id_groups_post.py @@ -49,13 +49,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/create_project_projects_post.py b/src/galileo/resources/api/projects/create_project_projects_post.py index c3729909e..12f150ed8 100644 --- a/src/galileo/resources/api/projects/create_project_projects_post.py +++ b/src/galileo/resources/api/projects/create_project_projects_post.py @@ -33,17 +33,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: if response.status_code == 200: - response_200 = ProjectCreateResponse.from_dict(response.json()) + return ProjectCreateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/create_user_project_collaborators_projects_project_id_users_post.py b/src/galileo/resources/api/projects/create_user_project_collaborators_projects_project_id_users_post.py index 109afefc4..b42e35789 100644 --- a/src/galileo/resources/api/projects/create_user_project_collaborators_projects_project_id_users_post.py +++ b/src/galileo/resources/api/projects/create_user_project_collaborators_projects_project_id_users_post.py @@ -49,13 +49,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/delete_group_project_collaborator_projects_project_id_groups_group_id_delete.py b/src/galileo/resources/api/projects/delete_group_project_collaborator_projects_project_id_groups_group_id_delete.py index ef1e2919e..02ddf7145 100644 --- a/src/galileo/resources/api/projects/delete_group_project_collaborator_projects_project_id_groups_group_id_delete.py +++ b/src/galileo/resources/api/projects/delete_group_project_collaborator_projects_project_id_groups_group_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(project_id: str, group_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/projects/delete_project_projects_project_id_delete.py b/src/galileo/resources/api/projects/delete_project_projects_project_id_delete.py index eba2866f1..63d96a49d 100644 --- a/src/galileo/resources/api/projects/delete_project_projects_project_id_delete.py +++ b/src/galileo/resources/api/projects/delete_project_projects_project_id_delete.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: if response.status_code == 200: - response_200 = ProjectDeleteResponse.from_dict(response.json()) + return ProjectDeleteResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/delete_user_project_collaborator_projects_project_id_users_user_id_delete.py b/src/galileo/resources/api/projects/delete_user_project_collaborator_projects_project_id_users_user_id_delete.py index 7c76cd1dc..eef2f39d8 100644 --- a/src/galileo/resources/api/projects/delete_user_project_collaborator_projects_project_id_users_user_id_delete.py +++ b/src/galileo/resources/api/projects/delete_user_project_collaborator_projects_project_id_users_user_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(project_id: str, user_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/projects/get_all_projects_projects_all_get.py b/src/galileo/resources/api/projects/get_all_projects_projects_all_get.py index 80c8f3508..548874d03 100644 --- a/src/galileo/resources/api/projects/get_all_projects_projects_all_get.py +++ b/src/galileo/resources/api/projects/get_all_projects_projects_all_get.py @@ -56,13 +56,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/get_collaborator_roles_collaborator_roles_get.py b/src/galileo/resources/api/projects/get_collaborator_roles_collaborator_roles_get.py index 2b91ae564..914de4ca4 100644 --- a/src/galileo/resources/api/projects/get_collaborator_roles_collaborator_roles_get.py +++ b/src/galileo/resources/api/projects/get_collaborator_roles_collaborator_roles_get.py @@ -35,8 +35,7 @@ def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[ return response_200 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[list["CollaboratorRoleInfo"]]: diff --git a/src/galileo/resources/api/projects/get_project_projects_project_id_get.py b/src/galileo/resources/api/projects/get_project_projects_project_id_get.py index c4dc919d6..96c21de20 100644 --- a/src/galileo/resources/api/projects/get_project_projects_project_id_get.py +++ b/src/galileo/resources/api/projects/get_project_projects_project_id_get.py @@ -30,17 +30,14 @@ def _get_kwargs(project_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectDB]]: if response.status_code == 200: - response_200 = ProjectDB.from_dict(response.json()) + return ProjectDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ProjectDB]]: diff --git a/src/galileo/resources/api/projects/get_projects_count_projects_count_post.py b/src/galileo/resources/api/projects/get_projects_count_projects_count_post.py index 2a020f7d2..bf160d848 100644 --- a/src/galileo/resources/api/projects/get_projects_count_projects_count_post.py +++ b/src/galileo/resources/api/projects/get_projects_count_projects_count_post.py @@ -30,16 +30,13 @@ def _get_kwargs(*, body: ProjectCollectionParams) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, int]]: if response.status_code == 200: - response_200 = cast(int, response.json()) - return response_200 + return cast(int, response.json()) if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, int]]: diff --git a/src/galileo/resources/api/projects/get_projects_paginated_projects_paginated_post.py b/src/galileo/resources/api/projects/get_projects_paginated_projects_paginated_post.py index 98ce9da47..96f30c81b 100644 --- a/src/galileo/resources/api/projects/get_projects_paginated_projects_paginated_post.py +++ b/src/galileo/resources/api/projects/get_projects_paginated_projects_paginated_post.py @@ -62,17 +62,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = GetProjectsPaginatedResponse.from_dict(response.json()) + return GetProjectsPaginatedResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/get_projects_projects_get.py b/src/galileo/resources/api/projects/get_projects_projects_get.py index b2d007dfe..0f25aab75 100644 --- a/src/galileo/resources/api/projects/get_projects_projects_get.py +++ b/src/galileo/resources/api/projects/get_projects_projects_get.py @@ -22,10 +22,7 @@ def _get_kwargs( params: dict[str, Any] = {} json_project_name: Union[None, Unset, str] - if isinstance(project_name, Unset): - json_project_name = UNSET - else: - json_project_name = project_name + json_project_name = UNSET if isinstance(project_name, Unset) else project_name params["project_name"] = json_project_name json_type_: Union[None, Unset, str] @@ -65,13 +62,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/list_group_project_collaborators_projects_project_id_groups_get.py b/src/galileo/resources/api/projects/list_group_project_collaborators_projects_project_id_groups_get.py index 3a5c7e4bc..3d29fd073 100644 --- a/src/galileo/resources/api/projects/list_group_project_collaborators_projects_project_id_groups_get.py +++ b/src/galileo/resources/api/projects/list_group_project_collaborators_projects_project_id_groups_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: if response.status_code == 200: - response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) + return ListGroupCollaboratorsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/list_user_project_collaborators_projects_project_id_users_get.py b/src/galileo/resources/api/projects/list_user_project_collaborators_projects_project_id_users_get.py index 087c61f64..cb1b42bc4 100644 --- a/src/galileo/resources/api/projects/list_user_project_collaborators_projects_project_id_users_get.py +++ b/src/galileo/resources/api/projects/list_user_project_collaborators_projects_project_id_users_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: if response.status_code == 200: - response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) + return ListUserCollaboratorsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/update_group_project_collaborator_projects_project_id_groups_group_id_patch.py b/src/galileo/resources/api/projects/update_group_project_collaborator_projects_project_id_groups_group_id_patch.py index 54b6c9f57..e2fbd520a 100644 --- a/src/galileo/resources/api/projects/update_group_project_collaborator_projects_project_id_groups_group_id_patch.py +++ b/src/galileo/resources/api/projects/update_group_project_collaborator_projects_project_id_groups_group_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: if response.status_code == 200: - response_200 = GroupCollaborator.from_dict(response.json()) + return GroupCollaborator.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/update_project_projects_project_id_put.py b/src/galileo/resources/api/projects/update_project_projects_project_id_put.py index e304b20a2..3d98dcdd9 100644 --- a/src/galileo/resources/api/projects/update_project_projects_project_id_put.py +++ b/src/galileo/resources/api/projects/update_project_projects_project_id_put.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: if response.status_code == 200: - response_200 = ProjectUpdateResponse.from_dict(response.json()) + return ProjectUpdateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/update_user_project_collaborator_projects_project_id_users_user_id_patch.py b/src/galileo/resources/api/projects/update_user_project_collaborator_projects_project_id_users_user_id_patch.py index 1caa06468..39af285fc 100644 --- a/src/galileo/resources/api/projects/update_user_project_collaborator_projects_project_id_users_user_id_patch.py +++ b/src/galileo/resources/api/projects/update_user_project_collaborator_projects_project_id_users_user_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: if response.status_code == 200: - response_200 = UserCollaborator.from_dict(response.json()) + return UserCollaborator.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/projects/upload_file_projects_project_id_upload_file_post.py b/src/galileo/resources/api/projects/upload_file_projects_project_id_upload_file_post.py index 9171781d2..907ede5db 100644 --- a/src/galileo/resources/api/projects/upload_file_projects_project_id_upload_file_post.py +++ b/src/galileo/resources/api/projects/upload_file_projects_project_id_upload_file_post.py @@ -34,16 +34,13 @@ def _get_kwargs(project_id: str, *, body: BodyUploadFileProjectsProjectIdUploadF def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/prompts/bulk_delete_global_templates_templates_bulk_delete_delete.py b/src/galileo/resources/api/prompts/bulk_delete_global_templates_templates_bulk_delete_delete.py index 790d13c97..1df6ffc53 100644 --- a/src/galileo/resources/api/prompts/bulk_delete_global_templates_templates_bulk_delete_delete.py +++ b/src/galileo/resources/api/prompts/bulk_delete_global_templates_templates_bulk_delete_delete.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -25,19 +26,19 @@ def _get_kwargs(*, body: BulkDeletePromptTemplatesRequest) -> dict[str, Any]: headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[HTTPValidationError]: if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[HTTPValidationError]: diff --git a/src/galileo/resources/api/prompts/create_global_prompt_template_templates_post.py b/src/galileo/resources/api/prompts/create_global_prompt_template_templates_post.py index b1acf22ad..8b8d6a8ab 100644 --- a/src/galileo/resources/api/prompts/create_global_prompt_template_templates_post.py +++ b/src/galileo/resources/api/prompts/create_global_prompt_template_templates_post.py @@ -33,17 +33,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/create_global_prompt_template_version_templates_template_id_versions_post.py b/src/galileo/resources/api/prompts/create_global_prompt_template_version_templates_template_id_versions_post.py index 71600f73b..936127636 100644 --- a/src/galileo/resources/api/prompts/create_global_prompt_template_version_templates_template_id_versions_post.py +++ b/src/galileo/resources/api/prompts/create_global_prompt_template_version_templates_template_id_versions_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + return BasePromptTemplateVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/create_group_prompt_template_collaborators_templates_template_id_groups_post.py b/src/galileo/resources/api/prompts/create_group_prompt_template_collaborators_templates_template_id_groups_post.py index 652adbcc8..4d98d8670 100644 --- a/src/galileo/resources/api/prompts/create_group_prompt_template_collaborators_templates_template_id_groups_post.py +++ b/src/galileo/resources/api/prompts/create_group_prompt_template_collaborators_templates_template_id_groups_post.py @@ -49,13 +49,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/create_prompt_template_version_projects_project_id_templates_template_id_versions_post.py b/src/galileo/resources/api/prompts/create_prompt_template_version_projects_project_id_templates_template_id_versions_post.py index 137fdec72..c5286b547 100644 --- a/src/galileo/resources/api/prompts/create_prompt_template_version_projects_project_id_templates_template_id_versions_post.py +++ b/src/galileo/resources/api/prompts/create_prompt_template_version_projects_project_id_templates_template_id_versions_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + return BasePromptTemplateVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/create_prompt_template_with_version_projects_project_id_templates_post.py b/src/galileo/resources/api/prompts/create_prompt_template_with_version_projects_project_id_templates_post.py index b7f7d2964..9ab79abf4 100644 --- a/src/galileo/resources/api/prompts/create_prompt_template_with_version_projects_project_id_templates_post.py +++ b/src/galileo/resources/api/prompts/create_prompt_template_with_version_projects_project_id_templates_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/create_user_prompt_template_collaborators_templates_template_id_users_post.py b/src/galileo/resources/api/prompts/create_user_prompt_template_collaborators_templates_template_id_users_post.py index ca5f48a3b..1ecea270c 100644 --- a/src/galileo/resources/api/prompts/create_user_prompt_template_collaborators_templates_template_id_users_post.py +++ b/src/galileo/resources/api/prompts/create_user_prompt_template_collaborators_templates_template_id_users_post.py @@ -49,13 +49,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/delete_global_template_templates_template_id_delete.py b/src/galileo/resources/api/prompts/delete_global_template_templates_template_id_delete.py index 4d72c57fa..68df3f53b 100644 --- a/src/galileo/resources/api/prompts/delete_global_template_templates_template_id_delete.py +++ b/src/galileo/resources/api/prompts/delete_global_template_templates_template_id_delete.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = DeletePromptResponse.from_dict(response.json()) + return DeletePromptResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/delete_group_prompt_template_collaborator_templates_template_id_groups_group_id_delete.py b/src/galileo/resources/api/prompts/delete_group_prompt_template_collaborator_templates_template_id_groups_group_id_delete.py index 666c2e284..b36329ac4 100644 --- a/src/galileo/resources/api/prompts/delete_group_prompt_template_collaborator_templates_template_id_groups_group_id_delete.py +++ b/src/galileo/resources/api/prompts/delete_group_prompt_template_collaborator_templates_template_id_groups_group_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(template_id: str, group_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/prompts/delete_template_projects_project_id_templates_template_id_delete.py b/src/galileo/resources/api/prompts/delete_template_projects_project_id_templates_template_id_delete.py index b03acc060..18dd26fba 100644 --- a/src/galileo/resources/api/prompts/delete_template_projects_project_id_templates_template_id_delete.py +++ b/src/galileo/resources/api/prompts/delete_template_projects_project_id_templates_template_id_delete.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = DeletePromptResponse.from_dict(response.json()) + return DeletePromptResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/delete_user_prompt_template_collaborator_templates_template_id_users_user_id_delete.py b/src/galileo/resources/api/prompts/delete_user_prompt_template_collaborator_templates_template_id_users_user_id_delete.py index dca9625b9..30904aa05 100644 --- a/src/galileo/resources/api/prompts/delete_user_prompt_template_collaborator_templates_template_id_users_user_id_delete.py +++ b/src/galileo/resources/api/prompts/delete_user_prompt_template_collaborator_templates_template_id_users_user_id_delete.py @@ -29,16 +29,13 @@ def _get_kwargs(template_id: str, user_id: str) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/prompts/generate_template_input_stub_input_stub_post.py b/src/galileo/resources/api/prompts/generate_template_input_stub_input_stub_post.py index d692d1a34..db9dc77e0 100644 --- a/src/galileo/resources/api/prompts/generate_template_input_stub_input_stub_post.py +++ b/src/galileo/resources/api/prompts/generate_template_input_stub_input_stub_post.py @@ -30,16 +30,13 @@ def _get_kwargs(*, body: TemplateStubRequest) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/prompts/get_global_template_templates_template_id_get.py b/src/galileo/resources/api/prompts/get_global_template_templates_template_id_get.py index 0fcdc139f..8f96dccfb 100644 --- a/src/galileo/resources/api/prompts/get_global_template_templates_template_id_get.py +++ b/src/galileo/resources/api/prompts/get_global_template_templates_template_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/get_global_template_version_templates_template_id_versions_version_get.py b/src/galileo/resources/api/prompts/get_global_template_version_templates_template_id_versions_version_get.py index 778677a21..3eab53cad 100644 --- a/src/galileo/resources/api/prompts/get_global_template_version_templates_template_id_versions_version_get.py +++ b/src/galileo/resources/api/prompts/get_global_template_version_templates_template_id_versions_version_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + return BasePromptTemplateVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/get_project_templates_projects_project_id_templates_get.py b/src/galileo/resources/api/prompts/get_project_templates_projects_project_id_templates_get.py index 7e4da556d..7a5a667fe 100644 --- a/src/galileo/resources/api/prompts/get_project_templates_projects_project_id_templates_get.py +++ b/src/galileo/resources/api/prompts/get_project_templates_projects_project_id_templates_get.py @@ -41,13 +41,11 @@ def _parse_response( return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/get_template_from_project_projects_project_id_templates_template_id_get.py b/src/galileo/resources/api/prompts/get_template_from_project_projects_project_id_templates_template_id_get.py index 64e273985..adc27ecb2 100644 --- a/src/galileo/resources/api/prompts/get_template_from_project_projects_project_id_templates_template_id_get.py +++ b/src/galileo/resources/api/prompts/get_template_from_project_projects_project_id_templates_template_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/get_template_version_by_name_projects_project_id_templates_versions_get.py b/src/galileo/resources/api/prompts/get_template_version_by_name_projects_project_id_templates_versions_get.py index 34fa39482..546017daf 100644 --- a/src/galileo/resources/api/prompts/get_template_version_by_name_projects_project_id_templates_versions_get.py +++ b/src/galileo/resources/api/prompts/get_template_version_by_name_projects_project_id_templates_versions_get.py @@ -21,10 +21,7 @@ def _get_kwargs(project_id: str, *, template_name: str, version: Union[None, Uns params["template_name"] = template_name json_version: Union[None, Unset, int] - if isinstance(version, Unset): - json_version = UNSET - else: - json_version = version + json_version = UNSET if isinstance(version, Unset) else version params["version"] = json_version params = {k: v for k, v in params.items() if v is not UNSET and v is not None} @@ -46,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + return BasePromptTemplateVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/get_template_version_projects_project_id_templates_template_id_versions_version_get.py b/src/galileo/resources/api/prompts/get_template_version_projects_project_id_templates_template_id_versions_version_get.py index 92c5ca2d1..85ccf55fd 100644 --- a/src/galileo/resources/api/prompts/get_template_version_projects_project_id_templates_template_id_versions_version_get.py +++ b/src/galileo/resources/api/prompts/get_template_version_projects_project_id_templates_template_id_versions_version_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + return BasePromptTemplateVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/list_group_prompt_template_collaborators_templates_template_id_groups_get.py b/src/galileo/resources/api/prompts/list_group_prompt_template_collaborators_templates_template_id_groups_get.py index 6771f9166..1a91f92d2 100644 --- a/src/galileo/resources/api/prompts/list_group_prompt_template_collaborators_templates_template_id_groups_get.py +++ b/src/galileo/resources/api/prompts/list_group_prompt_template_collaborators_templates_template_id_groups_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: if response.status_code == 200: - response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) + return ListGroupCollaboratorsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/list_user_prompt_template_collaborators_templates_template_id_users_get.py b/src/galileo/resources/api/prompts/list_user_prompt_template_collaborators_templates_template_id_users_get.py index 9c4100d33..435e47af6 100644 --- a/src/galileo/resources/api/prompts/list_user_prompt_template_collaborators_templates_template_id_users_get.py +++ b/src/galileo/resources/api/prompts/list_user_prompt_template_collaborators_templates_template_id_users_get.py @@ -43,17 +43,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: if response.status_code == 200: - response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) + return ListUserCollaboratorsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/query_template_versions_templates_template_id_versions_query_post.py b/src/galileo/resources/api/prompts/query_template_versions_templates_template_id_versions_query_post.py index 29220b6d3..51c353d6c 100644 --- a/src/galileo/resources/api/prompts/query_template_versions_templates_template_id_versions_query_post.py +++ b/src/galileo/resources/api/prompts/query_template_versions_templates_template_id_versions_query_post.py @@ -52,17 +52,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: if response.status_code == 200: - response_200 = ListPromptTemplateVersionResponse.from_dict(response.json()) + return ListPromptTemplateVersionResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/query_templates_templates_query_post.py b/src/galileo/resources/api/prompts/query_templates_templates_query_post.py index dcc78f486..19d5b75b7 100644 --- a/src/galileo/resources/api/prompts/query_templates_templates_query_post.py +++ b/src/galileo/resources/api/prompts/query_templates_templates_query_post.py @@ -48,17 +48,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: if response.status_code == 200: - response_200 = ListPromptTemplateResponse.from_dict(response.json()) + return ListPromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/render_template_render_template_post.py b/src/galileo/resources/api/prompts/render_template_render_template_post.py index 62110dfcb..5022acefb 100644 --- a/src/galileo/resources/api/prompts/render_template_render_template_post.py +++ b/src/galileo/resources/api/prompts/render_template_render_template_post.py @@ -48,17 +48,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: if response.status_code == 200: - response_200 = RenderTemplateResponse.from_dict(response.json()) + return RenderTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/set_selected_global_template_version_templates_template_id_versions_version_put.py b/src/galileo/resources/api/prompts/set_selected_global_template_version_templates_template_id_versions_version_put.py index 5b7f64073..09a1701d4 100644 --- a/src/galileo/resources/api/prompts/set_selected_global_template_version_templates_template_id_versions_version_put.py +++ b/src/galileo/resources/api/prompts/set_selected_global_template_version_templates_template_id_versions_version_put.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/set_selected_template_version_projects_project_id_templates_template_id_versions_version_put.py b/src/galileo/resources/api/prompts/set_selected_template_version_projects_project_id_templates_template_id_versions_version_put.py index 1d7f29fd2..e6a13c636 100644 --- a/src/galileo/resources/api/prompts/set_selected_template_version_projects_project_id_templates_template_id_versions_version_put.py +++ b/src/galileo/resources/api/prompts/set_selected_template_version_projects_project_id_templates_template_id_versions_version_put.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/update_global_template_templates_template_id_patch.py b/src/galileo/resources/api/prompts/update_global_template_templates_template_id_patch.py index 3501fdd6b..31a39d3d8 100644 --- a/src/galileo/resources/api/prompts/update_global_template_templates_template_id_patch.py +++ b/src/galileo/resources/api/prompts/update_global_template_templates_template_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = BasePromptTemplateResponse.from_dict(response.json()) + return BasePromptTemplateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/update_group_prompt_template_collaborator_templates_template_id_groups_group_id_patch.py b/src/galileo/resources/api/prompts/update_group_prompt_template_collaborator_templates_template_id_groups_group_id_patch.py index 868c00094..9905aaf11 100644 --- a/src/galileo/resources/api/prompts/update_group_prompt_template_collaborator_templates_template_id_groups_group_id_patch.py +++ b/src/galileo/resources/api/prompts/update_group_prompt_template_collaborator_templates_template_id_groups_group_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: if response.status_code == 200: - response_200 = GroupCollaborator.from_dict(response.json()) + return GroupCollaborator.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/prompts/update_user_prompt_template_collaborator_templates_template_id_users_user_id_patch.py b/src/galileo/resources/api/prompts/update_user_prompt_template_collaborator_templates_template_id_users_user_id_patch.py index d0ec1e843..5c5ee76a7 100644 --- a/src/galileo/resources/api/prompts/update_user_prompt_template_collaborator_templates_template_id_users_user_id_patch.py +++ b/src/galileo/resources/api/prompts/update_user_prompt_template_collaborator_templates_template_id_users_user_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: if response.status_code == 200: - response_200 = UserCollaborator.from_dict(response.json()) + return UserCollaborator.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/protect/create_stage_projects_project_id_stages_post.py b/src/galileo/resources/api/protect/create_stage_projects_project_id_stages_post.py index 39b6b6b52..9a609fa61 100644 --- a/src/galileo/resources/api/protect/create_stage_projects_project_id_stages_post.py +++ b/src/galileo/resources/api/protect/create_stage_projects_project_id_stages_post.py @@ -35,17 +35,14 @@ def _get_kwargs(project_id: str, *, body: StageWithRulesets) -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: if response.status_code == 200: - response_200 = StageDB.from_dict(response.json()) + return StageDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, StageDB]]: diff --git a/src/galileo/resources/api/protect/get_stage_projects_project_id_stages_get.py b/src/galileo/resources/api/protect/get_stage_projects_project_id_stages_get.py index 6bf26ebe3..38ba325e6 100644 --- a/src/galileo/resources/api/protect/get_stage_projects_project_id_stages_get.py +++ b/src/galileo/resources/api/protect/get_stage_projects_project_id_stages_get.py @@ -21,17 +21,11 @@ def _get_kwargs( params: dict[str, Any] = {} json_stage_name: Union[None, Unset, str] - if isinstance(stage_name, Unset): - json_stage_name = UNSET - else: - json_stage_name = stage_name + json_stage_name = UNSET if isinstance(stage_name, Unset) else stage_name params["stage_name"] = json_stage_name json_stage_id: Union[None, Unset, str] - if isinstance(stage_id, Unset): - json_stage_id = UNSET - else: - json_stage_id = stage_id + json_stage_id = UNSET if isinstance(stage_id, Unset) else stage_id params["stage_id"] = json_stage_id params = {k: v for k, v in params.items() if v is not UNSET and v is not None} @@ -51,17 +45,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: if response.status_code == 200: - response_200 = StageDB.from_dict(response.json()) + return StageDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, StageDB]]: diff --git a/src/galileo/resources/api/protect/invoke_protect_invoke_post.py b/src/galileo/resources/api/protect/invoke_protect_invoke_post.py index d682648d4..8a0f354f4 100644 --- a/src/galileo/resources/api/protect/invoke_protect_invoke_post.py +++ b/src/galileo/resources/api/protect/invoke_protect_invoke_post.py @@ -1,27 +1,44 @@ from http import HTTPStatus -from typing import Any, Optional, Union +from typing import Any, Optional, Union, cast import httpx from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient - +from ...types import Response, UNSET from ... import errors + from ...models.http_validation_error import HTTPValidationError from ...models.invoke_response import InvokeResponse from ...models.protect_request import ProtectRequest from ...models.protect_response import ProtectResponse -from ...types import Response +from typing import cast +from typing import cast, Union + + +def _get_kwargs( + *, + body: ProtectRequest, -def _get_kwargs(*, body: ProtectRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/protect/invoke"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/protect/invoke", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -30,24 +47,25 @@ def _get_kwargs(*, body: ProtectRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, Union["InvokeResponse", "ProtectResponse"]]]: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: if response.status_code == 200: - - def _parse_response_200(data: object) -> Union["InvokeResponse", "ProtectResponse"]: + def _parse_response_200(data: object) -> Union['InvokeResponse', 'ProtectResponse']: try: if not isinstance(data, dict): raise TypeError() response_200_type_0 = ProtectResponse.from_dict(data) + + return response_200_type_0 - except: # noqa: E722 + except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() response_200_type_1 = InvokeResponse.from_dict(data) + + return response_200_type_1 response_200 = _parse_response_200(response.json()) @@ -56,6 +74,8 @@ def _parse_response_200(data: object) -> Union["InvokeResponse", "ProtectRespons if response.status_code == 422: response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -63,9 +83,7 @@ def _parse_response_200(data: object) -> Union["InvokeResponse", "ProtectRespons return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, Union["InvokeResponse", "ProtectResponse"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,9 +93,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: ProtectRequest -) -> Response[Union[HTTPValidationError, Union["InvokeResponse", "ProtectResponse"]]]: - """Invoke + *, + client: ApiClient, + body: ProtectRequest, + +) -> Response[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: + """ Invoke Args: body (ProtectRequest): Protect request schema with custom OpenAPI title. @@ -88,19 +109,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - *, client: ApiClient, body: ProtectRequest -) -> Optional[Union[HTTPValidationError, Union["InvokeResponse", "ProtectResponse"]]]: - """Invoke + *, + client: ApiClient, + body: ProtectRequest, + +) -> Optional[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: + """ Invoke Args: body (ProtectRequest): Protect request schema with custom OpenAPI title. @@ -111,15 +140,22 @@ def sync( Returns: Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: ProtectRequest -) -> Response[Union[HTTPValidationError, Union["InvokeResponse", "ProtectResponse"]]]: - """Invoke + *, + client: ApiClient, + body: ProtectRequest, + +) -> Response[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: + """ Invoke Args: body (ProtectRequest): Protect request schema with custom OpenAPI title. @@ -130,19 +166,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]] - """ + """ - kwargs = _get_kwargs(body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - *, client: ApiClient, body: ProtectRequest -) -> Optional[Union[HTTPValidationError, Union["InvokeResponse", "ProtectResponse"]]]: - """Invoke + *, + client: ApiClient, + body: ProtectRequest, + +) -> Optional[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: + """ Invoke Args: body (ProtectRequest): Protect request schema with custom OpenAPI title. @@ -153,6 +197,11 @@ async def asyncio( Returns: Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']] - """ + """ + + + return (await asyncio_detailed( + client=client, +body=body, - return (await asyncio_detailed(client=client, body=body)).parsed + )).parsed diff --git a/src/galileo/resources/api/protect/pause_stage_projects_project_id_stages_stage_id_put.py b/src/galileo/resources/api/protect/pause_stage_projects_project_id_stages_stage_id_put.py index edecb3680..b61466e7c 100644 --- a/src/galileo/resources/api/protect/pause_stage_projects_project_id_stages_stage_id_put.py +++ b/src/galileo/resources/api/protect/pause_stage_projects_project_id_stages_stage_id_put.py @@ -37,17 +37,14 @@ def _get_kwargs(project_id: str, stage_id: str, *, pause: Union[Unset, bool] = F def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: if response.status_code == 200: - response_200 = StageDB.from_dict(response.json()) + return StageDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, StageDB]]: diff --git a/src/galileo/resources/api/protect/update_stage_projects_project_id_stages_stage_id_post.py b/src/galileo/resources/api/protect/update_stage_projects_project_id_stages_stage_id_post.py index 1681a06d5..67870711c 100644 --- a/src/galileo/resources/api/protect/update_stage_projects_project_id_stages_stage_id_post.py +++ b/src/galileo/resources/api/protect/update_stage_projects_project_id_stages_stage_id_post.py @@ -35,17 +35,14 @@ def _get_kwargs(project_id: str, stage_id: str, *, body: RulesetsMixin) -> dict[ def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: if response.status_code == 200: - response_200 = StageDB.from_dict(response.json()) + return StageDB.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, StageDB]]: diff --git a/src/galileo/resources/api/run_scorer_settings/get_settings_projects_project_id_runs_run_id_scorer_settings_get.py b/src/galileo/resources/api/run_scorer_settings/get_settings_projects_project_id_runs_run_id_scorer_settings_get.py index 60bd8a2a0..43c5de42a 100644 --- a/src/galileo/resources/api/run_scorer_settings/get_settings_projects_project_id_runs_run_id_scorer_settings_get.py +++ b/src/galileo/resources/api/run_scorer_settings/get_settings_projects_project_id_runs_run_id_scorer_settings_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: if response.status_code == 200: - response_200 = RunScorerSettingsResponse.from_dict(response.json()) + return RunScorerSettingsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_patch.py b/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_patch.py index 2a83fb090..c94d286c4 100644 --- a/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_patch.py +++ b/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: if response.status_code == 200: - response_200 = RunScorerSettingsResponse.from_dict(response.json()) + return RunScorerSettingsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_post.py b/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_post.py index bcfab0b03..993d7ca45 100644 --- a/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_post.py +++ b/src/galileo/resources/api/run_scorer_settings/upsert_scorers_config_projects_project_id_runs_run_id_scorer_settings_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: if response.status_code == 200: - response_200 = RunScorerSettingsResponse.from_dict(response.json()) + return RunScorerSettingsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/create_session_projects_project_id_sessions_post.py b/src/galileo/resources/api/trace/create_session_projects_project_id_sessions_post.py index c429a803e..ad3e3d593 100644 --- a/src/galileo/resources/api/trace/create_session_projects_project_id_sessions_post.py +++ b/src/galileo/resources/api/trace/create_session_projects_project_id_sessions_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: if response.status_code == 200: - response_200 = SessionCreateResponse.from_dict(response.json()) + return SessionCreateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/delete_sessions_projects_project_id_sessions_delete_post.py b/src/galileo/resources/api/trace/delete_sessions_projects_project_id_sessions_delete_post.py index 1eaf28cc8..91897ae8f 100644 --- a/src/galileo/resources/api/trace/delete_sessions_projects_project_id_sessions_delete_post.py +++ b/src/galileo/resources/api/trace/delete_sessions_projects_project_id_sessions_delete_post.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -26,6 +27,8 @@ def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs @@ -34,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: if response.status_code == 200: - response_200 = LogRecordsDeleteResponse.from_dict(response.json()) + return LogRecordsDeleteResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/delete_spans_projects_project_id_spans_delete_post.py b/src/galileo/resources/api/trace/delete_spans_projects_project_id_spans_delete_post.py index e35169d78..2f77b6540 100644 --- a/src/galileo/resources/api/trace/delete_spans_projects_project_id_spans_delete_post.py +++ b/src/galileo/resources/api/trace/delete_spans_projects_project_id_spans_delete_post.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -26,6 +27,8 @@ def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs @@ -34,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: if response.status_code == 200: - response_200 = LogRecordsDeleteResponse.from_dict(response.json()) + return LogRecordsDeleteResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/delete_traces_projects_project_id_traces_delete_post.py b/src/galileo/resources/api/trace/delete_traces_projects_project_id_traces_delete_post.py index 8d6a4fe95..f1561b5c3 100644 --- a/src/galileo/resources/api/trace/delete_traces_projects_project_id_traces_delete_post.py +++ b/src/galileo/resources/api/trace/delete_traces_projects_project_id_traces_delete_post.py @@ -3,6 +3,7 @@ import httpx +from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient @@ -26,6 +27,8 @@ def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, headers["Content-Type"] = "application/json" + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" + _kwargs["content_headers"] = headers return _kwargs @@ -34,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: if response.status_code == 200: - response_200 = LogRecordsDeleteResponse.from_dict(response.json()) + return LogRecordsDeleteResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/export_records_projects_project_id_export_records_post.py b/src/galileo/resources/api/trace/export_records_projects_project_id_export_records_post.py index 597bcf4ba..54e5e4203 100644 --- a/src/galileo/resources/api/trace/export_records_projects_project_id_export_records_post.py +++ b/src/galileo/resources/api/trace/export_records_projects_project_id_export_records_post.py @@ -34,16 +34,13 @@ def _get_kwargs(project_id: str, *, body: LogRecordsExportRequest) -> dict[str, def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: @@ -55,6 +52,12 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) +def stream_detailed(project_id: str, *, client: ApiClient, body: LogRecordsExportRequest) -> httpx.Response: + kwargs = _get_kwargs(project_id=project_id, body=body) + + return client.request(**kwargs) + + def sync_detailed( project_id: str, *, client: ApiClient, body: LogRecordsExportRequest ) -> Response[Union[Any, HTTPValidationError]]: @@ -80,14 +83,6 @@ def sync_detailed( return _build_response(client=client, response=response) -def stream_detailed(project_id: str, *, client: ApiClient, body: LogRecordsExportRequest) -> httpx.Response: - kwargs = _get_kwargs(project_id=project_id, body=body) - - response = client.request(**kwargs) - - return response - - def sync( project_id: str, *, client: ApiClient, body: LogRecordsExportRequest ) -> Optional[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/trace/get_aggregated_trace_view_projects_project_id_traces_aggregated_post.py b/src/galileo/resources/api/trace/get_aggregated_trace_view_projects_project_id_traces_aggregated_post.py index 0c170d997..1156b88d3 100644 --- a/src/galileo/resources/api/trace/get_aggregated_trace_view_projects_project_id_traces_aggregated_post.py +++ b/src/galileo/resources/api/trace/get_aggregated_trace_view_projects_project_id_traces_aggregated_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: if response.status_code == 200: - response_200 = AggregatedTraceViewResponse.from_dict(response.json()) + return AggregatedTraceViewResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/get_session_projects_project_id_sessions_session_id_get.py b/src/galileo/resources/api/trace/get_session_projects_project_id_sessions_session_id_get.py index ba7982716..4dc6c139a 100644 --- a/src/galileo/resources/api/trace/get_session_projects_project_id_sessions_session_id_get.py +++ b/src/galileo/resources/api/trace/get_session_projects_project_id_sessions_session_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExtendedSessionRecordWithChildren.from_dict(response.json()) + return ExtendedSessionRecordWithChildren.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/get_span_projects_project_id_spans_span_id_get.py b/src/galileo/resources/api/trace/get_span_projects_project_id_spans_span_id_get.py index 826473f67..f3a827e28 100644 --- a/src/galileo/resources/api/trace/get_span_projects_project_id_spans_span_id_get.py +++ b/src/galileo/resources/api/trace/get_span_projects_project_id_spans_span_id_get.py @@ -1,98 +1,99 @@ from http import HTTPStatus -from typing import Any, Optional, Union +from typing import Any, Optional, Union, cast import httpx from galileo.utils.headers_data import get_package_version from galileo_core.constants.request_method import RequestMethod from galileo_core.helpers.api_client import ApiClient - +from ...types import Response, UNSET from ... import errors + from ...models.extended_agent_span_record_with_children import ExtendedAgentSpanRecordWithChildren from ...models.extended_llm_span_record import ExtendedLlmSpanRecord from ...models.extended_retriever_span_record_with_children import ExtendedRetrieverSpanRecordWithChildren from ...models.extended_tool_span_record_with_children import ExtendedToolSpanRecordWithChildren from ...models.extended_workflow_span_record_with_children import ExtendedWorkflowSpanRecordWithChildren from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast +from typing import cast, Union + -def _get_kwargs(project_id: str, span_id: str) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + span_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/spans/{span_id}", + "path": "/projects/{project_id}/spans/{span_id}".format(project_id=project_id,span_id=span_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[ - Union[ - HTTPValidationError, - Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ], - ] -]: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: if response.status_code == 200: - - def _parse_response_200( - data: object, - ) -> Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ]: + def _parse_response_200(data: object) -> Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']: try: if not isinstance(data, dict): raise TypeError() response_200_type_0 = ExtendedAgentSpanRecordWithChildren.from_dict(data) + + return response_200_type_0 - except: # noqa: E722 + except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() response_200_type_1 = ExtendedWorkflowSpanRecordWithChildren.from_dict(data) + + return response_200_type_1 - except: # noqa: E722 + except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() response_200_type_2 = ExtendedLlmSpanRecord.from_dict(data) + + return response_200_type_2 - except: # noqa: E722 + except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() response_200_type_3 = ExtendedToolSpanRecordWithChildren.from_dict(data) + + return response_200_type_3 - except: # noqa: E722 + except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() response_200_type_4 = ExtendedRetrieverSpanRecordWithChildren.from_dict(data) + + return response_200_type_4 response_200 = _parse_response_200(response.json()) @@ -101,6 +102,8 @@ def _parse_response_200( if response.status_code == 422: response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -108,20 +111,7 @@ def _parse_response_200( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[ - Union[ - HTTPValidationError, - Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ], - ] -]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -131,20 +121,13 @@ def _build_response( def sync_detailed( - project_id: str, span_id: str, *, client: ApiClient -) -> Response[ - Union[ - HTTPValidationError, - Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ], - ] -]: - """Get Span + project_id: str, + span_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: + """ Get Span Args: project_id (str): @@ -156,30 +139,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, span_id=span_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +span_id=span_id, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, span_id: str, *, client: ApiClient -) -> Optional[ - Union[ - HTTPValidationError, - Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ], - ] -]: - """Get Span + project_id: str, + span_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: + """ Get Span Args: project_id (str): @@ -191,26 +173,24 @@ def sync( Returns: Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']] - """ + """ + - return sync_detailed(project_id=project_id, span_id=span_id, client=client).parsed + return sync_detailed( + project_id=project_id, +span_id=span_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, span_id: str, *, client: ApiClient -) -> Response[ - Union[ - HTTPValidationError, - Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ], - ] -]: - """Get Span + project_id: str, + span_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: + """ Get Span Args: project_id (str): @@ -222,30 +202,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, span_id=span_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +span_id=span_id, - return _build_response(client=client, response=response) + ) + response = await client.arequest( + **kwargs + ) + + return _build_response(client=client, response=response) async def asyncio( - project_id: str, span_id: str, *, client: ApiClient -) -> Optional[ - Union[ - HTTPValidationError, - Union[ - "ExtendedAgentSpanRecordWithChildren", - "ExtendedLlmSpanRecord", - "ExtendedRetrieverSpanRecordWithChildren", - "ExtendedToolSpanRecordWithChildren", - "ExtendedWorkflowSpanRecordWithChildren", - ], - ] -]: - """Get Span + project_id: str, + span_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: + """ Get Span Args: project_id (str): @@ -257,6 +236,12 @@ async def asyncio( Returns: Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +span_id=span_id, +client=client, - return (await asyncio_detailed(project_id=project_id, span_id=span_id, client=client)).parsed + )).parsed diff --git a/src/galileo/resources/api/trace/get_trace_projects_project_id_traces_trace_id_get.py b/src/galileo/resources/api/trace/get_trace_projects_project_id_traces_trace_id_get.py index 8489683d2..6e5ac14b3 100644 --- a/src/galileo/resources/api/trace/get_trace_projects_project_id_traces_trace_id_get.py +++ b/src/galileo/resources/api/trace/get_trace_projects_project_id_traces_trace_id_get.py @@ -32,17 +32,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: if response.status_code == 200: - response_200 = ExtendedTraceRecordWithChildren.from_dict(response.json()) + return ExtendedTraceRecordWithChildren.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/log_spans_projects_project_id_spans_post.py b/src/galileo/resources/api/trace/log_spans_projects_project_id_spans_post.py index 63d1e0a2f..980a1d95d 100644 --- a/src/galileo/resources/api/trace/log_spans_projects_project_id_spans_post.py +++ b/src/galileo/resources/api/trace/log_spans_projects_project_id_spans_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: if response.status_code == 200: - response_200 = LogSpansIngestResponse.from_dict(response.json()) + return LogSpansIngestResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/log_traces_projects_project_id_traces_post.py b/src/galileo/resources/api/trace/log_traces_projects_project_id_traces_post.py index 2fa97b7ba..b0087e414 100644 --- a/src/galileo/resources/api/trace/log_traces_projects_project_id_traces_post.py +++ b/src/galileo/resources/api/trace/log_traces_projects_project_id_traces_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: if response.status_code == 200: - response_200 = LogTracesIngestResponse.from_dict(response.json()) + return LogTracesIngestResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/metrics_testing_available_columns_projects_project_id_metrics_testing_available_columns_post.py b/src/galileo/resources/api/trace/metrics_testing_available_columns_projects_project_id_metrics_testing_available_columns_post.py index dad211212..048105f53 100644 --- a/src/galileo/resources/api/trace/metrics_testing_available_columns_projects_project_id_metrics_testing_available_columns_post.py +++ b/src/galileo/resources/api/trace/metrics_testing_available_columns_projects_project_id_metrics_testing_available_columns_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: if response.status_code == 200: - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/query_metrics_projects_project_id_metrics_search_post.py b/src/galileo/resources/api/trace/query_metrics_projects_project_id_metrics_search_post.py index 34f0284a0..eebf5984f 100644 --- a/src/galileo/resources/api/trace/query_metrics_projects_project_id_metrics_search_post.py +++ b/src/galileo/resources/api/trace/query_metrics_projects_project_id_metrics_search_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: if response.status_code == 200: - response_200 = LogRecordsMetricsResponse.from_dict(response.json()) + return LogRecordsMetricsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/query_sessions_projects_project_id_sessions_search_post.py b/src/galileo/resources/api/trace/query_sessions_projects_project_id_sessions_search_post.py index a475c529e..2671bee45 100644 --- a/src/galileo/resources/api/trace/query_sessions_projects_project_id_sessions_search_post.py +++ b/src/galileo/resources/api/trace/query_sessions_projects_project_id_sessions_search_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: if response.status_code == 200: - response_200 = LogRecordsQueryResponse.from_dict(response.json()) + return LogRecordsQueryResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/query_spans_projects_project_id_spans_search_post.py b/src/galileo/resources/api/trace/query_spans_projects_project_id_spans_search_post.py index b486ca60e..7df799d39 100644 --- a/src/galileo/resources/api/trace/query_spans_projects_project_id_spans_search_post.py +++ b/src/galileo/resources/api/trace/query_spans_projects_project_id_spans_search_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: if response.status_code == 200: - response_200 = LogRecordsQueryResponse.from_dict(response.json()) + return LogRecordsQueryResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/query_traces_projects_project_id_traces_search_post.py b/src/galileo/resources/api/trace/query_traces_projects_project_id_traces_search_post.py index 02ededd50..75558e8b0 100644 --- a/src/galileo/resources/api/trace/query_traces_projects_project_id_traces_search_post.py +++ b/src/galileo/resources/api/trace/query_traces_projects_project_id_traces_search_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: if response.status_code == 200: - response_200 = LogRecordsQueryResponse.from_dict(response.json()) + return LogRecordsQueryResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/recompute_metrics_projects_project_id_recompute_metrics_post.py b/src/galileo/resources/api/trace/recompute_metrics_projects_project_id_recompute_metrics_post.py index 76195d3b2..b2a2901f0 100644 --- a/src/galileo/resources/api/trace/recompute_metrics_projects_project_id_recompute_metrics_post.py +++ b/src/galileo/resources/api/trace/recompute_metrics_projects_project_id_recompute_metrics_post.py @@ -34,16 +34,13 @@ def _get_kwargs(project_id: str, *, body: RecomputeLogRecordsMetricsRequest) -> def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: if response.status_code == 200: - response_200 = response.json() - return response_200 + return response.json() if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]: diff --git a/src/galileo/resources/api/trace/sessions_available_columns_projects_project_id_sessions_available_columns_post.py b/src/galileo/resources/api/trace/sessions_available_columns_projects_project_id_sessions_available_columns_post.py index aeecbb1cf..a18d99f6d 100644 --- a/src/galileo/resources/api/trace/sessions_available_columns_projects_project_id_sessions_available_columns_post.py +++ b/src/galileo/resources/api/trace/sessions_available_columns_projects_project_id_sessions_available_columns_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: if response.status_code == 200: - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/spans_available_columns_projects_project_id_spans_available_columns_post.py b/src/galileo/resources/api/trace/spans_available_columns_projects_project_id_spans_available_columns_post.py index 39d59ce0d..7003332cb 100644 --- a/src/galileo/resources/api/trace/spans_available_columns_projects_project_id_spans_available_columns_post.py +++ b/src/galileo/resources/api/trace/spans_available_columns_projects_project_id_spans_available_columns_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: if response.status_code == 200: - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/traces_available_columns_projects_project_id_traces_available_columns_post.py b/src/galileo/resources/api/trace/traces_available_columns_projects_project_id_traces_available_columns_post.py index 6116530de..d1b589500 100644 --- a/src/galileo/resources/api/trace/traces_available_columns_projects_project_id_traces_available_columns_post.py +++ b/src/galileo/resources/api/trace/traces_available_columns_projects_project_id_traces_available_columns_post.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: if response.status_code == 200: - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/update_span_projects_project_id_spans_span_id_patch.py b/src/galileo/resources/api/trace/update_span_projects_project_id_spans_span_id_patch.py index 7b5aa1178..52b8d1956 100644 --- a/src/galileo/resources/api/trace/update_span_projects_project_id_spans_span_id_patch.py +++ b/src/galileo/resources/api/trace/update_span_projects_project_id_spans_span_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: if response.status_code == 200: - response_200 = LogSpanUpdateResponse.from_dict(response.json()) + return LogSpanUpdateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/api/trace/update_trace_projects_project_id_traces_trace_id_patch.py b/src/galileo/resources/api/trace/update_trace_projects_project_id_traces_trace_id_patch.py index 59959439d..43aacdb98 100644 --- a/src/galileo/resources/api/trace/update_trace_projects_project_id_traces_trace_id_patch.py +++ b/src/galileo/resources/api/trace/update_trace_projects_project_id_traces_trace_id_patch.py @@ -37,17 +37,14 @@ def _parse_response( *, client: ApiClient, response: httpx.Response ) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: if response.status_code == 200: - response_200 = LogTraceUpdateResponse.from_dict(response.json()) + return LogTraceUpdateResponse.from_dict(response.json()) - return response_200 if response.status_code == 422: - response_422 = HTTPValidationError.from_dict(response.json()) + return HTTPValidationError.from_dict(response.json()) - return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) - else: - return None + return None def _build_response( diff --git a/src/galileo/resources/models/__init__.py b/src/galileo/resources/models/__init__.py index b52a8b13f..255ec21ae 100644 --- a/src/galileo/resources/models/__init__.py +++ b/src/galileo/resources/models/__init__.py @@ -48,6 +48,10 @@ from .bucketed_metric import BucketedMetric from .bucketed_metric_buckets import BucketedMetricBuckets from .bucketed_metrics import BucketedMetrics +from .bulk_delete_datasets_request import BulkDeleteDatasetsRequest +from .bulk_delete_datasets_response import BulkDeleteDatasetsResponse +from .bulk_delete_failure import BulkDeleteFailure +from .bulk_delete_prompt_templates_request import BulkDeletePromptTemplatesRequest from .chain_poll_template import ChainPollTemplate from .chunk_attribution_utilization_scorer import ChunkAttributionUtilizationScorer from .chunk_attribution_utilization_scorer_type import ChunkAttributionUtilizationScorerType @@ -184,6 +188,7 @@ from .dataset_version_db import DatasetVersionDB from .dataset_version_index_sort import DatasetVersionIndexSort from .delete_prompt_response import DeletePromptResponse +from .delete_run_response import DeleteRunResponse from .delete_scorer_response import DeleteScorerResponse from .document import Document from .document_metadata import DocumentMetadata @@ -278,7 +283,7 @@ from .feedback_aggregate import FeedbackAggregate from .feedback_rating_db import FeedbackRatingDB from .feedback_rating_info import FeedbackRatingInfo -from .feedback_rating_info_feedback_type import FeedbackRatingInfoFeedbackType +from .feedback_type import FeedbackType from .few_shot_example import FewShotExample from .fine_tuned_scorer import FineTunedScorer from .fine_tuned_scorer_action import FineTunedScorerAction @@ -353,6 +358,8 @@ from .log_records_boolean_filter import LogRecordsBooleanFilter from .log_records_date_filter import LogRecordsDateFilter from .log_records_date_filter_operator import LogRecordsDateFilterOperator +from .log_records_delete_request import LogRecordsDeleteRequest +from .log_records_delete_response import LogRecordsDeleteResponse from .log_records_export_request import LogRecordsExportRequest from .log_records_id_filter import LogRecordsIDFilter from .log_records_id_filter_operator import LogRecordsIDFilterOperator @@ -510,6 +517,7 @@ from .run_params_map import RunParamsMap from .run_scorer_settings_patch_request import RunScorerSettingsPatchRequest from .run_scorer_settings_response import RunScorerSettingsResponse +from .run_tag_create_request import RunTagCreateRequest from .run_tag_db import RunTagDB from .score_aggregate import ScoreAggregate from .score_rating import ScoreRating @@ -602,16 +610,16 @@ __all__ = ( "ActionResult", "ActionType", + "AgentSpan", + "AgentSpanDatasetMetadata", + "AgentSpanUserMetadata", + "AgentType", "AgenticSessionSuccessScorer", "AgenticSessionSuccessScorerType", "AgenticSessionSuccessTemplate", "AgenticWorkflowSuccessScorer", "AgenticWorkflowSuccessScorerType", "AgenticWorkflowSuccessTemplate", - "AgentSpan", - "AgentSpanDatasetMetadata", - "AgentSpanUserMetadata", - "AgentType", "AggregatedTraceViewEdge", "AggregatedTraceViewGraph", "AggregatedTraceViewNode", @@ -642,6 +650,10 @@ "BucketedMetric", "BucketedMetricBuckets", "BucketedMetrics", + "BulkDeleteDatasetsRequest", + "BulkDeleteDatasetsResponse", + "BulkDeleteFailure", + "BulkDeletePromptTemplatesRequest", "ChainPollTemplate", "ChunkAttributionUtilizationScorer", "ChunkAttributionUtilizationScorerType", @@ -685,12 +697,12 @@ "CustomizedFactualityGPTScorer", "CustomizedFactualityGPTScorerAggregatesType0", "CustomizedFactualityGPTScorerExtraType0", - "CustomizedGroundednessGPTScorer", - "CustomizedGroundednessGPTScorerAggregatesType0", - "CustomizedGroundednessGPTScorerExtraType0", "CustomizedGroundTruthAdherenceGPTScorer", "CustomizedGroundTruthAdherenceGPTScorerAggregatesType0", "CustomizedGroundTruthAdherenceGPTScorerExtraType0", + "CustomizedGroundednessGPTScorer", + "CustomizedGroundednessGPTScorerAggregatesType0", + "CustomizedGroundednessGPTScorerExtraType0", "CustomizedInputSexistGPTScorer", "CustomizedInputSexistGPTScorerAggregatesType0", "CustomizedInputSexistGPTScorerExtraType0", @@ -715,6 +727,9 @@ "CustomizedToxicityGPTScorer", "CustomizedToxicityGPTScorerAggregatesType0", "CustomizedToxicityGPTScorerExtraType0", + "DataType", + "DataTypeOptions", + "DataUnit", "DatasetAction", "DatasetAppendRow", "DatasetAppendRowValues", @@ -724,8 +739,8 @@ "DatasetContentFilterOperator", "DatasetContentSortClause", "DatasetCreatedAtSort", - "DatasetData", "DatasetDB", + "DatasetData", "DatasetDeleteRow", "DatasetDraftFilter", "DatasetDraftFilterOperator", @@ -745,21 +760,19 @@ "DatasetProjectsSort", "DatasetRow", "DatasetRowMetadata", - "DatasetRowsSort", "DatasetRowValuesDict", "DatasetRowValuesDictAdditionalPropertyType3", "DatasetRowValuesItemType3", - "DatasetUpdatedAtSort", + "DatasetRowsSort", "DatasetUpdateRow", "DatasetUpdateRowValues", "DatasetUpdateRowValuesAdditionalPropertyType3", + "DatasetUpdatedAtSort", "DatasetUsedInProjectFilter", "DatasetVersionDB", "DatasetVersionIndexSort", - "DataType", - "DataTypeOptions", - "DataUnit", "DeletePromptResponse", + "DeleteRunResponse", "DeleteScorerResponse", "Document", "DocumentMetadata", @@ -775,8 +788,8 @@ "ExperimentResponseAggregateFeedback", "ExperimentResponseAggregateMetrics", "ExperimentResponseTags", - "ExperimentsAvailableColumnsResponse", "ExperimentUpdateRequest", + "ExperimentsAvailableColumnsResponse", "ExtendedAgentSpanRecord", "ExtendedAgentSpanRecordDatasetMetadata", "ExtendedAgentSpanRecordMetricInfoType0", @@ -836,7 +849,7 @@ "FeedbackAggregate", "FeedbackRatingDB", "FeedbackRatingInfo", - "FeedbackRatingInfoFeedbackType", + "FeedbackType", "FewShotExample", "FineTunedScorer", "FineTunedScorerAction", @@ -848,19 +861,19 @@ "GenerationResponse", "GetProjectsPaginatedResponse", "GetProjectsPaginatedResponseV2", - "GroundednessTemplate", "GroundTruthAdherenceScorer", "GroundTruthAdherenceTemplate", + "GroundednessTemplate", "GroupAction", "GroupCollaborator", "GroupCollaboratorCreate", "GroupMemberAction", + "HTTPValidationError", "HallucinationSegment", "HealthcheckResponse", "Histogram", "HistogramBucket", "HistogramStrategy", - "HTTPValidationError", "InputMap", "InputPIIScorer", "InputSexistScorer", @@ -882,6 +895,8 @@ "JobDB", "JobDBRequestData", "JobProgress", + "LLMExportFormat", + "LLMIntegration", "LikeDislikeAggregate", "LikeDislikeRating", "ListDatasetParams", @@ -895,23 +910,22 @@ "ListPromptTemplateResponse", "ListPromptTemplateVersionParams", "ListPromptTemplateVersionResponse", + "ListScorerVersionsResponse", "ListScorersRequest", "ListScorersResponse", - "ListScorerVersionsResponse", "ListUserCollaboratorsResponse", - "LLMExportFormat", - "LLMIntegration", "LlmMetrics", "LlmSpan", "LlmSpanDatasetMetadata", "LlmSpanToolsType0Item", "LlmSpanUserMetadata", - "LoggingMethod", "LogRecordsAvailableColumnsRequest", "LogRecordsAvailableColumnsResponse", "LogRecordsBooleanFilter", "LogRecordsDateFilter", "LogRecordsDateFilterOperator", + "LogRecordsDeleteRequest", + "LogRecordsDeleteResponse", "LogRecordsExportRequest", "LogRecordsIDFilter", "LogRecordsIDFilterOperator", @@ -926,17 +940,18 @@ "LogRecordsSortClause", "LogRecordsTextFilter", "LogRecordsTextFilterOperator", - "LogSpansIngestRequest", - "LogSpansIngestResponse", "LogSpanUpdateRequest", "LogSpanUpdateResponse", + "LogSpansIngestRequest", + "LogSpansIngestResponse", "LogStreamCreateRequest", "LogStreamResponse", "LogStreamUpdateRequest", - "LogTracesIngestRequest", - "LogTracesIngestResponse", "LogTraceUpdateRequest", "LogTraceUpdateResponse", + "LogTracesIngestRequest", + "LogTracesIngestResponse", + "LoggingMethod", "ManualLlmValidateScorersLlmValidatePostBody", "Message", "MessageRole", @@ -956,10 +971,10 @@ "MetricNotApplicable", "MetricNotComputed", "MetricPending", - "Metrics", - "MetricsTestingAvailableColumnsRequest", "MetricSuccess", "MetricThreshold", + "Metrics", + "MetricsTestingAvailableColumnsRequest", "Model", "ModelCostBy", "ModelType", @@ -988,10 +1003,10 @@ "ProjectBookmarkSort", "ProjectCollectionParams", "ProjectCreate", + "ProjectCreateResponse", "ProjectCreatedAtFilter", "ProjectCreatedAtFilterOperator", "ProjectCreatedAtSort", - "ProjectCreateResponse", "ProjectCreatorFilter", "ProjectCreatorFilterOperator", "ProjectDB", @@ -1012,10 +1027,10 @@ "ProjectTypeFilterOperator", "ProjectTypeSort", "ProjectUpdate", + "ProjectUpdateResponse", "ProjectUpdatedAtFilter", "ProjectUpdatedAtFilterOperator", "ProjectUpdatedAtSort", - "ProjectUpdateResponse", "PromptDatasetDB", "PromptInjectionScorer", "PromptInjectionScorerType", @@ -1048,9 +1063,9 @@ "RecomputeSettingsRuns", "RegisteredScorer", "RegisteredScorerAction", - "RenderedTemplate", "RenderTemplateRequest", "RenderTemplateResponse", + "RenderedTemplate", "RetrieverSpan", "RetrieverSpanDatasetMetadata", "RetrieverSpanUserMetadata", @@ -1068,6 +1083,7 @@ "RunParamsMap", "RunScorerSettingsPatchRequest", "RunScorerSettingsResponse", + "RunTagCreateRequest", "RunTagDB", "ScoreAggregate", "ScoreRating", @@ -1084,7 +1100,6 @@ "ScorerNameFilterOperator", "ScorerNameSort", "ScorerResponse", - "ScorersConfiguration", "ScorerTagsFilter", "ScorerTagsFilterOperator", "ScorerType", @@ -1093,6 +1108,7 @@ "ScorerTypes", "ScorerUpdatedAtFilter", "ScorerUpdatedAtFilterOperator", + "ScorersConfiguration", "Segment", "SegmentFilter", "SessionCreateRequest", @@ -1108,10 +1124,10 @@ "StepType", "StringData", "SubscriptionConfig", - "SyntheticDatasetExtensionRequest", - "SyntheticDatasetExtensionResponse", "SyntheticDataSourceDataset", "SyntheticDataTypes", + "SyntheticDatasetExtensionRequest", + "SyntheticDatasetExtensionResponse", "SystemMetricInfo", "TagsAggregate", "TagsAggregateCounts", diff --git a/src/galileo/resources/models/agent_span.py b/src/galileo/resources/models/agent_span.py index 06302d68b..e4c9aa3e2 100644 --- a/src/galileo/resources/models/agent_span.py +++ b/src/galileo/resources/models/agent_span.py @@ -94,32 +94,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -154,10 +142,7 @@ def to_dict(self) -> dict[str, Any]: output = self.output parent_id: Union[None, Unset, str] - if isinstance(self.parent_id, Unset): - parent_id = UNSET - else: - parent_id = self.parent_id + parent_id = UNSET if isinstance(self.parent_id, Unset) else self.parent_id redacted_input: Union[None, Unset, list[dict[str, Any]], str] if isinstance(self.redacted_input, Unset): @@ -186,23 +171,14 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, AgentSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, WorkflowSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, LlmSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, RetrieverSpan): + if isinstance(spans_item_data, (AgentSpan, WorkflowSpan, LlmSpan, RetrieverSpan)): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -210,26 +186,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -302,17 +269,11 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) _agent_type = d.pop("agent_type", UNSET) agent_type: Union[Unset, AgentType] - if isinstance(_agent_type, Unset): - agent_type = UNSET - else: - agent_type = AgentType(_agent_type) + agent_type = UNSET if isinstance(_agent_type, Unset) else AgentType(_agent_type) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -379,10 +340,7 @@ def _parse_input_(data: object) -> Union[Unset, list["Message"], str]: _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) name = d.pop("name", UNSET) @@ -394,9 +352,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -455,9 +412,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -496,40 +452,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = AgentSpan.from_dict(data) + return AgentSpan.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = WorkflowSpan.from_dict(data) + return WorkflowSpan.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = LlmSpan.from_dict(data) + return LlmSpan.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = RetrieverSpan.from_dict(data) + return RetrieverSpan.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ToolSpan.from_dict(data) - - return spans_item_type_4 + return ToolSpan.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -570,10 +520,7 @@ def _parse_trace_id(data: object) -> Union[None, Unset, str]: _user_metadata = d.pop("user_metadata", UNSET) user_metadata: Union[Unset, AgentSpanUserMetadata] - if isinstance(_user_metadata, Unset): - user_metadata = UNSET - else: - user_metadata = AgentSpanUserMetadata.from_dict(_user_metadata) + user_metadata = UNSET if isinstance(_user_metadata, Unset) else AgentSpanUserMetadata.from_dict(_user_metadata) agent_span = cls( agent_type=agent_type, diff --git a/src/galileo/resources/models/agentic_session_success_scorer.py b/src/galileo/resources/models/agentic_session_success_scorer.py index 80561160e..e068c582a 100644 --- a/src/galileo/resources/models/agentic_session_success_scorer.py +++ b/src/galileo/resources/models/agentic_session_success_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, AgenticSessionSuccessScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = AgenticSessionSuccessScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else AgenticSessionSuccessScorerType(_type_) agentic_session_success_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/agentic_workflow_success_scorer.py b/src/galileo/resources/models/agentic_workflow_success_scorer.py index a36ef0c3c..96307b4a8 100644 --- a/src/galileo/resources/models/agentic_workflow_success_scorer.py +++ b/src/galileo/resources/models/agentic_workflow_success_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, AgenticWorkflowSuccessScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = AgenticWorkflowSuccessScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else AgenticWorkflowSuccessScorerType(_type_) agentic_workflow_success_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/aggregated_trace_view_request.py b/src/galileo/resources/models/aggregated_trace_view_request.py index ad9c32f77..7b2d0e3b3 100644 --- a/src/galileo/resources/models/aggregated_trace_view_request.py +++ b/src/galileo/resources/models/aggregated_trace_view_request.py @@ -55,13 +55,10 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -103,40 +100,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) diff --git a/src/galileo/resources/models/aggregated_trace_view_response.py b/src/galileo/resources/models/aggregated_trace_view_response.py index ef3a4ab5c..77b7306c8 100644 --- a/src/galileo/resources/models/aggregated_trace_view_response.py +++ b/src/galileo/resources/models/aggregated_trace_view_response.py @@ -93,9 +93,8 @@ def _parse_end_time(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - end_time_type_0 = isoparse(data) + return isoparse(data) - return end_time_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) @@ -110,9 +109,8 @@ def _parse_start_time(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - start_time_type_0 = isoparse(data) + return isoparse(data) - return start_time_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/base_finetuned_scorer_db.py b/src/galileo/resources/models/base_finetuned_scorer_db.py index 9df3391d4..79f849ad1 100644 --- a/src/galileo/resources/models/base_finetuned_scorer_db.py +++ b/src/galileo/resources/models/base_finetuned_scorer_db.py @@ -73,9 +73,8 @@ def _parse_executor(data: object) -> Union[CoreScorerName, None, Unset]: try: if not isinstance(data, str): raise TypeError() - executor_type_0 = CoreScorerName(data) + return CoreScorerName(data) - return executor_type_0 except: # noqa: E722 pass return cast(Union[CoreScorerName, None, Unset], data) diff --git a/src/galileo/resources/models/base_generated_scorer_db.py b/src/galileo/resources/models/base_generated_scorer_db.py index d8062e0e2..8270de714 100644 --- a/src/galileo/resources/models/base_generated_scorer_db.py +++ b/src/galileo/resources/models/base_generated_scorer_db.py @@ -40,16 +40,10 @@ def to_dict(self) -> dict[str, Any]: name = self.name instructions: Union[None, Unset, str] - if isinstance(self.instructions, Unset): - instructions = UNSET - else: - instructions = self.instructions + instructions = UNSET if isinstance(self.instructions, Unset) else self.instructions user_prompt: Union[None, Unset, str] - if isinstance(self.user_prompt, Unset): - user_prompt = UNSET - else: - user_prompt = self.user_prompt + user_prompt = UNSET if isinstance(self.user_prompt, Unset) else self.user_prompt field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/base_prompt_template_response.py b/src/galileo/resources/models/base_prompt_template_response.py index 11b33674b..76ff61086 100644 --- a/src/galileo/resources/models/base_prompt_template_response.py +++ b/src/galileo/resources/models/base_prompt_template_response.py @@ -72,10 +72,7 @@ def to_dict(self) -> dict[str, Any]: max_version = self.max_version name: Union[dict[str, Any], str] - if isinstance(self.name, Name): - name = self.name.to_dict() - else: - name = self.name + name = self.name.to_dict() if isinstance(self.name, Name) else self.name selected_version = self.selected_version.to_dict() @@ -143,9 +140,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) @@ -160,9 +156,8 @@ def _parse_name(data: object) -> Union["Name", str]: try: if not isinstance(data, dict): raise TypeError() - name_type_1 = Name.from_dict(data) + return Name.from_dict(data) - return name_type_1 except: # noqa: E722 pass return cast(Union["Name", str], data) diff --git a/src/galileo/resources/models/base_prompt_template_version.py b/src/galileo/resources/models/base_prompt_template_version.py index 64eeea6d7..cb9795a76 100644 --- a/src/galileo/resources/models/base_prompt_template_version.py +++ b/src/galileo/resources/models/base_prompt_template_version.py @@ -44,10 +44,7 @@ def to_dict(self) -> dict[str, Any]: template = self.template output_type: Union[None, Unset, str] - if isinstance(self.output_type, Unset): - output_type = UNSET - else: - output_type = self.output_type + output_type = UNSET if isinstance(self.output_type, Unset) else self.output_type raw = self.raw @@ -56,10 +53,7 @@ def to_dict(self) -> dict[str, Any]: settings = self.settings.to_dict() version: Union[None, Unset, int] - if isinstance(self.version, Unset): - version = UNSET - else: - version = self.version + version = UNSET if isinstance(self.version, Unset) else self.version field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -113,10 +107,7 @@ def _parse_output_type(data: object) -> Union[None, Unset, str]: _settings = d.pop("settings", UNSET) settings: Union[Unset, PromptRunSettings] - if isinstance(_settings, Unset): - settings = UNSET - else: - settings = PromptRunSettings.from_dict(_settings) + settings = UNSET if isinstance(_settings, Unset) else PromptRunSettings.from_dict(_settings) def _parse_version(data: object) -> Union[None, Unset, int]: if data is None: diff --git a/src/galileo/resources/models/base_prompt_template_version_response.py b/src/galileo/resources/models/base_prompt_template_version_response.py index 92abb28ac..9dcaadf3f 100644 --- a/src/galileo/resources/models/base_prompt_template_version_response.py +++ b/src/galileo/resources/models/base_prompt_template_version_response.py @@ -98,10 +98,7 @@ def to_dict(self) -> dict[str, Any]: lines_removed = self.lines_removed output_type: Union[None, Unset, str] - if isinstance(self.output_type, Unset): - output_type = UNSET - else: - output_type = self.output_type + output_type = UNSET if isinstance(self.output_type, Unset) else self.output_type raw = self.raw @@ -151,9 +148,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) diff --git a/src/galileo/resources/models/base_scorer.py b/src/galileo/resources/models/base_scorer.py index 6fdcc1342..bdeedda67 100644 --- a/src/galileo/resources/models/base_scorer.py +++ b/src/galileo/resources/models/base_scorer.py @@ -103,10 +103,7 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[None, Unset, dict[str, Any]] if isinstance(self.chainpoll_template, Unset): @@ -117,16 +114,10 @@ def to_dict(self) -> dict[str, Any]: chainpoll_template = self.chainpoll_template cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -154,16 +145,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -183,30 +168,18 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias: Union[None, Unset, str] - if isinstance(self.model_alias, Unset): - model_alias = UNSET - else: - model_alias = self.model_alias + model_alias = UNSET if isinstance(self.model_alias, Unset) else self.model_alias name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -217,18 +190,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -334,9 +301,8 @@ def _parse_aggregate_keys(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - aggregate_keys_type_0 = cast(list[str], data) + return cast(list[str], data) - return aggregate_keys_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -351,9 +317,8 @@ def _parse_aggregates(data: object) -> Union["BaseScorerAggregatesType0", None, try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = BaseScorerAggregatesType0.from_dict(data) + return BaseScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["BaseScorerAggregatesType0", None, Unset], data) @@ -377,9 +342,8 @@ def _parse_chainpoll_template(data: object) -> Union["ChainPollTemplate", None, try: if not isinstance(data, dict): raise TypeError() - chainpoll_template_type_0 = ChainPollTemplate.from_dict(data) + return ChainPollTemplate.from_dict(data) - return chainpoll_template_type_0 except: # noqa: E722 pass return cast(Union["ChainPollTemplate", None, Unset], data) @@ -412,9 +376,8 @@ def _parse_extra(data: object) -> Union["BaseScorerExtraType0", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = BaseScorerExtraType0.from_dict(data) + return BaseScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["BaseScorerExtraType0", None, Unset], data) @@ -437,16 +400,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -485,9 +445,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -502,9 +461,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -557,9 +515,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -618,9 +575,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/base_scorer_version_db.py b/src/galileo/resources/models/base_scorer_version_db.py index 64e958262..2fe3e11ce 100644 --- a/src/galileo/resources/models/base_scorer_version_db.py +++ b/src/galileo/resources/models/base_scorer_version_db.py @@ -62,10 +62,7 @@ def to_dict(self) -> dict[str, Any]: version = self.version cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled finetuned_scorer: Union[None, Unset, dict[str, Any]] if isinstance(self.finetuned_scorer, Unset): @@ -92,16 +89,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -180,9 +171,8 @@ def _parse_finetuned_scorer(data: object) -> Union["BaseFinetunedScorerDB", None try: if not isinstance(data, dict): raise TypeError() - finetuned_scorer_type_0 = BaseFinetunedScorerDB.from_dict(data) + return BaseFinetunedScorerDB.from_dict(data) - return finetuned_scorer_type_0 except: # noqa: E722 pass return cast(Union["BaseFinetunedScorerDB", None, Unset], data) @@ -197,9 +187,8 @@ def _parse_generated_scorer(data: object) -> Union["BaseGeneratedScorerDB", None try: if not isinstance(data, dict): raise TypeError() - generated_scorer_type_0 = BaseGeneratedScorerDB.from_dict(data) + return BaseGeneratedScorerDB.from_dict(data) - return generated_scorer_type_0 except: # noqa: E722 pass return cast(Union["BaseGeneratedScorerDB", None, Unset], data) @@ -214,9 +203,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -249,9 +237,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -266,9 +253,8 @@ def _parse_registered_scorer(data: object) -> Union["BaseRegisteredScorerDB", No try: if not isinstance(data, dict): raise TypeError() - registered_scorer_type_0 = BaseRegisteredScorerDB.from_dict(data) + return BaseRegisteredScorerDB.from_dict(data) - return registered_scorer_type_0 except: # noqa: E722 pass return cast(Union["BaseRegisteredScorerDB", None, Unset], data) @@ -283,9 +269,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/base_scorer_version_response.py b/src/galileo/resources/models/base_scorer_version_response.py index fae9de1b8..292e21c4c 100644 --- a/src/galileo/resources/models/base_scorer_version_response.py +++ b/src/galileo/resources/models/base_scorer_version_response.py @@ -80,10 +80,7 @@ def to_dict(self) -> dict[str, Any]: chain_poll_template = self.chain_poll_template cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled finetuned_scorer: Union[None, Unset, dict[str, Any]] if isinstance(self.finetuned_scorer, Unset): @@ -110,16 +107,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -196,9 +187,8 @@ def _parse_chain_poll_template(data: object) -> Union["ChainPollTemplate", None, try: if not isinstance(data, dict): raise TypeError() - chain_poll_template_type_0 = ChainPollTemplate.from_dict(data) + return ChainPollTemplate.from_dict(data) - return chain_poll_template_type_0 except: # noqa: E722 pass return cast(Union["ChainPollTemplate", None, Unset], data) @@ -222,9 +212,8 @@ def _parse_finetuned_scorer(data: object) -> Union["FineTunedScorerResponse", No try: if not isinstance(data, dict): raise TypeError() - finetuned_scorer_type_0 = FineTunedScorerResponse.from_dict(data) + return FineTunedScorerResponse.from_dict(data) - return finetuned_scorer_type_0 except: # noqa: E722 pass return cast(Union["FineTunedScorerResponse", None, Unset], data) @@ -239,9 +228,8 @@ def _parse_generated_scorer(data: object) -> Union["GeneratedScorerResponse", No try: if not isinstance(data, dict): raise TypeError() - generated_scorer_type_0 = GeneratedScorerResponse.from_dict(data) + return GeneratedScorerResponse.from_dict(data) - return generated_scorer_type_0 except: # noqa: E722 pass return cast(Union["GeneratedScorerResponse", None, Unset], data) @@ -256,9 +244,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -291,9 +278,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -308,9 +294,8 @@ def _parse_registered_scorer(data: object) -> Union["CreateUpdateRegisteredScore try: if not isinstance(data, dict): raise TypeError() - registered_scorer_type_0 = CreateUpdateRegisteredScorerResponse.from_dict(data) + return CreateUpdateRegisteredScorerResponse.from_dict(data) - return registered_scorer_type_0 except: # noqa: E722 pass return cast(Union["CreateUpdateRegisteredScorerResponse", None, Unset], data) @@ -325,9 +310,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/bleu_scorer.py b/src/galileo/resources/models/bleu_scorer.py index 9e3c4295c..3c366ac77 100644 --- a/src/galileo/resources/models/bleu_scorer.py +++ b/src/galileo/resources/models/bleu_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/body_create_dataset_datasets_post.py b/src/galileo/resources/models/body_create_dataset_datasets_post.py index 1a432cda3..0b3bcebfd 100644 --- a/src/galileo/resources/models/body_create_dataset_datasets_post.py +++ b/src/galileo/resources/models/body_create_dataset_datasets_post.py @@ -37,10 +37,7 @@ def to_dict(self) -> dict[str, Any]: append_suffix_if_duplicate = self.append_suffix_if_duplicate copy_from_dataset_id: Union[None, Unset, str] - if isinstance(self.copy_from_dataset_id, Unset): - copy_from_dataset_id = UNSET - else: - copy_from_dataset_id = self.copy_from_dataset_id + copy_from_dataset_id = UNSET if isinstance(self.copy_from_dataset_id, Unset) else self.copy_from_dataset_id copy_from_dataset_version_index: Union[None, Unset, int] if isinstance(self.copy_from_dataset_version_index, Unset): @@ -62,10 +59,7 @@ def to_dict(self) -> dict[str, Any]: hidden = self.hidden name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -175,9 +169,8 @@ def _parse_file(data: object) -> Union[File, None, Unset]: try: if not isinstance(data, bytes): raise TypeError() - file_type_0 = File(payload=BytesIO(data)) + return File(payload=BytesIO(data)) - return file_type_0 except: # noqa: E722 pass return cast(Union[File, None, Unset], data) diff --git a/src/galileo/resources/models/body_login_email_login_post.py b/src/galileo/resources/models/body_login_email_login_post.py index 756cb64be..47ef8688b 100644 --- a/src/galileo/resources/models/body_login_email_login_post.py +++ b/src/galileo/resources/models/body_login_email_login_post.py @@ -35,22 +35,13 @@ def to_dict(self) -> dict[str, Any]: username = self.username client_id: Union[None, Unset, str] - if isinstance(self.client_id, Unset): - client_id = UNSET - else: - client_id = self.client_id + client_id = UNSET if isinstance(self.client_id, Unset) else self.client_id client_secret: Union[None, Unset, str] - if isinstance(self.client_secret, Unset): - client_secret = UNSET - else: - client_secret = self.client_secret + client_secret = UNSET if isinstance(self.client_secret, Unset) else self.client_secret grant_type: Union[None, Unset, str] - if isinstance(self.grant_type, Unset): - grant_type = UNSET - else: - grant_type = self.grant_type + grant_type = UNSET if isinstance(self.grant_type, Unset) else self.grant_type scope = self.scope diff --git a/src/galileo/resources/models/body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py b/src/galileo/resources/models/body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py index 74aca1eb1..e7a0c9a8b 100644 --- a/src/galileo/resources/models/body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py +++ b/src/galileo/resources/models/body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put.py @@ -85,9 +85,8 @@ def _parse_column_names(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - column_names_type_0 = cast(list[str], data) + return cast(list[str], data) - return column_names_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -102,9 +101,8 @@ def _parse_file(data: object) -> Union[File, None, Unset]: try: if not isinstance(data, bytes): raise TypeError() - file_type_0 = File(payload=BytesIO(data)) + return File(payload=BytesIO(data)) - return file_type_0 except: # noqa: E722 pass return cast(Union[File, None, Unset], data) diff --git a/src/galileo/resources/models/bucketed_metric.py b/src/galileo/resources/models/bucketed_metric.py index 01baae035..77e044414 100644 --- a/src/galileo/resources/models/bucketed_metric.py +++ b/src/galileo/resources/models/bucketed_metric.py @@ -33,10 +33,7 @@ def to_dict(self) -> dict[str, Any]: name = self.name average: Union[None, Unset, float] - if isinstance(self.average, Unset): - average = UNSET - else: - average = self.average + average = UNSET if isinstance(self.average, Unset) else self.average field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/chain_poll_template.py b/src/galileo/resources/models/chain_poll_template.py index 5880d2d30..1d2757caa 100644 --- a/src/galileo/resources/models/chain_poll_template.py +++ b/src/galileo/resources/models/chain_poll_template.py @@ -43,10 +43,7 @@ def to_dict(self) -> dict[str, Any]: explanation_field_name = self.explanation_field_name metric_description: Union[None, Unset, str] - if isinstance(self.metric_description, Unset): - metric_description = UNSET - else: - metric_description = self.metric_description + metric_description = UNSET if isinstance(self.metric_description, Unset) else self.metric_description metric_few_shot_examples: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.metric_few_shot_examples, Unset): @@ -56,10 +53,7 @@ def to_dict(self) -> dict[str, Any]: metric_few_shot_examples.append(metric_few_shot_examples_item) metric_system_prompt: Union[None, Unset, str] - if isinstance(self.metric_system_prompt, Unset): - metric_system_prompt = UNSET - else: - metric_system_prompt = self.metric_system_prompt + metric_system_prompt = UNSET if isinstance(self.metric_system_prompt, Unset) else self.metric_system_prompt value_field_name = self.value_field_name diff --git a/src/galileo/resources/models/chunk_attribution_utilization_scorer.py b/src/galileo/resources/models/chunk_attribution_utilization_scorer.py index ccaa0ed1f..7b1e0aae3 100644 --- a/src/galileo/resources/models/chunk_attribution_utilization_scorer.py +++ b/src/galileo/resources/models/chunk_attribution_utilization_scorer.py @@ -54,10 +54,7 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name @@ -102,16 +99,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -139,10 +133,7 @@ def _parse_model_name(data: object) -> Union[None, Unset, str]: _type_ = d.pop("type", UNSET) type_: Union[Unset, ChunkAttributionUtilizationScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = ChunkAttributionUtilizationScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else ChunkAttributionUtilizationScorerType(_type_) chunk_attribution_utilization_scorer = cls(filters=filters, model_name=model_name, name=name, type_=type_) diff --git a/src/galileo/resources/models/chunk_attribution_utilization_template.py b/src/galileo/resources/models/chunk_attribution_utilization_template.py index ccaf17d95..ba35b3c68 100644 --- a/src/galileo/resources/models/chunk_attribution_utilization_template.py +++ b/src/galileo/resources/models/chunk_attribution_utilization_template.py @@ -52,10 +52,7 @@ def to_dict(self) -> dict[str, Any]: explanation_field_name = self.explanation_field_name metric_description: Union[None, Unset, str] - if isinstance(self.metric_description, Unset): - metric_description = UNSET - else: - metric_description = self.metric_description + metric_description = UNSET if isinstance(self.metric_description, Unset) else self.metric_description metric_few_shot_examples: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.metric_few_shot_examples, Unset): @@ -65,10 +62,7 @@ def to_dict(self) -> dict[str, Any]: metric_few_shot_examples.append(metric_few_shot_examples_item) metric_system_prompt: Union[None, Unset, str] - if isinstance(self.metric_system_prompt, Unset): - metric_system_prompt = UNSET - else: - metric_system_prompt = self.metric_system_prompt + metric_system_prompt = UNSET if isinstance(self.metric_system_prompt, Unset) else self.metric_system_prompt template = self.template diff --git a/src/galileo/resources/models/column_info.py b/src/galileo/resources/models/column_info.py index 8ff946a63..5cdb093dd 100644 --- a/src/galileo/resources/models/column_info.py +++ b/src/galileo/resources/models/column_info.py @@ -112,18 +112,12 @@ def to_dict(self) -> dict[str, Any]: data_unit = self.data_unit description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description filterable = self.filterable group_label: Union[None, Unset, str] - if isinstance(self.group_label, Unset): - group_label = UNSET - else: - group_label = self.group_label + group_label = UNSET if isinstance(self.group_label, Unset) else self.group_label insight_type: Union[None, Unset, str] if isinstance(self.insight_type, Unset): @@ -138,10 +132,7 @@ def to_dict(self) -> dict[str, Any]: is_optional = self.is_optional label: Union[None, Unset, str] - if isinstance(self.label, Unset): - label = UNSET - else: - label = self.label + label = UNSET if isinstance(self.label, Unset) else self.label multi_valued = self.multi_valued @@ -154,10 +145,7 @@ def to_dict(self) -> dict[str, Any]: scorer_config = self.scorer_config scorer_id: Union[None, Unset, str] - if isinstance(self.scorer_id, Unset): - scorer_id = UNSET - else: - scorer_id = self.scorer_id + scorer_id = UNSET if isinstance(self.scorer_id, Unset) else self.scorer_id sortable = self.sortable @@ -227,9 +215,8 @@ def _parse_allowed_values(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - allowed_values_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return allowed_values_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) @@ -253,9 +240,8 @@ def _parse_data_type(data: object) -> Union[DataType, None, Unset]: try: if not isinstance(data, str): raise TypeError() - data_type_type_0 = DataType(data) + return DataType(data) - return data_type_type_0 except: # noqa: E722 pass return cast(Union[DataType, None, Unset], data) @@ -270,9 +256,8 @@ def _parse_data_unit(data: object) -> Union[DataUnit, None, Unset]: try: if not isinstance(data, str): raise TypeError() - data_unit_type_0 = DataUnit(data) + return DataUnit(data) - return data_unit_type_0 except: # noqa: E722 pass return cast(Union[DataUnit, None, Unset], data) @@ -307,9 +292,8 @@ def _parse_insight_type(data: object) -> Union[InsightType, None, Unset]: try: if not isinstance(data, str): raise TypeError() - insight_type_type_0 = InsightType(data) + return InsightType(data) - return insight_type_type_0 except: # noqa: E722 pass return cast(Union[InsightType, None, Unset], data) @@ -339,9 +323,8 @@ def _parse_scorer_config(data: object) -> Union["ScorerConfig", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - scorer_config_type_0 = ScorerConfig.from_dict(data) + return ScorerConfig.from_dict(data) - return scorer_config_type_0 except: # noqa: E722 pass return cast(Union["ScorerConfig", None, Unset], data) @@ -367,9 +350,8 @@ def _parse_threshold(data: object) -> Union["MetricThreshold", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - threshold_type_0 = MetricThreshold.from_dict(data) + return MetricThreshold.from_dict(data) - return threshold_type_0 except: # noqa: E722 pass return cast(Union["MetricThreshold", None, Unset], data) diff --git a/src/galileo/resources/models/column_mapping.py b/src/galileo/resources/models/column_mapping.py index 713854832..f9730a325 100644 --- a/src/galileo/resources/models/column_mapping.py +++ b/src/galileo/resources/models/column_mapping.py @@ -73,17 +73,15 @@ def _parse_input_(data: object) -> Union["ColumnMappingConfig", None, list[str]] try: if not isinstance(data, dict): raise TypeError() - input_type_0 = ColumnMappingConfig.from_dict(data) + return ColumnMappingConfig.from_dict(data) - return input_type_0 except: # noqa: E722 pass try: if not isinstance(data, list): raise TypeError() - input_type_1 = cast(list[str], data) + return cast(list[str], data) - return input_type_1 except: # noqa: E722 pass return cast(Union["ColumnMappingConfig", None, list[str]], data) @@ -96,17 +94,15 @@ def _parse_metadata(data: object) -> Union["ColumnMappingConfig", None, list[str try: if not isinstance(data, dict): raise TypeError() - metadata_type_0 = ColumnMappingConfig.from_dict(data) + return ColumnMappingConfig.from_dict(data) - return metadata_type_0 except: # noqa: E722 pass try: if not isinstance(data, list): raise TypeError() - metadata_type_1 = cast(list[str], data) + return cast(list[str], data) - return metadata_type_1 except: # noqa: E722 pass return cast(Union["ColumnMappingConfig", None, list[str]], data) @@ -119,17 +115,15 @@ def _parse_output(data: object) -> Union["ColumnMappingConfig", None, list[str]] try: if not isinstance(data, dict): raise TypeError() - output_type_0 = ColumnMappingConfig.from_dict(data) + return ColumnMappingConfig.from_dict(data) - return output_type_0 except: # noqa: E722 pass try: if not isinstance(data, list): raise TypeError() - output_type_1 = cast(list[str], data) + return cast(list[str], data) - return output_type_1 except: # noqa: E722 pass return cast(Union["ColumnMappingConfig", None, list[str]], data) diff --git a/src/galileo/resources/models/completeness_scorer.py b/src/galileo/resources/models/completeness_scorer.py index 594949f53..17308645c 100644 --- a/src/galileo/resources/models/completeness_scorer.py +++ b/src/galileo/resources/models/completeness_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, CompletenessScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = CompletenessScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else CompletenessScorerType(_type_) completeness_scorer = cls(filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_) diff --git a/src/galileo/resources/models/completeness_template.py b/src/galileo/resources/models/completeness_template.py index 53e6c7e7f..3b07bcff9 100644 --- a/src/galileo/resources/models/completeness_template.py +++ b/src/galileo/resources/models/completeness_template.py @@ -52,10 +52,7 @@ def to_dict(self) -> dict[str, Any]: explanation_field_name = self.explanation_field_name metric_description: Union[None, Unset, str] - if isinstance(self.metric_description, Unset): - metric_description = UNSET - else: - metric_description = self.metric_description + metric_description = UNSET if isinstance(self.metric_description, Unset) else self.metric_description metric_few_shot_examples: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.metric_few_shot_examples, Unset): @@ -65,10 +62,7 @@ def to_dict(self) -> dict[str, Any]: metric_few_shot_examples.append(metric_few_shot_examples_item) metric_system_prompt: Union[None, Unset, str] - if isinstance(self.metric_system_prompt, Unset): - metric_system_prompt = UNSET - else: - metric_system_prompt = self.metric_system_prompt + metric_system_prompt = UNSET if isinstance(self.metric_system_prompt, Unset) else self.metric_system_prompt template = self.template diff --git a/src/galileo/resources/models/context_adherence_scorer.py b/src/galileo/resources/models/context_adherence_scorer.py index d886c753f..21ccf8f65 100644 --- a/src/galileo/resources/models/context_adherence_scorer.py +++ b/src/galileo/resources/models/context_adherence_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, ContextAdherenceScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = ContextAdherenceScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else ContextAdherenceScorerType(_type_) context_adherence_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/context_relevance_scorer.py b/src/galileo/resources/models/context_relevance_scorer.py index 86844abfa..86d93c525 100644 --- a/src/galileo/resources/models/context_relevance_scorer.py +++ b/src/galileo/resources/models/context_relevance_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/correctness_scorer.py b/src/galileo/resources/models/correctness_scorer.py index cf4bb5706..f30abd709 100644 --- a/src/galileo/resources/models/correctness_scorer.py +++ b/src/galileo/resources/models/correctness_scorer.py @@ -54,18 +54,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_ = self.type_ @@ -108,16 +102,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/create_custom_luna_scorer_version_request.py b/src/galileo/resources/models/create_custom_luna_scorer_version_request.py index 2692892eb..a8c596efb 100644 --- a/src/galileo/resources/models/create_custom_luna_scorer_version_request.py +++ b/src/galileo/resources/models/create_custom_luna_scorer_version_request.py @@ -61,9 +61,8 @@ def _parse_executor(data: object) -> Union[CoreScorerName, None, Unset]: try: if not isinstance(data, str): raise TypeError() - executor_type_0 = CoreScorerName(data) + return CoreScorerName(data) - return executor_type_0 except: # noqa: E722 pass return cast(Union[CoreScorerName, None, Unset], data) diff --git a/src/galileo/resources/models/create_job_request.py b/src/galileo/resources/models/create_job_request.py index ad3454bee..a65c2a846 100644 --- a/src/galileo/resources/models/create_job_request.py +++ b/src/galileo/resources/models/create_job_request.py @@ -79,6 +79,7 @@ class CreateJobRequest: job_id (Union[None, Unset, str]): job_name (Union[Unset, str]): Default: 'default'. labels (Union[Unset, list[list[str]], list[str]]): + luna_model (Union[None, Unset, str]): metric_critique_configuration (Union['MetricCritiqueJobConfiguration', None, Unset]): migration_name (Union[None, Unset, str]): monitor_batch_id (Union[None, Unset, str]): @@ -132,6 +133,7 @@ class CreateJobRequest: job_id: Union[None, Unset, str] = UNSET job_name: Union[Unset, str] = "default" labels: Union[Unset, list[list[str]], list[str]] = UNSET + luna_model: Union[None, Unset, str] = UNSET metric_critique_configuration: Union["MetricCritiqueJobConfiguration", None, Unset] = UNSET migration_name: Union[None, Unset, str] = UNSET monitor_batch_id: Union[None, Unset, str] = UNSET @@ -270,16 +272,10 @@ def to_dict(self) -> dict[str, Any]: run_id = self.run_id dataset_id: Union[None, Unset, str] - if isinstance(self.dataset_id, Unset): - dataset_id = UNSET - else: - dataset_id = self.dataset_id + dataset_id = UNSET if isinstance(self.dataset_id, Unset) else self.dataset_id dataset_version_index: Union[None, Unset, int] - if isinstance(self.dataset_version_index, Unset): - dataset_version_index = UNSET - else: - dataset_version_index = self.dataset_version_index + dataset_version_index = UNSET if isinstance(self.dataset_version_index, Unset) else self.dataset_version_index epoch = self.epoch @@ -293,10 +289,7 @@ def to_dict(self) -> dict[str, Any]: feature_names = self.feature_names job_id: Union[None, Unset, str] - if isinstance(self.job_id, Unset): - job_id = UNSET - else: - job_id = self.job_id + job_id = UNSET if isinstance(self.job_id, Unset) else self.job_id job_name = self.job_name @@ -313,6 +306,9 @@ def to_dict(self) -> dict[str, Any]: else: labels = self.labels + luna_model: Union[None, Unset, str] + luna_model = UNSET if isinstance(self.luna_model, Unset) else self.luna_model + metric_critique_configuration: Union[None, Unset, dict[str, Any]] if isinstance(self.metric_critique_configuration, Unset): metric_critique_configuration = UNSET @@ -322,16 +318,10 @@ def to_dict(self) -> dict[str, Any]: metric_critique_configuration = self.metric_critique_configuration migration_name: Union[None, Unset, str] - if isinstance(self.migration_name, Unset): - migration_name = UNSET - else: - migration_name = self.migration_name + migration_name = UNSET if isinstance(self.migration_name, Unset) else self.migration_name monitor_batch_id: Union[None, Unset, str] - if isinstance(self.monitor_batch_id, Unset): - monitor_batch_id = UNSET - else: - monitor_batch_id = self.monitor_batch_id + monitor_batch_id = UNSET if isinstance(self.monitor_batch_id, Unset) else self.monitor_batch_id ner_labels: Union[None, Unset, list[str]] if isinstance(self.ner_labels, Unset): @@ -356,86 +346,27 @@ def to_dict(self) -> dict[str, Any]: ) in self.prompt_customized_scorers_configuration: prompt_customized_scorers_configuration_type_0_item: dict[str, Any] if isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedAgenticSessionSuccessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedAgenticWorkflowSuccessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( prompt_customized_scorers_configuration_type_0_item_data, - CustomizedChunkAttributionUtilizationGPTScorer, - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedCompletenessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedFactualityGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedGroundednessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedInstructionAdherenceGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedGroundTruthAdherenceGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedPromptInjectionGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance(prompt_customized_scorers_configuration_type_0_item_data, CustomizedSexistGPTScorer): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedInputSexistGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedToolSelectionQualityGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedToolErrorRateGPTScorer + ( + CustomizedAgenticSessionSuccessGPTScorer, + CustomizedAgenticWorkflowSuccessGPTScorer, + CustomizedChunkAttributionUtilizationGPTScorer, + CustomizedCompletenessGPTScorer, + CustomizedFactualityGPTScorer, + CustomizedGroundednessGPTScorer, + CustomizedInstructionAdherenceGPTScorer, + CustomizedGroundTruthAdherenceGPTScorer, + CustomizedPromptInjectionGPTScorer, + CustomizedSexistGPTScorer, + CustomizedInputSexistGPTScorer, + CustomizedToolSelectionQualityGPTScorer, + CustomizedToolErrorRateGPTScorer, + CustomizedToxicityGPTScorer, + ), ): prompt_customized_scorers_configuration_type_0_item = ( prompt_customized_scorers_configuration_type_0_item_data.to_dict() ) - elif isinstance(prompt_customized_scorers_configuration_type_0_item_data, CustomizedToxicityGPTScorer): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) else: prompt_customized_scorers_configuration_type_0_item = ( prompt_customized_scorers_configuration_type_0_item_data.to_dict() @@ -447,10 +378,7 @@ def to_dict(self) -> dict[str, Any]: prompt_customized_scorers_configuration = self.prompt_customized_scorers_configuration prompt_dataset_id: Union[None, Unset, str] - if isinstance(self.prompt_dataset_id, Unset): - prompt_dataset_id = UNSET - else: - prompt_dataset_id = self.prompt_dataset_id + prompt_dataset_id = UNSET if isinstance(self.prompt_dataset_id, Unset) else self.prompt_dataset_id prompt_finetuned_scorers_configuration: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.prompt_finetuned_scorers_configuration, Unset): @@ -539,10 +467,7 @@ def to_dict(self) -> dict[str, Any]: protect_scorer_payload = self.protect_scorer_payload protect_trace_id: Union[None, Unset, str] - if isinstance(self.protect_trace_id, Unset): - protect_trace_id = UNSET - else: - protect_trace_id = self.protect_trace_id + protect_trace_id = UNSET if isinstance(self.protect_trace_id, Unset) else self.protect_trace_id resource_limits: Union[None, Unset, dict[str, Any]] if isinstance(self.resource_limits, Unset): @@ -573,51 +498,34 @@ def to_dict(self) -> dict[str, Any]: scorers = [] for scorers_type_1_item_data in self.scorers: scorers_type_1_item: dict[str, Any] - if isinstance(scorers_type_1_item_data, AgenticWorkflowSuccessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, AgenticSessionSuccessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, BleuScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ChunkAttributionUtilizationScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, CompletenessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ContextAdherenceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ContextRelevanceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, CorrectnessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, GroundTruthAdherenceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputPIIScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputSexistScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputToneScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputToxicityScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InstructionAdherenceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputPIIScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputSexistScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputToneScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputToxicityScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, PromptInjectionScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, PromptPerplexityScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, RougeScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ToolErrorRateScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ToolSelectionQualityScorer): + if isinstance( + scorers_type_1_item_data, + ( + AgenticWorkflowSuccessScorer, + AgenticSessionSuccessScorer, + BleuScorer, + ChunkAttributionUtilizationScorer, + CompletenessScorer, + ContextAdherenceScorer, + ContextRelevanceScorer, + CorrectnessScorer, + GroundTruthAdherenceScorer, + InputPIIScorer, + InputSexistScorer, + InputToneScorer, + InputToxicityScorer, + InstructionAdherenceScorer, + OutputPIIScorer, + OutputSexistScorer, + OutputToneScorer, + OutputToxicityScorer, + PromptInjectionScorer, + PromptPerplexityScorer, + RougeScorer, + ToolErrorRateScorer, + ToolSelectionQualityScorer, + ), + ): scorers_type_1_item = scorers_type_1_item_data.to_dict() else: scorers_type_1_item = scorers_type_1_item_data.to_dict() @@ -668,10 +576,7 @@ def to_dict(self) -> dict[str, Any]: upload_data_in_separate_task = self.upload_data_in_separate_task user_id: Union[None, Unset, str] - if isinstance(self.user_id, Unset): - user_id = UNSET - else: - user_id = self.user_id + user_id = UNSET if isinstance(self.user_id, Unset) else self.user_id xray = self.xray @@ -692,6 +597,8 @@ def to_dict(self) -> dict[str, Any]: field_dict["job_name"] = job_name if labels is not UNSET: field_dict["labels"] = labels + if luna_model is not UNSET: + field_dict["luna_model"] = luna_model if metric_critique_configuration is not UNSET: field_dict["metric_critique_configuration"] = metric_critique_configuration if migration_name is not UNSET: @@ -840,9 +747,8 @@ def _parse_feature_names(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - feature_names_type_0 = cast(list[str], data) + return cast(list[str], data) - return feature_names_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -878,12 +784,19 @@ def _parse_labels(data: object) -> Union[Unset, list[list[str]], list[str]]: pass if not isinstance(data, list): raise TypeError() - labels_type_1 = cast(list[str], data) - - return labels_type_1 + return cast(list[str], data) labels = _parse_labels(d.pop("labels", UNSET)) + def _parse_luna_model(data: object) -> Union[None, Unset, str]: + if data is None: + return data + if isinstance(data, Unset): + return data + return cast(Union[None, Unset, str], data) + + luna_model = _parse_luna_model(d.pop("luna_model", UNSET)) + def _parse_metric_critique_configuration(data: object) -> Union["MetricCritiqueJobConfiguration", None, Unset]: if data is None: return data @@ -892,9 +805,8 @@ def _parse_metric_critique_configuration(data: object) -> Union["MetricCritiqueJ try: if not isinstance(data, dict): raise TypeError() - metric_critique_configuration_type_0 = MetricCritiqueJobConfiguration.from_dict(data) + return MetricCritiqueJobConfiguration.from_dict(data) - return metric_critique_configuration_type_0 except: # noqa: E722 pass return cast(Union["MetricCritiqueJobConfiguration", None, Unset], data) @@ -929,9 +841,8 @@ def _parse_ner_labels(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - ner_labels_type_0 = cast(list[str], data) + return cast(list[str], data) - return ner_labels_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -1002,150 +913,104 @@ def _parse_prompt_customized_scorers_configuration_type_0_item( try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_0 = ( - CustomizedAgenticSessionSuccessGPTScorer.from_dict(data) - ) + return CustomizedAgenticSessionSuccessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_1 = ( - CustomizedAgenticWorkflowSuccessGPTScorer.from_dict(data) - ) + return CustomizedAgenticWorkflowSuccessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_2 = ( - CustomizedChunkAttributionUtilizationGPTScorer.from_dict(data) - ) + return CustomizedChunkAttributionUtilizationGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_3 = ( - CustomizedCompletenessGPTScorer.from_dict(data) - ) + return CustomizedCompletenessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_4 = ( - CustomizedFactualityGPTScorer.from_dict(data) - ) + return CustomizedFactualityGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_5 = ( - CustomizedGroundednessGPTScorer.from_dict(data) - ) + return CustomizedGroundednessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_6 = ( - CustomizedInstructionAdherenceGPTScorer.from_dict(data) - ) + return CustomizedInstructionAdherenceGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_7 = ( - CustomizedGroundTruthAdherenceGPTScorer.from_dict(data) - ) + return CustomizedGroundTruthAdherenceGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_8 = ( - CustomizedPromptInjectionGPTScorer.from_dict(data) - ) + return CustomizedPromptInjectionGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_9 = ( - CustomizedSexistGPTScorer.from_dict(data) - ) + return CustomizedSexistGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_10 = ( - CustomizedInputSexistGPTScorer.from_dict(data) - ) + return CustomizedInputSexistGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_10 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_11 = ( - CustomizedToolSelectionQualityGPTScorer.from_dict(data) - ) + return CustomizedToolSelectionQualityGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_11 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_12 = ( - CustomizedToolErrorRateGPTScorer.from_dict(data) - ) + return CustomizedToolErrorRateGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_12 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_13 = ( - CustomizedToxicityGPTScorer.from_dict(data) - ) + return CustomizedToxicityGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_13 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_14 = ( - CustomizedInputToxicityGPTScorer.from_dict(data) - ) - - return prompt_customized_scorers_configuration_type_0_item_type_14 + return CustomizedInputToxicityGPTScorer.from_dict(data) prompt_customized_scorers_configuration_type_0_item = ( _parse_prompt_customized_scorers_configuration_type_0_item( @@ -1238,9 +1103,8 @@ def _parse_prompt_generated_scorers_configuration(data: object) -> Union[None, U try: if not isinstance(data, list): raise TypeError() - prompt_generated_scorers_configuration_type_0 = cast(list[str], data) + return cast(list[str], data) - return prompt_generated_scorers_configuration_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -1259,9 +1123,8 @@ def _parse_prompt_optimization_configuration( try: if not isinstance(data, dict): raise TypeError() - prompt_optimization_configuration_type_0 = PromptOptimizationConfiguration.from_dict(data) + return PromptOptimizationConfiguration.from_dict(data) - return prompt_optimization_configuration_type_0 except: # noqa: E722 pass return cast(Union["PromptOptimizationConfiguration", None, Unset], data) @@ -1310,9 +1173,8 @@ def _parse_prompt_scorer_settings(data: object) -> Union["BaseScorer", None, Uns try: if not isinstance(data, dict): raise TypeError() - prompt_scorer_settings_type_0 = BaseScorer.from_dict(data) + return BaseScorer.from_dict(data) - return prompt_scorer_settings_type_0 except: # noqa: E722 pass return cast(Union["BaseScorer", None, Unset], data) @@ -1327,9 +1189,8 @@ def _parse_prompt_scorers_configuration(data: object) -> Union["ScorersConfigura try: if not isinstance(data, dict): raise TypeError() - prompt_scorers_configuration_type_0 = ScorersConfiguration.from_dict(data) + return ScorersConfiguration.from_dict(data) - return prompt_scorers_configuration_type_0 except: # noqa: E722 pass return cast(Union["ScorersConfiguration", None, Unset], data) @@ -1344,9 +1205,8 @@ def _parse_prompt_settings(data: object) -> Union["PromptRunSettings", None, Uns try: if not isinstance(data, dict): raise TypeError() - prompt_settings_type_0 = PromptRunSettings.from_dict(data) + return PromptRunSettings.from_dict(data) - return prompt_settings_type_0 except: # noqa: E722 pass return cast(Union["PromptRunSettings", None, Unset], data) @@ -1370,9 +1230,8 @@ def _parse_protect_scorer_payload(data: object) -> Union[File, None, Unset]: try: if not isinstance(data, bytes): raise TypeError() - protect_scorer_payload_type_0 = File(payload=BytesIO(data)) + return File(payload=BytesIO(data)) - return protect_scorer_payload_type_0 except: # noqa: E722 pass return cast(Union[File, None, Unset], data) @@ -1396,9 +1255,8 @@ def _parse_resource_limits(data: object) -> Union["TaskResourceLimits", None, Un try: if not isinstance(data, dict): raise TypeError() - resource_limits_type_0 = TaskResourceLimits.from_dict(data) + return TaskResourceLimits.from_dict(data) - return resource_limits_type_0 except: # noqa: E722 pass return cast(Union["TaskResourceLimits", None, Unset], data) @@ -1413,9 +1271,8 @@ def _parse_scorer_config(data: object) -> Union["ScorerConfig", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - scorer_config_type_0 = ScorerConfig.from_dict(data) + return ScorerConfig.from_dict(data) - return scorer_config_type_0 except: # noqa: E722 pass return cast(Union["ScorerConfig", None, Unset], data) @@ -1512,192 +1369,167 @@ def _parse_scorers_type_1_item( try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_0 = AgenticWorkflowSuccessScorer.from_dict(data) + return AgenticWorkflowSuccessScorer.from_dict(data) - return scorers_type_1_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_1 = AgenticSessionSuccessScorer.from_dict(data) + return AgenticSessionSuccessScorer.from_dict(data) - return scorers_type_1_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_2 = BleuScorer.from_dict(data) + return BleuScorer.from_dict(data) - return scorers_type_1_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_3 = ChunkAttributionUtilizationScorer.from_dict(data) + return ChunkAttributionUtilizationScorer.from_dict(data) - return scorers_type_1_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_4 = CompletenessScorer.from_dict(data) + return CompletenessScorer.from_dict(data) - return scorers_type_1_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_5 = ContextAdherenceScorer.from_dict(data) + return ContextAdherenceScorer.from_dict(data) - return scorers_type_1_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_6 = ContextRelevanceScorer.from_dict(data) + return ContextRelevanceScorer.from_dict(data) - return scorers_type_1_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_7 = CorrectnessScorer.from_dict(data) + return CorrectnessScorer.from_dict(data) - return scorers_type_1_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_8 = GroundTruthAdherenceScorer.from_dict(data) + return GroundTruthAdherenceScorer.from_dict(data) - return scorers_type_1_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_9 = InputPIIScorer.from_dict(data) + return InputPIIScorer.from_dict(data) - return scorers_type_1_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_10 = InputSexistScorer.from_dict(data) + return InputSexistScorer.from_dict(data) - return scorers_type_1_item_type_10 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_11 = InputToneScorer.from_dict(data) + return InputToneScorer.from_dict(data) - return scorers_type_1_item_type_11 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_12 = InputToxicityScorer.from_dict(data) + return InputToxicityScorer.from_dict(data) - return scorers_type_1_item_type_12 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_13 = InstructionAdherenceScorer.from_dict(data) + return InstructionAdherenceScorer.from_dict(data) - return scorers_type_1_item_type_13 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_14 = OutputPIIScorer.from_dict(data) + return OutputPIIScorer.from_dict(data) - return scorers_type_1_item_type_14 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_15 = OutputSexistScorer.from_dict(data) + return OutputSexistScorer.from_dict(data) - return scorers_type_1_item_type_15 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_16 = OutputToneScorer.from_dict(data) + return OutputToneScorer.from_dict(data) - return scorers_type_1_item_type_16 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_17 = OutputToxicityScorer.from_dict(data) + return OutputToxicityScorer.from_dict(data) - return scorers_type_1_item_type_17 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_18 = PromptInjectionScorer.from_dict(data) + return PromptInjectionScorer.from_dict(data) - return scorers_type_1_item_type_18 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_19 = PromptPerplexityScorer.from_dict(data) + return PromptPerplexityScorer.from_dict(data) - return scorers_type_1_item_type_19 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_20 = RougeScorer.from_dict(data) + return RougeScorer.from_dict(data) - return scorers_type_1_item_type_20 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_21 = ToolErrorRateScorer.from_dict(data) + return ToolErrorRateScorer.from_dict(data) - return scorers_type_1_item_type_21 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_22 = ToolSelectionQualityScorer.from_dict(data) + return ToolSelectionQualityScorer.from_dict(data) - return scorers_type_1_item_type_22 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_23 = UncertaintyScorer.from_dict(data) - - return scorers_type_1_item_type_23 + return UncertaintyScorer.from_dict(data) scorers_type_1_item = _parse_scorers_type_1_item(scorers_type_1_item_data) @@ -1784,9 +1616,8 @@ def _parse_task_type(data: object) -> Union[None, TaskType, Unset]: try: if not isinstance(data, int): raise TypeError() - task_type_type_0 = TaskType(data) + return TaskType(data) - return task_type_type_0 except: # noqa: E722 pass return cast(Union[None, TaskType, Unset], data) @@ -1801,9 +1632,8 @@ def _parse_tasks(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - tasks_type_0 = cast(list[str], data) + return cast(list[str], data) - return tasks_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -1833,6 +1663,7 @@ def _parse_user_id(data: object) -> Union[None, Unset, str]: job_id=job_id, job_name=job_name, labels=labels, + luna_model=luna_model, metric_critique_configuration=metric_critique_configuration, migration_name=migration_name, monitor_batch_id=monitor_batch_id, diff --git a/src/galileo/resources/models/create_job_response.py b/src/galileo/resources/models/create_job_response.py index d1036faa8..4afc5a443 100644 --- a/src/galileo/resources/models/create_job_response.py +++ b/src/galileo/resources/models/create_job_response.py @@ -81,6 +81,7 @@ class CreateJobResponse: job_id (Union[None, Unset, str]): job_name (Union[Unset, str]): Default: 'default'. labels (Union[Unset, list[list[str]], list[str]]): + luna_model (Union[None, Unset, str]): metric_critique_configuration (Union['MetricCritiqueJobConfiguration', None, Unset]): migration_name (Union[None, Unset, str]): monitor_batch_id (Union[None, Unset, str]): @@ -136,6 +137,7 @@ class CreateJobResponse: job_id: Union[None, Unset, str] = UNSET job_name: Union[Unset, str] = "default" labels: Union[Unset, list[list[str]], list[str]] = UNSET + luna_model: Union[None, Unset, str] = UNSET metric_critique_configuration: Union["MetricCritiqueJobConfiguration", None, Unset] = UNSET migration_name: Union[None, Unset, str] = UNSET monitor_batch_id: Union[None, Unset, str] = UNSET @@ -278,16 +280,10 @@ def to_dict(self) -> dict[str, Any]: run_id = self.run_id dataset_id: Union[None, Unset, str] - if isinstance(self.dataset_id, Unset): - dataset_id = UNSET - else: - dataset_id = self.dataset_id + dataset_id = UNSET if isinstance(self.dataset_id, Unset) else self.dataset_id dataset_version_index: Union[None, Unset, int] - if isinstance(self.dataset_version_index, Unset): - dataset_version_index = UNSET - else: - dataset_version_index = self.dataset_version_index + dataset_version_index = UNSET if isinstance(self.dataset_version_index, Unset) else self.dataset_version_index epoch = self.epoch @@ -301,10 +297,7 @@ def to_dict(self) -> dict[str, Any]: feature_names = self.feature_names job_id: Union[None, Unset, str] - if isinstance(self.job_id, Unset): - job_id = UNSET - else: - job_id = self.job_id + job_id = UNSET if isinstance(self.job_id, Unset) else self.job_id job_name = self.job_name @@ -321,6 +314,9 @@ def to_dict(self) -> dict[str, Any]: else: labels = self.labels + luna_model: Union[None, Unset, str] + luna_model = UNSET if isinstance(self.luna_model, Unset) else self.luna_model + metric_critique_configuration: Union[None, Unset, dict[str, Any]] if isinstance(self.metric_critique_configuration, Unset): metric_critique_configuration = UNSET @@ -330,16 +326,10 @@ def to_dict(self) -> dict[str, Any]: metric_critique_configuration = self.metric_critique_configuration migration_name: Union[None, Unset, str] - if isinstance(self.migration_name, Unset): - migration_name = UNSET - else: - migration_name = self.migration_name + migration_name = UNSET if isinstance(self.migration_name, Unset) else self.migration_name monitor_batch_id: Union[None, Unset, str] - if isinstance(self.monitor_batch_id, Unset): - monitor_batch_id = UNSET - else: - monitor_batch_id = self.monitor_batch_id + monitor_batch_id = UNSET if isinstance(self.monitor_batch_id, Unset) else self.monitor_batch_id ner_labels: Union[None, Unset, list[str]] if isinstance(self.ner_labels, Unset): @@ -364,86 +354,27 @@ def to_dict(self) -> dict[str, Any]: ) in self.prompt_customized_scorers_configuration: prompt_customized_scorers_configuration_type_0_item: dict[str, Any] if isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedAgenticSessionSuccessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedAgenticWorkflowSuccessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( prompt_customized_scorers_configuration_type_0_item_data, - CustomizedChunkAttributionUtilizationGPTScorer, - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedCompletenessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedFactualityGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedGroundednessGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedInstructionAdherenceGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedGroundTruthAdherenceGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedPromptInjectionGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance(prompt_customized_scorers_configuration_type_0_item_data, CustomizedSexistGPTScorer): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedInputSexistGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedToolSelectionQualityGPTScorer - ): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) - elif isinstance( - prompt_customized_scorers_configuration_type_0_item_data, CustomizedToolErrorRateGPTScorer + ( + CustomizedAgenticSessionSuccessGPTScorer, + CustomizedAgenticWorkflowSuccessGPTScorer, + CustomizedChunkAttributionUtilizationGPTScorer, + CustomizedCompletenessGPTScorer, + CustomizedFactualityGPTScorer, + CustomizedGroundednessGPTScorer, + CustomizedInstructionAdherenceGPTScorer, + CustomizedGroundTruthAdherenceGPTScorer, + CustomizedPromptInjectionGPTScorer, + CustomizedSexistGPTScorer, + CustomizedInputSexistGPTScorer, + CustomizedToolSelectionQualityGPTScorer, + CustomizedToolErrorRateGPTScorer, + CustomizedToxicityGPTScorer, + ), ): prompt_customized_scorers_configuration_type_0_item = ( prompt_customized_scorers_configuration_type_0_item_data.to_dict() ) - elif isinstance(prompt_customized_scorers_configuration_type_0_item_data, CustomizedToxicityGPTScorer): - prompt_customized_scorers_configuration_type_0_item = ( - prompt_customized_scorers_configuration_type_0_item_data.to_dict() - ) else: prompt_customized_scorers_configuration_type_0_item = ( prompt_customized_scorers_configuration_type_0_item_data.to_dict() @@ -455,10 +386,7 @@ def to_dict(self) -> dict[str, Any]: prompt_customized_scorers_configuration = self.prompt_customized_scorers_configuration prompt_dataset_id: Union[None, Unset, str] - if isinstance(self.prompt_dataset_id, Unset): - prompt_dataset_id = UNSET - else: - prompt_dataset_id = self.prompt_dataset_id + prompt_dataset_id = UNSET if isinstance(self.prompt_dataset_id, Unset) else self.prompt_dataset_id prompt_finetuned_scorers_configuration: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.prompt_finetuned_scorers_configuration, Unset): @@ -547,10 +475,7 @@ def to_dict(self) -> dict[str, Any]: protect_scorer_payload = self.protect_scorer_payload protect_trace_id: Union[None, Unset, str] - if isinstance(self.protect_trace_id, Unset): - protect_trace_id = UNSET - else: - protect_trace_id = self.protect_trace_id + protect_trace_id = UNSET if isinstance(self.protect_trace_id, Unset) else self.protect_trace_id resource_limits: Union[None, Unset, dict[str, Any]] if isinstance(self.resource_limits, Unset): @@ -581,51 +506,34 @@ def to_dict(self) -> dict[str, Any]: scorers = [] for scorers_type_1_item_data in self.scorers: scorers_type_1_item: dict[str, Any] - if isinstance(scorers_type_1_item_data, AgenticWorkflowSuccessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, AgenticSessionSuccessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, BleuScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ChunkAttributionUtilizationScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, CompletenessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ContextAdherenceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ContextRelevanceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, CorrectnessScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, GroundTruthAdherenceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputPIIScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputSexistScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputToneScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InputToxicityScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, InstructionAdherenceScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputPIIScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputSexistScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputToneScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, OutputToxicityScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, PromptInjectionScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, PromptPerplexityScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, RougeScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ToolErrorRateScorer): - scorers_type_1_item = scorers_type_1_item_data.to_dict() - elif isinstance(scorers_type_1_item_data, ToolSelectionQualityScorer): + if isinstance( + scorers_type_1_item_data, + ( + AgenticWorkflowSuccessScorer, + AgenticSessionSuccessScorer, + BleuScorer, + ChunkAttributionUtilizationScorer, + CompletenessScorer, + ContextAdherenceScorer, + ContextRelevanceScorer, + CorrectnessScorer, + GroundTruthAdherenceScorer, + InputPIIScorer, + InputSexistScorer, + InputToneScorer, + InputToxicityScorer, + InstructionAdherenceScorer, + OutputPIIScorer, + OutputSexistScorer, + OutputToneScorer, + OutputToxicityScorer, + PromptInjectionScorer, + PromptPerplexityScorer, + RougeScorer, + ToolErrorRateScorer, + ToolSelectionQualityScorer, + ), + ): scorers_type_1_item = scorers_type_1_item_data.to_dict() else: scorers_type_1_item = scorers_type_1_item_data.to_dict() @@ -676,10 +584,7 @@ def to_dict(self) -> dict[str, Any]: upload_data_in_separate_task = self.upload_data_in_separate_task user_id: Union[None, Unset, str] - if isinstance(self.user_id, Unset): - user_id = UNSET - else: - user_id = self.user_id + user_id = UNSET if isinstance(self.user_id, Unset) else self.user_id xray = self.xray @@ -700,6 +605,8 @@ def to_dict(self) -> dict[str, Any]: field_dict["job_name"] = job_name if labels is not UNSET: field_dict["labels"] = labels + if luna_model is not UNSET: + field_dict["luna_model"] = luna_model if metric_critique_configuration is not UNSET: field_dict["metric_critique_configuration"] = metric_critique_configuration if migration_name is not UNSET: @@ -852,9 +759,8 @@ def _parse_feature_names(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - feature_names_type_0 = cast(list[str], data) + return cast(list[str], data) - return feature_names_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -890,12 +796,19 @@ def _parse_labels(data: object) -> Union[Unset, list[list[str]], list[str]]: pass if not isinstance(data, list): raise TypeError() - labels_type_1 = cast(list[str], data) - - return labels_type_1 + return cast(list[str], data) labels = _parse_labels(d.pop("labels", UNSET)) + def _parse_luna_model(data: object) -> Union[None, Unset, str]: + if data is None: + return data + if isinstance(data, Unset): + return data + return cast(Union[None, Unset, str], data) + + luna_model = _parse_luna_model(d.pop("luna_model", UNSET)) + def _parse_metric_critique_configuration(data: object) -> Union["MetricCritiqueJobConfiguration", None, Unset]: if data is None: return data @@ -904,9 +817,8 @@ def _parse_metric_critique_configuration(data: object) -> Union["MetricCritiqueJ try: if not isinstance(data, dict): raise TypeError() - metric_critique_configuration_type_0 = MetricCritiqueJobConfiguration.from_dict(data) + return MetricCritiqueJobConfiguration.from_dict(data) - return metric_critique_configuration_type_0 except: # noqa: E722 pass return cast(Union["MetricCritiqueJobConfiguration", None, Unset], data) @@ -941,9 +853,8 @@ def _parse_ner_labels(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - ner_labels_type_0 = cast(list[str], data) + return cast(list[str], data) - return ner_labels_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -1014,150 +925,104 @@ def _parse_prompt_customized_scorers_configuration_type_0_item( try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_0 = ( - CustomizedAgenticSessionSuccessGPTScorer.from_dict(data) - ) + return CustomizedAgenticSessionSuccessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_1 = ( - CustomizedAgenticWorkflowSuccessGPTScorer.from_dict(data) - ) + return CustomizedAgenticWorkflowSuccessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_2 = ( - CustomizedChunkAttributionUtilizationGPTScorer.from_dict(data) - ) + return CustomizedChunkAttributionUtilizationGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_3 = ( - CustomizedCompletenessGPTScorer.from_dict(data) - ) + return CustomizedCompletenessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_4 = ( - CustomizedFactualityGPTScorer.from_dict(data) - ) + return CustomizedFactualityGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_5 = ( - CustomizedGroundednessGPTScorer.from_dict(data) - ) + return CustomizedGroundednessGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_6 = ( - CustomizedInstructionAdherenceGPTScorer.from_dict(data) - ) + return CustomizedInstructionAdherenceGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_7 = ( - CustomizedGroundTruthAdherenceGPTScorer.from_dict(data) - ) + return CustomizedGroundTruthAdherenceGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_8 = ( - CustomizedPromptInjectionGPTScorer.from_dict(data) - ) + return CustomizedPromptInjectionGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_9 = ( - CustomizedSexistGPTScorer.from_dict(data) - ) + return CustomizedSexistGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_10 = ( - CustomizedInputSexistGPTScorer.from_dict(data) - ) + return CustomizedInputSexistGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_10 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_11 = ( - CustomizedToolSelectionQualityGPTScorer.from_dict(data) - ) + return CustomizedToolSelectionQualityGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_11 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_12 = ( - CustomizedToolErrorRateGPTScorer.from_dict(data) - ) + return CustomizedToolErrorRateGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_12 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_13 = ( - CustomizedToxicityGPTScorer.from_dict(data) - ) + return CustomizedToxicityGPTScorer.from_dict(data) - return prompt_customized_scorers_configuration_type_0_item_type_13 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - prompt_customized_scorers_configuration_type_0_item_type_14 = ( - CustomizedInputToxicityGPTScorer.from_dict(data) - ) - - return prompt_customized_scorers_configuration_type_0_item_type_14 + return CustomizedInputToxicityGPTScorer.from_dict(data) prompt_customized_scorers_configuration_type_0_item = ( _parse_prompt_customized_scorers_configuration_type_0_item( @@ -1250,9 +1115,8 @@ def _parse_prompt_generated_scorers_configuration(data: object) -> Union[None, U try: if not isinstance(data, list): raise TypeError() - prompt_generated_scorers_configuration_type_0 = cast(list[str], data) + return cast(list[str], data) - return prompt_generated_scorers_configuration_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -1271,9 +1135,8 @@ def _parse_prompt_optimization_configuration( try: if not isinstance(data, dict): raise TypeError() - prompt_optimization_configuration_type_0 = PromptOptimizationConfiguration.from_dict(data) + return PromptOptimizationConfiguration.from_dict(data) - return prompt_optimization_configuration_type_0 except: # noqa: E722 pass return cast(Union["PromptOptimizationConfiguration", None, Unset], data) @@ -1322,9 +1185,8 @@ def _parse_prompt_scorer_settings(data: object) -> Union["BaseScorer", None, Uns try: if not isinstance(data, dict): raise TypeError() - prompt_scorer_settings_type_0 = BaseScorer.from_dict(data) + return BaseScorer.from_dict(data) - return prompt_scorer_settings_type_0 except: # noqa: E722 pass return cast(Union["BaseScorer", None, Unset], data) @@ -1339,9 +1201,8 @@ def _parse_prompt_scorers_configuration(data: object) -> Union["ScorersConfigura try: if not isinstance(data, dict): raise TypeError() - prompt_scorers_configuration_type_0 = ScorersConfiguration.from_dict(data) + return ScorersConfiguration.from_dict(data) - return prompt_scorers_configuration_type_0 except: # noqa: E722 pass return cast(Union["ScorersConfiguration", None, Unset], data) @@ -1356,9 +1217,8 @@ def _parse_prompt_settings(data: object) -> Union["PromptRunSettings", None, Uns try: if not isinstance(data, dict): raise TypeError() - prompt_settings_type_0 = PromptRunSettings.from_dict(data) + return PromptRunSettings.from_dict(data) - return prompt_settings_type_0 except: # noqa: E722 pass return cast(Union["PromptRunSettings", None, Unset], data) @@ -1382,9 +1242,8 @@ def _parse_protect_scorer_payload(data: object) -> Union[File, None, Unset]: try: if not isinstance(data, bytes): raise TypeError() - protect_scorer_payload_type_0 = File(payload=BytesIO(data)) + return File(payload=BytesIO(data)) - return protect_scorer_payload_type_0 except: # noqa: E722 pass return cast(Union[File, None, Unset], data) @@ -1408,9 +1267,8 @@ def _parse_resource_limits(data: object) -> Union["TaskResourceLimits", None, Un try: if not isinstance(data, dict): raise TypeError() - resource_limits_type_0 = TaskResourceLimits.from_dict(data) + return TaskResourceLimits.from_dict(data) - return resource_limits_type_0 except: # noqa: E722 pass return cast(Union["TaskResourceLimits", None, Unset], data) @@ -1425,9 +1283,8 @@ def _parse_scorer_config(data: object) -> Union["ScorerConfig", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - scorer_config_type_0 = ScorerConfig.from_dict(data) + return ScorerConfig.from_dict(data) - return scorer_config_type_0 except: # noqa: E722 pass return cast(Union["ScorerConfig", None, Unset], data) @@ -1524,192 +1381,167 @@ def _parse_scorers_type_1_item( try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_0 = AgenticWorkflowSuccessScorer.from_dict(data) + return AgenticWorkflowSuccessScorer.from_dict(data) - return scorers_type_1_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_1 = AgenticSessionSuccessScorer.from_dict(data) + return AgenticSessionSuccessScorer.from_dict(data) - return scorers_type_1_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_2 = BleuScorer.from_dict(data) + return BleuScorer.from_dict(data) - return scorers_type_1_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_3 = ChunkAttributionUtilizationScorer.from_dict(data) + return ChunkAttributionUtilizationScorer.from_dict(data) - return scorers_type_1_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_4 = CompletenessScorer.from_dict(data) + return CompletenessScorer.from_dict(data) - return scorers_type_1_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_5 = ContextAdherenceScorer.from_dict(data) + return ContextAdherenceScorer.from_dict(data) - return scorers_type_1_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_6 = ContextRelevanceScorer.from_dict(data) + return ContextRelevanceScorer.from_dict(data) - return scorers_type_1_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_7 = CorrectnessScorer.from_dict(data) + return CorrectnessScorer.from_dict(data) - return scorers_type_1_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_8 = GroundTruthAdherenceScorer.from_dict(data) + return GroundTruthAdherenceScorer.from_dict(data) - return scorers_type_1_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_9 = InputPIIScorer.from_dict(data) + return InputPIIScorer.from_dict(data) - return scorers_type_1_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_10 = InputSexistScorer.from_dict(data) + return InputSexistScorer.from_dict(data) - return scorers_type_1_item_type_10 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_11 = InputToneScorer.from_dict(data) + return InputToneScorer.from_dict(data) - return scorers_type_1_item_type_11 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_12 = InputToxicityScorer.from_dict(data) + return InputToxicityScorer.from_dict(data) - return scorers_type_1_item_type_12 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_13 = InstructionAdherenceScorer.from_dict(data) + return InstructionAdherenceScorer.from_dict(data) - return scorers_type_1_item_type_13 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_14 = OutputPIIScorer.from_dict(data) + return OutputPIIScorer.from_dict(data) - return scorers_type_1_item_type_14 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_15 = OutputSexistScorer.from_dict(data) + return OutputSexistScorer.from_dict(data) - return scorers_type_1_item_type_15 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_16 = OutputToneScorer.from_dict(data) + return OutputToneScorer.from_dict(data) - return scorers_type_1_item_type_16 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_17 = OutputToxicityScorer.from_dict(data) + return OutputToxicityScorer.from_dict(data) - return scorers_type_1_item_type_17 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_18 = PromptInjectionScorer.from_dict(data) + return PromptInjectionScorer.from_dict(data) - return scorers_type_1_item_type_18 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_19 = PromptPerplexityScorer.from_dict(data) + return PromptPerplexityScorer.from_dict(data) - return scorers_type_1_item_type_19 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_20 = RougeScorer.from_dict(data) + return RougeScorer.from_dict(data) - return scorers_type_1_item_type_20 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_21 = ToolErrorRateScorer.from_dict(data) + return ToolErrorRateScorer.from_dict(data) - return scorers_type_1_item_type_21 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_22 = ToolSelectionQualityScorer.from_dict(data) + return ToolSelectionQualityScorer.from_dict(data) - return scorers_type_1_item_type_22 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - scorers_type_1_item_type_23 = UncertaintyScorer.from_dict(data) - - return scorers_type_1_item_type_23 + return UncertaintyScorer.from_dict(data) scorers_type_1_item = _parse_scorers_type_1_item(scorers_type_1_item_data) @@ -1796,9 +1628,8 @@ def _parse_task_type(data: object) -> Union[None, TaskType, Unset]: try: if not isinstance(data, int): raise TypeError() - task_type_type_0 = TaskType(data) + return TaskType(data) - return task_type_type_0 except: # noqa: E722 pass return cast(Union[None, TaskType, Unset], data) @@ -1813,9 +1644,8 @@ def _parse_tasks(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - tasks_type_0 = cast(list[str], data) + return cast(list[str], data) - return tasks_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -1847,6 +1677,7 @@ def _parse_user_id(data: object) -> Union[None, Unset, str]: job_id=job_id, job_name=job_name, labels=labels, + luna_model=luna_model, metric_critique_configuration=metric_critique_configuration, migration_name=migration_name, monitor_batch_id=monitor_batch_id, diff --git a/src/galileo/resources/models/create_llm_scorer_version_request.py b/src/galileo/resources/models/create_llm_scorer_version_request.py index 31a75d701..1637a3bfe 100644 --- a/src/galileo/resources/models/create_llm_scorer_version_request.py +++ b/src/galileo/resources/models/create_llm_scorer_version_request.py @@ -53,10 +53,7 @@ def to_dict(self) -> dict[str, Any]: chain_poll_template = self.chain_poll_template cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled input_type: Union[None, Unset, str] if isinstance(self.input_type, Unset): @@ -67,22 +64,13 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type instructions: Union[None, Unset, str] - if isinstance(self.instructions, Unset): - instructions = UNSET - else: - instructions = self.instructions + instructions = UNSET if isinstance(self.instructions, Unset) else self.instructions model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -102,10 +90,7 @@ def to_dict(self) -> dict[str, Any]: scoreable_node_types = self.scoreable_node_types user_prompt: Union[None, Unset, str] - if isinstance(self.user_prompt, Unset): - user_prompt = UNSET - else: - user_prompt = self.user_prompt + user_prompt = UNSET if isinstance(self.user_prompt, Unset) else self.user_prompt field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -145,9 +130,8 @@ def _parse_chain_poll_template(data: object) -> Union["ChainPollTemplate", None, try: if not isinstance(data, dict): raise TypeError() - chain_poll_template_type_0 = ChainPollTemplate.from_dict(data) + return ChainPollTemplate.from_dict(data) - return chain_poll_template_type_0 except: # noqa: E722 pass return cast(Union["ChainPollTemplate", None, Unset], data) @@ -171,9 +155,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -215,9 +198,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -232,9 +214,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/create_prompt_template_with_version_request_body.py b/src/galileo/resources/models/create_prompt_template_with_version_request_body.py index a45bc7baa..3bd4a33eb 100644 --- a/src/galileo/resources/models/create_prompt_template_with_version_request_body.py +++ b/src/galileo/resources/models/create_prompt_template_with_version_request_body.py @@ -44,10 +44,7 @@ def to_dict(self) -> dict[str, Any]: from ..models.name import Name name: Union[dict[str, Any], str] - if isinstance(self.name, Name): - name = self.name.to_dict() - else: - name = self.name + name = self.name.to_dict() if isinstance(self.name, Name) else self.name template: Union[list[dict[str, Any]], str] if isinstance(self.template, list): @@ -62,10 +59,7 @@ def to_dict(self) -> dict[str, Any]: hidden = self.hidden output_type: Union[None, Unset, str] - if isinstance(self.output_type, Unset): - output_type = UNSET - else: - output_type = self.output_type + output_type = UNSET if isinstance(self.output_type, Unset) else self.output_type raw = self.raw @@ -74,10 +68,7 @@ def to_dict(self) -> dict[str, Any]: settings = self.settings.to_dict() version: Union[None, Unset, int] - if isinstance(self.version, Unset): - version = UNSET - else: - version = self.version + version = UNSET if isinstance(self.version, Unset) else self.version field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -107,9 +98,8 @@ def _parse_name(data: object) -> Union["Name", str]: try: if not isinstance(data, dict): raise TypeError() - name_type_1 = Name.from_dict(data) + return Name.from_dict(data) - return name_type_1 except: # noqa: E722 pass return cast(Union["Name", str], data) @@ -149,10 +139,7 @@ def _parse_output_type(data: object) -> Union[None, Unset, str]: _settings = d.pop("settings", UNSET) settings: Union[Unset, PromptRunSettings] - if isinstance(_settings, Unset): - settings = UNSET - else: - settings = PromptRunSettings.from_dict(_settings) + settings = UNSET if isinstance(_settings, Unset) else PromptRunSettings.from_dict(_settings) def _parse_version(data: object) -> Union[None, Unset, int]: if data is None: diff --git a/src/galileo/resources/models/create_scorer_request.py b/src/galileo/resources/models/create_scorer_request.py index 8e35c9452..801deb0fc 100644 --- a/src/galileo/resources/models/create_scorer_request.py +++ b/src/galileo/resources/models/create_scorer_request.py @@ -27,6 +27,7 @@ class CreateScorerRequest: ground_truth (Union[None, Unset, bool]): model_type (Union[ModelType, None, Unset]): tags (Union[Unset, list[str]]): + user_prompt (Union[None, Unset, str]): """ name: str @@ -37,6 +38,7 @@ class CreateScorerRequest: ground_truth: Union[None, Unset, bool] = UNSET model_type: Union[ModelType, None, Unset] = UNSET tags: Union[Unset, list[str]] = UNSET + user_prompt: Union[None, Unset, str] = UNSET additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict) def to_dict(self) -> dict[str, Any]: @@ -47,10 +49,7 @@ def to_dict(self) -> dict[str, Any]: scorer_type = self.scorer_type.value default_version_id: Union[None, Unset, str] - if isinstance(self.default_version_id, Unset): - default_version_id = UNSET - else: - default_version_id = self.default_version_id + default_version_id = UNSET if isinstance(self.default_version_id, Unset) else self.default_version_id defaults: Union[None, Unset, dict[str, Any]] if isinstance(self.defaults, Unset): @@ -63,10 +62,7 @@ def to_dict(self) -> dict[str, Any]: description = self.description ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth model_type: Union[None, Unset, str] if isinstance(self.model_type, Unset): @@ -80,6 +76,9 @@ def to_dict(self) -> dict[str, Any]: if not isinstance(self.tags, Unset): tags = self.tags + user_prompt: Union[None, Unset, str] + user_prompt = UNSET if isinstance(self.user_prompt, Unset) else self.user_prompt + field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) field_dict.update({"name": name, "scorer_type": scorer_type}) @@ -95,6 +94,8 @@ def to_dict(self) -> dict[str, Any]: field_dict["model_type"] = model_type if tags is not UNSET: field_dict["tags"] = tags + if user_prompt is not UNSET: + field_dict["user_prompt"] = user_prompt return field_dict @@ -124,9 +125,8 @@ def _parse_defaults(data: object) -> Union["ScorerDefaults", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - defaults_type_0 = ScorerDefaults.from_dict(data) + return ScorerDefaults.from_dict(data) - return defaults_type_0 except: # noqa: E722 pass return cast(Union["ScorerDefaults", None, Unset], data) @@ -152,9 +152,8 @@ def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: try: if not isinstance(data, str): raise TypeError() - model_type_type_0 = ModelType(data) + return ModelType(data) - return model_type_type_0 except: # noqa: E722 pass return cast(Union[ModelType, None, Unset], data) @@ -163,6 +162,15 @@ def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: tags = cast(list[str], d.pop("tags", UNSET)) + def _parse_user_prompt(data: object) -> Union[None, Unset, str]: + if data is None: + return data + if isinstance(data, Unset): + return data + return cast(Union[None, Unset, str], data) + + user_prompt = _parse_user_prompt(d.pop("user_prompt", UNSET)) + create_scorer_request = cls( name=name, scorer_type=scorer_type, @@ -172,6 +180,7 @@ def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: ground_truth=ground_truth, model_type=model_type, tags=tags, + user_prompt=user_prompt, ) create_scorer_request.additional_properties = d diff --git a/src/galileo/resources/models/create_scorer_version_request.py b/src/galileo/resources/models/create_scorer_version_request.py index c184cbb94..3a5a371a7 100644 --- a/src/galileo/resources/models/create_scorer_version_request.py +++ b/src/galileo/resources/models/create_scorer_version_request.py @@ -33,10 +33,7 @@ class CreateScorerVersionRequest: def to_dict(self) -> dict[str, Any]: cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled input_type: Union[None, Unset, str] if isinstance(self.input_type, Unset): @@ -47,16 +44,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -114,9 +105,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -149,9 +139,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -166,9 +155,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/create_update_registered_scorer_response.py b/src/galileo/resources/models/create_update_registered_scorer_response.py index 4716be810..793af8a82 100644 --- a/src/galileo/resources/models/create_update_registered_scorer_response.py +++ b/src/galileo/resources/models/create_update_registered_scorer_response.py @@ -41,10 +41,7 @@ def to_dict(self) -> dict[str, Any]: created_by = self.created_by data_type: Union[None, str] - if isinstance(self.data_type, DataTypeOptions): - data_type = self.data_type.value - else: - data_type = self.data_type + data_type = self.data_type.value if isinstance(self.data_type, DataTypeOptions) else self.data_type id = self.id @@ -92,9 +89,8 @@ def _parse_data_type(data: object) -> Union[DataTypeOptions, None]: try: if not isinstance(data, str): raise TypeError() - data_type_type_0 = DataTypeOptions(data) + return DataTypeOptions(data) - return data_type_type_0 except: # noqa: E722 pass return cast(Union[DataTypeOptions, None], data) @@ -118,9 +114,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, list[str]], data) diff --git a/src/galileo/resources/models/customized_agentic_session_success_gpt_scorer.py b/src/galileo/resources/models/customized_agentic_session_success_gpt_scorer.py index 125270e2a..463ba3889 100644 --- a/src/galileo/resources/models/customized_agentic_session_success_gpt_scorer.py +++ b/src/galileo/resources/models/customized_agentic_session_success_gpt_scorer.py @@ -108,26 +108,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -155,16 +146,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -184,16 +169,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -210,18 +189,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -334,9 +307,8 @@ def _parse_aggregates( try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedAgenticSessionSuccessGPTScorerAggregatesType0.from_dict(data) + return CustomizedAgenticSessionSuccessGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedAgenticSessionSuccessGPTScorerAggregatesType0", None, Unset], data) @@ -385,9 +357,8 @@ def _parse_extra(data: object) -> Union["CustomizedAgenticSessionSuccessGPTScore try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedAgenticSessionSuccessGPTScorerExtraType0.from_dict(data) + return CustomizedAgenticSessionSuccessGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedAgenticSessionSuccessGPTScorerExtraType0", None, Unset], data) @@ -410,16 +381,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -458,9 +426,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -475,9 +442,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -518,9 +484,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -581,9 +546,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_agentic_workflow_success_gpt_scorer.py b/src/galileo/resources/models/customized_agentic_workflow_success_gpt_scorer.py index 59c29a35c..46e2665e5 100644 --- a/src/galileo/resources/models/customized_agentic_workflow_success_gpt_scorer.py +++ b/src/galileo/resources/models/customized_agentic_workflow_success_gpt_scorer.py @@ -108,26 +108,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -155,16 +146,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -184,16 +169,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -210,18 +189,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -334,9 +307,8 @@ def _parse_aggregates( try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedAgenticWorkflowSuccessGPTScorerAggregatesType0.from_dict(data) + return CustomizedAgenticWorkflowSuccessGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedAgenticWorkflowSuccessGPTScorerAggregatesType0", None, Unset], data) @@ -385,9 +357,8 @@ def _parse_extra(data: object) -> Union["CustomizedAgenticWorkflowSuccessGPTScor try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedAgenticWorkflowSuccessGPTScorerExtraType0.from_dict(data) + return CustomizedAgenticWorkflowSuccessGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedAgenticWorkflowSuccessGPTScorerExtraType0", None, Unset], data) @@ -410,16 +381,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -458,9 +426,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -475,9 +442,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -518,9 +484,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -583,9 +548,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_chunk_attribution_utilization_gpt_scorer.py b/src/galileo/resources/models/customized_chunk_attribution_utilization_gpt_scorer.py index dd532f116..56c87eb97 100644 --- a/src/galileo/resources/models/customized_chunk_attribution_utilization_gpt_scorer.py +++ b/src/galileo/resources/models/customized_chunk_attribution_utilization_gpt_scorer.py @@ -108,26 +108,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -155,16 +146,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -184,16 +169,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -210,18 +189,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -334,9 +307,8 @@ def _parse_aggregates( try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedChunkAttributionUtilizationGPTScorerAggregatesType0.from_dict(data) + return CustomizedChunkAttributionUtilizationGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedChunkAttributionUtilizationGPTScorerAggregatesType0", None, Unset], data) @@ -387,9 +359,8 @@ def _parse_extra( try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedChunkAttributionUtilizationGPTScorerExtraType0.from_dict(data) + return CustomizedChunkAttributionUtilizationGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedChunkAttributionUtilizationGPTScorerExtraType0", None, Unset], data) @@ -412,16 +383,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -460,9 +428,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -477,9 +444,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -520,9 +486,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -587,9 +552,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_completeness_gpt_scorer.py b/src/galileo/resources/models/customized_completeness_gpt_scorer.py index 819f4257e..a86a9ef30 100644 --- a/src/galileo/resources/models/customized_completeness_gpt_scorer.py +++ b/src/galileo/resources/models/customized_completeness_gpt_scorer.py @@ -101,26 +101,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -148,16 +139,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -177,16 +162,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -203,18 +182,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -323,9 +296,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedCompletenessGPTScorerAgg try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedCompletenessGPTScorerAggregatesType0.from_dict(data) + return CustomizedCompletenessGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedCompletenessGPTScorerAggregatesType0", None, Unset], data) @@ -374,9 +346,8 @@ def _parse_extra(data: object) -> Union["CustomizedCompletenessGPTScorerExtraTyp try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedCompletenessGPTScorerExtraType0.from_dict(data) + return CustomizedCompletenessGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedCompletenessGPTScorerExtraType0", None, Unset], data) @@ -399,16 +370,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -447,9 +415,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -464,9 +431,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -507,9 +473,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -570,9 +535,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_factuality_gpt_scorer.py b/src/galileo/resources/models/customized_factuality_gpt_scorer.py index ebbbb016d..49ab780ab 100644 --- a/src/galileo/resources/models/customized_factuality_gpt_scorer.py +++ b/src/galileo/resources/models/customized_factuality_gpt_scorer.py @@ -101,26 +101,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -150,16 +141,10 @@ def to_dict(self) -> dict[str, Any]: function_explanation_param_name = self.function_explanation_param_name generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -179,16 +164,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -205,18 +184,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -327,9 +300,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedFactualityGPTScorerAggre try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedFactualityGPTScorerAggregatesType0.from_dict(data) + return CustomizedFactualityGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedFactualityGPTScorerAggregatesType0", None, Unset], data) @@ -378,9 +350,8 @@ def _parse_extra(data: object) -> Union["CustomizedFactualityGPTScorerExtraType0 try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedFactualityGPTScorerExtraType0.from_dict(data) + return CustomizedFactualityGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedFactualityGPTScorerExtraType0", None, Unset], data) @@ -403,16 +374,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -453,9 +421,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -470,9 +437,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -513,9 +479,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -576,9 +541,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_ground_truth_adherence_gpt_scorer.py b/src/galileo/resources/models/customized_ground_truth_adherence_gpt_scorer.py index d8449d404..606da4df9 100644 --- a/src/galileo/resources/models/customized_ground_truth_adherence_gpt_scorer.py +++ b/src/galileo/resources/models/customized_ground_truth_adherence_gpt_scorer.py @@ -106,26 +106,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -153,16 +144,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -182,16 +167,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -208,18 +187,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -332,9 +305,8 @@ def _parse_aggregates( try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedGroundTruthAdherenceGPTScorerAggregatesType0.from_dict(data) + return CustomizedGroundTruthAdherenceGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedGroundTruthAdherenceGPTScorerAggregatesType0", None, Unset], data) @@ -383,9 +355,8 @@ def _parse_extra(data: object) -> Union["CustomizedGroundTruthAdherenceGPTScorer try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedGroundTruthAdherenceGPTScorerExtraType0.from_dict(data) + return CustomizedGroundTruthAdherenceGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedGroundTruthAdherenceGPTScorerExtraType0", None, Unset], data) @@ -408,16 +379,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -456,9 +424,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -473,9 +440,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -516,9 +482,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -579,9 +544,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_groundedness_gpt_scorer.py b/src/galileo/resources/models/customized_groundedness_gpt_scorer.py index 3e6cad5fb..72487d2d3 100644 --- a/src/galileo/resources/models/customized_groundedness_gpt_scorer.py +++ b/src/galileo/resources/models/customized_groundedness_gpt_scorer.py @@ -102,26 +102,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -149,16 +140,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -178,16 +163,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -204,18 +183,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -324,9 +297,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedGroundednessGPTScorerAgg try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedGroundednessGPTScorerAggregatesType0.from_dict(data) + return CustomizedGroundednessGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedGroundednessGPTScorerAggregatesType0", None, Unset], data) @@ -375,9 +347,8 @@ def _parse_extra(data: object) -> Union["CustomizedGroundednessGPTScorerExtraTyp try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedGroundednessGPTScorerExtraType0.from_dict(data) + return CustomizedGroundednessGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedGroundednessGPTScorerExtraType0", None, Unset], data) @@ -400,16 +371,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -448,9 +416,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -465,9 +432,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -508,9 +474,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -571,9 +536,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_input_sexist_gpt_scorer.py b/src/galileo/resources/models/customized_input_sexist_gpt_scorer.py index b3bef1814..2ac28a613 100644 --- a/src/galileo/resources/models/customized_input_sexist_gpt_scorer.py +++ b/src/galileo/resources/models/customized_input_sexist_gpt_scorer.py @@ -102,26 +102,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -149,16 +140,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -178,16 +163,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -204,18 +183,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -324,9 +297,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedInputSexistGPTScorerAggr try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedInputSexistGPTScorerAggregatesType0.from_dict(data) + return CustomizedInputSexistGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedInputSexistGPTScorerAggregatesType0", None, Unset], data) @@ -375,9 +347,8 @@ def _parse_extra(data: object) -> Union["CustomizedInputSexistGPTScorerExtraType try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedInputSexistGPTScorerExtraType0.from_dict(data) + return CustomizedInputSexistGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedInputSexistGPTScorerExtraType0", None, Unset], data) @@ -400,16 +371,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -448,9 +416,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -465,9 +432,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -508,9 +474,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -571,9 +536,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_input_toxicity_gpt_scorer.py b/src/galileo/resources/models/customized_input_toxicity_gpt_scorer.py index 2af4fd910..7389c2383 100644 --- a/src/galileo/resources/models/customized_input_toxicity_gpt_scorer.py +++ b/src/galileo/resources/models/customized_input_toxicity_gpt_scorer.py @@ -105,26 +105,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -152,16 +143,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -181,16 +166,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -207,18 +186,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -329,9 +302,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedInputToxicityGPTScorerAg try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedInputToxicityGPTScorerAggregatesType0.from_dict(data) + return CustomizedInputToxicityGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedInputToxicityGPTScorerAggregatesType0", None, Unset], data) @@ -380,9 +352,8 @@ def _parse_extra(data: object) -> Union["CustomizedInputToxicityGPTScorerExtraTy try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedInputToxicityGPTScorerExtraType0.from_dict(data) + return CustomizedInputToxicityGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedInputToxicityGPTScorerExtraType0", None, Unset], data) @@ -405,16 +376,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -453,9 +421,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -470,9 +437,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -513,9 +479,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -576,9 +541,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_instruction_adherence_gpt_scorer.py b/src/galileo/resources/models/customized_instruction_adherence_gpt_scorer.py index 98fca3fe9..e7268cd38 100644 --- a/src/galileo/resources/models/customized_instruction_adherence_gpt_scorer.py +++ b/src/galileo/resources/models/customized_instruction_adherence_gpt_scorer.py @@ -108,26 +108,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -157,16 +148,10 @@ def to_dict(self) -> dict[str, Any]: function_explanation_param_name = self.function_explanation_param_name generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -186,16 +171,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -212,18 +191,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -338,9 +311,8 @@ def _parse_aggregates( try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedInstructionAdherenceGPTScorerAggregatesType0.from_dict(data) + return CustomizedInstructionAdherenceGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedInstructionAdherenceGPTScorerAggregatesType0", None, Unset], data) @@ -389,9 +361,8 @@ def _parse_extra(data: object) -> Union["CustomizedInstructionAdherenceGPTScorer try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedInstructionAdherenceGPTScorerExtraType0.from_dict(data) + return CustomizedInstructionAdherenceGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedInstructionAdherenceGPTScorerExtraType0", None, Unset], data) @@ -414,16 +385,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -464,9 +432,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -481,9 +448,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -524,9 +490,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -587,9 +552,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_prompt_injection_gpt_scorer.py b/src/galileo/resources/models/customized_prompt_injection_gpt_scorer.py index 1818519c2..b76ed25fa 100644 --- a/src/galileo/resources/models/customized_prompt_injection_gpt_scorer.py +++ b/src/galileo/resources/models/customized_prompt_injection_gpt_scorer.py @@ -107,26 +107,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -154,16 +145,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -183,16 +168,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -209,18 +188,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -331,9 +304,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedPromptInjectionGPTScorer try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedPromptInjectionGPTScorerAggregatesType0.from_dict(data) + return CustomizedPromptInjectionGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedPromptInjectionGPTScorerAggregatesType0", None, Unset], data) @@ -382,9 +354,8 @@ def _parse_extra(data: object) -> Union["CustomizedPromptInjectionGPTScorerExtra try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedPromptInjectionGPTScorerExtraType0.from_dict(data) + return CustomizedPromptInjectionGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedPromptInjectionGPTScorerExtraType0", None, Unset], data) @@ -407,16 +378,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -455,9 +423,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -472,9 +439,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -515,9 +481,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -578,9 +543,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_sexist_gpt_scorer.py b/src/galileo/resources/models/customized_sexist_gpt_scorer.py index 94f0169e4..33151055d 100644 --- a/src/galileo/resources/models/customized_sexist_gpt_scorer.py +++ b/src/galileo/resources/models/customized_sexist_gpt_scorer.py @@ -98,26 +98,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -145,16 +136,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -174,16 +159,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -200,18 +179,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -318,9 +291,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedSexistGPTScorerAggregate try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedSexistGPTScorerAggregatesType0.from_dict(data) + return CustomizedSexistGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedSexistGPTScorerAggregatesType0", None, Unset], data) @@ -369,9 +341,8 @@ def _parse_extra(data: object) -> Union["CustomizedSexistGPTScorerExtraType0", N try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedSexistGPTScorerExtraType0.from_dict(data) + return CustomizedSexistGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedSexistGPTScorerExtraType0", None, Unset], data) @@ -394,16 +365,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -442,9 +410,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -459,9 +426,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -502,9 +468,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -565,9 +530,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_tool_error_rate_gpt_scorer.py b/src/galileo/resources/models/customized_tool_error_rate_gpt_scorer.py index cbaae2e95..d2ad6fa00 100644 --- a/src/galileo/resources/models/customized_tool_error_rate_gpt_scorer.py +++ b/src/galileo/resources/models/customized_tool_error_rate_gpt_scorer.py @@ -104,26 +104,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -151,16 +142,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -180,16 +165,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -206,18 +185,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -328,9 +301,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedToolErrorRateGPTScorerAg try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedToolErrorRateGPTScorerAggregatesType0.from_dict(data) + return CustomizedToolErrorRateGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedToolErrorRateGPTScorerAggregatesType0", None, Unset], data) @@ -379,9 +351,8 @@ def _parse_extra(data: object) -> Union["CustomizedToolErrorRateGPTScorerExtraTy try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedToolErrorRateGPTScorerExtraType0.from_dict(data) + return CustomizedToolErrorRateGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedToolErrorRateGPTScorerExtraType0", None, Unset], data) @@ -404,16 +375,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -452,9 +420,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -469,9 +436,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -512,9 +478,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -575,9 +540,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_tool_selection_quality_gpt_scorer.py b/src/galileo/resources/models/customized_tool_selection_quality_gpt_scorer.py index 459dd5401..c3f7afdf0 100644 --- a/src/galileo/resources/models/customized_tool_selection_quality_gpt_scorer.py +++ b/src/galileo/resources/models/customized_tool_selection_quality_gpt_scorer.py @@ -107,26 +107,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -154,16 +145,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -183,16 +168,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -209,18 +188,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -333,9 +306,8 @@ def _parse_aggregates( try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedToolSelectionQualityGPTScorerAggregatesType0.from_dict(data) + return CustomizedToolSelectionQualityGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedToolSelectionQualityGPTScorerAggregatesType0", None, Unset], data) @@ -384,9 +356,8 @@ def _parse_extra(data: object) -> Union["CustomizedToolSelectionQualityGPTScorer try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedToolSelectionQualityGPTScorerExtraType0.from_dict(data) + return CustomizedToolSelectionQualityGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedToolSelectionQualityGPTScorerExtraType0", None, Unset], data) @@ -409,16 +380,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -457,9 +425,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -474,9 +441,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -517,9 +483,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -580,9 +545,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/customized_toxicity_gpt_scorer.py b/src/galileo/resources/models/customized_toxicity_gpt_scorer.py index 1358bbab2..e6718f3e4 100644 --- a/src/galileo/resources/models/customized_toxicity_gpt_scorer.py +++ b/src/galileo/resources/models/customized_toxicity_gpt_scorer.py @@ -98,26 +98,17 @@ def to_dict(self) -> dict[str, Any]: aggregates = self.aggregates can_copy_to_llm: Union[None, Unset, bool] - if isinstance(self.can_copy_to_llm, Unset): - can_copy_to_llm = UNSET - else: - can_copy_to_llm = self.can_copy_to_llm + can_copy_to_llm = UNSET if isinstance(self.can_copy_to_llm, Unset) else self.can_copy_to_llm chainpoll_template: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.chainpoll_template, Unset): chainpoll_template = self.chainpoll_template.to_dict() cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description extra: Union[None, Unset, dict[str, Any]] if isinstance(self.extra, Unset): @@ -145,16 +136,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters generated_scorer_id: Union[None, Unset, str] - if isinstance(self.generated_scorer_id, Unset): - generated_scorer_id = UNSET - else: - generated_scorer_id = self.generated_scorer_id + generated_scorer_id = UNSET if isinstance(self.generated_scorer_id, Unset) else self.generated_scorer_id ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth indices: Union[None, Unset, list[int]] if isinstance(self.indices, Unset): @@ -174,16 +159,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type lora_task_id: Union[None, Unset, int] - if isinstance(self.lora_task_id, Unset): - lora_task_id = UNSET - else: - lora_task_id = self.lora_task_id + lora_task_id = UNSET if isinstance(self.lora_task_id, Unset) else self.lora_task_id metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name + metric_name = UNSET if isinstance(self.metric_name, Unset) else self.metric_name model_alias = self.model_alias @@ -200,18 +179,12 @@ def to_dict(self) -> dict[str, Any]: output_type = self.output_type prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt regex_field = self.regex_field registered_scorer_id: Union[None, Unset, str] - if isinstance(self.registered_scorer_id, Unset): - registered_scorer_id = UNSET - else: - registered_scorer_id = self.registered_scorer_id + registered_scorer_id = UNSET if isinstance(self.registered_scorer_id, Unset) else self.registered_scorer_id scoreable_node_types: Union[None, Unset, list[str]] if isinstance(self.scoreable_node_types, Unset): @@ -318,9 +291,8 @@ def _parse_aggregates(data: object) -> Union["CustomizedToxicityGPTScorerAggrega try: if not isinstance(data, dict): raise TypeError() - aggregates_type_0 = CustomizedToxicityGPTScorerAggregatesType0.from_dict(data) + return CustomizedToxicityGPTScorerAggregatesType0.from_dict(data) - return aggregates_type_0 except: # noqa: E722 pass return cast(Union["CustomizedToxicityGPTScorerAggregatesType0", None, Unset], data) @@ -369,9 +341,8 @@ def _parse_extra(data: object) -> Union["CustomizedToxicityGPTScorerExtraType0", try: if not isinstance(data, dict): raise TypeError() - extra_type_0 = CustomizedToxicityGPTScorerExtraType0.from_dict(data) + return CustomizedToxicityGPTScorerExtraType0.from_dict(data) - return extra_type_0 except: # noqa: E722 pass return cast(Union["CustomizedToxicityGPTScorerExtraType0", None, Unset], data) @@ -394,16 +365,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -442,9 +410,8 @@ def _parse_indices(data: object) -> Union[None, Unset, list[int]]: try: if not isinstance(data, list): raise TypeError() - indices_type_0 = cast(list[int], data) + return cast(list[int], data) - return indices_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[int]], data) @@ -459,9 +426,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -502,9 +468,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -565,9 +530,8 @@ def _parse_scores(data: object) -> Union[None, Unset, list[Any]]: try: if not isinstance(data, list): raise TypeError() - scores_type_0 = cast(list[Any], data) + return cast(list[Any], data) - return scores_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[Any]], data) diff --git a/src/galileo/resources/models/dataset_append_row.py b/src/galileo/resources/models/dataset_append_row.py index c4bec327d..2ea5ab2dd 100644 --- a/src/galileo/resources/models/dataset_append_row.py +++ b/src/galileo/resources/models/dataset_append_row.py @@ -33,10 +33,7 @@ def to_dict(self) -> dict[str, Any]: edit_type = self.edit_type row_id: Union[None, Unset, str] - if isinstance(self.row_id, Unset): - row_id = UNSET - else: - row_id = self.row_id + row_id = UNSET if isinstance(self.row_id, Unset) else self.row_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/dataset_append_row_values.py b/src/galileo/resources/models/dataset_append_row_values.py index 3519fb85a..23ef29a3f 100644 --- a/src/galileo/resources/models/dataset_append_row_values.py +++ b/src/galileo/resources/models/dataset_append_row_values.py @@ -55,9 +55,8 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = DatasetAppendRowValuesAdditionalPropertyType3.from_dict(data) + return DatasetAppendRowValuesAdditionalPropertyType3.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass return cast(Union["DatasetAppendRowValuesAdditionalPropertyType3", None, float, int, str], data) diff --git a/src/galileo/resources/models/dataset_content.py b/src/galileo/resources/models/dataset_content.py index 4d50cb214..ff11501b1 100644 --- a/src/galileo/resources/models/dataset_content.py +++ b/src/galileo/resources/models/dataset_content.py @@ -43,10 +43,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated @@ -60,10 +57,7 @@ def to_dict(self) -> dict[str, Any]: starting_token = self.starting_token warning_message: Union[None, Unset, str] - if isinstance(self.warning_message, Unset): - warning_message = UNSET - else: - warning_message = self.warning_message + warning_message = UNSET if isinstance(self.warning_message, Unset) else self.warning_message field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/dataset_content_filter.py b/src/galileo/resources/models/dataset_content_filter.py index 9a2cb3b58..b52468675 100644 --- a/src/galileo/resources/models/dataset_content_filter.py +++ b/src/galileo/resources/models/dataset_content_filter.py @@ -50,10 +50,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _operator = d.pop("operator", UNSET) operator: Union[Unset, DatasetContentFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = DatasetContentFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else DatasetContentFilterOperator(_operator) dataset_content_filter = cls(column_name=column_name, value=value, operator=operator) diff --git a/src/galileo/resources/models/dataset_data.py b/src/galileo/resources/models/dataset_data.py index 28782dab5..9be24f461 100644 --- a/src/galileo/resources/models/dataset_data.py +++ b/src/galileo/resources/models/dataset_data.py @@ -25,10 +25,7 @@ def to_dict(self) -> dict[str, Any]: dataset_id = self.dataset_id dataset_version_index: Union[None, Unset, int] - if isinstance(self.dataset_version_index, Unset): - dataset_version_index = UNSET - else: - dataset_version_index = self.dataset_version_index + dataset_version_index = UNSET if isinstance(self.dataset_version_index, Unset) else self.dataset_version_index field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/dataset_db.py b/src/galileo/resources/models/dataset_db.py index c59712ae6..b7a5bcb38 100644 --- a/src/galileo/resources/models/dataset_db.py +++ b/src/galileo/resources/models/dataset_db.py @@ -50,11 +50,7 @@ def to_dict(self) -> dict[str, Any]: from ..models.user_info import UserInfo column_names: Union[None, list[str]] - if isinstance(self.column_names, list): - column_names = self.column_names - - else: - column_names = self.column_names + column_names = self.column_names if isinstance(self.column_names, list) else self.column_names created_at = self.created_at.isoformat() @@ -120,9 +116,8 @@ def _parse_column_names(data: object) -> Union[None, list[str]]: try: if not isinstance(data, list): raise TypeError() - column_names_type_0 = cast(list[str], data) + return cast(list[str], data) - return column_names_type_0 except: # noqa: E722 pass return cast(Union[None, list[str]], data) @@ -137,9 +132,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) diff --git a/src/galileo/resources/models/dataset_id_filter.py b/src/galileo/resources/models/dataset_id_filter.py index c843c065b..df0f67b37 100644 --- a/src/galileo/resources/models/dataset_id_filter.py +++ b/src/galileo/resources/models/dataset_id_filter.py @@ -26,11 +26,7 @@ class DatasetIDFilter: def to_dict(self) -> dict[str, Any]: value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value name = self.name @@ -56,9 +52,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) @@ -71,10 +66,7 @@ def _parse_value(data: object) -> Union[list[str], str]: _operator = d.pop("operator", UNSET) operator: Union[Unset, DatasetIDFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = DatasetIDFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else DatasetIDFilterOperator(_operator) dataset_id_filter = cls(value=value, name=name, operator=operator) diff --git a/src/galileo/resources/models/dataset_name_filter.py b/src/galileo/resources/models/dataset_name_filter.py index 131b12885..6d89c1cdd 100644 --- a/src/galileo/resources/models/dataset_name_filter.py +++ b/src/galileo/resources/models/dataset_name_filter.py @@ -30,11 +30,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value case_sensitive = self.case_sensitive @@ -59,9 +55,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) diff --git a/src/galileo/resources/models/dataset_prepend_row.py b/src/galileo/resources/models/dataset_prepend_row.py index 53527a752..e7d3a68ca 100644 --- a/src/galileo/resources/models/dataset_prepend_row.py +++ b/src/galileo/resources/models/dataset_prepend_row.py @@ -33,10 +33,7 @@ def to_dict(self) -> dict[str, Any]: edit_type = self.edit_type row_id: Union[None, Unset, str] - if isinstance(self.row_id, Unset): - row_id = UNSET - else: - row_id = self.row_id + row_id = UNSET if isinstance(self.row_id, Unset) else self.row_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/dataset_prepend_row_values.py b/src/galileo/resources/models/dataset_prepend_row_values.py index 314dbb541..f40bbede6 100644 --- a/src/galileo/resources/models/dataset_prepend_row_values.py +++ b/src/galileo/resources/models/dataset_prepend_row_values.py @@ -55,9 +55,8 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = DatasetPrependRowValuesAdditionalPropertyType3.from_dict(data) + return DatasetPrependRowValuesAdditionalPropertyType3.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass return cast(Union["DatasetPrependRowValuesAdditionalPropertyType3", None, float, int, str], data) diff --git a/src/galileo/resources/models/dataset_project.py b/src/galileo/resources/models/dataset_project.py index 6c20cfb82..ccc16558b 100644 --- a/src/galileo/resources/models/dataset_project.py +++ b/src/galileo/resources/models/dataset_project.py @@ -75,9 +75,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) diff --git a/src/galileo/resources/models/dataset_row.py b/src/galileo/resources/models/dataset_row.py index 24108561b..c610d9def 100644 --- a/src/galileo/resources/models/dataset_row.py +++ b/src/galileo/resources/models/dataset_row.py @@ -38,10 +38,7 @@ def to_dict(self) -> dict[str, Any]: index = self.index metadata: Union[None, dict[str, Any]] - if isinstance(self.metadata, DatasetRowMetadata): - metadata = self.metadata.to_dict() - else: - metadata = self.metadata + metadata = self.metadata.to_dict() if isinstance(self.metadata, DatasetRowMetadata) else self.metadata row_id = self.row_id @@ -79,9 +76,8 @@ def _parse_metadata(data: object) -> Union["DatasetRowMetadata", None]: try: if not isinstance(data, dict): raise TypeError() - metadata_type_0 = DatasetRowMetadata.from_dict(data) + return DatasetRowMetadata.from_dict(data) - return metadata_type_0 except: # noqa: E722 pass return cast(Union["DatasetRowMetadata", None], data) @@ -100,9 +96,8 @@ def _parse_values_item(data: object) -> Union["DatasetRowValuesItemType3", None, try: if not isinstance(data, dict): raise TypeError() - values_item_type_3 = DatasetRowValuesItemType3.from_dict(data) + return DatasetRowValuesItemType3.from_dict(data) - return values_item_type_3 except: # noqa: E722 pass return cast(Union["DatasetRowValuesItemType3", None, float, int, str], data) diff --git a/src/galileo/resources/models/dataset_row_metadata.py b/src/galileo/resources/models/dataset_row_metadata.py index 8d8892fdc..83364ca33 100644 --- a/src/galileo/resources/models/dataset_row_metadata.py +++ b/src/galileo/resources/models/dataset_row_metadata.py @@ -84,9 +84,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) @@ -103,9 +102,8 @@ def _parse_updated_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - updated_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return updated_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) diff --git a/src/galileo/resources/models/dataset_row_values_dict.py b/src/galileo/resources/models/dataset_row_values_dict.py index 255d309ed..b5686cde5 100644 --- a/src/galileo/resources/models/dataset_row_values_dict.py +++ b/src/galileo/resources/models/dataset_row_values_dict.py @@ -53,9 +53,8 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = DatasetRowValuesDictAdditionalPropertyType3.from_dict(data) + return DatasetRowValuesDictAdditionalPropertyType3.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass return cast(Union["DatasetRowValuesDictAdditionalPropertyType3", None, float, int, str], data) diff --git a/src/galileo/resources/models/dataset_update_row_values.py b/src/galileo/resources/models/dataset_update_row_values.py index edaf5ffe5..48f055261 100644 --- a/src/galileo/resources/models/dataset_update_row_values.py +++ b/src/galileo/resources/models/dataset_update_row_values.py @@ -55,9 +55,8 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = DatasetUpdateRowValuesAdditionalPropertyType3.from_dict(data) + return DatasetUpdateRowValuesAdditionalPropertyType3.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass return cast(Union["DatasetUpdateRowValuesAdditionalPropertyType3", None, float, int, str], data) diff --git a/src/galileo/resources/models/dataset_version_db.py b/src/galileo/resources/models/dataset_version_db.py index 9201df429..aa590ae71 100644 --- a/src/galileo/resources/models/dataset_version_db.py +++ b/src/galileo/resources/models/dataset_version_db.py @@ -119,9 +119,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None], data) diff --git a/src/galileo/resources/models/document.py b/src/galileo/resources/models/document.py index 8da8265be..2e5f4f5c5 100644 --- a/src/galileo/resources/models/document.py +++ b/src/galileo/resources/models/document.py @@ -47,11 +47,6 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _metadata = d.pop("metadata", UNSET) metadata: Union[Unset, DocumentMetadata] - if isinstance(_metadata, Unset): - metadata = UNSET - else: - metadata = DocumentMetadata.from_dict(_metadata) + metadata = UNSET if isinstance(_metadata, Unset) else DocumentMetadata.from_dict(_metadata) - document = cls(page_content=page_content, metadata=metadata) - - return document + return cls(page_content=page_content, metadata=metadata) diff --git a/src/galileo/resources/models/experiment_create_request.py b/src/galileo/resources/models/experiment_create_request.py index 3d03c47de..fa7e261bc 100644 --- a/src/galileo/resources/models/experiment_create_request.py +++ b/src/galileo/resources/models/experiment_create_request.py @@ -45,22 +45,13 @@ def to_dict(self) -> dict[str, Any]: dataset = self.dataset playground_id: Union[None, Unset, str] - if isinstance(self.playground_id, Unset): - playground_id = UNSET - else: - playground_id = self.playground_id + playground_id = UNSET if isinstance(self.playground_id, Unset) else self.playground_id playground_prompt_id: Union[None, Unset, str] - if isinstance(self.playground_prompt_id, Unset): - playground_prompt_id = UNSET - else: - playground_prompt_id = self.playground_prompt_id + playground_prompt_id = UNSET if isinstance(self.playground_prompt_id, Unset) else self.playground_prompt_id task_type: Union[Literal[16], Literal[17], Unset] - if isinstance(self.task_type, Unset): - task_type = UNSET - else: - task_type = self.task_type + task_type = UNSET if isinstance(self.task_type, Unset) else self.task_type field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -91,9 +82,8 @@ def _parse_dataset(data: object) -> Union["ExperimentDatasetRequest", None, Unse try: if not isinstance(data, dict): raise TypeError() - dataset_type_0 = ExperimentDatasetRequest.from_dict(data) + return ExperimentDatasetRequest.from_dict(data) - return dataset_type_0 except: # noqa: E722 pass return cast(Union["ExperimentDatasetRequest", None, Unset], data) diff --git a/src/galileo/resources/models/experiment_dataset.py b/src/galileo/resources/models/experiment_dataset.py index c67723dee..d90f107e4 100644 --- a/src/galileo/resources/models/experiment_dataset.py +++ b/src/galileo/resources/models/experiment_dataset.py @@ -25,22 +25,13 @@ class ExperimentDataset: def to_dict(self) -> dict[str, Any]: dataset_id: Union[None, Unset, str] - if isinstance(self.dataset_id, Unset): - dataset_id = UNSET - else: - dataset_id = self.dataset_id + dataset_id = UNSET if isinstance(self.dataset_id, Unset) else self.dataset_id name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name version_index: Union[None, Unset, int] - if isinstance(self.version_index, Unset): - version_index = UNSET - else: - version_index = self.version_index + version_index = UNSET if isinstance(self.version_index, Unset) else self.version_index field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/experiment_metrics_request.py b/src/galileo/resources/models/experiment_metrics_request.py index 920a6ee71..1805bfe3b 100644 --- a/src/galileo/resources/models/experiment_metrics_request.py +++ b/src/galileo/resources/models/experiment_metrics_request.py @@ -50,13 +50,10 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -96,40 +93,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) diff --git a/src/galileo/resources/models/experiment_playground.py b/src/galileo/resources/models/experiment_playground.py index 491ea1214..4bd116ac5 100644 --- a/src/galileo/resources/models/experiment_playground.py +++ b/src/galileo/resources/models/experiment_playground.py @@ -23,16 +23,10 @@ class ExperimentPlayground: def to_dict(self) -> dict[str, Any]: name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name playground_id: Union[None, Unset, str] - if isinstance(self.playground_id, Unset): - playground_id = UNSET - else: - playground_id = self.playground_id + playground_id = UNSET if isinstance(self.playground_id, Unset) else self.playground_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/experiment_prompt.py b/src/galileo/resources/models/experiment_prompt.py index e51288fc9..339a6e246 100644 --- a/src/galileo/resources/models/experiment_prompt.py +++ b/src/galileo/resources/models/experiment_prompt.py @@ -27,28 +27,16 @@ class ExperimentPrompt: def to_dict(self) -> dict[str, Any]: content: Union[None, Unset, str] - if isinstance(self.content, Unset): - content = UNSET - else: - content = self.content + content = UNSET if isinstance(self.content, Unset) else self.content name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name prompt_template_id: Union[None, Unset, str] - if isinstance(self.prompt_template_id, Unset): - prompt_template_id = UNSET - else: - prompt_template_id = self.prompt_template_id + prompt_template_id = UNSET if isinstance(self.prompt_template_id, Unset) else self.prompt_template_id version_index: Union[None, Unset, int] - if isinstance(self.version_index, Unset): - version_index = UNSET - else: - version_index = self.version_index + version_index = UNSET if isinstance(self.version_index, Unset) else self.version_index field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/experiment_response.py b/src/galileo/resources/models/experiment_response.py index 1db0e4c62..f37180e33 100644 --- a/src/galileo/resources/models/experiment_response.py +++ b/src/galileo/resources/models/experiment_response.py @@ -96,10 +96,7 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by dataset: Union[None, Unset, dict[str, Any]] if isinstance(self.dataset, Unset): @@ -120,10 +117,7 @@ def to_dict(self) -> dict[str, Any]: playground = self.playground playground_id: Union[None, Unset, str] - if isinstance(self.playground_id, Unset): - playground_id = UNSET - else: - playground_id = self.playground_id + playground_id = UNSET if isinstance(self.playground_id, Unset) else self.playground_id prompt: Union[None, Unset, dict[str, Any]] if isinstance(self.prompt, Unset): @@ -134,10 +128,7 @@ def to_dict(self) -> dict[str, Any]: prompt = self.prompt prompt_model: Union[None, Unset, str] - if isinstance(self.prompt_model, Unset): - prompt_model = UNSET - else: - prompt_model = self.prompt_model + prompt_model = UNSET if isinstance(self.prompt_model, Unset) else self.prompt_model prompt_run_settings: Union[None, Unset, dict[str, Any]] if isinstance(self.prompt_run_settings, Unset): @@ -148,16 +139,10 @@ def to_dict(self) -> dict[str, Any]: prompt_run_settings = self.prompt_run_settings rank: Union[None, Unset, int] - if isinstance(self.rank, Unset): - rank = UNSET - else: - rank = self.rank + rank = UNSET if isinstance(self.rank, Unset) else self.rank ranking_score: Union[None, Unset, float] - if isinstance(self.ranking_score, Unset): - ranking_score = UNSET - else: - ranking_score = self.ranking_score + ranking_score = UNSET if isinstance(self.ranking_score, Unset) else self.ranking_score tags: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.tags, Unset): @@ -172,10 +157,7 @@ def to_dict(self) -> dict[str, Any]: updated_at = self.updated_at winner: Union[None, Unset, bool] - if isinstance(self.winner, Unset): - winner = UNSET - else: - winner = self.winner + winner = UNSET if isinstance(self.winner, Unset) else self.winner field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -248,10 +230,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_created_by(data: object) -> Union[None, Unset, str]: if data is None: @@ -270,9 +249,8 @@ def _parse_dataset(data: object) -> Union["ExperimentDataset", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - dataset_type_0 = ExperimentDataset.from_dict(data) + return ExperimentDataset.from_dict(data) - return dataset_type_0 except: # noqa: E722 pass return cast(Union["ExperimentDataset", None, Unset], data) @@ -289,9 +267,8 @@ def _parse_playground(data: object) -> Union["ExperimentPlayground", None, Unset try: if not isinstance(data, dict): raise TypeError() - playground_type_0 = ExperimentPlayground.from_dict(data) + return ExperimentPlayground.from_dict(data) - return playground_type_0 except: # noqa: E722 pass return cast(Union["ExperimentPlayground", None, Unset], data) @@ -315,9 +292,8 @@ def _parse_prompt(data: object) -> Union["ExperimentPrompt", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - prompt_type_0 = ExperimentPrompt.from_dict(data) + return ExperimentPrompt.from_dict(data) - return prompt_type_0 except: # noqa: E722 pass return cast(Union["ExperimentPrompt", None, Unset], data) @@ -341,9 +317,8 @@ def _parse_prompt_run_settings(data: object) -> Union["PromptRunSettings", None, try: if not isinstance(data, dict): raise TypeError() - prompt_run_settings_type_0 = PromptRunSettings.from_dict(data) + return PromptRunSettings.from_dict(data) - return prompt_run_settings_type_0 except: # noqa: E722 pass return cast(Union["PromptRunSettings", None, Unset], data) @@ -370,10 +345,7 @@ def _parse_ranking_score(data: object) -> Union[None, Unset, float]: _tags = d.pop("tags", UNSET) tags: Union[Unset, ExperimentResponseTags] - if isinstance(_tags, Unset): - tags = UNSET - else: - tags = ExperimentResponseTags.from_dict(_tags) + tags = UNSET if isinstance(_tags, Unset) else ExperimentResponseTags.from_dict(_tags) def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: if data is None: @@ -383,9 +355,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/experiment_update_request.py b/src/galileo/resources/models/experiment_update_request.py index 3d017539a..f634a7143 100644 --- a/src/galileo/resources/models/experiment_update_request.py +++ b/src/galileo/resources/models/experiment_update_request.py @@ -25,10 +25,7 @@ def to_dict(self) -> dict[str, Any]: name = self.name task_type: Union[Literal[16], Literal[17], Unset] - if isinstance(self.task_type, Unset): - task_type = UNSET - else: - task_type = self.task_type + task_type = UNSET if isinstance(self.task_type, Unset) else self.task_type field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/extended_agent_span_record.py b/src/galileo/resources/models/extended_agent_span_record.py index da4f53a53..e3465a6bb 100644 --- a/src/galileo/resources/models/extended_agent_span_record.py +++ b/src/galileo/resources/models/extended_agent_span_record.py @@ -114,32 +114,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -168,10 +156,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -216,32 +201,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -335,17 +308,11 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _agent_type = d.pop("agent_type", UNSET) agent_type: Union[Unset, AgentType] - if isinstance(_agent_type, Unset): - agent_type = UNSET - else: - agent_type = AgentType(_agent_type) + agent_type = UNSET if isinstance(_agent_type, Unset) else AgentType(_agent_type) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -420,9 +387,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedAgentSpanRecordMetricInfo try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedAgentSpanRecordMetricInfoType0.from_dict(data) + return ExtendedAgentSpanRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedAgentSpanRecordMetricInfoType0", None, Unset], data) @@ -431,10 +397,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedAgentSpanRecordMetricInfo _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -455,9 +418,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -507,9 +469,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -579,9 +540,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_agent_span_record_metric_info_type_0.py b/src/galileo/resources/models/extended_agent_span_record_metric_info_type_0.py index 6d595c3fe..4ca32aa44 100644 --- a/src/galileo/resources/models/extended_agent_span_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_agent_span_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_agent_span_record_with_children.py b/src/galileo/resources/models/extended_agent_span_record_with_children.py index 241c0deb7..e8d27c689 100644 --- a/src/galileo/resources/models/extended_agent_span_record_with_children.py +++ b/src/galileo/resources/models/extended_agent_span_record_with_children.py @@ -145,32 +145,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -199,10 +187,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -247,23 +232,22 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, ExtendedAgentSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedWorkflowSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedLlmSpanRecord): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedToolSpanRecordWithChildren): + if isinstance( + spans_item_data, + ( + ExtendedAgentSpanRecordWithChildren, + ExtendedWorkflowSpanRecordWithChildren, + ExtendedLlmSpanRecord, + ExtendedToolSpanRecordWithChildren, + ), + ): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -271,26 +255,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -396,17 +371,11 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _agent_type = d.pop("agent_type", UNSET) agent_type: Union[Unset, AgentType] - if isinstance(_agent_type, Unset): - agent_type = UNSET - else: - agent_type = AgentType(_agent_type) + agent_type = UNSET if isinstance(_agent_type, Unset) else AgentType(_agent_type) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -483,9 +452,8 @@ def _parse_metric_info( try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedAgentSpanRecordWithChildrenMetricInfoType0.from_dict(data) + return ExtendedAgentSpanRecordWithChildrenMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedAgentSpanRecordWithChildrenMetricInfoType0", None, Unset], data) @@ -494,10 +462,7 @@ def _parse_metric_info( _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -518,9 +483,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -570,9 +534,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -617,40 +580,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = ExtendedAgentSpanRecordWithChildren.from_dict(data) + return ExtendedAgentSpanRecordWithChildren.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = ExtendedWorkflowSpanRecordWithChildren.from_dict(data) + return ExtendedWorkflowSpanRecordWithChildren.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = ExtendedLlmSpanRecord.from_dict(data) + return ExtendedLlmSpanRecord.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = ExtendedToolSpanRecordWithChildren.from_dict(data) + return ExtendedToolSpanRecordWithChildren.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ExtendedRetrieverSpanRecordWithChildren.from_dict(data) - - return spans_item_type_4 + return ExtendedRetrieverSpanRecordWithChildren.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -697,9 +654,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_agent_span_record_with_children_metric_info_type_0.py b/src/galileo/resources/models/extended_agent_span_record_with_children_metric_info_type_0.py index 5604729c2..4031275df 100644 --- a/src/galileo/resources/models/extended_agent_span_record_with_children_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_agent_span_record_with_children_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_llm_span_record.py b/src/galileo/resources/models/extended_llm_span_record.py index 83b6c88ac..08273b887 100644 --- a/src/galileo/resources/models/extended_llm_span_record.py +++ b/src/galileo/resources/models/extended_llm_span_record.py @@ -116,38 +116,23 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id finish_reason: Union[None, Unset, str] - if isinstance(self.finish_reason, Unset): - finish_reason = UNSET - else: - finish_reason = self.finish_reason + finish_reason = UNSET if isinstance(self.finish_reason, Unset) else self.finish_reason has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.input_, Unset): @@ -171,16 +156,10 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id model: Union[None, Unset, str] - if isinstance(self.model, Unset): - model = UNSET - else: - model = self.model + model = UNSET if isinstance(self.model, Unset) else self.model name = self.name @@ -209,32 +188,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags temperature: Union[None, Unset, float] - if isinstance(self.temperature, Unset): - temperature = UNSET - else: - temperature = self.temperature + temperature = UNSET if isinstance(self.temperature, Unset) else self.temperature tools: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.tools, Unset): @@ -249,10 +216,7 @@ def to_dict(self) -> dict[str, Any]: tools = self.tools trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -352,10 +316,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -426,9 +387,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedLlmSpanRecordMetricInfoTy try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedLlmSpanRecordMetricInfoType0.from_dict(data) + return ExtendedLlmSpanRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedLlmSpanRecordMetricInfoType0", None, Unset], data) @@ -437,10 +397,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedLlmSpanRecordMetricInfoTy _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, LlmMetrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = LlmMetrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else LlmMetrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -464,10 +421,7 @@ def _parse_model(data: object) -> Union[None, Unset, str]: _output = d.pop("output", UNSET) output: Union[Unset, Message] - if isinstance(_output, Unset): - output = UNSET - else: - output = Message.from_dict(_output) + output = UNSET if isinstance(_output, Unset) else Message.from_dict(_output) def _parse_redacted_input(data: object) -> Union[None, Unset, list["Message"]]: if data is None: @@ -499,9 +453,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_0 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_0 except: # noqa: E722 pass return cast(Union["Message", None, Unset], data) @@ -589,9 +542,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_llm_span_record_metric_info_type_0.py b/src/galileo/resources/models/extended_llm_span_record_metric_info_type_0.py index f0101f9ea..2d2cdc19f 100644 --- a/src/galileo/resources/models/extended_llm_span_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_llm_span_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_retriever_span_record.py b/src/galileo/resources/models/extended_retriever_span_record.py index e25abcf21..f6f89b547 100644 --- a/src/galileo/resources/models/extended_retriever_span_record.py +++ b/src/galileo/resources/models/extended_retriever_span_record.py @@ -108,32 +108,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_ = self.input_ @@ -152,10 +140,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -167,10 +152,7 @@ def to_dict(self) -> dict[str, Any]: output.append(output_item) redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.redacted_output, Unset): @@ -185,32 +167,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -303,10 +273,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -363,9 +330,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedRetrieverSpanRecordMetric try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedRetrieverSpanRecordMetricInfoType0.from_dict(data) + return ExtendedRetrieverSpanRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedRetrieverSpanRecordMetricInfoType0", None, Unset], data) @@ -374,10 +340,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedRetrieverSpanRecordMetric _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -478,9 +441,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_retriever_span_record_metric_info_type_0.py b/src/galileo/resources/models/extended_retriever_span_record_metric_info_type_0.py index 86d54bb96..8347008b1 100644 --- a/src/galileo/resources/models/extended_retriever_span_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_retriever_span_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_retriever_span_record_with_children.py b/src/galileo/resources/models/extended_retriever_span_record_with_children.py index 14b11add6..fad700e34 100644 --- a/src/galileo/resources/models/extended_retriever_span_record_with_children.py +++ b/src/galileo/resources/models/extended_retriever_span_record_with_children.py @@ -137,32 +137,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_ = self.input_ @@ -181,10 +169,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -196,10 +181,7 @@ def to_dict(self) -> dict[str, Any]: output.append(output_item) redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.redacted_output, Unset): @@ -214,23 +196,22 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, ExtendedAgentSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedWorkflowSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedLlmSpanRecord): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedToolSpanRecordWithChildren): + if isinstance( + spans_item_data, + ( + ExtendedAgentSpanRecordWithChildren, + ExtendedWorkflowSpanRecordWithChildren, + ExtendedLlmSpanRecord, + ExtendedToolSpanRecordWithChildren, + ), + ): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -238,26 +219,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -360,10 +332,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -422,9 +391,8 @@ def _parse_metric_info( try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedRetrieverSpanRecordWithChildrenMetricInfoType0.from_dict(data) + return ExtendedRetrieverSpanRecordWithChildrenMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedRetrieverSpanRecordWithChildrenMetricInfoType0", None, Unset], data) @@ -433,10 +401,7 @@ def _parse_metric_info( _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -512,40 +477,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = ExtendedAgentSpanRecordWithChildren.from_dict(data) + return ExtendedAgentSpanRecordWithChildren.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = ExtendedWorkflowSpanRecordWithChildren.from_dict(data) + return ExtendedWorkflowSpanRecordWithChildren.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = ExtendedLlmSpanRecord.from_dict(data) + return ExtendedLlmSpanRecord.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = ExtendedToolSpanRecordWithChildren.from_dict(data) + return ExtendedToolSpanRecordWithChildren.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ExtendedRetrieverSpanRecordWithChildren.from_dict(data) - - return spans_item_type_4 + return ExtendedRetrieverSpanRecordWithChildren.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -592,9 +551,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_retriever_span_record_with_children_metric_info_type_0.py b/src/galileo/resources/models/extended_retriever_span_record_with_children_metric_info_type_0.py index 062742bfd..e0fd3399f 100644 --- a/src/galileo/resources/models/extended_retriever_span_record_with_children_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_retriever_span_record_with_children_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_session_record.py b/src/galileo/resources/models/extended_session_record.py index 883f69248..7ae838c3a 100644 --- a/src/galileo/resources/models/extended_session_record.py +++ b/src/galileo/resources/models/extended_session_record.py @@ -99,32 +99,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -151,10 +139,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -173,10 +158,7 @@ def to_dict(self) -> dict[str, Any]: output = self.output previous_session_id: Union[None, Unset, str] - if isinstance(self.previous_session_id, Unset): - previous_session_id = UNSET - else: - previous_session_id = self.previous_session_id + previous_session_id = UNSET if isinstance(self.previous_session_id, Unset) else self.previous_session_id redacted_input: Union[None, Unset, list[dict[str, Any]], str] if isinstance(self.redacted_input, Unset): @@ -205,32 +187,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -316,10 +286,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -392,9 +359,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedSessionRecordMetricInfoTy try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedSessionRecordMetricInfoType0.from_dict(data) + return ExtendedSessionRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedSessionRecordMetricInfoType0", None, Unset], data) @@ -403,10 +369,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedSessionRecordMetricInfoTy _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -427,9 +390,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -488,9 +450,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -560,9 +521,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_session_record_metric_info_type_0.py b/src/galileo/resources/models/extended_session_record_metric_info_type_0.py index c52e69a78..1f88f43fc 100644 --- a/src/galileo/resources/models/extended_session_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_session_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_session_record_with_children.py b/src/galileo/resources/models/extended_session_record_with_children.py index c9efae0d9..98952c14d 100644 --- a/src/galileo/resources/models/extended_session_record_with_children.py +++ b/src/galileo/resources/models/extended_session_record_with_children.py @@ -111,32 +111,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -163,10 +151,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -185,10 +170,7 @@ def to_dict(self) -> dict[str, Any]: output = self.output previous_session_id: Union[None, Unset, str] - if isinstance(self.previous_session_id, Unset): - previous_session_id = UNSET - else: - previous_session_id = self.previous_session_id + previous_session_id = UNSET if isinstance(self.previous_session_id, Unset) else self.previous_session_id redacted_input: Union[None, Unset, list[dict[str, Any]], str] if isinstance(self.redacted_input, Unset): @@ -217,32 +199,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id traces: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.traces, Unset): @@ -344,10 +314,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -420,9 +387,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedSessionRecordWithChildren try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedSessionRecordWithChildrenMetricInfoType0.from_dict(data) + return ExtendedSessionRecordWithChildrenMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedSessionRecordWithChildrenMetricInfoType0", None, Unset], data) @@ -431,10 +397,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedSessionRecordWithChildren _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -455,9 +418,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -516,9 +478,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -595,9 +556,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_session_record_with_children_metric_info_type_0.py b/src/galileo/resources/models/extended_session_record_with_children_metric_info_type_0.py index 705772719..c44b11967 100644 --- a/src/galileo/resources/models/extended_session_record_with_children_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_session_record_with_children_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_tool_span_record.py b/src/galileo/resources/models/extended_tool_span_record.py index 14a531557..0b051c8e8 100644 --- a/src/galileo/resources/models/extended_tool_span_record.py +++ b/src/galileo/resources/models/extended_tool_span_record.py @@ -106,32 +106,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_ = self.input_ @@ -150,64 +138,37 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, str] - if isinstance(self.redacted_output, Unset): - redacted_output = UNSET - else: - redacted_output = self.redacted_output + redacted_output = UNSET if isinstance(self.redacted_output, Unset) else self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags tool_call_id: Union[None, Unset, str] - if isinstance(self.tool_call_id, Unset): - tool_call_id = UNSET - else: - tool_call_id = self.tool_call_id + tool_call_id = UNSET if isinstance(self.tool_call_id, Unset) else self.tool_call_id trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -299,10 +260,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -359,9 +317,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedToolSpanRecordMetricInfoT try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedToolSpanRecordMetricInfoType0.from_dict(data) + return ExtendedToolSpanRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedToolSpanRecordMetricInfoType0", None, Unset], data) @@ -370,10 +327,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedToolSpanRecordMetricInfoT _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -472,9 +426,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_tool_span_record_metric_info_type_0.py b/src/galileo/resources/models/extended_tool_span_record_metric_info_type_0.py index 6dd4c11cb..4256bb303 100644 --- a/src/galileo/resources/models/extended_tool_span_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_tool_span_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_tool_span_record_with_children.py b/src/galileo/resources/models/extended_tool_span_record_with_children.py index 60c530903..756a1e76f 100644 --- a/src/galileo/resources/models/extended_tool_span_record_with_children.py +++ b/src/galileo/resources/models/extended_tool_span_record_with_children.py @@ -137,32 +137,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_ = self.input_ @@ -181,49 +169,36 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, str] - if isinstance(self.redacted_output, Unset): - redacted_output = UNSET - else: - redacted_output = self.redacted_output + redacted_output = UNSET if isinstance(self.redacted_output, Unset) else self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, ExtendedAgentSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedWorkflowSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedLlmSpanRecord): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedToolSpanRecordWithChildren): + if isinstance( + spans_item_data, + ( + ExtendedAgentSpanRecordWithChildren, + ExtendedWorkflowSpanRecordWithChildren, + ExtendedLlmSpanRecord, + ExtendedToolSpanRecordWithChildren, + ), + ): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -231,32 +206,20 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags tool_call_id: Union[None, Unset, str] - if isinstance(self.tool_call_id, Unset): - tool_call_id = UNSET - else: - tool_call_id = self.tool_call_id + tool_call_id = UNSET if isinstance(self.tool_call_id, Unset) else self.tool_call_id trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -360,10 +323,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -420,9 +380,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedToolSpanRecordWithChildre try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedToolSpanRecordWithChildrenMetricInfoType0.from_dict(data) + return ExtendedToolSpanRecordWithChildrenMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedToolSpanRecordWithChildrenMetricInfoType0", None, Unset], data) @@ -431,10 +390,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedToolSpanRecordWithChildre _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -499,40 +455,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = ExtendedAgentSpanRecordWithChildren.from_dict(data) + return ExtendedAgentSpanRecordWithChildren.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = ExtendedWorkflowSpanRecordWithChildren.from_dict(data) + return ExtendedWorkflowSpanRecordWithChildren.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = ExtendedLlmSpanRecord.from_dict(data) + return ExtendedLlmSpanRecord.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = ExtendedToolSpanRecordWithChildren.from_dict(data) + return ExtendedToolSpanRecordWithChildren.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ExtendedRetrieverSpanRecordWithChildren.from_dict(data) - - return spans_item_type_4 + return ExtendedRetrieverSpanRecordWithChildren.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -588,9 +538,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_tool_span_record_with_children_metric_info_type_0.py b/src/galileo/resources/models/extended_tool_span_record_with_children_metric_info_type_0.py index f65d3ad05..15113cd91 100644 --- a/src/galileo/resources/models/extended_tool_span_record_with_children_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_tool_span_record_with_children_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_trace_record.py b/src/galileo/resources/models/extended_trace_record.py index 667fdb325..3b51284d9 100644 --- a/src/galileo/resources/models/extended_trace_record.py +++ b/src/galileo/resources/models/extended_trace_record.py @@ -103,36 +103,24 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id feedback_rating_info: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.feedback_rating_info, Unset): feedback_rating_info = self.feedback_rating_info.to_dict() has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_ = self.input_ @@ -151,42 +139,24 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, str] - if isinstance(self.redacted_output, Unset): - redacted_output = UNSET - else: - redacted_output = self.redacted_output + redacted_output = UNSET if isinstance(self.redacted_output, Unset) else self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): @@ -279,10 +249,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -346,9 +313,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedTraceRecordMetricInfoType try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedTraceRecordMetricInfoType0.from_dict(data) + return ExtendedTraceRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedTraceRecordMetricInfoType0", None, Unset], data) @@ -357,10 +323,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedTraceRecordMetricInfoType _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -432,9 +395,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_trace_record_metric_info_type_0.py b/src/galileo/resources/models/extended_trace_record_metric_info_type_0.py index 0302e4c3a..67731a48f 100644 --- a/src/galileo/resources/models/extended_trace_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_trace_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_trace_record_with_children.py b/src/galileo/resources/models/extended_trace_record_with_children.py index 222cd20a5..ac2e0b3b7 100644 --- a/src/galileo/resources/models/extended_trace_record_with_children.py +++ b/src/galileo/resources/models/extended_trace_record_with_children.py @@ -136,36 +136,24 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id feedback_rating_info: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.feedback_rating_info, Unset): feedback_rating_info = self.feedback_rating_info.to_dict() has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_ = self.input_ @@ -184,49 +172,36 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, str] - if isinstance(self.redacted_output, Unset): - redacted_output = UNSET - else: - redacted_output = self.redacted_output + redacted_output = UNSET if isinstance(self.redacted_output, Unset) else self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, ExtendedAgentSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedWorkflowSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedLlmSpanRecord): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedToolSpanRecordWithChildren): + if isinstance( + spans_item_data, + ( + ExtendedAgentSpanRecordWithChildren, + ExtendedWorkflowSpanRecordWithChildren, + ExtendedLlmSpanRecord, + ExtendedToolSpanRecordWithChildren, + ), + ): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -234,10 +209,7 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): @@ -345,10 +317,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -412,9 +381,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedTraceRecordWithChildrenMe try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedTraceRecordWithChildrenMetricInfoType0.from_dict(data) + return ExtendedTraceRecordWithChildrenMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedTraceRecordWithChildrenMetricInfoType0", None, Unset], data) @@ -423,10 +391,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedTraceRecordWithChildrenMe _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -491,40 +456,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = ExtendedAgentSpanRecordWithChildren.from_dict(data) + return ExtendedAgentSpanRecordWithChildren.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = ExtendedWorkflowSpanRecordWithChildren.from_dict(data) + return ExtendedWorkflowSpanRecordWithChildren.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = ExtendedLlmSpanRecord.from_dict(data) + return ExtendedLlmSpanRecord.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = ExtendedToolSpanRecordWithChildren.from_dict(data) + return ExtendedToolSpanRecordWithChildren.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ExtendedRetrieverSpanRecordWithChildren.from_dict(data) - - return spans_item_type_4 + return ExtendedRetrieverSpanRecordWithChildren.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -553,9 +512,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_trace_record_with_children_metric_info_type_0.py b/src/galileo/resources/models/extended_trace_record_with_children_metric_info_type_0.py index 2e10ee058..4d214ea0a 100644 --- a/src/galileo/resources/models/extended_trace_record_with_children_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_trace_record_with_children_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_workflow_span_record.py b/src/galileo/resources/models/extended_workflow_span_record.py index b2f5e38e1..3eab7ca1b 100644 --- a/src/galileo/resources/models/extended_workflow_span_record.py +++ b/src/galileo/resources/models/extended_workflow_span_record.py @@ -108,32 +108,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -162,10 +150,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -210,32 +195,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -327,10 +300,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -405,9 +375,8 @@ def _parse_metric_info(data: object) -> Union["ExtendedWorkflowSpanRecordMetricI try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedWorkflowSpanRecordMetricInfoType0.from_dict(data) + return ExtendedWorkflowSpanRecordMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedWorkflowSpanRecordMetricInfoType0", None, Unset], data) @@ -416,10 +385,7 @@ def _parse_metric_info(data: object) -> Union["ExtendedWorkflowSpanRecordMetricI _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -440,9 +406,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -492,9 +457,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -564,9 +528,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_workflow_span_record_metric_info_type_0.py b/src/galileo/resources/models/extended_workflow_span_record_metric_info_type_0.py index 2777b8d75..ca0ffb673 100644 --- a/src/galileo/resources/models/extended_workflow_span_record_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_workflow_span_record_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/extended_workflow_span_record_with_children.py b/src/galileo/resources/models/extended_workflow_span_record_with_children.py index ec5675c96..feacea124 100644 --- a/src/galileo/resources/models/extended_workflow_span_record_with_children.py +++ b/src/galileo/resources/models/extended_workflow_span_record_with_children.py @@ -138,32 +138,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id has_children: Union[None, Unset, bool] - if isinstance(self.has_children, Unset): - has_children = UNSET - else: - has_children = self.has_children + has_children = UNSET if isinstance(self.has_children, Unset) else self.has_children input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -192,10 +180,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() metrics_batch_id: Union[None, Unset, str] - if isinstance(self.metrics_batch_id, Unset): - metrics_batch_id = UNSET - else: - metrics_batch_id = self.metrics_batch_id + metrics_batch_id = UNSET if isinstance(self.metrics_batch_id, Unset) else self.metrics_batch_id name = self.name @@ -240,23 +225,22 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_batch_id: Union[None, Unset, str] - if isinstance(self.session_batch_id, Unset): - session_batch_id = UNSET - else: - session_batch_id = self.session_batch_id + session_batch_id = UNSET if isinstance(self.session_batch_id, Unset) else self.session_batch_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, ExtendedAgentSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedWorkflowSpanRecordWithChildren): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedLlmSpanRecord): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, ExtendedToolSpanRecordWithChildren): + if isinstance( + spans_item_data, + ( + ExtendedAgentSpanRecordWithChildren, + ExtendedWorkflowSpanRecordWithChildren, + ExtendedLlmSpanRecord, + ExtendedToolSpanRecordWithChildren, + ), + ): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -264,26 +248,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -387,10 +362,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -467,9 +439,8 @@ def _parse_metric_info( try: if not isinstance(data, dict): raise TypeError() - metric_info_type_0 = ExtendedWorkflowSpanRecordWithChildrenMetricInfoType0.from_dict(data) + return ExtendedWorkflowSpanRecordWithChildrenMetricInfoType0.from_dict(data) - return metric_info_type_0 except: # noqa: E722 pass return cast(Union["ExtendedWorkflowSpanRecordWithChildrenMetricInfoType0", None, Unset], data) @@ -478,10 +449,7 @@ def _parse_metric_info( _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) def _parse_metrics_batch_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -502,9 +470,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -554,9 +521,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -601,40 +567,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = ExtendedAgentSpanRecordWithChildren.from_dict(data) + return ExtendedAgentSpanRecordWithChildren.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = ExtendedWorkflowSpanRecordWithChildren.from_dict(data) + return ExtendedWorkflowSpanRecordWithChildren.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = ExtendedLlmSpanRecord.from_dict(data) + return ExtendedLlmSpanRecord.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = ExtendedToolSpanRecordWithChildren.from_dict(data) + return ExtendedToolSpanRecordWithChildren.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ExtendedRetrieverSpanRecordWithChildren.from_dict(data) - - return spans_item_type_4 + return ExtendedRetrieverSpanRecordWithChildren.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -681,9 +641,8 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/extended_workflow_span_record_with_children_metric_info_type_0.py b/src/galileo/resources/models/extended_workflow_span_record_with_children_metric_info_type_0.py index 360616682..32ffab858 100644 --- a/src/galileo/resources/models/extended_workflow_span_record_with_children_metric_info_type_0.py +++ b/src/galileo/resources/models/extended_workflow_span_record_with_children_metric_info_type_0.py @@ -44,17 +44,10 @@ def to_dict(self) -> dict[str, Any]: field_dict: dict[str, Any] = {} for prop_name, prop in self.additional_properties.items(): - if isinstance(prop, MetricNotComputed): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricPending): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricComputing): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricNotApplicable): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricSuccess): - field_dict[prop_name] = prop.to_dict() - elif isinstance(prop, MetricError): + if isinstance( + prop, + (MetricNotComputed, MetricPending, MetricComputing, MetricNotApplicable, MetricSuccess, MetricError), + ): field_dict[prop_name] = prop.to_dict() else: field_dict[prop_name] = prop.to_dict() @@ -91,56 +84,48 @@ def _parse_additional_property( try: if not isinstance(data, dict): raise TypeError() - additional_property_type_0 = MetricNotComputed.from_dict(data) + return MetricNotComputed.from_dict(data) - return additional_property_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_1 = MetricPending.from_dict(data) + return MetricPending.from_dict(data) - return additional_property_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_2 = MetricComputing.from_dict(data) + return MetricComputing.from_dict(data) - return additional_property_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_3 = MetricNotApplicable.from_dict(data) + return MetricNotApplicable.from_dict(data) - return additional_property_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_4 = MetricSuccess.from_dict(data) + return MetricSuccess.from_dict(data) - return additional_property_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - additional_property_type_5 = MetricError.from_dict(data) + return MetricError.from_dict(data) - return additional_property_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - additional_property_type_6 = MetricFailed.from_dict(data) - - return additional_property_type_6 + return MetricFailed.from_dict(data) additional_property = _parse_additional_property(prop_dict) diff --git a/src/galileo/resources/models/factuality_template.py b/src/galileo/resources/models/factuality_template.py index 8b8728d17..52763a397 100644 --- a/src/galileo/resources/models/factuality_template.py +++ b/src/galileo/resources/models/factuality_template.py @@ -69,10 +69,7 @@ def to_dict(self) -> dict[str, Any]: explanation_field_name = self.explanation_field_name metric_description: Union[None, Unset, str] - if isinstance(self.metric_description, Unset): - metric_description = UNSET - else: - metric_description = self.metric_description + metric_description = UNSET if isinstance(self.metric_description, Unset) else self.metric_description metric_few_shot_examples: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.metric_few_shot_examples, Unset): diff --git a/src/galileo/resources/models/feedback_aggregate.py b/src/galileo/resources/models/feedback_aggregate.py index ec37b637a..d2b3fab03 100644 --- a/src/galileo/resources/models/feedback_aggregate.py +++ b/src/galileo/resources/models/feedback_aggregate.py @@ -30,11 +30,7 @@ def to_dict(self) -> dict[str, Any]: from ..models.star_aggregate import StarAggregate aggregate: dict[str, Any] - if isinstance(self.aggregate, LikeDislikeAggregate): - aggregate = self.aggregate.to_dict() - elif isinstance(self.aggregate, StarAggregate): - aggregate = self.aggregate.to_dict() - elif isinstance(self.aggregate, ScoreAggregate): + if isinstance(self.aggregate, (LikeDislikeAggregate, StarAggregate, ScoreAggregate)): aggregate = self.aggregate.to_dict() else: aggregate = self.aggregate.to_dict() @@ -60,32 +56,27 @@ def _parse_aggregate( try: if not isinstance(data, dict): raise TypeError() - aggregate_type_0 = LikeDislikeAggregate.from_dict(data) + return LikeDislikeAggregate.from_dict(data) - return aggregate_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - aggregate_type_1 = StarAggregate.from_dict(data) + return StarAggregate.from_dict(data) - return aggregate_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - aggregate_type_2 = ScoreAggregate.from_dict(data) + return ScoreAggregate.from_dict(data) - return aggregate_type_2 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - aggregate_type_3 = TagsAggregate.from_dict(data) - - return aggregate_type_3 + return TagsAggregate.from_dict(data) aggregate = _parse_aggregate(d.pop("aggregate")) diff --git a/src/galileo/resources/models/feedback_rating_db.py b/src/galileo/resources/models/feedback_rating_db.py index e26f1d0e9..a8d99d5e8 100644 --- a/src/galileo/resources/models/feedback_rating_db.py +++ b/src/galileo/resources/models/feedback_rating_db.py @@ -47,22 +47,13 @@ def to_dict(self) -> dict[str, Any]: created_by = self.created_by rating: dict[str, Any] - if isinstance(self.rating, LikeDislikeRating): - rating = self.rating.to_dict() - elif isinstance(self.rating, StarRating): - rating = self.rating.to_dict() - elif isinstance(self.rating, ScoreRating): - rating = self.rating.to_dict() - elif isinstance(self.rating, TagsRating): + if isinstance(self.rating, (LikeDislikeRating, StarRating, ScoreRating, TagsRating)): rating = self.rating.to_dict() else: rating = self.rating.to_dict() explanation: Union[None, Unset, str] - if isinstance(self.explanation, Unset): - explanation = UNSET - else: - explanation = self.explanation + explanation = UNSET if isinstance(self.explanation, Unset) else self.explanation field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -96,40 +87,34 @@ def _parse_rating( try: if not isinstance(data, dict): raise TypeError() - rating_type_0 = LikeDislikeRating.from_dict(data) + return LikeDislikeRating.from_dict(data) - return rating_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - rating_type_1 = StarRating.from_dict(data) + return StarRating.from_dict(data) - return rating_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - rating_type_2 = ScoreRating.from_dict(data) + return ScoreRating.from_dict(data) - return rating_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - rating_type_3 = TagsRating.from_dict(data) + return TagsRating.from_dict(data) - return rating_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - rating_type_4 = TextRating.from_dict(data) - - return rating_type_4 + return TextRating.from_dict(data) rating = _parse_rating(d.pop("rating")) diff --git a/src/galileo/resources/models/feedback_rating_info.py b/src/galileo/resources/models/feedback_rating_info.py index d11f0f143..af4cfaa0b 100644 --- a/src/galileo/resources/models/feedback_rating_info.py +++ b/src/galileo/resources/models/feedback_rating_info.py @@ -4,7 +4,7 @@ from attrs import define as _attrs_define from attrs import field as _attrs_field -from ..models.feedback_rating_info_feedback_type import FeedbackRatingInfoFeedbackType +from ..models.feedback_type import FeedbackType T = TypeVar("T", bound="FeedbackRatingInfo") @@ -13,27 +13,24 @@ class FeedbackRatingInfo: """ Attributes: - explanation (str): - feedback_type (FeedbackRatingInfoFeedbackType): + explanation (Union[None, str]): + feedback_type (FeedbackType): value (Union[bool, int, list[str], str]): """ - explanation: str - feedback_type: FeedbackRatingInfoFeedbackType + explanation: Union[None, str] + feedback_type: FeedbackType value: Union[bool, int, list[str], str] additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict) def to_dict(self) -> dict[str, Any]: + explanation: Union[None, str] explanation = self.explanation feedback_type = self.feedback_type.value value: Union[bool, int, list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -44,17 +41,22 @@ def to_dict(self) -> dict[str, Any]: @classmethod def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) - explanation = d.pop("explanation") - feedback_type = FeedbackRatingInfoFeedbackType(d.pop("feedback_type")) + def _parse_explanation(data: object) -> Union[None, str]: + if data is None: + return data + return cast(Union[None, str], data) + + explanation = _parse_explanation(d.pop("explanation")) + + feedback_type = FeedbackType(d.pop("feedback_type")) def _parse_value(data: object) -> Union[bool, int, list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_2 = cast(list[str], data) + return cast(list[str], data) - return value_type_2 except: # noqa: E722 pass return cast(Union[bool, int, list[str], str], data) diff --git a/src/galileo/resources/models/feedback_rating_info_feedback_type.py b/src/galileo/resources/models/feedback_type.py similarity index 79% rename from src/galileo/resources/models/feedback_rating_info_feedback_type.py rename to src/galileo/resources/models/feedback_type.py index 58aa00c1e..83328c285 100644 --- a/src/galileo/resources/models/feedback_rating_info_feedback_type.py +++ b/src/galileo/resources/models/feedback_type.py @@ -1,7 +1,7 @@ from enum import Enum -class FeedbackRatingInfoFeedbackType(str, Enum): +class FeedbackType(str, Enum): LIKE_DISLIKE = "like_dislike" SCORE = "score" STAR = "star" diff --git a/src/galileo/resources/models/fine_tuned_scorer.py b/src/galileo/resources/models/fine_tuned_scorer.py index 95b03d3a3..b113a95ec 100644 --- a/src/galileo/resources/models/fine_tuned_scorer.py +++ b/src/galileo/resources/models/fine_tuned_scorer.py @@ -49,16 +49,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -95,16 +89,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/fine_tuned_scorer_response.py b/src/galileo/resources/models/fine_tuned_scorer_response.py index b52c1f37b..23727e3e4 100644 --- a/src/galileo/resources/models/fine_tuned_scorer_response.py +++ b/src/galileo/resources/models/fine_tuned_scorer_response.py @@ -103,9 +103,8 @@ def _parse_executor(data: object) -> Union[CoreScorerName, None, Unset]: try: if not isinstance(data, str): raise TypeError() - executor_type_0 = CoreScorerName(data) + return CoreScorerName(data) - return executor_type_0 except: # noqa: E722 pass return cast(Union[CoreScorerName, None, Unset], data) diff --git a/src/galileo/resources/models/generated_scorer_configuration.py b/src/galileo/resources/models/generated_scorer_configuration.py index 94af14efa..5aa87081c 100644 --- a/src/galileo/resources/models/generated_scorer_configuration.py +++ b/src/galileo/resources/models/generated_scorer_configuration.py @@ -70,10 +70,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _output_type = d.pop("output_type", UNSET) output_type: Union[Unset, OutputTypeEnum] - if isinstance(_output_type, Unset): - output_type = UNSET - else: - output_type = OutputTypeEnum(_output_type) + output_type = UNSET if isinstance(_output_type, Unset) else OutputTypeEnum(_output_type) scoreable_node_types = cast(list[str], d.pop("scoreable_node_types", UNSET)) diff --git a/src/galileo/resources/models/generated_scorer_response.py b/src/galileo/resources/models/generated_scorer_response.py index 3f72fc96c..fedf37f86 100644 --- a/src/galileo/resources/models/generated_scorer_response.py +++ b/src/galileo/resources/models/generated_scorer_response.py @@ -72,16 +72,10 @@ def to_dict(self) -> dict[str, Any]: updated_at = self.updated_at.isoformat() instructions: Union[None, Unset, str] - if isinstance(self.instructions, Unset): - instructions = UNSET - else: - instructions = self.instructions + instructions = UNSET if isinstance(self.instructions, Unset) else self.instructions user_prompt: Union[None, Unset, str] - if isinstance(self.user_prompt, Unset): - user_prompt = UNSET - else: - user_prompt = self.user_prompt + user_prompt = UNSET if isinstance(self.user_prompt, Unset) else self.user_prompt field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/get_projects_paginated_response.py b/src/galileo/resources/models/get_projects_paginated_response.py index d6bff9687..d02165f90 100644 --- a/src/galileo/resources/models/get_projects_paginated_response.py +++ b/src/galileo/resources/models/get_projects_paginated_response.py @@ -40,10 +40,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/get_projects_paginated_response_v2.py b/src/galileo/resources/models/get_projects_paginated_response_v2.py index 1687f7d99..70d748738 100644 --- a/src/galileo/resources/models/get_projects_paginated_response_v2.py +++ b/src/galileo/resources/models/get_projects_paginated_response_v2.py @@ -45,10 +45,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/ground_truth_adherence_scorer.py b/src/galileo/resources/models/ground_truth_adherence_scorer.py index bc43baa6f..df97ee336 100644 --- a/src/galileo/resources/models/ground_truth_adherence_scorer.py +++ b/src/galileo/resources/models/ground_truth_adherence_scorer.py @@ -54,18 +54,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_ = self.type_ @@ -108,16 +102,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/group_collaborator_create.py b/src/galileo/resources/models/group_collaborator_create.py index a1edcb9ca..e7e722d6d 100644 --- a/src/galileo/resources/models/group_collaborator_create.py +++ b/src/galileo/resources/models/group_collaborator_create.py @@ -44,10 +44,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _role = d.pop("role", UNSET) role: Union[Unset, CollaboratorRole] - if isinstance(_role, Unset): - role = UNSET - else: - role = CollaboratorRole(_role) + role = UNSET if isinstance(_role, Unset) else CollaboratorRole(_role) group_collaborator_create = cls(group_id=group_id, role=role) diff --git a/src/galileo/resources/models/input_pii_scorer.py b/src/galileo/resources/models/input_pii_scorer.py index 8204fbe02..7d52279ea 100644 --- a/src/galileo/resources/models/input_pii_scorer.py +++ b/src/galileo/resources/models/input_pii_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/input_sexist_scorer.py b/src/galileo/resources/models/input_sexist_scorer.py index b53a1018b..863bffe7e 100644 --- a/src/galileo/resources/models/input_sexist_scorer.py +++ b/src/galileo/resources/models/input_sexist_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, InputSexistScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = InputSexistScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else InputSexistScorerType(_type_) input_sexist_scorer = cls(filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_) diff --git a/src/galileo/resources/models/input_tone_scorer.py b/src/galileo/resources/models/input_tone_scorer.py index ae46deb98..8ca5f6e7b 100644 --- a/src/galileo/resources/models/input_tone_scorer.py +++ b/src/galileo/resources/models/input_tone_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/input_toxicity_scorer.py b/src/galileo/resources/models/input_toxicity_scorer.py index a1f585876..f5953bee8 100644 --- a/src/galileo/resources/models/input_toxicity_scorer.py +++ b/src/galileo/resources/models/input_toxicity_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, InputToxicityScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = InputToxicityScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else InputToxicityScorerType(_type_) input_toxicity_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/instruction_adherence_scorer.py b/src/galileo/resources/models/instruction_adherence_scorer.py index 6c73b7c9e..1b72b5902 100644 --- a/src/galileo/resources/models/instruction_adherence_scorer.py +++ b/src/galileo/resources/models/instruction_adherence_scorer.py @@ -54,18 +54,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_ = self.type_ @@ -108,16 +102,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/invoke_response.py b/src/galileo/resources/models/invoke_response.py index 909ac015e..17a4c8bea 100644 --- a/src/galileo/resources/models/invoke_response.py +++ b/src/galileo/resources/models/invoke_response.py @@ -149,9 +149,8 @@ def _parse_headers(data: object) -> Union["InvokeResponseHeadersType0", None, Un try: if not isinstance(data, dict): raise TypeError() - headers_type_0 = InvokeResponseHeadersType0.from_dict(data) + return InvokeResponseHeadersType0.from_dict(data) - return headers_type_0 except: # noqa: E722 pass return cast(Union["InvokeResponseHeadersType0", None, Unset], data) @@ -166,9 +165,8 @@ def _parse_metadata(data: object) -> Union["InvokeResponseMetadataType0", None, try: if not isinstance(data, dict): raise TypeError() - metadata_type_0 = InvokeResponseMetadataType0.from_dict(data) + return InvokeResponseMetadataType0.from_dict(data) - return metadata_type_0 except: # noqa: E722 pass return cast(Union["InvokeResponseMetadataType0", None, Unset], data) @@ -191,10 +189,7 @@ def _parse_metadata(data: object) -> Union["InvokeResponseMetadataType0", None, _status = d.pop("status", UNSET) status: Union[Unset, ExecutionStatus] - if isinstance(_status, Unset): - status = UNSET - else: - status = ExecutionStatus(_status) + status = UNSET if isinstance(_status, Unset) else ExecutionStatus(_status) invoke_response = cls( action_result=action_result, diff --git a/src/galileo/resources/models/job_db.py b/src/galileo/resources/models/job_db.py index edd88a949..0a3a9c38b 100644 --- a/src/galileo/resources/models/job_db.py +++ b/src/galileo/resources/models/job_db.py @@ -89,10 +89,7 @@ def to_dict(self) -> dict[str, Any]: completed_at = self.completed_at error_message: Union[None, Unset, str] - if isinstance(self.error_message, Unset): - error_message = UNSET - else: - error_message = self.error_message + error_message = UNSET if isinstance(self.error_message, Unset) else self.error_message failed_at: Union[None, Unset, str] if isinstance(self.failed_at, Unset): @@ -103,16 +100,10 @@ def to_dict(self) -> dict[str, Any]: failed_at = self.failed_at migration_name: Union[None, Unset, str] - if isinstance(self.migration_name, Unset): - migration_name = UNSET - else: - migration_name = self.migration_name + migration_name = UNSET if isinstance(self.migration_name, Unset) else self.migration_name monitor_batch_id: Union[None, Unset, str] - if isinstance(self.monitor_batch_id, Unset): - monitor_batch_id = UNSET - else: - monitor_batch_id = self.monitor_batch_id + monitor_batch_id = UNSET if isinstance(self.monitor_batch_id, Unset) else self.monitor_batch_id processing_started: Union[None, Unset, str] if isinstance(self.processing_started, Unset): @@ -123,10 +114,7 @@ def to_dict(self) -> dict[str, Any]: processing_started = self.processing_started progress_message: Union[None, Unset, str] - if isinstance(self.progress_message, Unset): - progress_message = UNSET - else: - progress_message = self.progress_message + progress_message = UNSET if isinstance(self.progress_message, Unset) else self.progress_message progress_percent = self.progress_percent @@ -203,9 +191,8 @@ def _parse_completed_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - completed_at_type_0 = isoparse(data) + return isoparse(data) - return completed_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) @@ -229,9 +216,8 @@ def _parse_failed_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - failed_at_type_0 = isoparse(data) + return isoparse(data) - return failed_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) @@ -264,9 +250,8 @@ def _parse_processing_started(data: object) -> Union[None, Unset, datetime.datet try: if not isinstance(data, str): raise TypeError() - processing_started_type_0 = isoparse(data) + return isoparse(data) - return processing_started_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) diff --git a/src/galileo/resources/models/job_progress.py b/src/galileo/resources/models/job_progress.py index 09d8ea9c1..30fdf2b89 100644 --- a/src/galileo/resources/models/job_progress.py +++ b/src/galileo/resources/models/job_progress.py @@ -25,22 +25,13 @@ class JobProgress: def to_dict(self) -> dict[str, Any]: progress_message: Union[None, Unset, str] - if isinstance(self.progress_message, Unset): - progress_message = UNSET - else: - progress_message = self.progress_message + progress_message = UNSET if isinstance(self.progress_message, Unset) else self.progress_message steps_completed: Union[None, Unset, int] - if isinstance(self.steps_completed, Unset): - steps_completed = UNSET - else: - steps_completed = self.steps_completed + steps_completed = UNSET if isinstance(self.steps_completed, Unset) else self.steps_completed steps_total: Union[None, Unset, int] - if isinstance(self.steps_total, Unset): - steps_total = UNSET - else: - steps_total = self.steps_total + steps_total = UNSET if isinstance(self.steps_total, Unset) else self.steps_total field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/list_dataset_params.py b/src/galileo/resources/models/list_dataset_params.py index 14967b20a..393bda4d8 100644 --- a/src/galileo/resources/models/list_dataset_params.py +++ b/src/galileo/resources/models/list_dataset_params.py @@ -67,11 +67,7 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, DatasetNameFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, DatasetDraftFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, DatasetUsedInProjectFilter): + if isinstance(filters_item_data, (DatasetNameFilter, DatasetDraftFilter, DatasetUsedInProjectFilter)): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -81,19 +77,18 @@ def to_dict(self) -> dict[str, Any]: sort: Union[None, Unset, dict[str, Any]] if isinstance(self.sort, Unset): sort = UNSET - elif isinstance(self.sort, DatasetNameSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, DatasetCreatedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, DatasetUpdatedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, DatasetProjectLastUsedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, DatasetProjectsSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, DatasetRowsSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, DatasetLastEditedByUserAtSort): + elif isinstance( + self.sort, + ( + DatasetNameSort, + DatasetCreatedAtSort, + DatasetUpdatedAtSort, + DatasetProjectLastUsedAtSort, + DatasetProjectsSort, + DatasetRowsSort, + DatasetLastEditedByUserAtSort, + ), + ): sort = self.sort.to_dict() else: sort = self.sort @@ -133,32 +128,27 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = DatasetNameFilter.from_dict(data) + return DatasetNameFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = DatasetDraftFilter.from_dict(data) + return DatasetDraftFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = DatasetUsedInProjectFilter.from_dict(data) + return DatasetUsedInProjectFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = DatasetIDFilter.from_dict(data) - - return filters_item_type_3 + return DatasetIDFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -184,57 +174,50 @@ def _parse_sort( try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_0 = DatasetNameSort.from_dict(data) + return DatasetNameSort.from_dict(data) - return sort_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_1 = DatasetCreatedAtSort.from_dict(data) + return DatasetCreatedAtSort.from_dict(data) - return sort_type_0_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_2 = DatasetUpdatedAtSort.from_dict(data) + return DatasetUpdatedAtSort.from_dict(data) - return sort_type_0_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_3 = DatasetProjectLastUsedAtSort.from_dict(data) + return DatasetProjectLastUsedAtSort.from_dict(data) - return sort_type_0_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_4 = DatasetProjectsSort.from_dict(data) + return DatasetProjectsSort.from_dict(data) - return sort_type_0_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_5 = DatasetRowsSort.from_dict(data) + return DatasetRowsSort.from_dict(data) - return sort_type_0_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_6 = DatasetLastEditedByUserAtSort.from_dict(data) + return DatasetLastEditedByUserAtSort.from_dict(data) - return sort_type_0_type_6 except: # noqa: E722 pass return cast( diff --git a/src/galileo/resources/models/list_dataset_projects_response.py b/src/galileo/resources/models/list_dataset_projects_response.py index 4f3b35e9b..66914d0ee 100644 --- a/src/galileo/resources/models/list_dataset_projects_response.py +++ b/src/galileo/resources/models/list_dataset_projects_response.py @@ -35,10 +35,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_dataset_response.py b/src/galileo/resources/models/list_dataset_response.py index 1f6af4093..12bf77e65 100644 --- a/src/galileo/resources/models/list_dataset_response.py +++ b/src/galileo/resources/models/list_dataset_response.py @@ -42,10 +42,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_dataset_version_params.py b/src/galileo/resources/models/list_dataset_version_params.py index a45c9a1a6..018f6489c 100644 --- a/src/galileo/resources/models/list_dataset_version_params.py +++ b/src/galileo/resources/models/list_dataset_version_params.py @@ -56,9 +56,8 @@ def _parse_sort(data: object) -> Union["DatasetVersionIndexSort", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - sort_type_0 = DatasetVersionIndexSort.from_dict(data) + return DatasetVersionIndexSort.from_dict(data) - return sort_type_0 except: # noqa: E722 pass return cast(Union["DatasetVersionIndexSort", None, Unset], data) diff --git a/src/galileo/resources/models/list_dataset_version_response.py b/src/galileo/resources/models/list_dataset_version_response.py index 6e3ab6281..8d038d69e 100644 --- a/src/galileo/resources/models/list_dataset_version_response.py +++ b/src/galileo/resources/models/list_dataset_version_response.py @@ -40,10 +40,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_group_collaborators_response.py b/src/galileo/resources/models/list_group_collaborators_response.py index 274496992..f407e132a 100644 --- a/src/galileo/resources/models/list_group_collaborators_response.py +++ b/src/galileo/resources/models/list_group_collaborators_response.py @@ -40,10 +40,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_prompt_dataset_response.py b/src/galileo/resources/models/list_prompt_dataset_response.py index 4c280e65d..773f72aed 100644 --- a/src/galileo/resources/models/list_prompt_dataset_response.py +++ b/src/galileo/resources/models/list_prompt_dataset_response.py @@ -42,10 +42,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_prompt_template_params.py b/src/galileo/resources/models/list_prompt_template_params.py index 5f7a9976a..55a0617b8 100644 --- a/src/galileo/resources/models/list_prompt_template_params.py +++ b/src/galileo/resources/models/list_prompt_template_params.py @@ -49,9 +49,7 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, PromptTemplateNameFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, PromptTemplateCreatedByFilter): + if isinstance(filters_item_data, (PromptTemplateNameFilter, PromptTemplateCreatedByFilter)): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -61,11 +59,7 @@ def to_dict(self) -> dict[str, Any]: sort: Union[None, Unset, dict[str, Any]] if isinstance(self.sort, Unset): sort = UNSET - elif isinstance(self.sort, PromptTemplateNameSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, PromptTemplateCreatedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, PromptTemplateUpdatedAtSort): + elif isinstance(self.sort, (PromptTemplateNameSort, PromptTemplateCreatedAtSort, PromptTemplateUpdatedAtSort)): sort = self.sort.to_dict() else: sort = self.sort @@ -102,24 +96,20 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = PromptTemplateNameFilter.from_dict(data) + return PromptTemplateNameFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = PromptTemplateCreatedByFilter.from_dict(data) + return PromptTemplateCreatedByFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = PromptTemplateUsedInProjectFilter.from_dict(data) - - return filters_item_type_2 + return PromptTemplateUsedInProjectFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -135,25 +125,22 @@ def _parse_sort( try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_0 = PromptTemplateNameSort.from_dict(data) + return PromptTemplateNameSort.from_dict(data) - return sort_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_1 = PromptTemplateCreatedAtSort.from_dict(data) + return PromptTemplateCreatedAtSort.from_dict(data) - return sort_type_0_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_2 = PromptTemplateUpdatedAtSort.from_dict(data) + return PromptTemplateUpdatedAtSort.from_dict(data) - return sort_type_0_type_2 except: # noqa: E722 pass return cast( diff --git a/src/galileo/resources/models/list_prompt_template_response.py b/src/galileo/resources/models/list_prompt_template_response.py index 143bee513..b2c575f79 100644 --- a/src/galileo/resources/models/list_prompt_template_response.py +++ b/src/galileo/resources/models/list_prompt_template_response.py @@ -35,10 +35,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_prompt_template_version_params.py b/src/galileo/resources/models/list_prompt_template_version_params.py index f24944279..00474ae10 100644 --- a/src/galileo/resources/models/list_prompt_template_version_params.py +++ b/src/galileo/resources/models/list_prompt_template_version_params.py @@ -40,11 +40,10 @@ def to_dict(self) -> dict[str, Any]: sort: Union[None, Unset, dict[str, Any]] if isinstance(self.sort, Unset): sort = UNSET - elif isinstance(self.sort, PromptTemplateVersionNumberSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, PromptTemplateVersionCreatedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, PromptTemplateVersionUpdatedAtSort): + elif isinstance( + self.sort, + (PromptTemplateVersionNumberSort, PromptTemplateVersionCreatedAtSort, PromptTemplateVersionUpdatedAtSort), + ): sort = self.sort.to_dict() else: sort = self.sort @@ -81,25 +80,22 @@ def _parse_sort( try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_0 = PromptTemplateVersionNumberSort.from_dict(data) + return PromptTemplateVersionNumberSort.from_dict(data) - return sort_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_1 = PromptTemplateVersionCreatedAtSort.from_dict(data) + return PromptTemplateVersionCreatedAtSort.from_dict(data) - return sort_type_0_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_2 = PromptTemplateVersionUpdatedAtSort.from_dict(data) + return PromptTemplateVersionUpdatedAtSort.from_dict(data) - return sort_type_0_type_2 except: # noqa: E722 pass return cast( diff --git a/src/galileo/resources/models/list_prompt_template_version_response.py b/src/galileo/resources/models/list_prompt_template_version_response.py index 585110dd7..5956ade2e 100644 --- a/src/galileo/resources/models/list_prompt_template_version_response.py +++ b/src/galileo/resources/models/list_prompt_template_version_response.py @@ -35,10 +35,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_scorer_versions_response.py b/src/galileo/resources/models/list_scorer_versions_response.py index e534db245..954c880d0 100644 --- a/src/galileo/resources/models/list_scorer_versions_response.py +++ b/src/galileo/resources/models/list_scorer_versions_response.py @@ -35,10 +35,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_scorers_request.py b/src/galileo/resources/models/list_scorers_request.py index 3ef865351..b13128f09 100644 --- a/src/galileo/resources/models/list_scorers_request.py +++ b/src/galileo/resources/models/list_scorers_request.py @@ -62,17 +62,17 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, ScorerNameFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ScorerTypeFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ScorerModelTypeFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ScorerTagsFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ScorerCreatorFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ScorerCreatedAtFilter): + if isinstance( + filters_item_data, + ( + ScorerNameFilter, + ScorerTypeFilter, + ScorerModelTypeFilter, + ScorerTagsFilter, + ScorerCreatorFilter, + ScorerCreatedAtFilter, + ), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -82,9 +82,7 @@ def to_dict(self) -> dict[str, Any]: sort: Union[None, Unset, dict[str, Any]] if isinstance(self.sort, Unset): sort = UNSET - elif isinstance(self.sort, ScorerNameSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, ScorerEnabledInRunSort): + elif isinstance(self.sort, (ScorerNameSort, ScorerEnabledInRunSort)): sort = self.sort.to_dict() else: sort = self.sort @@ -130,56 +128,48 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = ScorerNameFilter.from_dict(data) + return ScorerNameFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = ScorerTypeFilter.from_dict(data) + return ScorerTypeFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = ScorerModelTypeFilter.from_dict(data) + return ScorerModelTypeFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = ScorerTagsFilter.from_dict(data) + return ScorerTagsFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = ScorerCreatorFilter.from_dict(data) + return ScorerCreatorFilter.from_dict(data) - return filters_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_5 = ScorerCreatedAtFilter.from_dict(data) + return ScorerCreatedAtFilter.from_dict(data) - return filters_item_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_6 = ScorerUpdatedAtFilter.from_dict(data) - - return filters_item_type_6 + return ScorerUpdatedAtFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -193,17 +183,15 @@ def _parse_sort(data: object) -> Union["ScorerEnabledInRunSort", "ScorerNameSort try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_0 = ScorerNameSort.from_dict(data) + return ScorerNameSort.from_dict(data) - return sort_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_1 = ScorerEnabledInRunSort.from_dict(data) + return ScorerEnabledInRunSort.from_dict(data) - return sort_type_0_type_1 except: # noqa: E722 pass return cast(Union["ScorerEnabledInRunSort", "ScorerNameSort", None, Unset], data) diff --git a/src/galileo/resources/models/list_scorers_response.py b/src/galileo/resources/models/list_scorers_response.py index 95d4c3b4b..7c7b980b8 100644 --- a/src/galileo/resources/models/list_scorers_response.py +++ b/src/galileo/resources/models/list_scorers_response.py @@ -35,10 +35,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/list_user_collaborators_response.py b/src/galileo/resources/models/list_user_collaborators_response.py index e0a03ced1..7d0962180 100644 --- a/src/galileo/resources/models/list_user_collaborators_response.py +++ b/src/galileo/resources/models/list_user_collaborators_response.py @@ -40,10 +40,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/llm_metrics.py b/src/galileo/resources/models/llm_metrics.py index 3bde568b4..bc5c1cb5b 100644 --- a/src/galileo/resources/models/llm_metrics.py +++ b/src/galileo/resources/models/llm_metrics.py @@ -30,28 +30,16 @@ class LlmMetrics: def to_dict(self) -> dict[str, Any]: duration_ns: Union[None, Unset, int] - if isinstance(self.duration_ns, Unset): - duration_ns = UNSET - else: - duration_ns = self.duration_ns + duration_ns = UNSET if isinstance(self.duration_ns, Unset) else self.duration_ns num_input_tokens: Union[None, Unset, int] - if isinstance(self.num_input_tokens, Unset): - num_input_tokens = UNSET - else: - num_input_tokens = self.num_input_tokens + num_input_tokens = UNSET if isinstance(self.num_input_tokens, Unset) else self.num_input_tokens num_output_tokens: Union[None, Unset, int] - if isinstance(self.num_output_tokens, Unset): - num_output_tokens = UNSET - else: - num_output_tokens = self.num_output_tokens + num_output_tokens = UNSET if isinstance(self.num_output_tokens, Unset) else self.num_output_tokens num_total_tokens: Union[None, Unset, int] - if isinstance(self.num_total_tokens, Unset): - num_total_tokens = UNSET - else: - num_total_tokens = self.num_total_tokens + num_total_tokens = UNSET if isinstance(self.num_total_tokens, Unset) else self.num_total_tokens time_to_first_token_ns: Union[None, Unset, int] if isinstance(self.time_to_first_token_ns, Unset): diff --git a/src/galileo/resources/models/llm_span.py b/src/galileo/resources/models/llm_span.py index fa24e4c32..856153e5d 100644 --- a/src/galileo/resources/models/llm_span.py +++ b/src/galileo/resources/models/llm_span.py @@ -86,38 +86,23 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id finish_reason: Union[None, Unset, str] - if isinstance(self.finish_reason, Unset): - finish_reason = UNSET - else: - finish_reason = self.finish_reason + finish_reason = UNSET if isinstance(self.finish_reason, Unset) else self.finish_reason id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id input_: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.input_, Unset): @@ -131,10 +116,7 @@ def to_dict(self) -> dict[str, Any]: metrics = self.metrics.to_dict() model: Union[None, Unset, str] - if isinstance(self.model, Unset): - model = UNSET - else: - model = self.model + model = UNSET if isinstance(self.model, Unset) else self.model name = self.name @@ -143,10 +125,7 @@ def to_dict(self) -> dict[str, Any]: output = self.output.to_dict() parent_id: Union[None, Unset, str] - if isinstance(self.parent_id, Unset): - parent_id = UNSET - else: - parent_id = self.parent_id + parent_id = UNSET if isinstance(self.parent_id, Unset) else self.parent_id redacted_input: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.redacted_input, Unset): @@ -169,32 +148,20 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags temperature: Union[None, Unset, float] - if isinstance(self.temperature, Unset): - temperature = UNSET - else: - temperature = self.temperature + temperature = UNSET if isinstance(self.temperature, Unset) else self.temperature tools: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.tools, Unset): @@ -209,10 +176,7 @@ def to_dict(self) -> dict[str, Any]: tools = self.tools trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -285,10 +249,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -351,10 +312,7 @@ def _parse_id(data: object) -> Union[None, Unset, str]: _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, LlmMetrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = LlmMetrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else LlmMetrics.from_dict(_metrics) def _parse_model(data: object) -> Union[None, Unset, str]: if data is None: @@ -369,10 +327,7 @@ def _parse_model(data: object) -> Union[None, Unset, str]: _output = d.pop("output", UNSET) output: Union[Unset, Message] - if isinstance(_output, Unset): - output = UNSET - else: - output = Message.from_dict(_output) + output = UNSET if isinstance(_output, Unset) else Message.from_dict(_output) def _parse_parent_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -413,9 +368,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_0 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_0 except: # noqa: E722 pass return cast(Union["Message", None, Unset], data) @@ -497,10 +451,7 @@ def _parse_trace_id(data: object) -> Union[None, Unset, str]: _user_metadata = d.pop("user_metadata", UNSET) user_metadata: Union[Unset, LlmSpanUserMetadata] - if isinstance(_user_metadata, Unset): - user_metadata = UNSET - else: - user_metadata = LlmSpanUserMetadata.from_dict(_user_metadata) + user_metadata = UNSET if isinstance(_user_metadata, Unset) else LlmSpanUserMetadata.from_dict(_user_metadata) llm_span = cls( created_at=created_at, diff --git a/src/galileo/resources/models/log_records_available_columns_request.py b/src/galileo/resources/models/log_records_available_columns_request.py index 4a1f922a5..088907669 100644 --- a/src/galileo/resources/models/log_records_available_columns_request.py +++ b/src/galileo/resources/models/log_records_available_columns_request.py @@ -25,22 +25,13 @@ class LogRecordsAvailableColumnsRequest: def to_dict(self) -> dict[str, Any]: experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/log_records_delete_request.py b/src/galileo/resources/models/log_records_delete_request.py index 66e5489f8..3c3bb08e5 100644 --- a/src/galileo/resources/models/log_records_delete_request.py +++ b/src/galileo/resources/models/log_records_delete_request.py @@ -56,23 +56,17 @@ def to_dict(self) -> dict[str, Any]: from ..models.log_records_number_filter import LogRecordsNumberFilter experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id filters: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.filters, Unset): filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -80,16 +74,10 @@ def to_dict(self) -> dict[str, Any]: filters.append(filters_item) log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -140,40 +128,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) diff --git a/src/galileo/resources/models/log_records_export_request.py b/src/galileo/resources/models/log_records_export_request.py index 04d394f14..cd0c01a55 100644 --- a/src/galileo/resources/models/log_records_export_request.py +++ b/src/galileo/resources/models/log_records_export_request.py @@ -75,10 +75,7 @@ def to_dict(self) -> dict[str, Any]: column_ids = self.column_ids experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id export_format: Union[Unset, str] = UNSET if not isinstance(self.export_format, Unset): @@ -89,13 +86,10 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -103,16 +97,10 @@ def to_dict(self) -> dict[str, Any]: filters.append(filters_item) log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id sort: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.sort, Unset): @@ -158,9 +146,8 @@ def _parse_column_ids(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - column_ids_type_0 = cast(list[str], data) + return cast(list[str], data) - return column_ids_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -178,10 +165,7 @@ def _parse_experiment_id(data: object) -> Union[None, Unset, str]: _export_format = d.pop("export_format", UNSET) export_format: Union[Unset, LLMExportFormat] - if isinstance(_export_format, Unset): - export_format = UNSET - else: - export_format = LLMExportFormat(_export_format) + export_format = UNSET if isinstance(_export_format, Unset) else LLMExportFormat(_export_format) filters = [] _filters = d.pop("filters", UNSET) @@ -199,40 +183,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -258,10 +236,7 @@ def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: _sort = d.pop("sort", UNSET) sort: Union[Unset, LogRecordsSortClause] - if isinstance(_sort, Unset): - sort = UNSET - else: - sort = LogRecordsSortClause.from_dict(_sort) + sort = UNSET if isinstance(_sort, Unset) else LogRecordsSortClause.from_dict(_sort) log_records_export_request = cls( root_type=root_type, diff --git a/src/galileo/resources/models/log_records_id_filter.py b/src/galileo/resources/models/log_records_id_filter.py index 327e5db1e..7f1155da7 100644 --- a/src/galileo/resources/models/log_records_id_filter.py +++ b/src/galileo/resources/models/log_records_id_filter.py @@ -30,11 +30,7 @@ def to_dict(self) -> dict[str, Any]: column_id = self.column_id value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value operator: Union[Unset, str] = UNSET if not isinstance(self.operator, Unset): @@ -61,9 +57,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) @@ -72,10 +67,7 @@ def _parse_value(data: object) -> Union[list[str], str]: _operator = d.pop("operator", UNSET) operator: Union[Unset, LogRecordsIDFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = LogRecordsIDFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else LogRecordsIDFilterOperator(_operator) type_ = cast(Union[Literal["id"], Unset], d.pop("type", UNSET)) if type_ != "id" and not isinstance(type_, Unset): diff --git a/src/galileo/resources/models/log_records_metrics_query_request.py b/src/galileo/resources/models/log_records_metrics_query_request.py index 22d5a9a0e..b3fb4dc34 100644 --- a/src/galileo/resources/models/log_records_metrics_query_request.py +++ b/src/galileo/resources/models/log_records_metrics_query_request.py @@ -66,23 +66,17 @@ def to_dict(self) -> dict[str, Any]: start_time = self.start_time.isoformat() experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id filters: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.filters, Unset): filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -90,24 +84,15 @@ def to_dict(self) -> dict[str, Any]: filters.append(filters_item) group_by: Union[None, Unset, str] - if isinstance(self.group_by, Unset): - group_by = UNSET - else: - group_by = self.group_by + group_by = UNSET if isinstance(self.group_by, Unset) else self.group_by interval = self.interval log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -165,40 +150,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) diff --git a/src/galileo/resources/models/log_records_number_filter.py b/src/galileo/resources/models/log_records_number_filter.py index 7e9c0d3d1..54e5b0f5e 100644 --- a/src/galileo/resources/models/log_records_number_filter.py +++ b/src/galileo/resources/models/log_records_number_filter.py @@ -32,14 +32,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[float, int, list[float], list[int]] - if isinstance(self.value, list): - value = self.value - - elif isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, (list, list)) else self.value type_ = self.type_ @@ -62,17 +55,15 @@ def _parse_value(data: object) -> Union[float, int, list[float], list[int]]: try: if not isinstance(data, list): raise TypeError() - value_type_2 = cast(list[int], data) + return cast(list[int], data) - return value_type_2 except: # noqa: E722 pass try: if not isinstance(data, list): raise TypeError() - value_type_3 = cast(list[float], data) + return cast(list[float], data) - return value_type_3 except: # noqa: E722 pass return cast(Union[float, int, list[float], list[int]], data) diff --git a/src/galileo/resources/models/log_records_query_request.py b/src/galileo/resources/models/log_records_query_request.py index eab8b90d7..0cb4c7aaf 100644 --- a/src/galileo/resources/models/log_records_query_request.py +++ b/src/galileo/resources/models/log_records_query_request.py @@ -61,23 +61,17 @@ def to_dict(self) -> dict[str, Any]: from ..models.log_records_number_filter import LogRecordsNumberFilter experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id filters: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.filters, Unset): filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -87,16 +81,10 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id sort: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.sort, Unset): @@ -164,40 +152,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -225,10 +207,7 @@ def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: _sort = d.pop("sort", UNSET) sort: Union[Unset, LogRecordsSortClause] - if isinstance(_sort, Unset): - sort = UNSET - else: - sort = LogRecordsSortClause.from_dict(_sort) + sort = UNSET if isinstance(_sort, Unset) else LogRecordsSortClause.from_dict(_sort) starting_token = d.pop("starting_token", UNSET) diff --git a/src/galileo/resources/models/log_records_query_response.py b/src/galileo/resources/models/log_records_query_response.py index 43fa06802..c56fe9b0d 100644 --- a/src/galileo/resources/models/log_records_query_response.py +++ b/src/galileo/resources/models/log_records_query_response.py @@ -63,10 +63,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated @@ -75,17 +72,17 @@ def to_dict(self) -> dict[str, Any]: records = [] for records_item_data in self.records: records_item: dict[str, Any] - if isinstance(records_item_data, ExtendedTraceRecord): - records_item = records_item_data.to_dict() - elif isinstance(records_item_data, ExtendedAgentSpanRecord): - records_item = records_item_data.to_dict() - elif isinstance(records_item_data, ExtendedWorkflowSpanRecord): - records_item = records_item_data.to_dict() - elif isinstance(records_item_data, ExtendedLlmSpanRecord): - records_item = records_item_data.to_dict() - elif isinstance(records_item_data, ExtendedToolSpanRecord): - records_item = records_item_data.to_dict() - elif isinstance(records_item_data, ExtendedRetrieverSpanRecord): + if isinstance( + records_item_data, + ( + ExtendedTraceRecord, + ExtendedAgentSpanRecord, + ExtendedWorkflowSpanRecord, + ExtendedLlmSpanRecord, + ExtendedToolSpanRecord, + ExtendedRetrieverSpanRecord, + ), + ): records_item = records_item_data.to_dict() else: records_item = records_item_data.to_dict() @@ -152,56 +149,48 @@ def _parse_records_item( try: if not isinstance(data, dict): raise TypeError() - records_item_type_0 = ExtendedTraceRecord.from_dict(data) + return ExtendedTraceRecord.from_dict(data) - return records_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - records_item_type_1 = ExtendedAgentSpanRecord.from_dict(data) + return ExtendedAgentSpanRecord.from_dict(data) - return records_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - records_item_type_2 = ExtendedWorkflowSpanRecord.from_dict(data) + return ExtendedWorkflowSpanRecord.from_dict(data) - return records_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - records_item_type_3 = ExtendedLlmSpanRecord.from_dict(data) + return ExtendedLlmSpanRecord.from_dict(data) - return records_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - records_item_type_4 = ExtendedToolSpanRecord.from_dict(data) + return ExtendedToolSpanRecord.from_dict(data) - return records_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - records_item_type_5 = ExtendedRetrieverSpanRecord.from_dict(data) + return ExtendedRetrieverSpanRecord.from_dict(data) - return records_item_type_5 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - records_item_type_6 = ExtendedSessionRecord.from_dict(data) - - return records_item_type_6 + return ExtendedSessionRecord.from_dict(data) records_item = _parse_records_item(records_item_data) diff --git a/src/galileo/resources/models/log_records_text_filter.py b/src/galileo/resources/models/log_records_text_filter.py index 7922e3cfb..fcfed5f16 100644 --- a/src/galileo/resources/models/log_records_text_filter.py +++ b/src/galileo/resources/models/log_records_text_filter.py @@ -34,11 +34,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value case_sensitive = self.case_sensitive @@ -65,9 +61,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) diff --git a/src/galileo/resources/models/log_span_update_request.py b/src/galileo/resources/models/log_span_update_request.py index 50d7a1588..ca7b1e55c 100644 --- a/src/galileo/resources/models/log_span_update_request.py +++ b/src/galileo/resources/models/log_span_update_request.py @@ -56,16 +56,10 @@ def to_dict(self) -> dict[str, Any]: span_id = self.span_id client_version: Union[None, Unset, str] - if isinstance(self.client_version, Unset): - client_version = UNSET - else: - client_version = self.client_version + client_version = UNSET if isinstance(self.client_version, Unset) else self.client_version experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id input_: Union[None, Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -80,20 +74,14 @@ def to_dict(self) -> dict[str, Any]: input_ = self.input_ log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id logging_method: Union[Unset, str] = UNSET if not isinstance(self.logging_method, Unset): logging_method = self.logging_method.value metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id output: Union[None, Unset, dict[str, Any], list[dict[str, Any]], str] if isinstance(self.output, Unset): @@ -112,10 +100,7 @@ def to_dict(self) -> dict[str, Any]: reliable = self.reliable status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code tags: Union[None, Unset, list[str]] if isinstance(self.tags, Unset): @@ -211,10 +196,7 @@ def _parse_log_stream_id(data: object) -> Union[None, Unset, str]: _logging_method = d.pop("logging_method", UNSET) logging_method: Union[Unset, LoggingMethod] - if isinstance(_logging_method, Unset): - logging_method = UNSET - else: - logging_method = LoggingMethod(_logging_method) + logging_method = UNSET if isinstance(_logging_method, Unset) else LoggingMethod(_logging_method) def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -233,9 +215,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -274,9 +255,8 @@ def _parse_tags(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - tags_type_0 = cast(list[str], data) + return cast(list[str], data) - return tags_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/log_span_update_response.py b/src/galileo/resources/models/log_span_update_response.py index f2807db24..6dec85194 100644 --- a/src/galileo/resources/models/log_span_update_response.py +++ b/src/galileo/resources/models/log_span_update_response.py @@ -45,22 +45,13 @@ def to_dict(self) -> dict[str, Any]: span_id = self.span_id experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/log_spans_ingest_request.py b/src/galileo/resources/models/log_spans_ingest_request.py index 9e51e9758..b90597f71 100644 --- a/src/galileo/resources/models/log_spans_ingest_request.py +++ b/src/galileo/resources/models/log_spans_ingest_request.py @@ -59,13 +59,7 @@ def to_dict(self) -> dict[str, Any]: spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, AgentSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, WorkflowSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, LlmSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, RetrieverSpan): + if isinstance(spans_item_data, (AgentSpan, WorkflowSpan, LlmSpan, RetrieverSpan)): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -75,32 +69,20 @@ def to_dict(self) -> dict[str, Any]: trace_id = self.trace_id client_version: Union[None, Unset, str] - if isinstance(self.client_version, Unset): - client_version = UNSET - else: - client_version = self.client_version + client_version = UNSET if isinstance(self.client_version, Unset) else self.client_version experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id logging_method: Union[Unset, str] = UNSET if not isinstance(self.logging_method, Unset): logging_method = self.logging_method.value metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id reliable = self.reliable @@ -143,40 +125,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = AgentSpan.from_dict(data) + return AgentSpan.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = WorkflowSpan.from_dict(data) + return WorkflowSpan.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = LlmSpan.from_dict(data) + return LlmSpan.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = RetrieverSpan.from_dict(data) + return RetrieverSpan.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ToolSpan.from_dict(data) - - return spans_item_type_4 + return ToolSpan.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -213,10 +189,7 @@ def _parse_log_stream_id(data: object) -> Union[None, Unset, str]: _logging_method = d.pop("logging_method", UNSET) logging_method: Union[Unset, LoggingMethod] - if isinstance(_logging_method, Unset): - logging_method = UNSET - else: - logging_method = LoggingMethod(_logging_method) + logging_method = UNSET if isinstance(_logging_method, Unset) else LoggingMethod(_logging_method) def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: if data is None: diff --git a/src/galileo/resources/models/log_spans_ingest_response.py b/src/galileo/resources/models/log_spans_ingest_response.py index 426405e76..feca89b4b 100644 --- a/src/galileo/resources/models/log_spans_ingest_response.py +++ b/src/galileo/resources/models/log_spans_ingest_response.py @@ -49,22 +49,13 @@ def to_dict(self) -> dict[str, Any]: trace_id = self.trace_id experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/log_stream_response.py b/src/galileo/resources/models/log_stream_response.py index 57ad445c1..bd4c99ae4 100644 --- a/src/galileo/resources/models/log_stream_response.py +++ b/src/galileo/resources/models/log_stream_response.py @@ -45,10 +45,7 @@ def to_dict(self) -> dict[str, Any]: updated_at = self.updated_at.isoformat() created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by has_user_created_sessions = self.has_user_created_sessions diff --git a/src/galileo/resources/models/log_trace_update_request.py b/src/galileo/resources/models/log_trace_update_request.py index 5d9cf6dd9..ad2b8e59b 100644 --- a/src/galileo/resources/models/log_trace_update_request.py +++ b/src/galileo/resources/models/log_trace_update_request.py @@ -50,58 +50,34 @@ def to_dict(self) -> dict[str, Any]: trace_id = self.trace_id client_version: Union[None, Unset, str] - if isinstance(self.client_version, Unset): - client_version = UNSET - else: - client_version = self.client_version + client_version = UNSET if isinstance(self.client_version, Unset) else self.client_version experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id input_: Union[None, Unset, str] - if isinstance(self.input_, Unset): - input_ = UNSET - else: - input_ = self.input_ + input_ = UNSET if isinstance(self.input_, Unset) else self.input_ is_complete: Union[None, Unset, bool] - if isinstance(self.is_complete, Unset): - is_complete = UNSET - else: - is_complete = self.is_complete + is_complete = UNSET if isinstance(self.is_complete, Unset) else self.is_complete log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id logging_method: Union[Unset, str] = UNSET if not isinstance(self.logging_method, Unset): logging_method = self.logging_method.value metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output reliable = self.reliable status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code tags: Union[None, Unset, list[str]] if isinstance(self.tags, Unset): @@ -192,10 +168,7 @@ def _parse_log_stream_id(data: object) -> Union[None, Unset, str]: _logging_method = d.pop("logging_method", UNSET) logging_method: Union[Unset, LoggingMethod] - if isinstance(_logging_method, Unset): - logging_method = UNSET - else: - logging_method = LoggingMethod(_logging_method) + logging_method = UNSET if isinstance(_logging_method, Unset) else LoggingMethod(_logging_method) def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: if data is None: @@ -234,9 +207,8 @@ def _parse_tags(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - tags_type_0 = cast(list[str], data) + return cast(list[str], data) - return tags_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/log_trace_update_response.py b/src/galileo/resources/models/log_trace_update_response.py index e13071e5b..b8a74ad2f 100644 --- a/src/galileo/resources/models/log_trace_update_response.py +++ b/src/galileo/resources/models/log_trace_update_response.py @@ -45,22 +45,13 @@ def to_dict(self) -> dict[str, Any]: trace_id = self.trace_id experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/log_traces_ingest_request.py b/src/galileo/resources/models/log_traces_ingest_request.py index 582f3bacf..40c41b1bb 100644 --- a/src/galileo/resources/models/log_traces_ingest_request.py +++ b/src/galileo/resources/models/log_traces_ingest_request.py @@ -51,42 +51,27 @@ def to_dict(self) -> dict[str, Any]: traces.append(traces_item) client_version: Union[None, Unset, str] - if isinstance(self.client_version, Unset): - client_version = UNSET - else: - client_version = self.client_version + client_version = UNSET if isinstance(self.client_version, Unset) else self.client_version experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id is_complete = self.is_complete log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id logging_method: Union[Unset, str] = UNSET if not isinstance(self.logging_method, Unset): logging_method = self.logging_method.value metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id reliable = self.reliable session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -153,10 +138,7 @@ def _parse_log_stream_id(data: object) -> Union[None, Unset, str]: _logging_method = d.pop("logging_method", UNSET) logging_method: Union[Unset, LoggingMethod] - if isinstance(_logging_method, Unset): - logging_method = UNSET - else: - logging_method = LoggingMethod(_logging_method) + logging_method = UNSET if isinstance(_logging_method, Unset) else LoggingMethod(_logging_method) def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: if data is None: diff --git a/src/galileo/resources/models/log_traces_ingest_response.py b/src/galileo/resources/models/log_traces_ingest_response.py index 8ea510ca3..c65b3e7df 100644 --- a/src/galileo/resources/models/log_traces_ingest_response.py +++ b/src/galileo/resources/models/log_traces_ingest_response.py @@ -45,22 +45,13 @@ def to_dict(self) -> dict[str, Any]: traces_count = self.traces_count experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/message.py b/src/galileo/resources/models/message.py index 53c176ced..7e7b06530 100644 --- a/src/galileo/resources/models/message.py +++ b/src/galileo/resources/models/message.py @@ -36,10 +36,7 @@ def to_dict(self) -> dict[str, Any]: role = self.role.value tool_call_id: Union[None, Unset, str] - if isinstance(self.tool_call_id, Unset): - tool_call_id = UNSET - else: - tool_call_id = self.tool_call_id + tool_call_id = UNSET if isinstance(self.tool_call_id, Unset) else self.tool_call_id tool_calls: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.tool_calls, Unset): diff --git a/src/galileo/resources/models/messages_list_item.py b/src/galileo/resources/models/messages_list_item.py index d0559845b..47814fd5a 100644 --- a/src/galileo/resources/models/messages_list_item.py +++ b/src/galileo/resources/models/messages_list_item.py @@ -25,10 +25,7 @@ def to_dict(self) -> dict[str, Any]: content = self.content role: str - if isinstance(self.role, MessagesListItemRole): - role = self.role.value - else: - role = self.role + role = self.role.value if isinstance(self.role, MessagesListItemRole) else self.role field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -45,9 +42,8 @@ def _parse_role(data: object) -> Union[MessagesListItemRole, str]: try: if not isinstance(data, str): raise TypeError() - role_type_1 = MessagesListItemRole(data) + return MessagesListItemRole(data) - return role_type_1 except: # noqa: E722 pass return cast(Union[MessagesListItemRole, str], data) diff --git a/src/galileo/resources/models/metadata_filter.py b/src/galileo/resources/models/metadata_filter.py index 27e0bb630..754175b63 100644 --- a/src/galileo/resources/models/metadata_filter.py +++ b/src/galileo/resources/models/metadata_filter.py @@ -35,11 +35,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value filter_type = self.filter_type @@ -66,9 +62,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) diff --git a/src/galileo/resources/models/metric_computation.py b/src/galileo/resources/models/metric_computation.py index becb5aa7c..9d8fa9186 100644 --- a/src/galileo/resources/models/metric_computation.py +++ b/src/galileo/resources/models/metric_computation.py @@ -36,16 +36,10 @@ def to_dict(self) -> dict[str, Any]: from ..models.metric_computation_value_type_4 import MetricComputationValueType4 error_message: Union[None, Unset, str] - if isinstance(self.error_message, Unset): - error_message = UNSET - else: - error_message = self.error_message + error_message = UNSET if isinstance(self.error_message, Unset) else self.error_message execution_time: Union[None, Unset, float] - if isinstance(self.execution_time, Unset): - execution_time = UNSET - else: - execution_time = self.execution_time + execution_time = UNSET if isinstance(self.execution_time, Unset) else self.execution_time status: Union[None, Unset, str] if isinstance(self.status, Unset): @@ -116,9 +110,8 @@ def _parse_status(data: object) -> Union[MetricComputationStatus, None, Unset]: try: if not isinstance(data, str): raise TypeError() - status_type_0 = MetricComputationStatus(data) + return MetricComputationStatus(data) - return status_type_0 except: # noqa: E722 pass return cast(Union[MetricComputationStatus, None, Unset], data) @@ -154,9 +147,8 @@ def _parse_value_type_3_item(data: object) -> Union[None, float, int, str]: try: if not isinstance(data, dict): raise TypeError() - value_type_4 = MetricComputationValueType4.from_dict(data) + return MetricComputationValueType4.from_dict(data) - return value_type_4 except: # noqa: E722 pass return cast( diff --git a/src/galileo/resources/models/metric_computing.py b/src/galileo/resources/models/metric_computing.py index 7eee346c2..721558199 100644 --- a/src/galileo/resources/models/metric_computing.py +++ b/src/galileo/resources/models/metric_computing.py @@ -62,9 +62,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metric_critique_job_configuration.py b/src/galileo/resources/models/metric_critique_job_configuration.py index 607f7f3ff..48ccde9d2 100644 --- a/src/galileo/resources/models/metric_critique_job_configuration.py +++ b/src/galileo/resources/models/metric_critique_job_configuration.py @@ -59,22 +59,16 @@ def to_dict(self) -> dict[str, Any]: recompute_settings: Union[None, Unset, dict[str, Any]] if isinstance(self.recompute_settings, Unset): recompute_settings = UNSET - elif isinstance(self.recompute_settings, RecomputeSettingsRuns): - recompute_settings = self.recompute_settings.to_dict() - elif isinstance(self.recompute_settings, RecomputeSettingsProject): - recompute_settings = self.recompute_settings.to_dict() - elif isinstance(self.recompute_settings, RecomputeSettingsObserve): - recompute_settings = self.recompute_settings.to_dict() - elif isinstance(self.recompute_settings, RecomputeSettingsLogStream): + elif isinstance( + self.recompute_settings, + (RecomputeSettingsRuns, RecomputeSettingsProject, RecomputeSettingsObserve, RecomputeSettingsLogStream), + ): recompute_settings = self.recompute_settings.to_dict() else: recompute_settings = self.recompute_settings scorer_id: Union[None, Unset, str] - if isinstance(self.scorer_id, Unset): - scorer_id = UNSET - else: - scorer_id = self.scorer_id + scorer_id = UNSET if isinstance(self.scorer_id, Unset) else self.scorer_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -135,33 +129,29 @@ def _parse_recompute_settings( try: if not isinstance(data, dict): raise TypeError() - recompute_settings_type_0_type_0 = RecomputeSettingsRuns.from_dict(data) + return RecomputeSettingsRuns.from_dict(data) - return recompute_settings_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - recompute_settings_type_0_type_1 = RecomputeSettingsProject.from_dict(data) + return RecomputeSettingsProject.from_dict(data) - return recompute_settings_type_0_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - recompute_settings_type_0_type_2 = RecomputeSettingsObserve.from_dict(data) + return RecomputeSettingsObserve.from_dict(data) - return recompute_settings_type_0_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - recompute_settings_type_0_type_3 = RecomputeSettingsLogStream.from_dict(data) + return RecomputeSettingsLogStream.from_dict(data) - return recompute_settings_type_0_type_3 except: # noqa: E722 pass return cast( diff --git a/src/galileo/resources/models/metric_error.py b/src/galileo/resources/models/metric_error.py index 7c0d74c8c..738f6a652 100644 --- a/src/galileo/resources/models/metric_error.py +++ b/src/galileo/resources/models/metric_error.py @@ -26,10 +26,7 @@ class MetricError: def to_dict(self) -> dict[str, Any]: message: Union[None, Unset, str] - if isinstance(self.message, Unset): - message = UNSET - else: - message = self.message + message = UNSET if isinstance(self.message, Unset) else self.message scorer_type: Union[None, Unset, str] if isinstance(self.scorer_type, Unset): @@ -74,9 +71,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metric_failed.py b/src/galileo/resources/models/metric_failed.py index e4d1d5c9f..5b51db01a 100644 --- a/src/galileo/resources/models/metric_failed.py +++ b/src/galileo/resources/models/metric_failed.py @@ -26,10 +26,7 @@ class MetricFailed: def to_dict(self) -> dict[str, Any]: message: Union[None, Unset, str] - if isinstance(self.message, Unset): - message = UNSET - else: - message = self.message + message = UNSET if isinstance(self.message, Unset) else self.message scorer_type: Union[None, Unset, str] if isinstance(self.scorer_type, Unset): @@ -74,9 +71,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metric_not_applicable.py b/src/galileo/resources/models/metric_not_applicable.py index a4cfd3d06..61842649d 100644 --- a/src/galileo/resources/models/metric_not_applicable.py +++ b/src/galileo/resources/models/metric_not_applicable.py @@ -62,9 +62,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metric_not_computed.py b/src/galileo/resources/models/metric_not_computed.py index 556c00bb8..887882e0d 100644 --- a/src/galileo/resources/models/metric_not_computed.py +++ b/src/galileo/resources/models/metric_not_computed.py @@ -62,9 +62,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metric_pending.py b/src/galileo/resources/models/metric_pending.py index dab4479ba..bd53c0e9c 100644 --- a/src/galileo/resources/models/metric_pending.py +++ b/src/galileo/resources/models/metric_pending.py @@ -55,9 +55,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metric_success.py b/src/galileo/resources/models/metric_success.py index 84e7d5fee..57b385146 100644 --- a/src/galileo/resources/models/metric_success.py +++ b/src/galileo/resources/models/metric_success.py @@ -146,15 +146,7 @@ def to_dict(self) -> dict[str, Any]: value = str(self.value) elif isinstance(self.value, datetime.datetime): value = self.value.isoformat() - elif isinstance(self.value, Segment): - value = self.value.to_dict() - elif isinstance(self.value, HallucinationSegment): - value = self.value.to_dict() - elif isinstance(self.value, Document): - value = self.value.to_dict() - elif isinstance(self.value, FeedbackRatingDB): - value = self.value.to_dict() - elif isinstance(self.value, FeedbackAggregate): + elif isinstance(self.value, (Segment, HallucinationSegment, Document, FeedbackRatingDB, FeedbackAggregate)): value = self.value.to_dict() elif isinstance(self.value, list): value = [] @@ -164,15 +156,10 @@ def to_dict(self) -> dict[str, Any]: value_type_11_item = str(value_type_11_item_data) elif isinstance(value_type_11_item_data, datetime.datetime): value_type_11_item = value_type_11_item_data.isoformat() - elif isinstance(value_type_11_item_data, Segment): - value_type_11_item = value_type_11_item_data.to_dict() - elif isinstance(value_type_11_item_data, HallucinationSegment): - value_type_11_item = value_type_11_item_data.to_dict() - elif isinstance(value_type_11_item_data, Document): - value_type_11_item = value_type_11_item_data.to_dict() - elif isinstance(value_type_11_item_data, FeedbackRatingDB): - value_type_11_item = value_type_11_item_data.to_dict() - elif isinstance(value_type_11_item_data, FeedbackAggregate): + elif isinstance( + value_type_11_item_data, + (Segment, HallucinationSegment, Document, FeedbackRatingDB, FeedbackAggregate), + ): value_type_11_item = value_type_11_item_data.to_dict() else: value_type_11_item = value_type_11_item_data @@ -188,15 +175,10 @@ def to_dict(self) -> dict[str, Any]: value_type_12_item_item = str(value_type_12_item_item_data) elif isinstance(value_type_12_item_item_data, datetime.datetime): value_type_12_item_item = value_type_12_item_item_data.isoformat() - elif isinstance(value_type_12_item_item_data, Segment): - value_type_12_item_item = value_type_12_item_item_data.to_dict() - elif isinstance(value_type_12_item_item_data, HallucinationSegment): - value_type_12_item_item = value_type_12_item_item_data.to_dict() - elif isinstance(value_type_12_item_item_data, Document): - value_type_12_item_item = value_type_12_item_item_data.to_dict() - elif isinstance(value_type_12_item_item_data, FeedbackRatingDB): - value_type_12_item_item = value_type_12_item_item_data.to_dict() - elif isinstance(value_type_12_item_item_data, FeedbackAggregate): + elif isinstance( + value_type_12_item_item_data, + (Segment, HallucinationSegment, Document, FeedbackRatingDB, FeedbackAggregate), + ): value_type_12_item_item = value_type_12_item_item_data.to_dict() else: value_type_12_item_item = value_type_12_item_item_data @@ -216,15 +198,10 @@ def to_dict(self) -> dict[str, Any]: value_type_13_item_item_item = str(value_type_13_item_item_item_data) elif isinstance(value_type_13_item_item_item_data, datetime.datetime): value_type_13_item_item_item = value_type_13_item_item_item_data.isoformat() - elif isinstance(value_type_13_item_item_item_data, Segment): - value_type_13_item_item_item = value_type_13_item_item_item_data.to_dict() - elif isinstance(value_type_13_item_item_item_data, HallucinationSegment): - value_type_13_item_item_item = value_type_13_item_item_item_data.to_dict() - elif isinstance(value_type_13_item_item_item_data, Document): - value_type_13_item_item_item = value_type_13_item_item_item_data.to_dict() - elif isinstance(value_type_13_item_item_item_data, FeedbackRatingDB): - value_type_13_item_item_item = value_type_13_item_item_item_data.to_dict() - elif isinstance(value_type_13_item_item_item_data, FeedbackAggregate): + elif isinstance( + value_type_13_item_item_item_data, + (Segment, HallucinationSegment, Document, FeedbackRatingDB, FeedbackAggregate), + ): value_type_13_item_item_item = value_type_13_item_item_item_data.to_dict() else: value_type_13_item_item_item = value_type_13_item_item_item_data @@ -238,10 +215,7 @@ def to_dict(self) -> dict[str, Any]: value = self.value cost: Union[None, Unset, float] - if isinstance(self.cost, Unset): - cost = UNSET - else: - cost = self.cost + cost = UNSET if isinstance(self.cost, Unset) else self.cost critique: Union[None, Unset, dict[str, Any]] if isinstance(self.critique, Unset): @@ -252,34 +226,19 @@ def to_dict(self) -> dict[str, Any]: critique = self.critique display_value: Union[None, Unset, str] - if isinstance(self.display_value, Unset): - display_value = UNSET - else: - display_value = self.display_value + display_value = UNSET if isinstance(self.display_value, Unset) else self.display_value explanation: Union[None, Unset, str] - if isinstance(self.explanation, Unset): - explanation = UNSET - else: - explanation = self.explanation + explanation = UNSET if isinstance(self.explanation, Unset) else self.explanation model_alias: Union[None, Unset, str] - if isinstance(self.model_alias, Unset): - model_alias = UNSET - else: - model_alias = self.model_alias + model_alias = UNSET if isinstance(self.model_alias, Unset) else self.model_alias num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges rationale: Union[None, Unset, str] - if isinstance(self.rationale, Unset): - rationale = UNSET - else: - rationale = self.rationale + rationale = UNSET if isinstance(self.rationale, Unset) else self.rationale scorer_type: Union[None, Unset, str] if isinstance(self.scorer_type, Unset): @@ -401,57 +360,50 @@ def _parse_value( try: if not isinstance(data, str): raise TypeError() - value_type_4 = UUID(data) + return UUID(data) - return value_type_4 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - value_type_5 = isoparse(data) + return isoparse(data) - return value_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_6 = Segment.from_dict(data) + return Segment.from_dict(data) - return value_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_7 = HallucinationSegment.from_dict(data) + return HallucinationSegment.from_dict(data) - return value_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_8 = Document.from_dict(data) + return Document.from_dict(data) - return value_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_9 = FeedbackRatingDB.from_dict(data) + return FeedbackRatingDB.from_dict(data) - return value_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_10 = FeedbackAggregate.from_dict(data) + return FeedbackAggregate.from_dict(data) - return value_type_10 except: # noqa: E722 pass try: @@ -482,57 +434,50 @@ def _parse_value_type_11_item( try: if not isinstance(data, str): raise TypeError() - value_type_11_item_type_4 = UUID(data) + return UUID(data) - return value_type_11_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - value_type_11_item_type_5 = isoparse(data) + return isoparse(data) - return value_type_11_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_11_item_type_6 = Segment.from_dict(data) + return Segment.from_dict(data) - return value_type_11_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_11_item_type_7 = HallucinationSegment.from_dict(data) + return HallucinationSegment.from_dict(data) - return value_type_11_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_11_item_type_8 = Document.from_dict(data) + return Document.from_dict(data) - return value_type_11_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_11_item_type_9 = FeedbackRatingDB.from_dict(data) + return FeedbackRatingDB.from_dict(data) - return value_type_11_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_11_item_type_10 = FeedbackAggregate.from_dict(data) + return FeedbackAggregate.from_dict(data) - return value_type_11_item_type_10 except: # noqa: E722 pass return cast( @@ -591,57 +536,50 @@ def _parse_value_type_12_item_item( try: if not isinstance(data, str): raise TypeError() - value_type_12_item_item_type_4 = UUID(data) + return UUID(data) - return value_type_12_item_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - value_type_12_item_item_type_5 = isoparse(data) + return isoparse(data) - return value_type_12_item_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_12_item_item_type_6 = Segment.from_dict(data) + return Segment.from_dict(data) - return value_type_12_item_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_12_item_item_type_7 = HallucinationSegment.from_dict(data) + return HallucinationSegment.from_dict(data) - return value_type_12_item_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_12_item_item_type_8 = Document.from_dict(data) + return Document.from_dict(data) - return value_type_12_item_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_12_item_item_type_9 = FeedbackRatingDB.from_dict(data) + return FeedbackRatingDB.from_dict(data) - return value_type_12_item_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_12_item_item_type_10 = FeedbackAggregate.from_dict(data) + return FeedbackAggregate.from_dict(data) - return value_type_12_item_item_type_10 except: # noqa: E722 pass return cast( @@ -705,57 +643,50 @@ def _parse_value_type_13_item_item_item( try: if not isinstance(data, str): raise TypeError() - value_type_13_item_item_item_type_4 = UUID(data) + return UUID(data) - return value_type_13_item_item_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - value_type_13_item_item_item_type_5 = isoparse(data) + return isoparse(data) - return value_type_13_item_item_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_13_item_item_item_type_6 = Segment.from_dict(data) + return Segment.from_dict(data) - return value_type_13_item_item_item_type_6 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_13_item_item_item_type_7 = HallucinationSegment.from_dict(data) + return HallucinationSegment.from_dict(data) - return value_type_13_item_item_item_type_7 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_13_item_item_item_type_8 = Document.from_dict(data) + return Document.from_dict(data) - return value_type_13_item_item_item_type_8 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_13_item_item_item_type_9 = FeedbackRatingDB.from_dict(data) + return FeedbackRatingDB.from_dict(data) - return value_type_13_item_item_item_type_9 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - value_type_13_item_item_item_type_10 = FeedbackAggregate.from_dict(data) + return FeedbackAggregate.from_dict(data) - return value_type_13_item_item_item_type_10 except: # noqa: E722 pass return cast( @@ -880,9 +811,8 @@ def _parse_critique(data: object) -> Union["MetricCritiqueColumnar", None, Unset try: if not isinstance(data, dict): raise TypeError() - critique_type_0 = MetricCritiqueColumnar.from_dict(data) + return MetricCritiqueColumnar.from_dict(data) - return critique_type_0 except: # noqa: E722 pass return cast(Union["MetricCritiqueColumnar", None, Unset], data) @@ -942,9 +872,8 @@ def _parse_scorer_type(data: object) -> Union[None, ScorerType, Unset]: try: if not isinstance(data, str): raise TypeError() - scorer_type_type_0 = ScorerType(data) + return ScorerType(data) - return scorer_type_type_0 except: # noqa: E722 pass return cast(Union[None, ScorerType, Unset], data) diff --git a/src/galileo/resources/models/metrics.py b/src/galileo/resources/models/metrics.py index c88bce70f..9b3423046 100644 --- a/src/galileo/resources/models/metrics.py +++ b/src/galileo/resources/models/metrics.py @@ -22,10 +22,7 @@ class Metrics: def to_dict(self) -> dict[str, Any]: duration_ns: Union[None, Unset, int] - if isinstance(self.duration_ns, Unset): - duration_ns = UNSET - else: - duration_ns = self.duration_ns + duration_ns = UNSET if isinstance(self.duration_ns, Unset) else self.duration_ns field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/metrics_testing_available_columns_request.py b/src/galileo/resources/models/metrics_testing_available_columns_request.py index 6221bad75..e152ae612 100644 --- a/src/galileo/resources/models/metrics_testing_available_columns_request.py +++ b/src/galileo/resources/models/metrics_testing_available_columns_request.py @@ -38,22 +38,13 @@ def to_dict(self) -> dict[str, Any]: cot_enabled = self.cot_enabled experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id output_type: Union[Unset, str] = UNSET if not isinstance(self.output_type, Unset): @@ -111,10 +102,7 @@ def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: _output_type = d.pop("output_type", UNSET) output_type: Union[Unset, OutputTypeEnum] - if isinstance(_output_type, Unset): - output_type = UNSET - else: - output_type = OutputTypeEnum(_output_type) + output_type = UNSET if isinstance(_output_type, Unset) else OutputTypeEnum(_output_type) metrics_testing_available_columns_request = cls( name=name, diff --git a/src/galileo/resources/models/model.py b/src/galileo/resources/models/model.py index 9ca42e61a..d7f926025 100644 --- a/src/galileo/resources/models/model.py +++ b/src/galileo/resources/models/model.py @@ -83,16 +83,10 @@ def to_dict(self) -> dict[str, Any]: alternative_names = self.alternative_names api_version: Union[None, Unset, str] - if isinstance(self.api_version, Unset): - api_version = UNSET - else: - api_version = self.api_version + api_version = UNSET if isinstance(self.api_version, Unset) else self.api_version assistant_role: Union[None, Unset, str] - if isinstance(self.assistant_role, Unset): - assistant_role = UNSET - else: - assistant_role = self.assistant_role + assistant_role = UNSET if isinstance(self.assistant_role, Unset) else self.assistant_role cost_by: Union[Unset, str] = UNSET if not isinstance(self.cost_by, Unset): @@ -111,10 +105,7 @@ def to_dict(self) -> dict[str, Any]: input_price = self.input_price input_token_limit: Union[None, Unset, int] - if isinstance(self.input_token_limit, Unset): - input_token_limit = UNSET - else: - input_token_limit = self.input_token_limit + input_token_limit = UNSET if isinstance(self.input_token_limit, Unset) else self.input_token_limit integration: Union[Unset, str] = UNSET if not isinstance(self.integration, Unset): @@ -133,10 +124,7 @@ def to_dict(self) -> dict[str, Any]: output_price = self.output_price output_token_limit: Union[None, Unset, int] - if isinstance(self.output_token_limit, Unset): - output_token_limit = UNSET - else: - output_token_limit = self.output_token_limit + output_token_limit = UNSET if isinstance(self.output_token_limit, Unset) else self.output_token_limit params_map: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.params_map, Unset): @@ -149,16 +137,10 @@ def to_dict(self) -> dict[str, Any]: system_supported = self.system_supported token_limit: Union[None, Unset, int] - if isinstance(self.token_limit, Unset): - token_limit = UNSET - else: - token_limit = self.token_limit + token_limit = UNSET if isinstance(self.token_limit, Unset) else self.token_limit user_role: Union[None, Unset, str] - if isinstance(self.user_role, Unset): - user_role = UNSET - else: - user_role = self.user_role + user_role = UNSET if isinstance(self.user_role, Unset) else self.user_role field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -237,10 +219,7 @@ def _parse_assistant_role(data: object) -> Union[None, Unset, str]: _cost_by = d.pop("cost_by", UNSET) cost_by: Union[Unset, ModelCostBy] - if isinstance(_cost_by, Unset): - cost_by = UNSET - else: - cost_by = ModelCostBy(_cost_by) + cost_by = UNSET if isinstance(_cost_by, Unset) else ModelCostBy(_cost_by) formatting_tokens = d.pop("formatting_tokens", UNSET) @@ -252,9 +231,8 @@ def _parse_input_map(data: object) -> Union["InputMap", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - input_map_type_0 = InputMap.from_dict(data) + return InputMap.from_dict(data) - return input_map_type_0 except: # noqa: E722 pass return cast(Union["InputMap", None, Unset], data) @@ -274,10 +252,7 @@ def _parse_input_token_limit(data: object) -> Union[None, Unset, int]: _integration = d.pop("integration", UNSET) integration: Union[Unset, LLMIntegration] - if isinstance(_integration, Unset): - integration = UNSET - else: - integration = LLMIntegration(_integration) + integration = UNSET if isinstance(_integration, Unset) else LLMIntegration(_integration) is_chat = d.pop("is_chat", UNSET) @@ -289,9 +264,8 @@ def _parse_output_map(data: object) -> Union["OutputMap", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - output_map_type_0 = OutputMap.from_dict(data) + return OutputMap.from_dict(data) - return output_map_type_0 except: # noqa: E722 pass return cast(Union["OutputMap", None, Unset], data) @@ -311,10 +285,7 @@ def _parse_output_token_limit(data: object) -> Union[None, Unset, int]: _params_map = d.pop("params_map", UNSET) params_map: Union[Unset, RunParamsMap] - if isinstance(_params_map, Unset): - params_map = UNSET - else: - params_map = RunParamsMap.from_dict(_params_map) + params_map = UNSET if isinstance(_params_map, Unset) else RunParamsMap.from_dict(_params_map) provides_log_probs = d.pop("provides_log_probs", UNSET) diff --git a/src/galileo/resources/models/output_map.py b/src/galileo/resources/models/output_map.py index 571ac7031..060e65653 100644 --- a/src/galileo/resources/models/output_map.py +++ b/src/galileo/resources/models/output_map.py @@ -31,28 +31,16 @@ def to_dict(self) -> dict[str, Any]: response = self.response completion_reason: Union[None, Unset, str] - if isinstance(self.completion_reason, Unset): - completion_reason = UNSET - else: - completion_reason = self.completion_reason + completion_reason = UNSET if isinstance(self.completion_reason, Unset) else self.completion_reason input_token_count: Union[None, Unset, str] - if isinstance(self.input_token_count, Unset): - input_token_count = UNSET - else: - input_token_count = self.input_token_count + input_token_count = UNSET if isinstance(self.input_token_count, Unset) else self.input_token_count output_token_count: Union[None, Unset, str] - if isinstance(self.output_token_count, Unset): - output_token_count = UNSET - else: - output_token_count = self.output_token_count + output_token_count = UNSET if isinstance(self.output_token_count, Unset) else self.output_token_count token_count: Union[None, Unset, str] - if isinstance(self.token_count, Unset): - token_count = UNSET - else: - token_count = self.token_count + token_count = UNSET if isinstance(self.token_count, Unset) else self.token_count field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/output_pii_scorer.py b/src/galileo/resources/models/output_pii_scorer.py index 682b6a9f5..3c6852101 100644 --- a/src/galileo/resources/models/output_pii_scorer.py +++ b/src/galileo/resources/models/output_pii_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/output_sexist_scorer.py b/src/galileo/resources/models/output_sexist_scorer.py index b3a038164..1e643e9c4 100644 --- a/src/galileo/resources/models/output_sexist_scorer.py +++ b/src/galileo/resources/models/output_sexist_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, OutputSexistScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = OutputSexistScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else OutputSexistScorerType(_type_) output_sexist_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/output_tone_scorer.py b/src/galileo/resources/models/output_tone_scorer.py index 050d326b3..697a50315 100644 --- a/src/galileo/resources/models/output_tone_scorer.py +++ b/src/galileo/resources/models/output_tone_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/output_toxicity_scorer.py b/src/galileo/resources/models/output_toxicity_scorer.py index 7d97cbb74..a2f118c2b 100644 --- a/src/galileo/resources/models/output_toxicity_scorer.py +++ b/src/galileo/resources/models/output_toxicity_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, OutputToxicityScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = OutputToxicityScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else OutputToxicityScorerType(_type_) output_toxicity_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/output_type_enum.py b/src/galileo/resources/models/output_type_enum.py index 716a6f65d..e59d087a8 100644 --- a/src/galileo/resources/models/output_type_enum.py +++ b/src/galileo/resources/models/output_type_enum.py @@ -7,6 +7,7 @@ class OutputTypeEnum(str, Enum): COUNT = "count" DISCRETE = "discrete" FREEFORM = "freeform" + MULTILABEL = "multilabel" PERCENTAGE = "percentage" def __str__(self) -> str: diff --git a/src/galileo/resources/models/payload.py b/src/galileo/resources/models/payload.py index a1cedf0f8..f6e4a7027 100644 --- a/src/galileo/resources/models/payload.py +++ b/src/galileo/resources/models/payload.py @@ -23,16 +23,10 @@ class Payload: def to_dict(self) -> dict[str, Any]: input_: Union[None, Unset, str] - if isinstance(self.input_, Unset): - input_ = UNSET - else: - input_ = self.input_ + input_ = UNSET if isinstance(self.input_, Unset) else self.input_ output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/permission.py b/src/galileo/resources/models/permission.py index 5329f3135..0c7aefe5e 100644 --- a/src/galileo/resources/models/permission.py +++ b/src/galileo/resources/models/permission.py @@ -49,25 +49,21 @@ class Permission: def to_dict(self) -> dict[str, Any]: action: str - if isinstance(self.action, UserAction): - action = self.action.value - elif isinstance(self.action, GroupAction): - action = self.action.value - elif isinstance(self.action, GroupMemberAction): - action = self.action.value - elif isinstance(self.action, ProjectAction): - action = self.action.value - elif isinstance(self.action, RegisteredScorerAction): - action = self.action.value - elif isinstance(self.action, ApiKeyAction): - action = self.action.value - elif isinstance(self.action, GeneratedScorerAction): - action = self.action.value - elif isinstance(self.action, FineTunedScorerAction): - action = self.action.value - elif isinstance(self.action, DatasetAction): - action = self.action.value - elif isinstance(self.action, IntegrationAction): + if isinstance( + self.action, + ( + UserAction, + GroupAction, + GroupMemberAction, + ProjectAction, + RegisteredScorerAction, + ApiKeyAction, + GeneratedScorerAction, + FineTunedScorerAction, + DatasetAction, + IntegrationAction, + ), + ): action = self.action.value else: action = self.action.value @@ -75,10 +71,7 @@ def to_dict(self) -> dict[str, Any]: allowed = self.allowed message: Union[None, Unset, str] - if isinstance(self.message, Unset): - message = UNSET - else: - message = self.message + message = UNSET if isinstance(self.message, Unset) else self.message field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -110,88 +103,76 @@ def _parse_action( try: if not isinstance(data, str): raise TypeError() - action_type_0 = UserAction(data) + return UserAction(data) - return action_type_0 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_1 = GroupAction(data) + return GroupAction(data) - return action_type_1 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_2 = GroupMemberAction(data) + return GroupMemberAction(data) - return action_type_2 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_3 = ProjectAction(data) + return ProjectAction(data) - return action_type_3 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_4 = RegisteredScorerAction(data) + return RegisteredScorerAction(data) - return action_type_4 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_5 = ApiKeyAction(data) + return ApiKeyAction(data) - return action_type_5 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_6 = GeneratedScorerAction(data) + return GeneratedScorerAction(data) - return action_type_6 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_7 = FineTunedScorerAction(data) + return FineTunedScorerAction(data) - return action_type_7 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_8 = DatasetAction(data) + return DatasetAction(data) - return action_type_8 except: # noqa: E722 pass try: if not isinstance(data, str): raise TypeError() - action_type_9 = IntegrationAction(data) + return IntegrationAction(data) - return action_type_9 except: # noqa: E722 pass if not isinstance(data, str): raise TypeError() - action_type_10 = OrganizationAction(data) - - return action_type_10 + return OrganizationAction(data) action = _parse_action(d.pop("action")) diff --git a/src/galileo/resources/models/preview_dataset_request.py b/src/galileo/resources/models/preview_dataset_request.py index b0fe1e57a..beabb665f 100644 --- a/src/galileo/resources/models/preview_dataset_request.py +++ b/src/galileo/resources/models/preview_dataset_request.py @@ -56,9 +56,8 @@ def _parse_column_mapping(data: object) -> Union["ColumnMapping", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - column_mapping_type_0 = ColumnMapping.from_dict(data) + return ColumnMapping.from_dict(data) - return column_mapping_type_0 except: # noqa: E722 pass return cast(Union["ColumnMapping", None, Unset], data) diff --git a/src/galileo/resources/models/project_collection_params.py b/src/galileo/resources/models/project_collection_params.py index 053d73727..b3b3079b9 100644 --- a/src/galileo/resources/models/project_collection_params.py +++ b/src/galileo/resources/models/project_collection_params.py @@ -83,19 +83,18 @@ def to_dict(self) -> dict[str, Any]: filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, ProjectIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ProjectNameFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ProjectTypeFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ProjectCreatorFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ProjectCreatedAtFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ProjectUpdatedAtFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, ProjectRunsFilter): + if isinstance( + filters_item_data, + ( + ProjectIDFilter, + ProjectNameFilter, + ProjectTypeFilter, + ProjectCreatorFilter, + ProjectCreatedAtFilter, + ProjectUpdatedAtFilter, + ProjectRunsFilter, + ), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -105,17 +104,17 @@ def to_dict(self) -> dict[str, Any]: sort: Union[None, Unset, dict[str, Any]] if isinstance(self.sort, Unset): sort = UNSET - elif isinstance(self.sort, ProjectNameSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, ProjectTypeSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, ProjectCreatedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, ProjectUpdatedAtSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, ProjectRunsSort): - sort = self.sort.to_dict() - elif isinstance(self.sort, ProjectBookmarkSort): + elif isinstance( + self.sort, + ( + ProjectNameSort, + ProjectTypeSort, + ProjectCreatedAtSort, + ProjectUpdatedAtSort, + ProjectRunsSort, + ProjectBookmarkSort, + ), + ): sort = self.sort.to_dict() else: sort = self.sort @@ -167,64 +166,55 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = ProjectIDFilter.from_dict(data) + return ProjectIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = ProjectNameFilter.from_dict(data) + return ProjectNameFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = ProjectTypeFilter.from_dict(data) + return ProjectTypeFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = ProjectCreatorFilter.from_dict(data) + return ProjectCreatorFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = ProjectCreatedAtFilter.from_dict(data) + return ProjectCreatedAtFilter.from_dict(data) - return filters_item_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_5 = ProjectUpdatedAtFilter.from_dict(data) + return ProjectUpdatedAtFilter.from_dict(data) - return filters_item_type_5 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_6 = ProjectRunsFilter.from_dict(data) + return ProjectRunsFilter.from_dict(data) - return filters_item_type_6 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_7 = ProjectBookmarkFilter.from_dict(data) - - return filters_item_type_7 + return ProjectBookmarkFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -249,49 +239,43 @@ def _parse_sort( try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_0 = ProjectNameSort.from_dict(data) + return ProjectNameSort.from_dict(data) - return sort_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_1 = ProjectTypeSort.from_dict(data) + return ProjectTypeSort.from_dict(data) - return sort_type_0_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_2 = ProjectCreatedAtSort.from_dict(data) + return ProjectCreatedAtSort.from_dict(data) - return sort_type_0_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_3 = ProjectUpdatedAtSort.from_dict(data) + return ProjectUpdatedAtSort.from_dict(data) - return sort_type_0_type_3 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_4 = ProjectRunsSort.from_dict(data) + return ProjectRunsSort.from_dict(data) - return sort_type_0_type_4 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - sort_type_0_type_5 = ProjectBookmarkSort.from_dict(data) + return ProjectBookmarkSort.from_dict(data) - return sort_type_0_type_5 except: # noqa: E722 pass return cast( diff --git a/src/galileo/resources/models/project_create.py b/src/galileo/resources/models/project_create.py index 05127e5bc..686d9c11d 100644 --- a/src/galileo/resources/models/project_create.py +++ b/src/galileo/resources/models/project_create.py @@ -32,10 +32,7 @@ def to_dict(self) -> dict[str, Any]: create_example_templates = self.create_example_templates created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -71,10 +68,7 @@ def _parse_created_by(data: object) -> Union[None, Unset, str]: _type_ = d.pop("type", UNSET) type_: Union[Unset, ProjectType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = ProjectType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else ProjectType(_type_) project_create = cls( name=name, create_example_templates=create_example_templates, created_by=created_by, type_=type_ diff --git a/src/galileo/resources/models/project_create_response.py b/src/galileo/resources/models/project_create_response.py index 3a6d020a3..dbaef734c 100644 --- a/src/galileo/resources/models/project_create_response.py +++ b/src/galileo/resources/models/project_create_response.py @@ -40,16 +40,10 @@ def to_dict(self) -> dict[str, Any]: updated_at = self.updated_at.isoformat() created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name type_: Union[None, Unset, str] if isinstance(self.type_, Unset): @@ -106,9 +100,8 @@ def _parse_type_(data: object) -> Union[None, ProjectType, Unset]: try: if not isinstance(data, str): raise TypeError() - type_type_0 = ProjectType(data) + return ProjectType(data) - return type_type_0 except: # noqa: E722 pass return cast(Union[None, ProjectType, Unset], data) diff --git a/src/galileo/resources/models/project_creator_filter.py b/src/galileo/resources/models/project_creator_filter.py index c1c7969bb..ea04e772b 100644 --- a/src/galileo/resources/models/project_creator_filter.py +++ b/src/galileo/resources/models/project_creator_filter.py @@ -26,11 +26,7 @@ class ProjectCreatorFilter: def to_dict(self) -> dict[str, Any]: value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value name = self.name @@ -56,9 +52,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) @@ -71,10 +66,7 @@ def _parse_value(data: object) -> Union[list[str], str]: _operator = d.pop("operator", UNSET) operator: Union[Unset, ProjectCreatorFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = ProjectCreatorFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else ProjectCreatorFilterOperator(_operator) project_creator_filter = cls(value=value, name=name, operator=operator) diff --git a/src/galileo/resources/models/project_db.py b/src/galileo/resources/models/project_db.py index 8b88b63a5..9e40b605d 100644 --- a/src/galileo/resources/models/project_db.py +++ b/src/galileo/resources/models/project_db.py @@ -70,10 +70,7 @@ def to_dict(self) -> dict[str, Any]: bookmark = self.bookmark description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description labels: Union[Unset, list[str]] = UNSET if not isinstance(self.labels, Unset): @@ -83,10 +80,7 @@ def to_dict(self) -> dict[str, Any]: labels.append(labels_item) name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name permissions: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.permissions, Unset): @@ -196,9 +190,8 @@ def _parse_type_(data: object) -> Union[None, ProjectType, Unset]: try: if not isinstance(data, str): raise TypeError() - type_type_0 = ProjectType(data) + return ProjectType(data) - return type_type_0 except: # noqa: E722 pass return cast(Union[None, ProjectType, Unset], data) diff --git a/src/galileo/resources/models/project_db_thin.py b/src/galileo/resources/models/project_db_thin.py index ecd866c30..045ff1bf7 100644 --- a/src/galileo/resources/models/project_db_thin.py +++ b/src/galileo/resources/models/project_db_thin.py @@ -60,10 +60,7 @@ def to_dict(self) -> dict[str, Any]: bookmark = self.bookmark name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name permissions: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.permissions, Unset): @@ -143,9 +140,8 @@ def _parse_type_(data: object) -> Union[None, ProjectType, Unset]: try: if not isinstance(data, str): raise TypeError() - type_type_0 = ProjectType(data) + return ProjectType(data) - return type_type_0 except: # noqa: E722 pass return cast(Union[None, ProjectType, Unset], data) diff --git a/src/galileo/resources/models/project_id_filter.py b/src/galileo/resources/models/project_id_filter.py index 9c1822825..081cbce1f 100644 --- a/src/galileo/resources/models/project_id_filter.py +++ b/src/galileo/resources/models/project_id_filter.py @@ -26,11 +26,7 @@ class ProjectIDFilter: def to_dict(self) -> dict[str, Any]: value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value name = self.name @@ -56,9 +52,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) @@ -71,10 +66,7 @@ def _parse_value(data: object) -> Union[list[str], str]: _operator = d.pop("operator", UNSET) operator: Union[Unset, ProjectIDFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = ProjectIDFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else ProjectIDFilterOperator(_operator) project_id_filter = cls(value=value, name=name, operator=operator) diff --git a/src/galileo/resources/models/project_item.py b/src/galileo/resources/models/project_item.py index fe96c2bfa..b8d18782f 100644 --- a/src/galileo/resources/models/project_item.py +++ b/src/galileo/resources/models/project_item.py @@ -70,10 +70,7 @@ def to_dict(self) -> dict[str, Any]: created_by_user = self.created_by_user description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description labels: Union[Unset, list[str]] = UNSET if not isinstance(self.labels, Unset): @@ -83,16 +80,10 @@ def to_dict(self) -> dict[str, Any]: labels.append(labels_item) num_experiments: Union[None, Unset, int] - if isinstance(self.num_experiments, Unset): - num_experiments = UNSET - else: - num_experiments = self.num_experiments + num_experiments = UNSET if isinstance(self.num_experiments, Unset) else self.num_experiments num_logstreams: Union[None, Unset, int] - if isinstance(self.num_logstreams, Unset): - num_logstreams = UNSET - else: - num_logstreams = self.num_logstreams + num_logstreams = UNSET if isinstance(self.num_logstreams, Unset) else self.num_logstreams permissions: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.permissions, Unset): @@ -145,9 +136,8 @@ def _parse_created_by_user(data: object) -> Union["UserInfo", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - created_by_user_type_0 = UserInfo.from_dict(data) + return UserInfo.from_dict(data) - return created_by_user_type_0 except: # noqa: E722 pass return cast(Union["UserInfo", None, Unset], data) diff --git a/src/galileo/resources/models/project_name_filter.py b/src/galileo/resources/models/project_name_filter.py index d9803bd3c..c8879edcb 100644 --- a/src/galileo/resources/models/project_name_filter.py +++ b/src/galileo/resources/models/project_name_filter.py @@ -30,11 +30,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value case_sensitive = self.case_sensitive @@ -59,9 +55,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) diff --git a/src/galileo/resources/models/project_runs_filter.py b/src/galileo/resources/models/project_runs_filter.py index 1d0a23aa8..fa8a115f7 100644 --- a/src/galileo/resources/models/project_runs_filter.py +++ b/src/galileo/resources/models/project_runs_filter.py @@ -28,14 +28,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[float, int, list[float], list[int]] - if isinstance(self.value, list): - value = self.value - - elif isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, (list, list)) else self.value name = self.name @@ -56,17 +49,15 @@ def _parse_value(data: object) -> Union[float, int, list[float], list[int]]: try: if not isinstance(data, list): raise TypeError() - value_type_2 = cast(list[int], data) + return cast(list[int], data) - return value_type_2 except: # noqa: E722 pass try: if not isinstance(data, list): raise TypeError() - value_type_3 = cast(list[float], data) + return cast(list[float], data) - return value_type_3 except: # noqa: E722 pass return cast(Union[float, int, list[float], list[int]], data) diff --git a/src/galileo/resources/models/project_type_filter.py b/src/galileo/resources/models/project_type_filter.py index 781e658ac..39fef9326 100644 --- a/src/galileo/resources/models/project_type_filter.py +++ b/src/galileo/resources/models/project_type_filter.py @@ -56,9 +56,8 @@ def _parse_value(data: object) -> Union[ProjectType, list[ProjectType]]: try: if not isinstance(data, str): raise TypeError() - value_type_0 = ProjectType(data) + return ProjectType(data) - return value_type_0 except: # noqa: E722 pass if not isinstance(data, list): diff --git a/src/galileo/resources/models/project_update.py b/src/galileo/resources/models/project_update.py index 1e6172af0..7544a166c 100644 --- a/src/galileo/resources/models/project_update.py +++ b/src/galileo/resources/models/project_update.py @@ -30,16 +30,10 @@ class ProjectUpdate: def to_dict(self) -> dict[str, Any]: created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description labels: Union[None, Unset, list[str]] if isinstance(self.labels, Unset): @@ -51,10 +45,7 @@ def to_dict(self) -> dict[str, Any]: labels = self.labels name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name type_: Union[None, Unset, str] if isinstance(self.type_, Unset): @@ -110,9 +101,8 @@ def _parse_labels(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - labels_type_0 = cast(list[str], data) + return cast(list[str], data) - return labels_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -136,9 +126,8 @@ def _parse_type_(data: object) -> Union[None, ProjectType, Unset]: try: if not isinstance(data, str): raise TypeError() - type_type_0 = ProjectType(data) + return ProjectType(data) - return type_type_0 except: # noqa: E722 pass return cast(Union[None, ProjectType, Unset], data) diff --git a/src/galileo/resources/models/project_update_response.py b/src/galileo/resources/models/project_update_response.py index ea710b62f..654653459 100644 --- a/src/galileo/resources/models/project_update_response.py +++ b/src/galileo/resources/models/project_update_response.py @@ -45,16 +45,10 @@ def to_dict(self) -> dict[str, Any]: updated_at = self.updated_at.isoformat() created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description labels: Union[Unset, list[str]] = UNSET if not isinstance(self.labels, Unset): @@ -64,10 +58,7 @@ def to_dict(self) -> dict[str, Any]: labels.append(labels_item) name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name type_: Union[None, Unset, str] if isinstance(self.type_, Unset): @@ -144,9 +135,8 @@ def _parse_type_(data: object) -> Union[None, ProjectType, Unset]: try: if not isinstance(data, str): raise TypeError() - type_type_0 = ProjectType(data) + return ProjectType(data) - return type_type_0 except: # noqa: E722 pass return cast(Union[None, ProjectType, Unset], data) diff --git a/src/galileo/resources/models/prompt_dataset_db.py b/src/galileo/resources/models/prompt_dataset_db.py index 09ca737b1..01c836d7c 100644 --- a/src/galileo/resources/models/prompt_dataset_db.py +++ b/src/galileo/resources/models/prompt_dataset_db.py @@ -35,28 +35,16 @@ def to_dict(self) -> dict[str, Any]: id = self.id file_name: Union[None, Unset, str] - if isinstance(self.file_name, Unset): - file_name = UNSET - else: - file_name = self.file_name + file_name = UNSET if isinstance(self.file_name, Unset) else self.file_name message: Union[None, Unset, str] - if isinstance(self.message, Unset): - message = UNSET - else: - message = self.message + message = UNSET if isinstance(self.message, Unset) else self.message num_rows: Union[None, Unset, int] - if isinstance(self.num_rows, Unset): - num_rows = UNSET - else: - num_rows = self.num_rows + num_rows = UNSET if isinstance(self.num_rows, Unset) else self.num_rows rows: Union[None, Unset, int] - if isinstance(self.rows, Unset): - rows = UNSET - else: - rows = self.rows + rows = UNSET if isinstance(self.rows, Unset) else self.rows field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/prompt_injection_scorer.py b/src/galileo/resources/models/prompt_injection_scorer.py index 889929af7..b473d207a 100644 --- a/src/galileo/resources/models/prompt_injection_scorer.py +++ b/src/galileo/resources/models/prompt_injection_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, PromptInjectionScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = PromptInjectionScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else PromptInjectionScorerType(_type_) prompt_injection_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/prompt_optimization_configuration.py b/src/galileo/resources/models/prompt_optimization_configuration.py index b67bba333..1cab56502 100644 --- a/src/galileo/resources/models/prompt_optimization_configuration.py +++ b/src/galileo/resources/models/prompt_optimization_configuration.py @@ -71,16 +71,10 @@ def to_dict(self) -> dict[str, Any]: integration_name = self.integration_name.value reasoning_effort: Union[None, Unset, str] - if isinstance(self.reasoning_effort, Unset): - reasoning_effort = UNSET - else: - reasoning_effort = self.reasoning_effort + reasoning_effort = UNSET if isinstance(self.reasoning_effort, Unset) else self.reasoning_effort verbosity: Union[None, Unset, str] - if isinstance(self.verbosity, Unset): - verbosity = UNSET - else: - verbosity = self.verbosity + verbosity = UNSET if isinstance(self.verbosity, Unset) else self.verbosity field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -132,10 +126,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _integration_name = d.pop("integration_name", UNSET) integration_name: Union[Unset, LLMIntegration] - if isinstance(_integration_name, Unset): - integration_name = UNSET - else: - integration_name = LLMIntegration(_integration_name) + integration_name = UNSET if isinstance(_integration_name, Unset) else LLMIntegration(_integration_name) def _parse_reasoning_effort(data: object) -> Union[None, Unset, str]: if data is None: diff --git a/src/galileo/resources/models/prompt_perplexity_scorer.py b/src/galileo/resources/models/prompt_perplexity_scorer.py index 1a017b01e..506077743 100644 --- a/src/galileo/resources/models/prompt_perplexity_scorer.py +++ b/src/galileo/resources/models/prompt_perplexity_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/prompt_run_settings.py b/src/galileo/resources/models/prompt_run_settings.py index b3fa5d074..84fb7a579 100644 --- a/src/galileo/resources/models/prompt_run_settings.py +++ b/src/galileo/resources/models/prompt_run_settings.py @@ -27,7 +27,7 @@ class PromptRunSettings: known_models (Union[Unset, list['Model']]): logprobs (Union[Unset, bool]): Default: True. max_tokens (Union[Unset, int]): Default: 1024. - model_alias (Union[Unset, str]): Default: 'gpt-5-mini'. + model_alias (Union[Unset, str]): Default: 'GPT-4o'. n (Union[Unset, int]): Default: 1. presence_penalty (Union[Unset, float]): Default: 0.0. reasoning_effort (Union[Unset, str]): Default: 'medium'. @@ -48,7 +48,7 @@ class PromptRunSettings: known_models: Union[Unset, list["Model"]] = UNSET logprobs: Union[Unset, bool] = True max_tokens: Union[Unset, int] = 1024 - model_alias: Union[Unset, str] = "gpt-5-mini" + model_alias: Union[Unset, str] = "GPT-4o" n: Union[Unset, int] = 1 presence_penalty: Union[Unset, float] = 0.0 reasoning_effort: Union[Unset, str] = "medium" @@ -68,10 +68,7 @@ def to_dict(self) -> dict[str, Any]: from ..models.prompt_run_settings_response_format_type_0 import PromptRunSettingsResponseFormatType0 deployment_name: Union[None, Unset, str] - if isinstance(self.deployment_name, Unset): - deployment_name = UNSET - else: - deployment_name = self.deployment_name + deployment_name = UNSET if isinstance(self.deployment_name, Unset) else self.deployment_name echo = self.echo @@ -236,9 +233,8 @@ def _parse_response_format(data: object) -> Union["PromptRunSettingsResponseForm try: if not isinstance(data, dict): raise TypeError() - response_format_type_0 = PromptRunSettingsResponseFormatType0.from_dict(data) + return PromptRunSettingsResponseFormatType0.from_dict(data) - return response_format_type_0 except: # noqa: E722 pass return cast(Union["PromptRunSettingsResponseFormatType0", None, Unset], data) @@ -253,9 +249,8 @@ def _parse_stop_sequences(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - stop_sequences_type_0 = cast(list[str], data) + return cast(list[str], data) - return stop_sequences_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -272,9 +267,8 @@ def _parse_tool_choice(data: object) -> Union["OpenAIToolChoice", None, Unset, s try: if not isinstance(data, dict): raise TypeError() - tool_choice_type_1 = OpenAIToolChoice.from_dict(data) + return OpenAIToolChoice.from_dict(data) - return tool_choice_type_1 except: # noqa: E722 pass return cast(Union["OpenAIToolChoice", None, Unset, str], data) diff --git a/src/galileo/resources/models/prompt_template_created_by_filter.py b/src/galileo/resources/models/prompt_template_created_by_filter.py index 8700456aa..2dbed6eac 100644 --- a/src/galileo/resources/models/prompt_template_created_by_filter.py +++ b/src/galileo/resources/models/prompt_template_created_by_filter.py @@ -27,11 +27,7 @@ class PromptTemplateCreatedByFilter: def to_dict(self) -> dict[str, Any]: value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value name = self.name @@ -57,9 +53,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) @@ -72,10 +67,7 @@ def _parse_value(data: object) -> Union[list[str], str]: _operator = d.pop("operator", UNSET) operator: Union[Unset, PromptTemplateCreatedByFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = PromptTemplateCreatedByFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else PromptTemplateCreatedByFilterOperator(_operator) prompt_template_created_by_filter = cls(value=value, name=name, operator=operator) diff --git a/src/galileo/resources/models/prompt_template_name_filter.py b/src/galileo/resources/models/prompt_template_name_filter.py index 6ac55d2b2..473ac44b8 100644 --- a/src/galileo/resources/models/prompt_template_name_filter.py +++ b/src/galileo/resources/models/prompt_template_name_filter.py @@ -30,11 +30,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value case_sensitive = self.case_sensitive @@ -59,9 +55,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) diff --git a/src/galileo/resources/models/protect_request.py b/src/galileo/resources/models/protect_request.py index 1caeb5ccd..7d9bbda4e 100644 --- a/src/galileo/resources/models/protect_request.py +++ b/src/galileo/resources/models/protect_request.py @@ -79,34 +79,19 @@ def to_dict(self) -> dict[str, Any]: prioritized_rulesets.append(prioritized_rulesets_item) project_id: Union[None, Unset, str] - if isinstance(self.project_id, Unset): - project_id = UNSET - else: - project_id = self.project_id + project_id = UNSET if isinstance(self.project_id, Unset) else self.project_id project_name: Union[None, Unset, str] - if isinstance(self.project_name, Unset): - project_name = UNSET - else: - project_name = self.project_name + project_name = UNSET if isinstance(self.project_name, Unset) else self.project_name stage_id: Union[None, Unset, str] - if isinstance(self.stage_id, Unset): - stage_id = UNSET - else: - stage_id = self.stage_id + stage_id = UNSET if isinstance(self.stage_id, Unset) else self.stage_id stage_name: Union[None, Unset, str] - if isinstance(self.stage_name, Unset): - stage_name = UNSET - else: - stage_name = self.stage_name + stage_name = UNSET if isinstance(self.stage_name, Unset) else self.stage_name stage_version: Union[None, Unset, int] - if isinstance(self.stage_version, Unset): - stage_version = UNSET - else: - stage_version = self.stage_version + stage_version = UNSET if isinstance(self.stage_version, Unset) else self.stage_version timeout = self.timeout @@ -152,9 +137,8 @@ def _parse_headers(data: object) -> Union["ProtectRequestHeadersType0", None, Un try: if not isinstance(data, dict): raise TypeError() - headers_type_0 = ProtectRequestHeadersType0.from_dict(data) + return ProtectRequestHeadersType0.from_dict(data) - return headers_type_0 except: # noqa: E722 pass return cast(Union["ProtectRequestHeadersType0", None, Unset], data) @@ -169,9 +153,8 @@ def _parse_metadata(data: object) -> Union["ProtectRequestMetadataType0", None, try: if not isinstance(data, dict): raise TypeError() - metadata_type_0 = ProtectRequestMetadataType0.from_dict(data) + return ProtectRequestMetadataType0.from_dict(data) - return metadata_type_0 except: # noqa: E722 pass return cast(Union["ProtectRequestMetadataType0", None, Unset], data) diff --git a/src/galileo/resources/models/protect_response.py b/src/galileo/resources/models/protect_response.py index 1d1d6b422..26dd71c3f 100644 --- a/src/galileo/resources/models/protect_response.py +++ b/src/galileo/resources/models/protect_response.py @@ -57,10 +57,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _status = d.pop("status", UNSET) status: Union[Unset, ExecutionStatus] - if isinstance(_status, Unset): - status = UNSET - else: - status = ExecutionStatus(_status) + status = UNSET if isinstance(_status, Unset) else ExecutionStatus(_status) protect_response = cls(text=text, trace_metadata=trace_metadata, status=status) diff --git a/src/galileo/resources/models/query_dataset_params.py b/src/galileo/resources/models/query_dataset_params.py index df084fab8..01db8e331 100644 --- a/src/galileo/resources/models/query_dataset_params.py +++ b/src/galileo/resources/models/query_dataset_params.py @@ -75,9 +75,8 @@ def _parse_sort(data: object) -> Union["DatasetContentSortClause", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - sort_type_0 = DatasetContentSortClause.from_dict(data) + return DatasetContentSortClause.from_dict(data) - return sort_type_0 except: # noqa: E722 pass return cast(Union["DatasetContentSortClause", None, Unset], data) diff --git a/src/galileo/resources/models/recompute_log_records_metrics_request.py b/src/galileo/resources/models/recompute_log_records_metrics_request.py index fe99ee11b..f7be4b395 100644 --- a/src/galileo/resources/models/recompute_log_records_metrics_request.py +++ b/src/galileo/resources/models/recompute_log_records_metrics_request.py @@ -67,23 +67,17 @@ def to_dict(self) -> dict[str, Any]: scorer_ids = self.scorer_ids experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id filters: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.filters, Unset): filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -93,16 +87,10 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id sort: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.sort, Unset): @@ -171,40 +159,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -232,10 +214,7 @@ def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: _sort = d.pop("sort", UNSET) sort: Union[Unset, LogRecordsSortClause] - if isinstance(_sort, Unset): - sort = UNSET - else: - sort = LogRecordsSortClause.from_dict(_sort) + sort = UNSET if isinstance(_sort, Unset) else LogRecordsSortClause.from_dict(_sort) starting_token = d.pop("starting_token", UNSET) diff --git a/src/galileo/resources/models/registered_scorer.py b/src/galileo/resources/models/registered_scorer.py index cf7766843..8cdda1a83 100644 --- a/src/galileo/resources/models/registered_scorer.py +++ b/src/galileo/resources/models/registered_scorer.py @@ -49,16 +49,10 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -95,16 +89,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/render_template_request.py b/src/galileo/resources/models/render_template_request.py index 9a5a91080..13b4706e3 100644 --- a/src/galileo/resources/models/render_template_request.py +++ b/src/galileo/resources/models/render_template_request.py @@ -28,10 +28,7 @@ def to_dict(self) -> dict[str, Any]: from ..models.dataset_data import DatasetData data: dict[str, Any] - if isinstance(self.data, DatasetData): - data = self.data.to_dict() - else: - data = self.data.to_dict() + data = self.data.to_dict() if isinstance(self.data, DatasetData) else self.data.to_dict() template = self.template @@ -52,16 +49,13 @@ def _parse_data(data: object) -> Union["DatasetData", "StringData"]: try: if not isinstance(data, dict): raise TypeError() - data_type_0 = DatasetData.from_dict(data) + return DatasetData.from_dict(data) - return data_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - data_type_1 = StringData.from_dict(data) - - return data_type_1 + return StringData.from_dict(data) data = _parse_data(d.pop("data")) diff --git a/src/galileo/resources/models/render_template_response.py b/src/galileo/resources/models/render_template_response.py index 3c2981419..cf8bed251 100644 --- a/src/galileo/resources/models/render_template_response.py +++ b/src/galileo/resources/models/render_template_response.py @@ -40,10 +40,7 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit next_starting_token: Union[None, Unset, int] - if isinstance(self.next_starting_token, Unset): - next_starting_token = UNSET - else: - next_starting_token = self.next_starting_token + next_starting_token = UNSET if isinstance(self.next_starting_token, Unset) else self.next_starting_token paginated = self.paginated diff --git a/src/galileo/resources/models/rendered_template.py b/src/galileo/resources/models/rendered_template.py index f4ed1c9d3..a192e3b70 100644 --- a/src/galileo/resources/models/rendered_template.py +++ b/src/galileo/resources/models/rendered_template.py @@ -25,10 +25,7 @@ def to_dict(self) -> dict[str, Any]: result = self.result warning: Union[None, Unset, str] - if isinstance(self.warning, Unset): - warning = UNSET - else: - warning = self.warning + warning = UNSET if isinstance(self.warning, Unset) else self.warning field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/retriever_span.py b/src/galileo/resources/models/retriever_span.py index 5fc70691b..56b178985 100644 --- a/src/galileo/resources/models/retriever_span.py +++ b/src/galileo/resources/models/retriever_span.py @@ -86,32 +86,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id input_ = self.input_ @@ -129,16 +117,10 @@ def to_dict(self) -> dict[str, Any]: output.append(output_item) parent_id: Union[None, Unset, str] - if isinstance(self.parent_id, Unset): - parent_id = UNSET - else: - parent_id = self.parent_id + parent_id = UNSET if isinstance(self.parent_id, Unset) else self.parent_id redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.redacted_output, Unset): @@ -153,23 +135,14 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, AgentSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, WorkflowSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, LlmSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, RetrieverSpan): + if isinstance(spans_item_data, (AgentSpan, WorkflowSpan, LlmSpan, RetrieverSpan)): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -177,26 +150,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -266,10 +230,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -318,10 +279,7 @@ def _parse_id(data: object) -> Union[None, Unset, str]: _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) name = d.pop("name", UNSET) @@ -391,40 +349,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = AgentSpan.from_dict(data) + return AgentSpan.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = WorkflowSpan.from_dict(data) + return WorkflowSpan.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = LlmSpan.from_dict(data) + return LlmSpan.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = RetrieverSpan.from_dict(data) + return RetrieverSpan.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ToolSpan.from_dict(data) - - return spans_item_type_4 + return ToolSpan.from_dict(data) spans_item = _parse_spans_item(spans_item_data) diff --git a/src/galileo/resources/models/rouge_scorer.py b/src/galileo/resources/models/rouge_scorer.py index 1d52290b9..52b5364e7 100644 --- a/src/galileo/resources/models/rouge_scorer.py +++ b/src/galileo/resources/models/rouge_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/rule.py b/src/galileo/resources/models/rule.py index 25ed4f1d7..ca4012662 100644 --- a/src/galileo/resources/models/rule.py +++ b/src/galileo/resources/models/rule.py @@ -29,11 +29,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value target_value: Union[None, float, int, list[Any], str] - if isinstance(self.target_value, list): - target_value = self.target_value - - else: - target_value = self.target_value + target_value = self.target_value if isinstance(self.target_value, list) else self.target_value field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -54,9 +50,8 @@ def _parse_target_value(data: object) -> Union[None, float, int, list[Any], str] try: if not isinstance(data, list): raise TypeError() - target_value_type_3 = cast(list[Any], data) + return cast(list[Any], data) - return target_value_type_3 except: # noqa: E722 pass return cast(Union[None, float, int, list[Any], str], data) diff --git a/src/galileo/resources/models/rule_result.py b/src/galileo/resources/models/rule_result.py index 4314e9d97..cd70173cf 100644 --- a/src/galileo/resources/models/rule_result.py +++ b/src/galileo/resources/models/rule_result.py @@ -37,27 +37,17 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value target_value: Union[None, float, int, list[Any], str] - if isinstance(self.target_value, list): - target_value = self.target_value - - else: - target_value = self.target_value + target_value = self.target_value if isinstance(self.target_value, list) else self.target_value execution_time: Union[None, Unset, float] - if isinstance(self.execution_time, Unset): - execution_time = UNSET - else: - execution_time = self.execution_time + execution_time = UNSET if isinstance(self.execution_time, Unset) else self.execution_time status: Union[Unset, str] = UNSET if not isinstance(self.status, Unset): status = self.status.value value: Union[Any, None, Unset] - if isinstance(self.value, Unset): - value = UNSET - else: - value = self.value + value = UNSET if isinstance(self.value, Unset) else self.value field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -84,9 +74,8 @@ def _parse_target_value(data: object) -> Union[None, float, int, list[Any], str] try: if not isinstance(data, list): raise TypeError() - target_value_type_3 = cast(list[Any], data) + return cast(list[Any], data) - return target_value_type_3 except: # noqa: E722 pass return cast(Union[None, float, int, list[Any], str], data) @@ -104,10 +93,7 @@ def _parse_execution_time(data: object) -> Union[None, Unset, float]: _status = d.pop("status", UNSET) status: Union[Unset, ExecutionStatus] - if isinstance(_status, Unset): - status = UNSET - else: - status = ExecutionStatus(_status) + status = UNSET if isinstance(_status, Unset) else ExecutionStatus(_status) def _parse_value(data: object) -> Union[Any, None, Unset]: if data is None: diff --git a/src/galileo/resources/models/ruleset.py b/src/galileo/resources/models/ruleset.py index 779125ea5..44fe4586b 100644 --- a/src/galileo/resources/models/ruleset.py +++ b/src/galileo/resources/models/ruleset.py @@ -41,10 +41,7 @@ def to_dict(self) -> dict[str, Any]: action = self.action.to_dict() description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description rules: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.rules, Unset): @@ -79,16 +76,13 @@ def _parse_action(data: object) -> Union["OverrideAction", "PassthroughAction", try: if not isinstance(data, dict): raise TypeError() - action_type_0 = OverrideAction.from_dict(data) + return OverrideAction.from_dict(data) - return action_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - action_type_1 = PassthroughAction.from_dict(data) - - return action_type_1 + return PassthroughAction.from_dict(data) action = _parse_action(d.pop("action", UNSET)) diff --git a/src/galileo/resources/models/ruleset_result.py b/src/galileo/resources/models/ruleset_result.py index c8f121c70..7eb01938f 100644 --- a/src/galileo/resources/models/ruleset_result.py +++ b/src/galileo/resources/models/ruleset_result.py @@ -47,10 +47,7 @@ def to_dict(self) -> dict[str, Any]: action = self.action.to_dict() description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description rule_results: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.rule_results, Unset): @@ -101,16 +98,13 @@ def _parse_action(data: object) -> Union["OverrideAction", "PassthroughAction", try: if not isinstance(data, dict): raise TypeError() - action_type_0 = OverrideAction.from_dict(data) + return OverrideAction.from_dict(data) - return action_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - action_type_1 = PassthroughAction.from_dict(data) - - return action_type_1 + return PassthroughAction.from_dict(data) action = _parse_action(d.pop("action", UNSET)) @@ -139,10 +133,7 @@ def _parse_description(data: object) -> Union[None, Unset, str]: _status = d.pop("status", UNSET) status: Union[Unset, ExecutionStatus] - if isinstance(_status, Unset): - status = UNSET - else: - status = ExecutionStatus(_status) + status = UNSET if isinstance(_status, Unset) else ExecutionStatus(_status) ruleset_result = cls( action=action, description=description, rule_results=rule_results, rules=rules, status=status diff --git a/src/galileo/resources/models/run_db.py b/src/galileo/resources/models/run_db.py index a595880c8..4d38d0a60 100644 --- a/src/galileo/resources/models/run_db.py +++ b/src/galileo/resources/models/run_db.py @@ -81,34 +81,19 @@ def to_dict(self) -> dict[str, Any]: winner = self.winner dataset_hash: Union[None, Unset, str] - if isinstance(self.dataset_hash, Unset): - dataset_hash = UNSET - else: - dataset_hash = self.dataset_hash + dataset_hash = UNSET if isinstance(self.dataset_hash, Unset) else self.dataset_hash dataset_version_id: Union[None, Unset, str] - if isinstance(self.dataset_version_id, Unset): - dataset_version_id = UNSET - else: - dataset_version_id = self.dataset_version_id + dataset_version_id = UNSET if isinstance(self.dataset_version_id, Unset) else self.dataset_version_id example_content_id: Union[None, Unset, str] - if isinstance(self.example_content_id, Unset): - example_content_id = UNSET - else: - example_content_id = self.example_content_id + example_content_id = UNSET if isinstance(self.example_content_id, Unset) else self.example_content_id name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name project_id: Union[None, Unset, str] - if isinstance(self.project_id, Unset): - project_id = UNSET - else: - project_id = self.project_id + project_id = UNSET if isinstance(self.project_id, Unset) else self.project_id run_tags: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.run_tags, Unset): @@ -244,9 +229,8 @@ def _parse_task_type(data: object) -> Union[None, TaskType, Unset]: try: if not isinstance(data, int): raise TypeError() - task_type_type_0 = TaskType(data) + return TaskType(data) - return task_type_type_0 except: # noqa: E722 pass return cast(Union[None, TaskType, Unset], data) diff --git a/src/galileo/resources/models/run_db_thin.py b/src/galileo/resources/models/run_db_thin.py index 7bb5ace16..a06d46daf 100644 --- a/src/galileo/resources/models/run_db_thin.py +++ b/src/galileo/resources/models/run_db_thin.py @@ -73,34 +73,19 @@ def to_dict(self) -> dict[str, Any]: winner = self.winner dataset_hash: Union[None, Unset, str] - if isinstance(self.dataset_hash, Unset): - dataset_hash = UNSET - else: - dataset_hash = self.dataset_hash + dataset_hash = UNSET if isinstance(self.dataset_hash, Unset) else self.dataset_hash dataset_version_id: Union[None, Unset, str] - if isinstance(self.dataset_version_id, Unset): - dataset_version_id = UNSET - else: - dataset_version_id = self.dataset_version_id + dataset_version_id = UNSET if isinstance(self.dataset_version_id, Unset) else self.dataset_version_id example_content_id: Union[None, Unset, str] - if isinstance(self.example_content_id, Unset): - example_content_id = UNSET - else: - example_content_id = self.example_content_id + example_content_id = UNSET if isinstance(self.example_content_id, Unset) else self.example_content_id name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name project_id: Union[None, Unset, str] - if isinstance(self.project_id, Unset): - project_id = UNSET - else: - project_id = self.project_id + project_id = UNSET if isinstance(self.project_id, Unset) else self.project_id run_tags: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.run_tags, Unset): @@ -230,9 +215,8 @@ def _parse_task_type(data: object) -> Union[None, TaskType, Unset]: try: if not isinstance(data, int): raise TypeError() - task_type_type_0 = TaskType(data) + return TaskType(data) - return task_type_type_0 except: # noqa: E722 pass return cast(Union[None, TaskType, Unset], data) diff --git a/src/galileo/resources/models/run_params_map.py b/src/galileo/resources/models/run_params_map.py index 3b3385eb8..5f1468749 100644 --- a/src/galileo/resources/models/run_params_map.py +++ b/src/galileo/resources/models/run_params_map.py @@ -59,118 +59,61 @@ class RunParamsMap: def to_dict(self) -> dict[str, Any]: api_version: Union[None, Unset, str] - if isinstance(self.api_version, Unset): - api_version = UNSET - else: - api_version = self.api_version + api_version = UNSET if isinstance(self.api_version, Unset) else self.api_version deployment_name: Union[None, Unset, str] - if isinstance(self.deployment_name, Unset): - deployment_name = UNSET - else: - deployment_name = self.deployment_name + deployment_name = UNSET if isinstance(self.deployment_name, Unset) else self.deployment_name echo: Union[None, Unset, str] - if isinstance(self.echo, Unset): - echo = UNSET - else: - echo = self.echo + echo = UNSET if isinstance(self.echo, Unset) else self.echo frequency_penalty: Union[None, Unset, str] - if isinstance(self.frequency_penalty, Unset): - frequency_penalty = UNSET - else: - frequency_penalty = self.frequency_penalty + frequency_penalty = UNSET if isinstance(self.frequency_penalty, Unset) else self.frequency_penalty logprobs: Union[None, Unset, str] - if isinstance(self.logprobs, Unset): - logprobs = UNSET - else: - logprobs = self.logprobs + logprobs = UNSET if isinstance(self.logprobs, Unset) else self.logprobs max_tokens: Union[None, Unset, str] - if isinstance(self.max_tokens, Unset): - max_tokens = UNSET - else: - max_tokens = self.max_tokens + max_tokens = UNSET if isinstance(self.max_tokens, Unset) else self.max_tokens model: Union[None, Unset, str] - if isinstance(self.model, Unset): - model = UNSET - else: - model = self.model + model = UNSET if isinstance(self.model, Unset) else self.model n: Union[None, Unset, str] - if isinstance(self.n, Unset): - n = UNSET - else: - n = self.n + n = UNSET if isinstance(self.n, Unset) else self.n presence_penalty: Union[None, Unset, str] - if isinstance(self.presence_penalty, Unset): - presence_penalty = UNSET - else: - presence_penalty = self.presence_penalty + presence_penalty = UNSET if isinstance(self.presence_penalty, Unset) else self.presence_penalty reasoning_effort: Union[None, Unset, str] - if isinstance(self.reasoning_effort, Unset): - reasoning_effort = UNSET - else: - reasoning_effort = self.reasoning_effort + reasoning_effort = UNSET if isinstance(self.reasoning_effort, Unset) else self.reasoning_effort response_format: Union[None, Unset, str] - if isinstance(self.response_format, Unset): - response_format = UNSET - else: - response_format = self.response_format + response_format = UNSET if isinstance(self.response_format, Unset) else self.response_format stop_sequences: Union[None, Unset, str] - if isinstance(self.stop_sequences, Unset): - stop_sequences = UNSET - else: - stop_sequences = self.stop_sequences + stop_sequences = UNSET if isinstance(self.stop_sequences, Unset) else self.stop_sequences temperature: Union[None, Unset, str] - if isinstance(self.temperature, Unset): - temperature = UNSET - else: - temperature = self.temperature + temperature = UNSET if isinstance(self.temperature, Unset) else self.temperature tool_choice: Union[None, Unset, str] - if isinstance(self.tool_choice, Unset): - tool_choice = UNSET - else: - tool_choice = self.tool_choice + tool_choice = UNSET if isinstance(self.tool_choice, Unset) else self.tool_choice tools: Union[None, Unset, str] - if isinstance(self.tools, Unset): - tools = UNSET - else: - tools = self.tools + tools = UNSET if isinstance(self.tools, Unset) else self.tools top_k: Union[None, Unset, str] - if isinstance(self.top_k, Unset): - top_k = UNSET - else: - top_k = self.top_k + top_k = UNSET if isinstance(self.top_k, Unset) else self.top_k top_logprobs: Union[None, Unset, str] - if isinstance(self.top_logprobs, Unset): - top_logprobs = UNSET - else: - top_logprobs = self.top_logprobs + top_logprobs = UNSET if isinstance(self.top_logprobs, Unset) else self.top_logprobs top_p: Union[None, Unset, str] - if isinstance(self.top_p, Unset): - top_p = UNSET - else: - top_p = self.top_p + top_p = UNSET if isinstance(self.top_p, Unset) else self.top_p verbosity: Union[None, Unset, str] - if isinstance(self.verbosity, Unset): - verbosity = UNSET - else: - verbosity = self.verbosity + verbosity = UNSET if isinstance(self.verbosity, Unset) else self.verbosity field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/scorer_config.py b/src/galileo/resources/models/scorer_config.py index b08de91ef..4545ff431 100644 --- a/src/galileo/resources/models/scorer_config.py +++ b/src/galileo/resources/models/scorer_config.py @@ -68,10 +68,7 @@ def to_dict(self) -> dict[str, Any]: scorer_type = self.scorer_type.value cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled filters: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.filters, Unset): @@ -99,10 +96,7 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name model_type: Union[None, Unset, str] if isinstance(self.model_type, Unset): @@ -113,16 +107,10 @@ def to_dict(self) -> dict[str, Any]: model_type = self.model_type name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -211,16 +199,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -241,9 +226,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -267,9 +251,8 @@ def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: try: if not isinstance(data, str): raise TypeError() - model_type_type_0 = ModelType(data) + return ModelType(data) - return model_type_type_0 except: # noqa: E722 pass return cast(Union[ModelType, None, Unset], data) @@ -302,9 +285,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -319,9 +301,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) @@ -336,9 +317,8 @@ def _parse_scorer_version(data: object) -> Union["BaseScorerVersionDB", None, Un try: if not isinstance(data, dict): raise TypeError() - scorer_version_type_0 = BaseScorerVersionDB.from_dict(data) + return BaseScorerVersionDB.from_dict(data) - return scorer_version_type_0 except: # noqa: E722 pass return cast(Union["BaseScorerVersionDB", None, Unset], data) diff --git a/src/galileo/resources/models/scorer_creator_filter.py b/src/galileo/resources/models/scorer_creator_filter.py index 8a42ba471..07babe9fd 100644 --- a/src/galileo/resources/models/scorer_creator_filter.py +++ b/src/galileo/resources/models/scorer_creator_filter.py @@ -26,11 +26,7 @@ class ScorerCreatorFilter: def to_dict(self) -> dict[str, Any]: value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value name = self.name @@ -56,9 +52,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) @@ -71,10 +66,7 @@ def _parse_value(data: object) -> Union[list[str], str]: _operator = d.pop("operator", UNSET) operator: Union[Unset, ScorerCreatorFilterOperator] - if isinstance(_operator, Unset): - operator = UNSET - else: - operator = ScorerCreatorFilterOperator(_operator) + operator = UNSET if isinstance(_operator, Unset) else ScorerCreatorFilterOperator(_operator) scorer_creator_filter = cls(value=value, name=name, operator=operator) diff --git a/src/galileo/resources/models/scorer_defaults.py b/src/galileo/resources/models/scorer_defaults.py index 5111448ec..bdfb10e60 100644 --- a/src/galileo/resources/models/scorer_defaults.py +++ b/src/galileo/resources/models/scorer_defaults.py @@ -47,10 +47,7 @@ def to_dict(self) -> dict[str, Any]: from ..models.node_name_filter import NodeNameFilter cot_enabled: Union[None, Unset, bool] - if isinstance(self.cot_enabled, Unset): - cot_enabled = UNSET - else: - cot_enabled = self.cot_enabled + cot_enabled = UNSET if isinstance(self.cot_enabled, Unset) else self.cot_enabled filters: Union[None, Unset, list[dict[str, Any]]] if isinstance(self.filters, Unset): @@ -78,16 +75,10 @@ def to_dict(self) -> dict[str, Any]: input_type = self.input_type model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges output_type: Union[None, Unset, str] if isinstance(self.output_type, Unset): @@ -158,16 +149,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -188,9 +176,8 @@ def _parse_input_type(data: object) -> Union[InputTypeEnum, None, Unset]: try: if not isinstance(data, str): raise TypeError() - input_type_type_0 = InputTypeEnum(data) + return InputTypeEnum(data) - return input_type_type_0 except: # noqa: E722 pass return cast(Union[InputTypeEnum, None, Unset], data) @@ -223,9 +210,8 @@ def _parse_output_type(data: object) -> Union[None, OutputTypeEnum, Unset]: try: if not isinstance(data, str): raise TypeError() - output_type_type_0 = OutputTypeEnum(data) + return OutputTypeEnum(data) - return output_type_type_0 except: # noqa: E722 pass return cast(Union[None, OutputTypeEnum, Unset], data) @@ -240,9 +226,8 @@ def _parse_scoreable_node_types(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - scoreable_node_types_type_0 = cast(list[str], data) + return cast(list[str], data) - return scoreable_node_types_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/scorer_name_filter.py b/src/galileo/resources/models/scorer_name_filter.py index 51df3876c..0bf0b2043 100644 --- a/src/galileo/resources/models/scorer_name_filter.py +++ b/src/galileo/resources/models/scorer_name_filter.py @@ -30,11 +30,7 @@ def to_dict(self) -> dict[str, Any]: operator = self.operator.value value: Union[list[str], str] - if isinstance(self.value, list): - value = self.value - - else: - value = self.value + value = self.value if isinstance(self.value, list) else self.value case_sensitive = self.case_sensitive @@ -59,9 +55,8 @@ def _parse_value(data: object) -> Union[list[str], str]: try: if not isinstance(data, list): raise TypeError() - value_type_1 = cast(list[str], data) + return cast(list[str], data) - return value_type_1 except: # noqa: E722 pass return cast(Union[list[str], str], data) diff --git a/src/galileo/resources/models/scorer_response.py b/src/galileo/resources/models/scorer_response.py index 39844a10e..da05746c9 100644 --- a/src/galileo/resources/models/scorer_response.py +++ b/src/galileo/resources/models/scorer_response.py @@ -12,7 +12,6 @@ if TYPE_CHECKING: from ..models.base_scorer_version_db import BaseScorerVersionDB - from ..models.chain_poll_template import ChainPollTemplate from ..models.scorer_defaults import ScorerDefaults @@ -27,7 +26,6 @@ class ScorerResponse: name (str): scorer_type (ScorerTypes): tags (list[str]): - chain_poll_template (Union['ChainPollTemplate', None, Unset]): created_at (Union[None, Unset, datetime.datetime]): created_by (Union[None, Unset, str]): default_version (Union['BaseScorerVersionDB', None, Unset]): @@ -39,16 +37,15 @@ class ScorerResponse: num_judges, etc. This enables the ui to know which fields a user can configure when they're setting a scorer label (Union[Unset, str]): Default: ''. latest_version (Union['BaseScorerVersionDB', None, Unset]): - metric_name (Union[None, Unset, str]): model_type (Union[ModelType, None, Unset]): updated_at (Union[None, Unset, datetime.datetime]): + user_prompt (Union[None, Unset, str]): """ id: str name: str scorer_type: ScorerTypes tags: list[str] - chain_poll_template: Union["ChainPollTemplate", None, Unset] = UNSET created_at: Union[None, Unset, datetime.datetime] = UNSET created_by: Union[None, Unset, str] = UNSET default_version: Union["BaseScorerVersionDB", None, Unset] = UNSET @@ -59,14 +56,13 @@ class ScorerResponse: included_fields: Union[Unset, list[str]] = UNSET label: Union[Unset, str] = "" latest_version: Union["BaseScorerVersionDB", None, Unset] = UNSET - metric_name: Union[None, Unset, str] = UNSET model_type: Union[ModelType, None, Unset] = UNSET updated_at: Union[None, Unset, datetime.datetime] = UNSET + user_prompt: Union[None, Unset, str] = UNSET additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict) def to_dict(self) -> dict[str, Any]: from ..models.base_scorer_version_db import BaseScorerVersionDB - from ..models.chain_poll_template import ChainPollTemplate from ..models.scorer_defaults import ScorerDefaults id = self.id @@ -77,14 +73,6 @@ def to_dict(self) -> dict[str, Any]: tags = self.tags - chain_poll_template: Union[None, Unset, dict[str, Any]] - if isinstance(self.chain_poll_template, Unset): - chain_poll_template = UNSET - elif isinstance(self.chain_poll_template, ChainPollTemplate): - chain_poll_template = self.chain_poll_template.to_dict() - else: - chain_poll_template = self.chain_poll_template - created_at: Union[None, Unset, str] if isinstance(self.created_at, Unset): created_at = UNSET @@ -94,10 +82,7 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at created_by: Union[None, Unset, str] - if isinstance(self.created_by, Unset): - created_by = UNSET - else: - created_by = self.created_by + created_by = UNSET if isinstance(self.created_by, Unset) else self.created_by default_version: Union[None, Unset, dict[str, Any]] if isinstance(self.default_version, Unset): @@ -108,10 +93,7 @@ def to_dict(self) -> dict[str, Any]: default_version = self.default_version default_version_id: Union[None, Unset, str] - if isinstance(self.default_version_id, Unset): - default_version_id = UNSET - else: - default_version_id = self.default_version_id + default_version_id = UNSET if isinstance(self.default_version_id, Unset) else self.default_version_id defaults: Union[None, Unset, dict[str, Any]] if isinstance(self.defaults, Unset): @@ -122,16 +104,10 @@ def to_dict(self) -> dict[str, Any]: defaults = self.defaults description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth included_fields: Union[Unset, list[str]] = UNSET if not isinstance(self.included_fields, Unset): @@ -147,12 +123,6 @@ def to_dict(self) -> dict[str, Any]: else: latest_version = self.latest_version - metric_name: Union[None, Unset, str] - if isinstance(self.metric_name, Unset): - metric_name = UNSET - else: - metric_name = self.metric_name - model_type: Union[None, Unset, str] if isinstance(self.model_type, Unset): model_type = UNSET @@ -169,11 +139,12 @@ def to_dict(self) -> dict[str, Any]: else: updated_at = self.updated_at + user_prompt: Union[None, Unset, str] + user_prompt = UNSET if isinstance(self.user_prompt, Unset) else self.user_prompt + field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) field_dict.update({"id": id, "name": name, "scorer_type": scorer_type, "tags": tags}) - if chain_poll_template is not UNSET: - field_dict["chain_poll_template"] = chain_poll_template if created_at is not UNSET: field_dict["created_at"] = created_at if created_by is not UNSET: @@ -194,19 +165,18 @@ def to_dict(self) -> dict[str, Any]: field_dict["label"] = label if latest_version is not UNSET: field_dict["latest_version"] = latest_version - if metric_name is not UNSET: - field_dict["metric_name"] = metric_name if model_type is not UNSET: field_dict["model_type"] = model_type if updated_at is not UNSET: field_dict["updated_at"] = updated_at + if user_prompt is not UNSET: + field_dict["user_prompt"] = user_prompt return field_dict @classmethod def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: from ..models.base_scorer_version_db import BaseScorerVersionDB - from ..models.chain_poll_template import ChainPollTemplate from ..models.scorer_defaults import ScorerDefaults d = dict(src_dict) @@ -218,23 +188,6 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: tags = cast(list[str], d.pop("tags")) - def _parse_chain_poll_template(data: object) -> Union["ChainPollTemplate", None, Unset]: - if data is None: - return data - if isinstance(data, Unset): - return data - try: - if not isinstance(data, dict): - raise TypeError() - chain_poll_template_type_0 = ChainPollTemplate.from_dict(data) - - return chain_poll_template_type_0 - except: # noqa: E722 - pass - return cast(Union["ChainPollTemplate", None, Unset], data) - - chain_poll_template = _parse_chain_poll_template(d.pop("chain_poll_template", UNSET)) - def _parse_created_at(data: object) -> Union[None, Unset, datetime.datetime]: if data is None: return data @@ -243,9 +196,8 @@ def _parse_created_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - created_at_type_0 = isoparse(data) + return isoparse(data) - return created_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) @@ -269,9 +221,8 @@ def _parse_default_version(data: object) -> Union["BaseScorerVersionDB", None, U try: if not isinstance(data, dict): raise TypeError() - default_version_type_0 = BaseScorerVersionDB.from_dict(data) + return BaseScorerVersionDB.from_dict(data) - return default_version_type_0 except: # noqa: E722 pass return cast(Union["BaseScorerVersionDB", None, Unset], data) @@ -295,9 +246,8 @@ def _parse_defaults(data: object) -> Union["ScorerDefaults", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - defaults_type_0 = ScorerDefaults.from_dict(data) + return ScorerDefaults.from_dict(data) - return defaults_type_0 except: # noqa: E722 pass return cast(Union["ScorerDefaults", None, Unset], data) @@ -334,24 +284,14 @@ def _parse_latest_version(data: object) -> Union["BaseScorerVersionDB", None, Un try: if not isinstance(data, dict): raise TypeError() - latest_version_type_0 = BaseScorerVersionDB.from_dict(data) + return BaseScorerVersionDB.from_dict(data) - return latest_version_type_0 except: # noqa: E722 pass return cast(Union["BaseScorerVersionDB", None, Unset], data) latest_version = _parse_latest_version(d.pop("latest_version", UNSET)) - def _parse_metric_name(data: object) -> Union[None, Unset, str]: - if data is None: - return data - if isinstance(data, Unset): - return data - return cast(Union[None, Unset, str], data) - - metric_name = _parse_metric_name(d.pop("metric_name", UNSET)) - def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: if data is None: return data @@ -360,9 +300,8 @@ def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: try: if not isinstance(data, str): raise TypeError() - model_type_type_0 = ModelType(data) + return ModelType(data) - return model_type_type_0 except: # noqa: E722 pass return cast(Union[ModelType, None, Unset], data) @@ -377,21 +316,28 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: try: if not isinstance(data, str): raise TypeError() - updated_at_type_0 = isoparse(data) + return isoparse(data) - return updated_at_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, datetime.datetime], data) updated_at = _parse_updated_at(d.pop("updated_at", UNSET)) + def _parse_user_prompt(data: object) -> Union[None, Unset, str]: + if data is None: + return data + if isinstance(data, Unset): + return data + return cast(Union[None, Unset, str], data) + + user_prompt = _parse_user_prompt(d.pop("user_prompt", UNSET)) + scorer_response = cls( id=id, name=name, scorer_type=scorer_type, tags=tags, - chain_poll_template=chain_poll_template, created_at=created_at, created_by=created_by, default_version=default_version, @@ -402,9 +348,9 @@ def _parse_updated_at(data: object) -> Union[None, Unset, datetime.datetime]: included_fields=included_fields, label=label, latest_version=latest_version, - metric_name=metric_name, model_type=model_type, updated_at=updated_at, + user_prompt=user_prompt, ) scorer_response.additional_properties = d diff --git a/src/galileo/resources/models/scorer_type_filter.py b/src/galileo/resources/models/scorer_type_filter.py index 1a29809c8..47b4de842 100644 --- a/src/galileo/resources/models/scorer_type_filter.py +++ b/src/galileo/resources/models/scorer_type_filter.py @@ -56,9 +56,8 @@ def _parse_value(data: object) -> Union[ScorerTypes, list[ScorerTypes]]: try: if not isinstance(data, str): raise TypeError() - value_type_0 = ScorerTypes(data) + return ScorerTypes(data) - return value_type_0 except: # noqa: E722 pass if not isinstance(data, list): diff --git a/src/galileo/resources/models/segment.py b/src/galileo/resources/models/segment.py index 7802a03ed..cf1e83c32 100644 --- a/src/galileo/resources/models/segment.py +++ b/src/galileo/resources/models/segment.py @@ -34,10 +34,7 @@ def to_dict(self) -> dict[str, Any]: value = self.value prob: Union[None, Unset, float] - if isinstance(self.prob, Unset): - prob = UNSET - else: - prob = self.prob + prob = UNSET if isinstance(self.prob, Unset) else self.prob field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/segment_filter.py b/src/galileo/resources/models/segment_filter.py index 68d8bb1c4..8b77572a3 100644 --- a/src/galileo/resources/models/segment_filter.py +++ b/src/galileo/resources/models/segment_filter.py @@ -38,9 +38,7 @@ def to_dict(self) -> dict[str, Any]: filter_: Union[None, Unset, dict[str, Any]] if isinstance(self.filter_, Unset): filter_ = UNSET - elif isinstance(self.filter_, NodeNameFilter): - filter_ = self.filter_.to_dict() - elif isinstance(self.filter_, MetadataFilter): + elif isinstance(self.filter_, (NodeNameFilter, MetadataFilter)): filter_ = self.filter_.to_dict() else: filter_ = self.filter_ @@ -73,17 +71,15 @@ def _parse_filter_(data: object) -> Union["MetadataFilter", "NodeNameFilter", No try: if not isinstance(data, dict): raise TypeError() - filter_type_0_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filter_type_0_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filter_type_0_type_1 = MetadataFilter.from_dict(data) + return MetadataFilter.from_dict(data) - return filter_type_0_type_1 except: # noqa: E722 pass return cast(Union["MetadataFilter", "NodeNameFilter", None, Unset], data) diff --git a/src/galileo/resources/models/session_create_request.py b/src/galileo/resources/models/session_create_request.py index 3319a610e..9bbd7732e 100644 --- a/src/galileo/resources/models/session_create_request.py +++ b/src/galileo/resources/models/session_create_request.py @@ -31,40 +31,22 @@ class SessionCreateRequest: def to_dict(self) -> dict[str, Any]: experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name previous_session_id: Union[None, Unset, str] - if isinstance(self.previous_session_id, Unset): - previous_session_id = UNSET - else: - previous_session_id = self.previous_session_id + previous_session_id = UNSET if isinstance(self.previous_session_id, Unset) else self.previous_session_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/session_create_response.py b/src/galileo/resources/models/session_create_response.py index ecfd5a820..c819f8f92 100644 --- a/src/galileo/resources/models/session_create_response.py +++ b/src/galileo/resources/models/session_create_response.py @@ -40,16 +40,10 @@ def to_dict(self) -> dict[str, Any]: project_name = self.project_name external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id previous_session_id: Union[None, Unset, str] - if isinstance(self.previous_session_id, Unset): - previous_session_id = UNSET - else: - previous_session_id = self.previous_session_id + previous_session_id = UNSET if isinstance(self.previous_session_id, Unset) else self.previous_session_id field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/stage_db.py b/src/galileo/resources/models/stage_db.py index d89f0ee97..27c2925b9 100644 --- a/src/galileo/resources/models/stage_db.py +++ b/src/galileo/resources/models/stage_db.py @@ -45,10 +45,7 @@ def to_dict(self) -> dict[str, Any]: project_id = self.project_id description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description paused = self.paused @@ -57,10 +54,7 @@ def to_dict(self) -> dict[str, Any]: type_ = self.type_.value version: Union[None, Unset, int] - if isinstance(self.version, Unset): - version = UNSET - else: - version = self.version + version = UNSET if isinstance(self.version, Unset) else self.version field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) @@ -100,10 +94,7 @@ def _parse_description(data: object) -> Union[None, Unset, str]: _type_ = d.pop("type", UNSET) type_: Union[Unset, StageType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = StageType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else StageType(_type_) def _parse_version(data: object) -> Union[None, Unset, int]: if data is None: diff --git a/src/galileo/resources/models/stage_with_rulesets.py b/src/galileo/resources/models/stage_with_rulesets.py index c49b5d076..a45e19391 100644 --- a/src/galileo/resources/models/stage_with_rulesets.py +++ b/src/galileo/resources/models/stage_with_rulesets.py @@ -41,10 +41,7 @@ def to_dict(self) -> dict[str, Any]: project_id = self.project_id description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description paused = self.paused @@ -102,10 +99,7 @@ def _parse_description(data: object) -> Union[None, Unset, str]: _type_ = d.pop("type", UNSET) type_: Union[Unset, StageType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = StageType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else StageType(_type_) stage_with_rulesets = cls( name=name, diff --git a/src/galileo/resources/models/synthetic_data_source_dataset.py b/src/galileo/resources/models/synthetic_data_source_dataset.py index ecf04371d..bb347b860 100644 --- a/src/galileo/resources/models/synthetic_data_source_dataset.py +++ b/src/galileo/resources/models/synthetic_data_source_dataset.py @@ -28,10 +28,7 @@ def to_dict(self) -> dict[str, Any]: dataset_id = self.dataset_id dataset_version_index: Union[None, Unset, int] - if isinstance(self.dataset_version_index, Unset): - dataset_version_index = UNSET - else: - dataset_version_index = self.dataset_version_index + dataset_version_index = UNSET if isinstance(self.dataset_version_index, Unset) else self.dataset_version_index row_ids: Union[None, Unset, list[str]] if isinstance(self.row_ids, Unset): @@ -74,9 +71,8 @@ def _parse_row_ids(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - row_ids_type_0 = cast(list[str], data) + return cast(list[str], data) - return row_ids_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) diff --git a/src/galileo/resources/models/synthetic_dataset_extension_request.py b/src/galileo/resources/models/synthetic_dataset_extension_request.py index 5f19a7096..3007713a6 100644 --- a/src/galileo/resources/models/synthetic_dataset_extension_request.py +++ b/src/galileo/resources/models/synthetic_dataset_extension_request.py @@ -60,16 +60,10 @@ def to_dict(self) -> dict[str, Any]: examples = self.examples instructions: Union[None, Unset, str] - if isinstance(self.instructions, Unset): - instructions = UNSET - else: - instructions = self.instructions + instructions = UNSET if isinstance(self.instructions, Unset) else self.instructions prompt: Union[None, Unset, str] - if isinstance(self.prompt, Unset): - prompt = UNSET - else: - prompt = self.prompt + prompt = UNSET if isinstance(self.prompt, Unset) else self.prompt prompt_settings: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.prompt_settings, Unset): @@ -168,9 +162,8 @@ def _parse_source_dataset(data: object) -> Union["SyntheticDataSourceDataset", N try: if not isinstance(data, dict): raise TypeError() - source_dataset_type_0 = SyntheticDataSourceDataset.from_dict(data) + return SyntheticDataSourceDataset.from_dict(data) - return source_dataset_type_0 except: # noqa: E722 pass return cast(Union["SyntheticDataSourceDataset", None, Unset], data) diff --git a/src/galileo/resources/models/system_metric_info.py b/src/galileo/resources/models/system_metric_info.py index b3dc6969a..c6c8b653f 100644 --- a/src/galileo/resources/models/system_metric_info.py +++ b/src/galileo/resources/models/system_metric_info.py @@ -64,52 +64,28 @@ def to_dict(self) -> dict[str, Any]: histogram = self.histogram max_: Union[None, Unset, float] - if isinstance(self.max_, Unset): - max_ = UNSET - else: - max_ = self.max_ + max_ = UNSET if isinstance(self.max_, Unset) else self.max_ mean: Union[None, Unset, float] - if isinstance(self.mean, Unset): - mean = UNSET - else: - mean = self.mean + mean = UNSET if isinstance(self.mean, Unset) else self.mean median: Union[None, Unset, float] - if isinstance(self.median, Unset): - median = UNSET - else: - median = self.median + median = UNSET if isinstance(self.median, Unset) else self.median min_: Union[None, Unset, float] - if isinstance(self.min_, Unset): - min_ = UNSET - else: - min_ = self.min_ + min_ = UNSET if isinstance(self.min_, Unset) else self.min_ p25: Union[None, Unset, float] - if isinstance(self.p25, Unset): - p25 = UNSET - else: - p25 = self.p25 + p25 = UNSET if isinstance(self.p25, Unset) else self.p25 p5: Union[None, Unset, float] - if isinstance(self.p5, Unset): - p5 = UNSET - else: - p5 = self.p5 + p5 = UNSET if isinstance(self.p5, Unset) else self.p5 p75: Union[None, Unset, float] - if isinstance(self.p75, Unset): - p75 = UNSET - else: - p75 = self.p75 + p75 = UNSET if isinstance(self.p75, Unset) else self.p75 p95: Union[None, Unset, float] - if isinstance(self.p95, Unset): - p95 = UNSET - else: - p95 = self.p95 + p95 = UNSET if isinstance(self.p95, Unset) else self.p95 unit: Union[None, Unset, str] if isinstance(self.unit, Unset): @@ -168,9 +144,8 @@ def _parse_histogram(data: object) -> Union["Histogram", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - histogram_type_0 = Histogram.from_dict(data) + return Histogram.from_dict(data) - return histogram_type_0 except: # noqa: E722 pass return cast(Union["Histogram", None, Unset], data) @@ -257,9 +232,8 @@ def _parse_unit(data: object) -> Union[DataUnit, None, Unset]: try: if not isinstance(data, str): raise TypeError() - unit_type_0 = DataUnit(data) + return DataUnit(data) - return unit_type_0 except: # noqa: E722 pass return cast(Union[DataUnit, None, Unset], data) diff --git a/src/galileo/resources/models/tool_error_rate_scorer.py b/src/galileo/resources/models/tool_error_rate_scorer.py index b0b195d99..45c21a0fd 100644 --- a/src/galileo/resources/models/tool_error_rate_scorer.py +++ b/src/galileo/resources/models/tool_error_rate_scorer.py @@ -53,10 +53,7 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name @@ -101,16 +98,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -138,10 +132,7 @@ def _parse_model_name(data: object) -> Union[None, Unset, str]: _type_ = d.pop("type", UNSET) type_: Union[Unset, ToolErrorRateScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = ToolErrorRateScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else ToolErrorRateScorerType(_type_) tool_error_rate_scorer = cls(filters=filters, model_name=model_name, name=name, type_=type_) diff --git a/src/galileo/resources/models/tool_selection_quality_scorer.py b/src/galileo/resources/models/tool_selection_quality_scorer.py index 34dc35988..7de7c8e1d 100644 --- a/src/galileo/resources/models/tool_selection_quality_scorer.py +++ b/src/galileo/resources/models/tool_selection_quality_scorer.py @@ -55,18 +55,12 @@ def to_dict(self) -> dict[str, Any]: filters = self.filters model_name: Union[None, Unset, str] - if isinstance(self.model_name, Unset): - model_name = UNSET - else: - model_name = self.model_name + model_name = UNSET if isinstance(self.model_name, Unset) else self.model_name name = self.name num_judges: Union[None, Unset, int] - if isinstance(self.num_judges, Unset): - num_judges = UNSET - else: - num_judges = self.num_judges + num_judges = UNSET if isinstance(self.num_judges, Unset) else self.num_judges type_: Union[Unset, str] = UNSET if not isinstance(self.type_, Unset): @@ -111,16 +105,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) @@ -157,10 +148,7 @@ def _parse_num_judges(data: object) -> Union[None, Unset, int]: _type_ = d.pop("type", UNSET) type_: Union[Unset, ToolSelectionQualityScorerType] - if isinstance(_type_, Unset): - type_ = UNSET - else: - type_ = ToolSelectionQualityScorerType(_type_) + type_ = UNSET if isinstance(_type_, Unset) else ToolSelectionQualityScorerType(_type_) tool_selection_quality_scorer = cls( filters=filters, model_name=model_name, name=name, num_judges=num_judges, type_=type_ diff --git a/src/galileo/resources/models/tool_span.py b/src/galileo/resources/models/tool_span.py index 76c4f15fd..a9cab4e0a 100644 --- a/src/galileo/resources/models/tool_span.py +++ b/src/galileo/resources/models/tool_span.py @@ -87,32 +87,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id input_ = self.input_ @@ -123,47 +111,26 @@ def to_dict(self) -> dict[str, Any]: name = self.name output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output parent_id: Union[None, Unset, str] - if isinstance(self.parent_id, Unset): - parent_id = UNSET - else: - parent_id = self.parent_id + parent_id = UNSET if isinstance(self.parent_id, Unset) else self.parent_id redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, str] - if isinstance(self.redacted_output, Unset): - redacted_output = UNSET - else: - redacted_output = self.redacted_output + redacted_output = UNSET if isinstance(self.redacted_output, Unset) else self.redacted_output session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, AgentSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, WorkflowSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, LlmSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, RetrieverSpan): + if isinstance(spans_item_data, (AgentSpan, WorkflowSpan, LlmSpan, RetrieverSpan)): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -171,32 +138,20 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags tool_call_id: Union[None, Unset, str] - if isinstance(self.tool_call_id, Unset): - tool_call_id = UNSET - else: - tool_call_id = self.tool_call_id + tool_call_id = UNSET if isinstance(self.tool_call_id, Unset) else self.tool_call_id trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -267,10 +222,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -319,10 +271,7 @@ def _parse_id(data: object) -> Union[None, Unset, str]: _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) name = d.pop("name", UNSET) @@ -381,40 +330,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = AgentSpan.from_dict(data) + return AgentSpan.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = WorkflowSpan.from_dict(data) + return WorkflowSpan.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = LlmSpan.from_dict(data) + return LlmSpan.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = RetrieverSpan.from_dict(data) + return RetrieverSpan.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ToolSpan.from_dict(data) - - return spans_item_type_4 + return ToolSpan.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -464,10 +407,7 @@ def _parse_trace_id(data: object) -> Union[None, Unset, str]: _user_metadata = d.pop("user_metadata", UNSET) user_metadata: Union[Unset, ToolSpanUserMetadata] - if isinstance(_user_metadata, Unset): - user_metadata = UNSET - else: - user_metadata = ToolSpanUserMetadata.from_dict(_user_metadata) + user_metadata = UNSET if isinstance(_user_metadata, Unset) else ToolSpanUserMetadata.from_dict(_user_metadata) tool_span = cls( created_at=created_at, diff --git a/src/galileo/resources/models/trace.py b/src/galileo/resources/models/trace.py index 80d59aa69..55243817e 100644 --- a/src/galileo/resources/models/trace.py +++ b/src/galileo/resources/models/trace.py @@ -86,32 +86,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id input_ = self.input_ @@ -122,47 +110,26 @@ def to_dict(self) -> dict[str, Any]: name = self.name output: Union[None, Unset, str] - if isinstance(self.output, Unset): - output = UNSET - else: - output = self.output + output = UNSET if isinstance(self.output, Unset) else self.output parent_id: Union[None, Unset, str] - if isinstance(self.parent_id, Unset): - parent_id = UNSET - else: - parent_id = self.parent_id + parent_id = UNSET if isinstance(self.parent_id, Unset) else self.parent_id redacted_input: Union[None, Unset, str] - if isinstance(self.redacted_input, Unset): - redacted_input = UNSET - else: - redacted_input = self.redacted_input + redacted_input = UNSET if isinstance(self.redacted_input, Unset) else self.redacted_input redacted_output: Union[None, Unset, str] - if isinstance(self.redacted_output, Unset): - redacted_output = UNSET - else: - redacted_output = self.redacted_output + redacted_output = UNSET if isinstance(self.redacted_output, Unset) else self.redacted_output session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, AgentSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, WorkflowSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, LlmSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, RetrieverSpan): + if isinstance(spans_item_data, (AgentSpan, WorkflowSpan, LlmSpan, RetrieverSpan)): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -170,26 +137,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -259,10 +217,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -311,10 +266,7 @@ def _parse_id(data: object) -> Union[None, Unset, str]: _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) name = d.pop("name", UNSET) @@ -373,40 +325,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = AgentSpan.from_dict(data) + return AgentSpan.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = WorkflowSpan.from_dict(data) + return WorkflowSpan.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = LlmSpan.from_dict(data) + return LlmSpan.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = RetrieverSpan.from_dict(data) + return RetrieverSpan.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ToolSpan.from_dict(data) - - return spans_item_type_4 + return ToolSpan.from_dict(data) spans_item = _parse_spans_item(spans_item_data) @@ -447,10 +393,7 @@ def _parse_trace_id(data: object) -> Union[None, Unset, str]: _user_metadata = d.pop("user_metadata", UNSET) user_metadata: Union[Unset, TraceUserMetadata] - if isinstance(_user_metadata, Unset): - user_metadata = UNSET - else: - user_metadata = TraceUserMetadata.from_dict(_user_metadata) + user_metadata = UNSET if isinstance(_user_metadata, Unset) else TraceUserMetadata.from_dict(_user_metadata) trace = cls( created_at=created_at, diff --git a/src/galileo/resources/models/uncertainty_scorer.py b/src/galileo/resources/models/uncertainty_scorer.py index bfd57fdab..67dbe61ce 100644 --- a/src/galileo/resources/models/uncertainty_scorer.py +++ b/src/galileo/resources/models/uncertainty_scorer.py @@ -82,16 +82,13 @@ def _parse_filters_type_0_item(data: object) -> Union["MetadataFilter", "NodeNam try: if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_0 = NodeNameFilter.from_dict(data) + return NodeNameFilter.from_dict(data) - return filters_type_0_item_type_0 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_type_0_item_type_1 = MetadataFilter.from_dict(data) - - return filters_type_0_item_type_1 + return MetadataFilter.from_dict(data) filters_type_0_item = _parse_filters_type_0_item(filters_type_0_item_data) diff --git a/src/galileo/resources/models/update_dataset_content_request.py b/src/galileo/resources/models/update_dataset_content_request.py index b81c4f817..d03e2657a 100644 --- a/src/galileo/resources/models/update_dataset_content_request.py +++ b/src/galileo/resources/models/update_dataset_content_request.py @@ -44,13 +44,7 @@ def to_dict(self) -> dict[str, Any]: edits = [] for edits_item_data in self.edits: edits_item: dict[str, Any] - if isinstance(edits_item_data, DatasetPrependRow): - edits_item = edits_item_data.to_dict() - elif isinstance(edits_item_data, DatasetAppendRow): - edits_item = edits_item_data.to_dict() - elif isinstance(edits_item_data, DatasetUpdateRow): - edits_item = edits_item_data.to_dict() - elif isinstance(edits_item_data, DatasetDeleteRow): + if isinstance(edits_item_data, (DatasetPrependRow, DatasetAppendRow, DatasetUpdateRow, DatasetDeleteRow)): edits_item = edits_item_data.to_dict() else: edits_item = edits_item_data.to_dict() @@ -84,40 +78,34 @@ def _parse_edits_item( try: if not isinstance(data, dict): raise TypeError() - edits_item_type_0 = DatasetPrependRow.from_dict(data) + return DatasetPrependRow.from_dict(data) - return edits_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - edits_item_type_1 = DatasetAppendRow.from_dict(data) + return DatasetAppendRow.from_dict(data) - return edits_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - edits_item_type_2 = DatasetUpdateRow.from_dict(data) + return DatasetUpdateRow.from_dict(data) - return edits_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - edits_item_type_3 = DatasetDeleteRow.from_dict(data) + return DatasetDeleteRow.from_dict(data) - return edits_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - edits_item_type_4 = DatasetFilterRows.from_dict(data) - - return edits_item_type_4 + return DatasetFilterRows.from_dict(data) edits_item = _parse_edits_item(edits_item_data) diff --git a/src/galileo/resources/models/update_dataset_request.py b/src/galileo/resources/models/update_dataset_request.py index eca5b4547..11e8b77aa 100644 --- a/src/galileo/resources/models/update_dataset_request.py +++ b/src/galileo/resources/models/update_dataset_request.py @@ -41,10 +41,7 @@ def to_dict(self) -> dict[str, Any]: column_mapping = self.column_mapping draft: Union[None, Unset, bool] - if isinstance(self.draft, Unset): - draft = UNSET - else: - draft = self.draft + draft = UNSET if isinstance(self.draft, Unset) else self.draft name: Union[None, Unset, dict[str, Any], str] if isinstance(self.name, Unset): @@ -81,9 +78,8 @@ def _parse_column_mapping(data: object) -> Union["ColumnMapping", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - column_mapping_type_0 = ColumnMapping.from_dict(data) + return ColumnMapping.from_dict(data) - return column_mapping_type_0 except: # noqa: E722 pass return cast(Union["ColumnMapping", None, Unset], data) @@ -107,9 +103,8 @@ def _parse_name(data: object) -> Union["Name", None, Unset, str]: try: if not isinstance(data, dict): raise TypeError() - name_type_1 = Name.from_dict(data) + return Name.from_dict(data) - return name_type_1 except: # noqa: E722 pass return cast(Union["Name", None, Unset, str], data) diff --git a/src/galileo/resources/models/update_dataset_version_request.py b/src/galileo/resources/models/update_dataset_version_request.py index dadee8b14..e128bdbb5 100644 --- a/src/galileo/resources/models/update_dataset_version_request.py +++ b/src/galileo/resources/models/update_dataset_version_request.py @@ -21,10 +21,7 @@ class UpdateDatasetVersionRequest: def to_dict(self) -> dict[str, Any]: name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/update_prompt_template_request.py b/src/galileo/resources/models/update_prompt_template_request.py index 3b19fbb91..5353c65b0 100644 --- a/src/galileo/resources/models/update_prompt_template_request.py +++ b/src/galileo/resources/models/update_prompt_template_request.py @@ -56,9 +56,8 @@ def _parse_name(data: object) -> Union["Name", None, Unset, str]: try: if not isinstance(data, dict): raise TypeError() - name_type_1 = Name.from_dict(data) + return Name.from_dict(data) - return name_type_1 except: # noqa: E722 pass return cast(Union["Name", None, Unset, str], data) diff --git a/src/galileo/resources/models/update_scorer_request.py b/src/galileo/resources/models/update_scorer_request.py index 31ddecd37..3cde3beee 100644 --- a/src/galileo/resources/models/update_scorer_request.py +++ b/src/galileo/resources/models/update_scorer_request.py @@ -25,6 +25,7 @@ class UpdateScorerRequest: model_type (Union[ModelType, None, Unset]): name (Union[None, Unset, str]): tags (Union[None, Unset, list[str]]): + user_prompt (Union[None, Unset, str]): """ default_version_id: Union[None, Unset, str] = UNSET @@ -34,16 +35,14 @@ class UpdateScorerRequest: model_type: Union[ModelType, None, Unset] = UNSET name: Union[None, Unset, str] = UNSET tags: Union[None, Unset, list[str]] = UNSET + user_prompt: Union[None, Unset, str] = UNSET additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict) def to_dict(self) -> dict[str, Any]: from ..models.scorer_defaults import ScorerDefaults default_version_id: Union[None, Unset, str] - if isinstance(self.default_version_id, Unset): - default_version_id = UNSET - else: - default_version_id = self.default_version_id + default_version_id = UNSET if isinstance(self.default_version_id, Unset) else self.default_version_id defaults: Union[None, Unset, dict[str, Any]] if isinstance(self.defaults, Unset): @@ -54,16 +53,10 @@ def to_dict(self) -> dict[str, Any]: defaults = self.defaults description: Union[None, Unset, str] - if isinstance(self.description, Unset): - description = UNSET - else: - description = self.description + description = UNSET if isinstance(self.description, Unset) else self.description ground_truth: Union[None, Unset, bool] - if isinstance(self.ground_truth, Unset): - ground_truth = UNSET - else: - ground_truth = self.ground_truth + ground_truth = UNSET if isinstance(self.ground_truth, Unset) else self.ground_truth model_type: Union[None, Unset, str] if isinstance(self.model_type, Unset): @@ -74,10 +67,7 @@ def to_dict(self) -> dict[str, Any]: model_type = self.model_type name: Union[None, Unset, str] - if isinstance(self.name, Unset): - name = UNSET - else: - name = self.name + name = UNSET if isinstance(self.name, Unset) else self.name tags: Union[None, Unset, list[str]] if isinstance(self.tags, Unset): @@ -88,6 +78,9 @@ def to_dict(self) -> dict[str, Any]: else: tags = self.tags + user_prompt: Union[None, Unset, str] + user_prompt = UNSET if isinstance(self.user_prompt, Unset) else self.user_prompt + field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) field_dict.update({}) @@ -105,6 +98,8 @@ def to_dict(self) -> dict[str, Any]: field_dict["name"] = name if tags is not UNSET: field_dict["tags"] = tags + if user_prompt is not UNSET: + field_dict["user_prompt"] = user_prompt return field_dict @@ -131,9 +126,8 @@ def _parse_defaults(data: object) -> Union["ScorerDefaults", None, Unset]: try: if not isinstance(data, dict): raise TypeError() - defaults_type_0 = ScorerDefaults.from_dict(data) + return ScorerDefaults.from_dict(data) - return defaults_type_0 except: # noqa: E722 pass return cast(Union["ScorerDefaults", None, Unset], data) @@ -166,9 +160,8 @@ def _parse_model_type(data: object) -> Union[ModelType, None, Unset]: try: if not isinstance(data, str): raise TypeError() - model_type_type_0 = ModelType(data) + return ModelType(data) - return model_type_type_0 except: # noqa: E722 pass return cast(Union[ModelType, None, Unset], data) @@ -192,15 +185,23 @@ def _parse_tags(data: object) -> Union[None, Unset, list[str]]: try: if not isinstance(data, list): raise TypeError() - tags_type_0 = cast(list[str], data) + return cast(list[str], data) - return tags_type_0 except: # noqa: E722 pass return cast(Union[None, Unset, list[str]], data) tags = _parse_tags(d.pop("tags", UNSET)) + def _parse_user_prompt(data: object) -> Union[None, Unset, str]: + if data is None: + return data + if isinstance(data, Unset): + return data + return cast(Union[None, Unset, str], data) + + user_prompt = _parse_user_prompt(d.pop("user_prompt", UNSET)) + update_scorer_request = cls( default_version_id=default_version_id, defaults=defaults, @@ -209,6 +210,7 @@ def _parse_tags(data: object) -> Union[None, Unset, list[str]]: model_type=model_type, name=name, tags=tags, + user_prompt=user_prompt, ) update_scorer_request.additional_properties = d diff --git a/src/galileo/resources/models/upsert_dataset_content_request.py b/src/galileo/resources/models/upsert_dataset_content_request.py index f7538c191..b1c5e3116 100644 --- a/src/galileo/resources/models/upsert_dataset_content_request.py +++ b/src/galileo/resources/models/upsert_dataset_content_request.py @@ -26,10 +26,7 @@ def to_dict(self) -> dict[str, Any]: dataset_id = self.dataset_id version_index: Union[None, Unset, int] - if isinstance(self.version_index, Unset): - version_index = UNSET - else: - version_index = self.version_index + version_index = UNSET if isinstance(self.version_index, Unset) else self.version_index field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/user_collaborator_create.py b/src/galileo/resources/models/user_collaborator_create.py index 8f68b25f7..f9595fc47 100644 --- a/src/galileo/resources/models/user_collaborator_create.py +++ b/src/galileo/resources/models/user_collaborator_create.py @@ -44,10 +44,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _role = d.pop("role", UNSET) role: Union[Unset, CollaboratorRole] - if isinstance(_role, Unset): - role = UNSET - else: - role = CollaboratorRole(_role) + role = UNSET if isinstance(_role, Unset) else CollaboratorRole(_role) user_collaborator_create = cls(user_id=user_id, role=role) diff --git a/src/galileo/resources/models/user_db.py b/src/galileo/resources/models/user_db.py index 131e06ad3..483c5ec2a 100644 --- a/src/galileo/resources/models/user_db.py +++ b/src/galileo/resources/models/user_db.py @@ -67,22 +67,13 @@ def to_dict(self) -> dict[str, Any]: auth_method = self.auth_method.value email_is_verified: Union[None, Unset, bool] - if isinstance(self.email_is_verified, Unset): - email_is_verified = UNSET - else: - email_is_verified = self.email_is_verified + email_is_verified = UNSET if isinstance(self.email_is_verified, Unset) else self.email_is_verified first_name: Union[None, Unset, str] - if isinstance(self.first_name, Unset): - first_name = UNSET - else: - first_name = self.first_name + first_name = UNSET if isinstance(self.first_name, Unset) else self.first_name last_name: Union[None, Unset, str] - if isinstance(self.last_name, Unset): - last_name = UNSET - else: - last_name = self.last_name + last_name = UNSET if isinstance(self.last_name, Unset) else self.last_name permissions: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.permissions, Unset): @@ -141,10 +132,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: _auth_method = d.pop("auth_method", UNSET) auth_method: Union[Unset, AuthMethod] - if isinstance(_auth_method, Unset): - auth_method = UNSET - else: - auth_method = AuthMethod(_auth_method) + auth_method = UNSET if isinstance(_auth_method, Unset) else AuthMethod(_auth_method) def _parse_email_is_verified(data: object) -> Union[None, Unset, bool]: if data is None: @@ -182,10 +170,7 @@ def _parse_last_name(data: object) -> Union[None, Unset, str]: _role = d.pop("role", UNSET) role: Union[Unset, UserRole] - if isinstance(_role, Unset): - role = UNSET - else: - role = UserRole(_role) + role = UNSET if isinstance(_role, Unset) else UserRole(_role) user_db = cls( created_at=created_at, diff --git a/src/galileo/resources/models/user_info.py b/src/galileo/resources/models/user_info.py index 128afea13..1cb503e12 100644 --- a/src/galileo/resources/models/user_info.py +++ b/src/galileo/resources/models/user_info.py @@ -32,16 +32,10 @@ def to_dict(self) -> dict[str, Any]: id = self.id first_name: Union[None, Unset, str] - if isinstance(self.first_name, Unset): - first_name = UNSET - else: - first_name = self.first_name + first_name = UNSET if isinstance(self.first_name, Unset) else self.first_name last_name: Union[None, Unset, str] - if isinstance(self.last_name, Unset): - last_name = UNSET - else: - last_name = self.last_name + last_name = UNSET if isinstance(self.last_name, Unset) else self.last_name field_dict: dict[str, Any] = {} field_dict.update(self.additional_properties) diff --git a/src/galileo/resources/models/validate_llm_scorer_log_record_request.py b/src/galileo/resources/models/validate_llm_scorer_log_record_request.py index 3b08bcd5d..c19988e91 100644 --- a/src/galileo/resources/models/validate_llm_scorer_log_record_request.py +++ b/src/galileo/resources/models/validate_llm_scorer_log_record_request.py @@ -86,23 +86,17 @@ def to_dict(self) -> dict[str, Any]: user_prompt = self.user_prompt experiment_id: Union[None, Unset, str] - if isinstance(self.experiment_id, Unset): - experiment_id = UNSET - else: - experiment_id = self.experiment_id + experiment_id = UNSET if isinstance(self.experiment_id, Unset) else self.experiment_id filters: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.filters, Unset): filters = [] for filters_item_data in self.filters: filters_item: dict[str, Any] - if isinstance(filters_item_data, LogRecordsIDFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsDateFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsNumberFilter): - filters_item = filters_item_data.to_dict() - elif isinstance(filters_item_data, LogRecordsBooleanFilter): + if isinstance( + filters_item_data, + (LogRecordsIDFilter, LogRecordsDateFilter, LogRecordsNumberFilter, LogRecordsBooleanFilter), + ): filters_item = filters_item_data.to_dict() else: filters_item = filters_item_data.to_dict() @@ -112,16 +106,10 @@ def to_dict(self) -> dict[str, Any]: limit = self.limit log_stream_id: Union[None, Unset, str] - if isinstance(self.log_stream_id, Unset): - log_stream_id = UNSET - else: - log_stream_id = self.log_stream_id + log_stream_id = UNSET if isinstance(self.log_stream_id, Unset) else self.log_stream_id metrics_testing_id: Union[None, Unset, str] - if isinstance(self.metrics_testing_id, Unset): - metrics_testing_id = UNSET - else: - metrics_testing_id = self.metrics_testing_id + metrics_testing_id = UNSET if isinstance(self.metrics_testing_id, Unset) else self.metrics_testing_id sort: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.sort, Unset): @@ -208,40 +196,34 @@ def _parse_filters_item( try: if not isinstance(data, dict): raise TypeError() - filters_item_type_0 = LogRecordsIDFilter.from_dict(data) + return LogRecordsIDFilter.from_dict(data) - return filters_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_1 = LogRecordsDateFilter.from_dict(data) + return LogRecordsDateFilter.from_dict(data) - return filters_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_2 = LogRecordsNumberFilter.from_dict(data) + return LogRecordsNumberFilter.from_dict(data) - return filters_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - filters_item_type_3 = LogRecordsBooleanFilter.from_dict(data) + return LogRecordsBooleanFilter.from_dict(data) - return filters_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - filters_item_type_4 = LogRecordsTextFilter.from_dict(data) - - return filters_item_type_4 + return LogRecordsTextFilter.from_dict(data) filters_item = _parse_filters_item(filters_item_data) @@ -269,10 +251,7 @@ def _parse_metrics_testing_id(data: object) -> Union[None, Unset, str]: _sort = d.pop("sort", UNSET) sort: Union[Unset, LogRecordsSortClause] - if isinstance(_sort, Unset): - sort = UNSET - else: - sort = LogRecordsSortClause.from_dict(_sort) + sort = UNSET if isinstance(_sort, Unset) else LogRecordsSortClause.from_dict(_sort) starting_token = d.pop("starting_token", UNSET) diff --git a/src/galileo/resources/models/workflow_span.py b/src/galileo/resources/models/workflow_span.py index 8daf42c06..2b31861f7 100644 --- a/src/galileo/resources/models/workflow_span.py +++ b/src/galileo/resources/models/workflow_span.py @@ -88,32 +88,20 @@ def to_dict(self) -> dict[str, Any]: created_at = self.created_at.isoformat() dataset_input: Union[None, Unset, str] - if isinstance(self.dataset_input, Unset): - dataset_input = UNSET - else: - dataset_input = self.dataset_input + dataset_input = UNSET if isinstance(self.dataset_input, Unset) else self.dataset_input dataset_metadata: Union[Unset, dict[str, Any]] = UNSET if not isinstance(self.dataset_metadata, Unset): dataset_metadata = self.dataset_metadata.to_dict() dataset_output: Union[None, Unset, str] - if isinstance(self.dataset_output, Unset): - dataset_output = UNSET - else: - dataset_output = self.dataset_output + dataset_output = UNSET if isinstance(self.dataset_output, Unset) else self.dataset_output external_id: Union[None, Unset, str] - if isinstance(self.external_id, Unset): - external_id = UNSET - else: - external_id = self.external_id + external_id = UNSET if isinstance(self.external_id, Unset) else self.external_id id: Union[None, Unset, str] - if isinstance(self.id, Unset): - id = UNSET - else: - id = self.id + id = UNSET if isinstance(self.id, Unset) else self.id input_: Union[Unset, list[dict[str, Any]], str] if isinstance(self.input_, Unset): @@ -148,10 +136,7 @@ def to_dict(self) -> dict[str, Any]: output = self.output parent_id: Union[None, Unset, str] - if isinstance(self.parent_id, Unset): - parent_id = UNSET - else: - parent_id = self.parent_id + parent_id = UNSET if isinstance(self.parent_id, Unset) else self.parent_id redacted_input: Union[None, Unset, list[dict[str, Any]], str] if isinstance(self.redacted_input, Unset): @@ -180,23 +165,14 @@ def to_dict(self) -> dict[str, Any]: redacted_output = self.redacted_output session_id: Union[None, Unset, str] - if isinstance(self.session_id, Unset): - session_id = UNSET - else: - session_id = self.session_id + session_id = UNSET if isinstance(self.session_id, Unset) else self.session_id spans: Union[Unset, list[dict[str, Any]]] = UNSET if not isinstance(self.spans, Unset): spans = [] for spans_item_data in self.spans: spans_item: dict[str, Any] - if isinstance(spans_item_data, AgentSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, WorkflowSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, LlmSpan): - spans_item = spans_item_data.to_dict() - elif isinstance(spans_item_data, RetrieverSpan): + if isinstance(spans_item_data, (AgentSpan, WorkflowSpan, LlmSpan, RetrieverSpan)): spans_item = spans_item_data.to_dict() else: spans_item = spans_item_data.to_dict() @@ -204,26 +180,17 @@ def to_dict(self) -> dict[str, Any]: spans.append(spans_item) status_code: Union[None, Unset, int] - if isinstance(self.status_code, Unset): - status_code = UNSET - else: - status_code = self.status_code + status_code = UNSET if isinstance(self.status_code, Unset) else self.status_code step_number: Union[None, Unset, int] - if isinstance(self.step_number, Unset): - step_number = UNSET - else: - step_number = self.step_number + step_number = UNSET if isinstance(self.step_number, Unset) else self.step_number tags: Union[Unset, list[str]] = UNSET if not isinstance(self.tags, Unset): tags = self.tags trace_id: Union[None, Unset, str] - if isinstance(self.trace_id, Unset): - trace_id = UNSET - else: - trace_id = self.trace_id + trace_id = UNSET if isinstance(self.trace_id, Unset) else self.trace_id type_ = self.type_ @@ -294,10 +261,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T: d = dict(src_dict) _created_at = d.pop("created_at", UNSET) created_at: Union[Unset, datetime.datetime] - if isinstance(_created_at, Unset): - created_at = UNSET - else: - created_at = isoparse(_created_at) + created_at = UNSET if isinstance(_created_at, Unset) else isoparse(_created_at) def _parse_dataset_input(data: object) -> Union[None, Unset, str]: if data is None: @@ -364,10 +328,7 @@ def _parse_input_(data: object) -> Union[Unset, list["Message"], str]: _metrics = d.pop("metrics", UNSET) metrics: Union[Unset, Metrics] - if isinstance(_metrics, Unset): - metrics = UNSET - else: - metrics = Metrics.from_dict(_metrics) + metrics = UNSET if isinstance(_metrics, Unset) else Metrics.from_dict(_metrics) name = d.pop("name", UNSET) @@ -379,9 +340,8 @@ def _parse_output(data: object) -> Union["Message", None, Unset, list["Document" try: if not isinstance(data, dict): raise TypeError() - output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return output_type_1 except: # noqa: E722 pass try: @@ -440,9 +400,8 @@ def _parse_redacted_output(data: object) -> Union["Message", None, Unset, list[" try: if not isinstance(data, dict): raise TypeError() - redacted_output_type_1 = Message.from_dict(data) + return Message.from_dict(data) - return redacted_output_type_1 except: # noqa: E722 pass try: @@ -481,40 +440,34 @@ def _parse_spans_item( try: if not isinstance(data, dict): raise TypeError() - spans_item_type_0 = AgentSpan.from_dict(data) + return AgentSpan.from_dict(data) - return spans_item_type_0 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_1 = WorkflowSpan.from_dict(data) + return WorkflowSpan.from_dict(data) - return spans_item_type_1 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_2 = LlmSpan.from_dict(data) + return LlmSpan.from_dict(data) - return spans_item_type_2 except: # noqa: E722 pass try: if not isinstance(data, dict): raise TypeError() - spans_item_type_3 = RetrieverSpan.from_dict(data) + return RetrieverSpan.from_dict(data) - return spans_item_type_3 except: # noqa: E722 pass if not isinstance(data, dict): raise TypeError() - spans_item_type_4 = ToolSpan.from_dict(data) - - return spans_item_type_4 + return ToolSpan.from_dict(data) spans_item = _parse_spans_item(spans_item_data)