From 12969c76bff7eaafacff3cc2d962652d13b6a70f Mon Sep 17 00:00:00 2001 From: Nachiket Paranjape Date: Wed, 24 Sep 2025 16:21:32 -0700 Subject: [PATCH 1/3] update openapi --- openapi.yaml | 852 +++++++++++++++--- .../auth/login_api_key_login_api_key_post.py | 107 ++- .../api/auth/login_email_login_post.py | 106 ++- ...gen_llm_scorer_scorers_llm_autogen_post.py | 107 ++- ...ion_scorers_scorer_id_version_code_post.py | 123 ++- ...sion_scorers_scorer_id_version_llm_post.py | 118 ++- ...ion_scorers_scorer_id_version_luna_post.py | 118 ++- ...n_scorers_scorer_id_version_preset_post.py | 118 ++- .../resources/api/data/create_scorers_post.py | 113 ++- .../delete_scorer_scorers_scorer_id_delete.py | 110 ++- .../data/get_scorer_scorers_scorer_id_get.py | 110 ++- ...code_scorers_scorer_id_version_code_get.py | 111 ++- ...or_latest_scorers_scorer_id_version_get.py | 120 ++- ...r_scorer_scorers_scorer_id_versions_get.py | 129 ++- ...er_route_scorers_scorer_id_projects_get.py | 129 ++- ...versions_scorer_version_id_projects_get.py | 116 ++- ..._scorers_with_filters_scorers_list_post.py | 101 ++- .../api/data/list_tags_scorers_tags_get.py | 93 +- ..._llm_validate_scorers_llm_validate_post.py | 107 ++- ...id_versions_version_number_restore_post.py | 117 ++- .../data/update_scorers_scorer_id_patch.py | 118 ++- ...rd_scorers_llm_validate_log_record_post.py | 107 ++- ...te_datasets_datasets_bulk_delete_delete.py | 110 ++- .../datasets/create_dataset_datasets_post.py | 89 +- ...orators_datasets_dataset_id_groups_post.py | 130 ++- ...borators_datasets_dataset_id_users_post.py | 130 ++- ...lete_dataset_datasets_dataset_id_delete.py | 102 ++- ...asets_dataset_id_groups_group_id_delete.py | 109 ++- ...ct_id_prompt_datasets_dataset_id_delete.py | 109 ++- ...atasets_dataset_id_users_user_id_delete.py | 109 ++- ...ataset_datasets_dataset_id_download_get.py | 102 ++- ...oject_id_prompt_datasets_dataset_id_get.py | 107 ++- ...nd_dataset_content_datasets_extend_post.py | 113 ++- ...content_datasets_dataset_id_content_get.py | 129 ++- .../get_dataset_datasets_dataset_id_get.py | 104 ++- ...d_status_datasets_extend_dataset_id_get.py | 112 ++- ...t_id_versions_version_index_content_get.py | 118 ++- ...ojects_datasets_dataset_id_projects_get.py | 129 ++- .../datasets/list_datasets_datasets_get.py | 97 +- ...borators_datasets_dataset_id_groups_get.py | 129 ++- ...projects_project_id_prompt_datasets_get.py | 129 ++- ...aborators_datasets_dataset_id_users_get.py | 129 ++- ...ataset_datasets_dataset_id_preview_post.py | 106 ++- ..._datasets_dataset_id_content_query_post.py | 112 ++- ...datasets_dataset_id_versions_query_post.py | 106 ++- .../query_datasets_datasets_query_post.py | 103 ++- ...ntent_datasets_dataset_id_content_patch.py | 119 ++- ...pdate_dataset_datasets_dataset_id_patch.py | 110 ++- ...dataset_id_versions_version_index_patch.py | 127 ++- ...tasets_dataset_id_groups_group_id_patch.py | 127 ++- ...oject_id_prompt_datasets_dataset_id_put.py | 142 +-- ...datasets_dataset_id_users_user_id_patch.py | 127 ++- ...rojects_project_id_prompt_datasets_post.py | 109 ++- ...content_datasets_dataset_id_content_put.py | 110 ++- ...nt_projects_project_id_experiments_post.py | 118 ++- ...ect_id_experiments_experiment_id_delete.py | 105 ++- ...t_id_experiments_available_columns_post.py | 108 ++- ..._experiments_experiment_id_metrics_post.py | 127 ++- ...roject_id_experiments_experiment_id_get.py | 117 ++- ...cts_project_id_experiments_metrics_post.py | 118 ++- ...nts_projects_project_id_experiments_get.py | 116 ++- ...roject_id_experiments_experiment_id_put.py | 127 ++- .../resources/api/experiment_tags/__init__.py | 1 + ...iments_experiment_id_tags_tag_id_delete.py | 134 ++- ...periments_experiment_id_tags_tag_id_get.py | 126 ++- ...t_id_experiments_experiment_id_tags_get.py | 133 ++- ..._id_experiments_experiment_id_tags_post.py | 122 ++- ...periments_experiment_id_tags_tag_id_put.py | 135 ++- .../api/health/healthcheck_healthcheck_get.py | 95 +- .../api/jobs/create_job_jobs_post.py | 113 ++- .../api/jobs/get_job_jobs_job_id_get.py | 108 ++- ...rojects_project_id_runs_run_id_jobs_get.py | 140 ++- ..._project_id_runs_run_id_jobs_latest_get.py | 131 ++- ...am_projects_project_id_log_streams_post.py | 118 ++- ...ect_id_log_streams_log_stream_id_delete.py | 105 ++- ...roject_id_log_streams_log_stream_id_get.py | 117 ++- ...ams_projects_project_id_log_streams_get.py | 116 ++- ...roject_id_log_streams_log_stream_id_put.py | 127 ++- ...orators_projects_project_id_groups_post.py | 130 ++- .../projects/create_project_projects_post.py | 113 ++- ...borators_projects_project_id_users_post.py | 130 ++- ...jects_project_id_groups_group_id_delete.py | 109 ++- ...lete_project_projects_project_id_delete.py | 110 ++- ...rojects_project_id_users_user_id_delete.py | 109 ++- .../get_all_projects_projects_all_get.py | 120 ++- ...llaborator_roles_collaborator_roles_get.py | 101 ++- .../get_project_projects_project_id_get.py | 104 ++- .../get_projects_count_projects_count_post.py | 103 ++- ...jects_paginated_projects_paginated_post.py | 103 ++- .../api/projects/get_projects_projects_get.py | 127 ++- ...borators_projects_project_id_groups_get.py | 129 ++- ...aborators_projects_project_id_users_get.py | 129 ++- ...ojects_project_id_groups_group_id_patch.py | 127 ++- .../update_project_projects_project_id_put.py | 118 ++- ...projects_project_id_users_user_id_patch.py | 127 ++- ...le_projects_project_id_upload_file_post.py | 113 ++- ..._templates_templates_bulk_delete_delete.py | 100 +- ...e_global_prompt_template_templates_post.py | 113 ++- ...ion_templates_template_id_versions_post.py | 118 ++- ...ators_templates_template_id_groups_post.py | 130 ++- ..._id_templates_template_id_versions_post.py | 127 ++- ...sion_projects_project_id_templates_post.py | 118 ++- ...rators_templates_template_id_users_post.py | 130 ++- ...l_template_templates_template_id_delete.py | 110 ++- ...ates_template_id_groups_group_id_delete.py | 109 ++- ...project_id_templates_template_id_delete.py | 117 ++- ...plates_template_id_users_user_id_delete.py | 109 ++- ...ate_template_input_stub_input_stub_post.py | 105 ++- ...obal_template_templates_template_id_get.py | 108 ++- ...plates_template_id_versions_version_get.py | 117 ++- ...lates_projects_project_id_templates_get.py | 118 ++- ...ts_project_id_templates_template_id_get.py | 117 ++- ...jects_project_id_templates_versions_get.py | 132 ++- ...plates_template_id_versions_version_get.py | 128 ++- ...rators_templates_template_id_groups_get.py | 129 ++- ...orators_templates_template_id_users_get.py | 129 ++- ...mplates_template_id_versions_query_post.py | 106 ++- .../query_templates_templates_query_post.py | 101 ++- .../render_template_render_template_post.py | 101 ++- ...plates_template_id_versions_version_put.py | 117 ++- ...plates_template_id_versions_version_put.py | 128 ++- ...al_template_templates_template_id_patch.py | 118 ++- ...lates_template_id_groups_group_id_patch.py | 127 ++- ...mplates_template_id_users_user_id_patch.py | 127 ++- ...e_stage_projects_project_id_stages_post.py | 110 ++- ...et_stage_projects_project_id_stages_get.py | 98 +- .../api/protect/invoke_protect_invoke_post.py | 129 ++- ...projects_project_id_stages_stage_id_put.py | 121 ++- ...rojects_project_id_stages_stage_id_post.py | 119 ++- ...ject_id_runs_run_id_scorer_settings_get.py | 117 ++- ...ct_id_runs_run_id_scorer_settings_patch.py | 127 ++- ...ect_id_runs_run_id_scorer_settings_post.py | 127 ++- ...ssion_projects_project_id_sessions_post.py | 118 ++- ...rojects_project_id_sessions_delete_post.py | 121 ++- ...s_projects_project_id_spans_delete_post.py | 121 ++- ..._projects_project_id_traces_delete_post.py | 121 ++- ...projects_project_id_export_records_post.py | 132 ++- ...jects_project_id_traces_aggregated_post.py | 118 ++- ...ects_project_id_sessions_session_id_get.py | 117 ++- ...n_projects_project_id_spans_span_id_get.py | 219 +++-- ...projects_project_id_traces_trace_id_get.py | 117 ++- ...og_spans_projects_project_id_spans_post.py | 118 ++- ..._traces_projects_project_id_traces_post.py | 118 ++- ..._metrics_testing_available_columns_post.py | 118 ++- ...projects_project_id_metrics_search_post.py | 118 ++- ...rojects_project_id_sessions_search_post.py | 118 ++- ...s_projects_project_id_spans_search_post.py | 118 ++- ..._projects_project_id_traces_search_post.py | 118 ++- ...jects_project_id_recompute_metrics_post.py | 108 ++- ...ject_id_sessions_available_columns_post.py | 118 ++- ...project_id_spans_available_columns_post.py | 118 ++- ...roject_id_traces_available_columns_post.py | 118 ++- ...projects_project_id_spans_span_id_patch.py | 127 ++- ...ojects_project_id_traces_trace_id_patch.py | 127 ++- src/galileo/resources/models/__init__.py | 84 +- src/galileo/resources/models/agent_span.py | 95 +- .../models/agentic_session_success_scorer.py | 22 +- .../models/agentic_workflow_success_scorer.py | 22 +- .../models/aggregated_trace_view_request.py | 27 +- .../models/aggregated_trace_view_response.py | 6 +- .../models/base_finetuned_scorer_db.py | 3 +- .../models/base_generated_scorer_db.py | 10 +- .../models/base_prompt_template_response.py | 11 +- .../models/base_prompt_template_version.py | 15 +- .../base_prompt_template_version_response.py | 8 +- src/galileo/resources/models/base_scorer.py | 86 +- .../models/base_scorer_version_db.py | 33 +- .../models/base_scorer_version_response.py | 36 +- src/galileo/resources/models/bleu_scorer.py | 7 +- .../body_create_dataset_datasets_post.py | 13 +- .../models/body_login_email_login_post.py | 15 +- ...oject_id_prompt_datasets_dataset_id_put.py | 6 +- .../resources/models/bucketed_metric.py | 5 +- .../resources/models/chain_poll_template.py | 10 +- .../chunk_attribution_utilization_scorer.py | 17 +- .../chunk_attribution_utilization_template.py | 10 +- src/galileo/resources/models/column_info.py | 38 +- .../resources/models/column_mapping.py | 18 +- .../resources/models/completeness_scorer.py | 22 +- .../resources/models/completeness_template.py | 10 +- .../models/context_adherence_scorer.py | 22 +- .../models/context_relevance_scorer.py | 7 +- .../resources/models/correctness_scorer.py | 17 +- ...eate_custom_luna_scorer_version_request.py | 3 +- .../resources/models/create_job_request.py | 413 +++------ .../resources/models/create_job_response.py | 413 +++------ .../create_llm_scorer_version_request.py | 37 +- ...ompt_template_with_version_request_body.py | 23 +- .../resources/models/create_scorer_request.py | 33 +- .../models/create_scorer_version_request.py | 24 +- ...reate_update_registered_scorer_response.py | 11 +- ...ized_agentic_session_success_gpt_scorer.py | 70 +- ...zed_agentic_workflow_success_gpt_scorer.py | 70 +- ...hunk_attribution_utilization_gpt_scorer.py | 70 +- .../customized_completeness_gpt_scorer.py | 70 +- .../customized_factuality_gpt_scorer.py | 70 +- ...mized_ground_truth_adherence_gpt_scorer.py | 70 +- .../customized_groundedness_gpt_scorer.py | 70 +- .../customized_input_sexist_gpt_scorer.py | 70 +- .../customized_input_toxicity_gpt_scorer.py | 70 +- ...omized_instruction_adherence_gpt_scorer.py | 70 +- .../customized_prompt_injection_gpt_scorer.py | 70 +- .../models/customized_sexist_gpt_scorer.py | 70 +- .../customized_tool_error_rate_gpt_scorer.py | 70 +- ...mized_tool_selection_quality_gpt_scorer.py | 70 +- .../models/customized_toxicity_gpt_scorer.py | 70 +- .../resources/models/dataset_append_row.py | 5 +- .../models/dataset_append_row_values.py | 3 +- .../resources/models/dataset_content.py | 10 +- .../models/dataset_content_filter.py | 5 +- src/galileo/resources/models/dataset_data.py | 5 +- src/galileo/resources/models/dataset_db.py | 12 +- .../resources/models/dataset_id_filter.py | 14 +- .../resources/models/dataset_name_filter.py | 9 +- .../resources/models/dataset_prepend_row.py | 5 +- .../models/dataset_prepend_row_values.py | 3 +- .../resources/models/dataset_project.py | 3 +- src/galileo/resources/models/dataset_row.py | 11 +- .../resources/models/dataset_row_metadata.py | 6 +- .../models/dataset_row_values_dict.py | 3 +- .../models/dataset_update_row_values.py | 3 +- .../resources/models/dataset_version_db.py | 3 +- src/galileo/resources/models/document.py | 9 +- .../models/experiment_create_request.py | 18 +- .../resources/models/experiment_dataset.py | 15 +- .../models/experiment_metrics_request.py | 27 +- .../resources/models/experiment_playground.py | 10 +- .../resources/models/experiment_prompt.py | 20 +- .../resources/models/experiment_response.py | 55 +- .../models/experiment_update_request.py | 5 +- .../models/extended_agent_span_record.py | 72 +- ...ed_agent_span_record_metric_info_type_0.py | 37 +- ...xtended_agent_span_record_with_children.py | 104 +-- ...record_with_children_metric_info_type_0.py | 37 +- .../models/extended_llm_span_record.py | 84 +- ...nded_llm_span_record_metric_info_type_0.py | 37 +- .../models/extended_retriever_span_record.py | 66 +- ...etriever_span_record_metric_info_type_0.py | 37 +- ...ded_retriever_span_record_with_children.py | 98 +- ...record_with_children_metric_info_type_0.py | 37 +- .../models/extended_session_record.py | 72 +- ...ended_session_record_metric_info_type_0.py | 37 +- .../extended_session_record_with_children.py | 72 +- ...record_with_children_metric_info_type_0.py | 37 +- .../models/extended_tool_span_record.py | 81 +- ...ded_tool_span_record_metric_info_type_0.py | 37 +- ...extended_tool_span_record_with_children.py | 113 +-- ...record_with_children_metric_info_type_0.py | 37 +- .../resources/models/extended_trace_record.py | 66 +- ...xtended_trace_record_metric_info_type_0.py | 37 +- .../extended_trace_record_with_children.py | 98 +- ...record_with_children_metric_info_type_0.py | 37 +- .../models/extended_workflow_span_record.py | 67 +- ...workflow_span_record_metric_info_type_0.py | 37 +- ...nded_workflow_span_record_with_children.py | 99 +- ...record_with_children_metric_info_type_0.py | 37 +- .../resources/models/factuality_template.py | 5 +- .../resources/models/feedback_aggregate.py | 19 +- .../resources/models/feedback_rating_db.py | 29 +- .../resources/models/feedback_rating_info.py | 30 +- ...info_feedback_type.py => feedback_type.py} | 2 +- .../resources/models/fine_tuned_scorer.py | 17 +- .../models/fine_tuned_scorer_response.py | 3 +- .../models/generated_scorer_configuration.py | 5 +- .../models/generated_scorer_response.py | 10 +- .../models/get_projects_paginated_response.py | 5 +- .../get_projects_paginated_response_v2.py | 5 +- .../models/ground_truth_adherence_scorer.py | 17 +- .../models/group_collaborator_create.py | 5 +- .../resources/models/input_pii_scorer.py | 7 +- .../resources/models/input_sexist_scorer.py | 22 +- .../resources/models/input_tone_scorer.py | 7 +- .../resources/models/input_toxicity_scorer.py | 22 +- .../models/instruction_adherence_scorer.py | 17 +- .../resources/models/invoke_response.py | 11 +- src/galileo/resources/models/job_db.py | 29 +- src/galileo/resources/models/job_progress.py | 15 +- .../resources/models/list_dataset_params.py | 65 +- .../models/list_dataset_projects_response.py | 5 +- .../resources/models/list_dataset_response.py | 5 +- .../models/list_dataset_version_params.py | 3 +- .../models/list_dataset_version_response.py | 5 +- .../list_group_collaborators_response.py | 5 +- .../models/list_prompt_dataset_response.py | 5 +- .../models/list_prompt_template_params.py | 29 +- .../models/list_prompt_template_response.py | 5 +- .../list_prompt_template_version_params.py | 18 +- .../list_prompt_template_version_response.py | 5 +- .../models/list_scorer_versions_response.py | 5 +- .../resources/models/list_scorers_request.py | 54 +- .../resources/models/list_scorers_response.py | 5 +- .../list_user_collaborators_response.py | 5 +- src/galileo/resources/models/llm_metrics.py | 20 +- src/galileo/resources/models/llm_span.py | 83 +- .../log_records_available_columns_request.py | 15 +- .../models/log_records_delete_request.py | 42 +- .../models/log_records_export_request.py | 55 +- .../resources/models/log_records_id_filter.py | 14 +- .../log_records_metrics_query_request.py | 47 +- .../models/log_records_number_filter.py | 15 +- .../models/log_records_query_request.py | 47 +- .../models/log_records_query_response.py | 49 +- .../models/log_records_text_filter.py | 9 +- .../models/log_span_update_request.py | 36 +- .../models/log_span_update_response.py | 15 +- .../models/log_spans_ingest_request.py | 49 +- .../models/log_spans_ingest_response.py | 15 +- .../resources/models/log_stream_response.py | 5 +- .../models/log_trace_update_request.py | 48 +- .../models/log_trace_update_response.py | 15 +- .../models/log_traces_ingest_request.py | 30 +- .../models/log_traces_ingest_response.py | 15 +- src/galileo/resources/models/message.py | 5 +- .../resources/models/messages_list_item.py | 8 +- .../resources/models/metadata_filter.py | 9 +- .../resources/models/metric_computation.py | 16 +- .../resources/models/metric_computing.py | 3 +- .../metric_critique_job_configuration.py | 28 +- src/galileo/resources/models/metric_error.py | 8 +- src/galileo/resources/models/metric_failed.py | 8 +- .../resources/models/metric_not_applicable.py | 3 +- .../resources/models/metric_not_computed.py | 3 +- .../resources/models/metric_pending.py | 3 +- .../resources/models/metric_success.py | 169 +--- src/galileo/resources/models/metrics.py | 5 +- ...trics_testing_available_columns_request.py | 20 +- src/galileo/resources/models/model.py | 51 +- src/galileo/resources/models/output_map.py | 20 +- .../resources/models/output_pii_scorer.py | 7 +- .../resources/models/output_sexist_scorer.py | 22 +- .../resources/models/output_tone_scorer.py | 7 +- .../models/output_toxicity_scorer.py | 22 +- .../resources/models/output_type_enum.py | 1 + src/galileo/resources/models/payload.py | 10 +- src/galileo/resources/models/permission.py | 73 +- .../models/preview_dataset_request.py | 3 +- .../models/project_collection_params.py | 90 +- .../resources/models/project_create.py | 10 +- .../models/project_create_response.py | 13 +- .../models/project_creator_filter.py | 14 +- src/galileo/resources/models/project_db.py | 13 +- .../resources/models/project_db_thin.py | 8 +- .../resources/models/project_id_filter.py | 14 +- src/galileo/resources/models/project_item.py | 18 +- .../resources/models/project_name_filter.py | 9 +- .../resources/models/project_runs_filter.py | 15 +- .../resources/models/project_type_filter.py | 3 +- .../resources/models/project_update.py | 21 +- .../models/project_update_response.py | 18 +- .../resources/models/prompt_dataset_db.py | 20 +- .../models/prompt_injection_scorer.py | 22 +- .../prompt_optimization_configuration.py | 15 +- .../models/prompt_perplexity_scorer.py | 7 +- .../resources/models/prompt_run_settings.py | 18 +- .../prompt_template_created_by_filter.py | 14 +- .../models/prompt_template_name_filter.py | 9 +- .../resources/models/protect_request.py | 31 +- .../resources/models/protect_response.py | 5 +- .../resources/models/query_dataset_params.py | 3 +- .../recompute_log_records_metrics_request.py | 47 +- .../resources/models/registered_scorer.py | 17 +- .../models/render_template_request.py | 12 +- .../models/render_template_response.py | 5 +- .../resources/models/rendered_template.py | 5 +- .../resources/models/retriever_span.py | 84 +- src/galileo/resources/models/rouge_scorer.py | 7 +- src/galileo/resources/models/rule.py | 9 +- src/galileo/resources/models/rule_result.py | 24 +- src/galileo/resources/models/ruleset.py | 12 +- .../resources/models/ruleset_result.py | 17 +- src/galileo/resources/models/run_db.py | 28 +- src/galileo/resources/models/run_db_thin.py | 28 +- .../resources/models/run_params_map.py | 95 +- src/galileo/resources/models/scorer_config.py | 42 +- .../resources/models/scorer_creator_filter.py | 14 +- .../resources/models/scorer_defaults.py | 31 +- .../resources/models/scorer_name_filter.py | 9 +- .../resources/models/scorer_response.py | 108 +-- .../resources/models/scorer_type_filter.py | 3 +- src/galileo/resources/models/segment.py | 5 +- .../resources/models/segment_filter.py | 10 +- .../models/session_create_request.py | 30 +- .../models/session_create_response.py | 10 +- src/galileo/resources/models/stage_db.py | 15 +- .../resources/models/stage_with_rulesets.py | 10 +- .../models/synthetic_data_source_dataset.py | 8 +- .../synthetic_dataset_extension_request.py | 13 +- .../resources/models/system_metric_info.py | 46 +- .../models/tool_error_rate_scorer.py | 17 +- .../models/tool_selection_quality_scorer.py | 22 +- src/galileo/resources/models/tool_span.py | 104 +-- src/galileo/resources/models/trace.py | 99 +- .../resources/models/uncertainty_scorer.py | 7 +- .../models/update_dataset_content_request.py | 24 +- .../models/update_dataset_request.py | 11 +- .../models/update_dataset_version_request.py | 5 +- .../models/update_prompt_template_request.py | 3 +- .../resources/models/update_scorer_request.py | 46 +- .../models/upsert_dataset_content_request.py | 5 +- .../models/user_collaborator_create.py | 5 +- src/galileo/resources/models/user_db.py | 25 +- src/galileo/resources/models/user_info.py | 10 +- .../validate_llm_scorer_log_record_request.py | 47 +- src/galileo/resources/models/workflow_span.py | 85 +- 404 files changed, 15761 insertions(+), 10767 deletions(-) create mode 100644 src/galileo/resources/api/experiment_tags/__init__.py rename src/galileo/resources/models/{feedback_rating_info_feedback_type.py => feedback_type.py} (79%) 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/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..622262229 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 @@ -1,26 +1,42 @@ 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.api_key_login_request import ApiKeyLoginRequest from ...models.http_validation_error import HTTPValidationError from ...models.token import Token -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: ApiKeyLoginRequest, -def _get_kwargs(*, body: ApiKeyLoginRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/login/api_key"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/login/api_key", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -30,13 +46,17 @@ 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: + if response.status_code == : response_200 = Token.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -53,8 +73,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient, body: ApiKeyLoginRequest) -> Response[Union[HTTPValidationError, Token]]: - """Login Api Key +def sync_detailed( + *, + client: ApiClient, + body: ApiKeyLoginRequest, + +) -> Response[Union[HTTPValidationError, Token]]: + """ Login Api Key Args: body (ApiKeyLoginRequest): @@ -65,17 +90,27 @@ def sync_detailed(*, client: ApiClient, body: ApiKeyLoginRequest) -> Response[Un Returns: Response[Union[HTTPValidationError, Token]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: ApiKeyLoginRequest, -def sync(*, client: ApiClient, body: ApiKeyLoginRequest) -> Optional[Union[HTTPValidationError, Token]]: - """Login Api Key +) -> Optional[Union[HTTPValidationError, Token]]: + """ Login Api Key Args: body (ApiKeyLoginRequest): @@ -86,15 +121,22 @@ def sync(*, client: ApiClient, body: ApiKeyLoginRequest) -> Optional[Union[HTTPV Returns: Union[HTTPValidationError, Token] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: ApiKeyLoginRequest + *, + client: ApiClient, + body: ApiKeyLoginRequest, + ) -> Response[Union[HTTPValidationError, Token]]: - """Login Api Key + """ Login Api Key Args: body (ApiKeyLoginRequest): @@ -105,17 +147,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Token]] - """ + """ - kwargs = _get_kwargs(body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, + body: ApiKeyLoginRequest, -async def asyncio(*, client: ApiClient, body: ApiKeyLoginRequest) -> Optional[Union[HTTPValidationError, Token]]: - """Login Api Key +) -> Optional[Union[HTTPValidationError, Token]]: + """ Login Api Key Args: body (ApiKeyLoginRequest): @@ -126,6 +178,11 @@ async def asyncio(*, client: ApiClient, body: ApiKeyLoginRequest) -> Optional[Un Returns: Union[HTTPValidationError, Token] - """ + """ + + + 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/auth/login_email_login_post.py b/src/galileo/resources/api/auth/login_email_login_post.py index 180480a90..5cac8f64c 100644 --- a/src/galileo/resources/api/auth/login_email_login_post.py +++ b/src/galileo/resources/api/auth/login_email_login_post.py @@ -1,23 +1,38 @@ 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.body_login_email_login_post import BodyLoginEmailLoginPost from ...models.http_validation_error import HTTPValidationError from ...models.token import Token -from ...types import Response +from typing import cast + -def _get_kwargs(*, body: BodyLoginEmailLoginPost) -> dict[str, Any]: +def _get_kwargs( + *, + body: BodyLoginEmailLoginPost, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/login"} + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/login", + } _kwargs["data"] = body.to_dict() @@ -30,13 +45,17 @@ 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: + if response.status_code == : response_200 = Token.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -53,8 +72,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Response[Union[HTTPValidationError, Token]]: - """Login Email +def sync_detailed( + *, + client: ApiClient, + body: BodyLoginEmailLoginPost, + +) -> Response[Union[HTTPValidationError, Token]]: + """ Login Email Args: body (BodyLoginEmailLoginPost): @@ -65,17 +89,27 @@ def sync_detailed(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Respon Returns: Response[Union[HTTPValidationError, Token]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: BodyLoginEmailLoginPost, -def sync(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Optional[Union[HTTPValidationError, Token]]: - """Login Email +) -> Optional[Union[HTTPValidationError, Token]]: + """ Login Email Args: body (BodyLoginEmailLoginPost): @@ -86,15 +120,22 @@ def sync(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Optional[Union[ Returns: Union[HTTPValidationError, Token] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: BodyLoginEmailLoginPost + *, + client: ApiClient, + body: BodyLoginEmailLoginPost, + ) -> Response[Union[HTTPValidationError, Token]]: - """Login Email + """ Login Email Args: body (BodyLoginEmailLoginPost): @@ -105,17 +146,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Token]] - """ + """ - kwargs = _get_kwargs(body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, + body: BodyLoginEmailLoginPost, -async def asyncio(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Optional[Union[HTTPValidationError, Token]]: - """Login Email +) -> Optional[Union[HTTPValidationError, Token]]: + """ Login Email Args: body (BodyLoginEmailLoginPost): @@ -126,6 +177,11 @@ async def asyncio(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Option Returns: Union[HTTPValidationError, Token] - """ + """ + + + 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/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..c4d3f4c45 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 @@ -1,21 +1,32 @@ 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.create_llm_scorer_autogen_request import CreateLLMScorerAutogenRequest from ...models.generation_response import GenerationResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: CreateLLMScorerAutogenRequest, -def _get_kwargs(*, body: CreateLLMScorerAutogenRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, @@ -25,6 +36,7 @@ def _get_kwargs(*, body: CreateLLMScorerAutogenRequest) -> dict[str, Any]: _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +45,18 @@ def _get_kwargs(*, body: CreateLLMScorerAutogenRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[GenerationResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GenerationResponse, HTTPValidationError]]: + if response.status_code == : response_200 = GenerationResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GenerationResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GenerationResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: CreateLLMScorerAutogenRequest + *, + client: ApiClient, + body: CreateLLMScorerAutogenRequest, + ) -> Response[Union[GenerationResponse, HTTPValidationError]]: - """Autogen Llm Scorer + """ Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -79,19 +94,27 @@ def sync_detailed( Returns: Response[Union[GenerationResponse, HTTPValidationError]] - """ + """ - 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: CreateLLMScorerAutogenRequest + *, + client: ApiClient, + body: CreateLLMScorerAutogenRequest, + ) -> Optional[Union[GenerationResponse, HTTPValidationError]]: - """Autogen Llm Scorer + """ Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -106,15 +129,22 @@ def sync( Returns: Union[GenerationResponse, HTTPValidationError] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: CreateLLMScorerAutogenRequest + *, + client: ApiClient, + body: CreateLLMScorerAutogenRequest, + ) -> Response[Union[GenerationResponse, HTTPValidationError]]: - """Autogen Llm Scorer + """ Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -129,19 +159,27 @@ async def asyncio_detailed( Returns: Response[Union[GenerationResponse, HTTPValidationError]] - """ + """ - 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: CreateLLMScorerAutogenRequest + *, + client: ApiClient, + body: CreateLLMScorerAutogenRequest, + ) -> Optional[Union[GenerationResponse, HTTPValidationError]]: - """Autogen Llm Scorer + """ Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -156,6 +194,11 @@ async def asyncio( Returns: Union[GenerationResponse, HTTPValidationError] - """ + """ + + + 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/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..1932de98e 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 @@ -1,48 +1,62 @@ 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.base_scorer_version_response import BaseScorerVersionResponse -from ...models.body_create_code_scorer_version_scorers_scorer_id_version_code_post import ( - BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, -) +from ...models.body_create_code_scorer_version_scorers_scorer_id_version_code_post import BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, + *, + body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, -def _get_kwargs(scorer_id: str, *, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/version/code", + "path": "/scorers/{scorer_id}/version/code".format(scorer_id=scorer_id,), } _kwargs["files"] = body.to_multipart() + + 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[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BaseScorerVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +74,13 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost + scorer_id: str, + *, + client: ApiClient, + body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Code Scorer Version + """ Create Code Scorer Version Args: scorer_id (str): @@ -76,19 +92,29 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost + scorer_id: str, + *, + client: ApiClient, + body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Code Scorer Version + """ Create Code Scorer Version Args: scorer_id (str): @@ -100,15 +126,24 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost + scorer_id: str, + *, + client: ApiClient, + body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Code Scorer Version + """ Create Code Scorer Version Args: scorer_id (str): @@ -120,19 +155,29 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost + scorer_id: str, + *, + client: ApiClient, + body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Code Scorer Version + """ Create Code Scorer Version Args: scorer_id (str): @@ -144,6 +189,12 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +body=body, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed + )).parsed 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..2af939226 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 @@ -1,30 +1,43 @@ 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.base_scorer_version_response import BaseScorerVersionResponse from ...models.create_llm_scorer_version_request import CreateLLMScorerVersionRequest from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, + *, + body: CreateLLMScorerVersionRequest, -def _get_kwargs(scorer_id: str, *, body: CreateLLMScorerVersionRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/version/llm", + "path": "/scorers/{scorer_id}/version/llm".format(scorer_id=scorer_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(scorer_id: str, *, body: CreateLLMScorerVersionRequest) -> dict[ return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BaseScorerVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateLLMScorerVersionRequest, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Llm Scorer Version + """ Create Llm Scorer Version Args: scorer_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateLLMScorerVersionRequest, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Llm Scorer Version + """ Create Llm Scorer Version Args: scorer_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateLLMScorerVersionRequest, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Llm Scorer Version + """ Create Llm Scorer Version Args: scorer_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateLLMScorerVersionRequest, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Llm Scorer Version + """ Create Llm Scorer Version Args: scorer_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +body=body, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed + )).parsed 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..18937b003 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 @@ -1,30 +1,43 @@ 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.base_scorer_version_response import BaseScorerVersionResponse from ...models.create_custom_luna_scorer_version_request import CreateCustomLunaScorerVersionRequest from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, + *, + body: CreateCustomLunaScorerVersionRequest, -def _get_kwargs(scorer_id: str, *, body: CreateCustomLunaScorerVersionRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/version/luna", + "path": "/scorers/{scorer_id}/version/luna".format(scorer_id=scorer_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(scorer_id: str, *, body: CreateCustomLunaScorerVersionRequest) - return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BaseScorerVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateCustomLunaScorerVersionRequest, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Luna Scorer Version + """ Create Luna Scorer Version Args: scorer_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateCustomLunaScorerVersionRequest, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Luna Scorer Version + """ Create Luna Scorer Version Args: scorer_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateCustomLunaScorerVersionRequest, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Luna Scorer Version + """ Create Luna Scorer Version Args: scorer_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateCustomLunaScorerVersionRequest, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Luna Scorer Version + """ Create Luna Scorer Version Args: scorer_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +body=body, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed + )).parsed 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..317568496 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 @@ -1,30 +1,43 @@ 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.base_scorer_version_response import BaseScorerVersionResponse from ...models.create_scorer_version_request import CreateScorerVersionRequest from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, + *, + body: CreateScorerVersionRequest, -def _get_kwargs(scorer_id: str, *, body: CreateScorerVersionRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/version/preset", + "path": "/scorers/{scorer_id}/version/preset".format(scorer_id=scorer_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(scorer_id: str, *, body: CreateScorerVersionRequest) -> dict[str return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BaseScorerVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateScorerVersionRequest, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Preset Scorer Version + """ Create Preset Scorer Version Create a preset scorer version. @@ -78,19 +95,29 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateScorerVersionRequest, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Preset Scorer Version + """ Create Preset Scorer Version Create a preset scorer version. @@ -104,15 +131,24 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateScorerVersionRequest, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Preset Scorer Version + """ Create Preset Scorer Version Create a preset scorer version. @@ -126,19 +162,29 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest + scorer_id: str, + *, + client: ApiClient, + body: CreateScorerVersionRequest, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Create Preset Scorer Version + """ Create Preset Scorer Version Create a preset scorer version. @@ -152,6 +198,12 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +body=body, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed + )).parsed diff --git a/src/galileo/resources/api/data/create_scorers_post.py b/src/galileo/resources/api/data/create_scorers_post.py index abded7ab7..365764895 100644 --- a/src/galileo/resources/api/data/create_scorers_post.py +++ b/src/galileo/resources/api/data/create_scorers_post.py @@ -1,26 +1,42 @@ 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.create_scorer_request import CreateScorerRequest from ...models.http_validation_error import HTTPValidationError from ...models.scorer_response import ScorerResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: CreateScorerRequest, -def _get_kwargs(*, body: CreateScorerRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/scorers"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/scorers", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,16 +45,18 @@ def _get_kwargs(*, body: CreateScorerRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + if response.status_code == : response_200 = ScorerResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -46,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ScorerResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ScorerResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -58,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: CreateScorerRequest + *, + client: ApiClient, + body: CreateScorerRequest, + ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """Create + """ Create Args: body (CreateScorerRequest): @@ -71,17 +90,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: CreateScorerRequest, -def sync(*, client: ApiClient, body: CreateScorerRequest) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """Create +) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + """ Create Args: body (CreateScorerRequest): @@ -92,15 +121,22 @@ def sync(*, client: ApiClient, body: CreateScorerRequest) -> Optional[Union[HTTP Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: CreateScorerRequest + *, + client: ApiClient, + body: CreateScorerRequest, + ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """Create + """ Create Args: body (CreateScorerRequest): @@ -111,19 +147,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ - 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: CreateScorerRequest + *, + client: ApiClient, + body: CreateScorerRequest, + ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """Create + """ Create Args: body (CreateScorerRequest): @@ -134,6 +178,11 @@ async def asyncio( Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ + + + 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/data/delete_scorer_scorers_scorer_id_delete.py b/src/galileo/resources/api/data/delete_scorer_scorers_scorer_id_delete.py index 7883e125b..cbeb656df 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 @@ -1,43 +1,56 @@ 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.delete_scorer_response import DeleteScorerResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, -def _get_kwargs(scorer_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/scorers/{scorer_id}", + "path": "/scorers/{scorer_id}".format(scorer_id=scorer_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[DeleteScorerResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: + if response.status_code == : response_200 = DeleteScorerResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -56,8 +67,13 @@ def _build_response( ) -def sync_detailed(scorer_id: str, *, client: ApiClient) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: - """Delete Scorer +def sync_detailed( + scorer_id: str, + *, + client: ApiClient, + +) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: + """ Delete Scorer Args: scorer_id (str): @@ -68,17 +84,27 @@ def sync_detailed(scorer_id: str, *, client: ApiClient) -> Response[Union[Delete Returns: Response[Union[DeleteScorerResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + scorer_id: str, + *, + client: ApiClient, -def sync(scorer_id: str, *, client: ApiClient) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: - """Delete Scorer +) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: + """ Delete Scorer Args: scorer_id (str): @@ -89,15 +115,22 @@ def sync(scorer_id: str, *, client: ApiClient) -> Optional[Union[DeleteScorerRes Returns: Union[DeleteScorerResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient + scorer_id: str, + *, + client: ApiClient, + ) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: - """Delete Scorer + """ Delete Scorer Args: scorer_id (str): @@ -108,17 +141,27 @@ async def asyncio_detailed( Returns: Response[Union[DeleteScorerResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + scorer_id: str, + *, + client: ApiClient, -async def asyncio(scorer_id: str, *, client: ApiClient) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: - """Delete Scorer +) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: + """ Delete Scorer Args: scorer_id (str): @@ -129,6 +172,11 @@ async def asyncio(scorer_id: str, *, client: ApiClient) -> Optional[Union[Delete Returns: Union[DeleteScorerResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, - return (await asyncio_detailed(scorer_id=scorer_id, client=client)).parsed + )).parsed 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..a38e004fb 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 @@ -1,43 +1,56 @@ 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.scorer_response import ScorerResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, -def _get_kwargs(scorer_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"/scorers/{scorer_id}", + "path": "/scorers/{scorer_id}".format(scorer_id=scorer_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, ScorerResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + if response.status_code == : response_200 = ScorerResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ScorerResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ScorerResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -56,8 +67,13 @@ def _build_response( ) -def sync_detailed(scorer_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """Get Scorer +def sync_detailed( + scorer_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, ScorerResponse]]: + """ Get Scorer Args: scorer_id (str): @@ -68,17 +84,27 @@ def sync_detailed(scorer_id: str, *, client: ApiClient) -> Response[Union[HTTPVa Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + scorer_id: str, + *, + client: ApiClient, -def sync(scorer_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """Get Scorer +) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + """ Get Scorer Args: scorer_id (str): @@ -89,15 +115,22 @@ def sync(scorer_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationE Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient + scorer_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """Get Scorer + """ Get Scorer Args: scorer_id (str): @@ -108,17 +141,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + scorer_id: str, + *, + client: ApiClient, -async def asyncio(scorer_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """Get Scorer +) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + """ Get Scorer Args: scorer_id (str): @@ -129,6 +172,11 @@ async def asyncio(scorer_id: str, *, client: ApiClient) -> Optional[Union[HTTPVa Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, - return (await asyncio_detailed(scorer_id=scorer_id, client=client)).parsed + )).parsed 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..a2a9d7a71 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 @@ -1,19 +1,32 @@ 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 ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + + +def _get_kwargs( + scorer_id: str, + *, + version: Union[None, Unset, int] = UNSET, -def _get_kwargs(scorer_id: str, *, version: Union[None, Unset, int] = UNSET) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,15 +37,18 @@ def _get_kwargs(scorer_id: str, *, version: Union[None, Unset, int] = UNSET) -> json_version = version params["version"] = json_version + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/version/code", + "path": "/scorers/{scorer_id}/version/code".format(scorer_id=scorer_id,), "params": params, } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,12 +56,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : 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 +81,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[None, Unset, int] = UNSET, + ) -> Response[Union[Any, HTTPValidationError]]: - """Get Scorer Version Code + """ Get Scorer Version Code Args: scorer_id (str): @@ -77,19 +99,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +version=version, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[None, Unset, int] = UNSET, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Get Scorer Version Code + """ Get Scorer Version Code Args: scorer_id (str): @@ -101,15 +133,24 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, version=version).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +version=version, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[None, Unset, int] = UNSET, + ) -> Response[Union[Any, HTTPValidationError]]: - """Get Scorer Version Code + """ Get Scorer Version Code Args: scorer_id (str): @@ -121,19 +162,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +version=version, - return _build_response(client=client, response=response) + ) + response = await client.arequest( + **kwargs + ) + + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[None, Unset, int] = UNSET, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Get Scorer Version Code + """ Get Scorer Version Code Args: scorer_id (str): @@ -145,6 +196,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +version=version, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, version=version)).parsed + )).parsed 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..45d153cb6 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 @@ -1,50 +1,67 @@ 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.base_scorer_version_response import BaseScorerVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + + +def _get_kwargs( + scorer_id: str, + *, + version: Union[Unset, int] = UNSET, -def _get_kwargs(scorer_id: str, *, version: Union[Unset, int] = UNSET) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} params["version"] = version + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/version", + "path": "/scorers/{scorer_id}/version".format(scorer_id=scorer_id,), "params": params, } + 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[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BaseScorerVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -52,9 +69,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -64,9 +79,13 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[Unset, int] = UNSET, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Get Scorer Version Or Latest + """ Get Scorer Version Or Latest Args: scorer_id (str): @@ -78,19 +97,29 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +version=version, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[Unset, int] = UNSET, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Get Scorer Version Or Latest + """ Get Scorer Version Or Latest Args: scorer_id (str): @@ -102,15 +131,24 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, version=version).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +version=version, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[Unset, int] = UNSET, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Get Scorer Version Or Latest + """ Get Scorer Version Or Latest Args: scorer_id (str): @@ -122,19 +160,29 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +version=version, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET + scorer_id: str, + *, + client: ApiClient, + version: Union[Unset, int] = UNSET, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Get Scorer Version Or Latest + """ Get Scorer Version Or Latest Args: scorer_id (str): @@ -146,6 +194,12 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +version=version, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, version=version)).parsed + )).parsed 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..a21e5bd30 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 @@ -1,22 +1,33 @@ 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.list_scorer_versions_response import ListScorerVersionsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - scorer_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/versions", + "path": "/scorers/{scorer_id}/versions".format(scorer_id=scorer_id,), "params": params, } + 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, ListScorerVersionsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: + if response.status_code == : response_200 = ListScorerVersionsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """List All Versions For Scorer + """ List All Versions For Scorer Args: scorer_id (str): @@ -83,19 +102,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListScorerVersionsResponse]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """List All Versions For Scorer + """ List All Versions For Scorer Args: scorer_id (str): @@ -108,15 +139,26 @@ def sync( Returns: Union[HTTPValidationError, ListScorerVersionsResponse] - """ + """ - return sync_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """List All Versions For Scorer + """ List All Versions For Scorer Args: scorer_id (str): @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListScorerVersionsResponse]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """List All Versions For Scorer + """ List All Versions For Scorer Args: scorer_id (str): @@ -154,8 +208,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListScorerVersionsResponse] - """ + """ - return ( - await asyncio_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..38cf5f125 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 @@ -1,22 +1,33 @@ 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.get_projects_paginated_response_v2 import GetProjectsPaginatedResponseV2 from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - scorer_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/projects", + "path": "/scorers/{scorer_id}/projects".format(scorer_id=scorer_id,), "params": params, } + 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[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: + if response.status_code == : response_200 = GetProjectsPaginatedResponseV2.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Route + """ List Projects For Scorer Route List all projects associated with a specific scorer. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Route + """ List Projects For Scorer Route List all projects associated with a specific scorer. @@ -112,15 +143,26 @@ def sync( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ + """ - return sync_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Route + """ List Projects For Scorer Route List all projects associated with a specific scorer. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Route + """ List Projects For Scorer Route List all projects associated with a specific scorer. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..cce7208e9 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 @@ -1,22 +1,34 @@ 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.get_projects_paginated_response_v2 import GetProjectsPaginatedResponseV2 from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - scorer_version_id: str, *, scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + scorer_version_id: str, + *, + scorer_id: str, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -26,31 +38,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/scorers/versions/{scorer_version_id}/projects", + "path": "/scorers/versions/{scorer_version_id}/projects".format(scorer_version_id=scorer_version_id,), "params": params, } + 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[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: + if response.status_code == : response_200 = GetProjectsPaginatedResponseV2.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,8 +91,9 @@ def sync_detailed( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Version Route + """ List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -93,17 +109,23 @@ def sync_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs( - scorer_version_id=scorer_version_id, scorer_id=scorer_id, starting_token=starting_token, limit=limit + scorer_version_id=scorer_version_id, +scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, + ) - response = client.request(**kwargs) + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) - def sync( scorer_version_id: str, *, @@ -111,8 +133,9 @@ def sync( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Version Route + """ List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -128,16 +151,17 @@ def sync( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ + """ + return sync_detailed( scorer_version_id=scorer_version_id, - client=client, - scorer_id=scorer_id, - starting_token=starting_token, - limit=limit, - ).parsed +client=client, +scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, + ).parsed async def asyncio_detailed( scorer_version_id: str, @@ -146,8 +170,9 @@ async def asyncio_detailed( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Version Route + """ List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -163,17 +188,23 @@ async def asyncio_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs( - scorer_version_id=scorer_version_id, scorer_id=scorer_id, starting_token=starting_token, limit=limit + scorer_version_id=scorer_version_id, +scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, + ) - response = await client.arequest(**kwargs) + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) - async def asyncio( scorer_version_id: str, *, @@ -181,8 +212,9 @@ async def asyncio( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """List Projects For Scorer Version Route + """ List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -198,14 +230,14 @@ async def asyncio( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ - - return ( - await asyncio_detailed( - scorer_version_id=scorer_version_id, - client=client, - scorer_id=scorer_id, - starting_token=starting_token, - limit=limit, - ) - ).parsed + """ + + + return (await asyncio_detailed( + scorer_version_id=scorer_version_id, +client=client, +scorer_id=scorer_id, +starting_token=starting_token, +limit=limit, + + )).parsed 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..f6cae9772 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 @@ -1,23 +1,34 @@ 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.list_scorers_request import ListScorersRequest from ...models.list_scorers_response import ListScorersResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - *, body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + *, + body: ListScorersRequest, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -25,8 +36,10 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -36,6 +49,7 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -44,16 +58,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: + if response.status_code == : response_200 = ListScorersResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -61,9 +77,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListScorersResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListScorersResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -78,8 +92,9 @@ def sync_detailed( body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListScorersResponse]]: - """List Scorers With Filters + """ List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -92,14 +107,21 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListScorersResponse]] - """ + """ - kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( *, @@ -107,8 +129,9 @@ def sync( body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: - """List Scorers With Filters + """ List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -121,10 +144,16 @@ def sync( Returns: Union[HTTPValidationError, ListScorersResponse] - """ + """ - return sync_detailed(client=client, body=body, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + client=client, +body=body, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( *, @@ -132,8 +161,9 @@ async def asyncio_detailed( body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListScorersResponse]]: - """List Scorers With Filters + """ List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -146,14 +176,21 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListScorersResponse]] - """ + """ - kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -161,8 +198,9 @@ async def asyncio( body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: - """List Scorers With Filters + """ List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -175,6 +213,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListScorersResponse] - """ + """ + + + return (await asyncio_detailed( + client=client, +body=body, +starting_token=starting_token, +limit=limit, - return (await asyncio_detailed(client=client, body=body, starting_token=starting_token, limit=limit)).parsed + )).parsed 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..9106bfade 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 @@ -1,20 +1,34 @@ from http import HTTPStatus -from typing import Any, Optional, cast +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 ...types import Response + +from typing import cast + -def _get_kwargs() -> dict[str, Any]: +def _get_kwargs( + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.GET, + "return_raw_response": True, + "path": "/scorers/tags", + } - _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": "/scorers/tags"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -23,7 +37,7 @@ def _get_kwargs() -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list[str]]: - if response.status_code == 200: + if response.status_code == : response_200 = cast(list[str], response.json()) return response_200 @@ -42,8 +56,12 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient) -> Response[list[str]]: - """List Tags +def sync_detailed( + *, + client: ApiClient, + +) -> Response[list[str]]: + """ List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -51,17 +69,25 @@ def sync_detailed(*, client: ApiClient) -> Response[list[str]]: Returns: Response[list[str]] - """ + """ - kwargs = _get_kwargs() - response = client.request(**kwargs) + kwargs = _get_kwargs( + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, -def sync(*, client: ApiClient) -> Optional[list[str]]: - """List Tags +) -> Optional[list[str]]: + """ List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -69,13 +95,20 @@ def sync(*, client: ApiClient) -> Optional[list[str]]: Returns: list[str] - """ + """ + - return sync_detailed(client=client).parsed + return sync_detailed( + client=client, + ).parsed -async def asyncio_detailed(*, client: ApiClient) -> Response[list[str]]: - """List Tags +async def asyncio_detailed( + *, + client: ApiClient, + +) -> Response[list[str]]: + """ List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -83,17 +116,25 @@ async def asyncio_detailed(*, client: ApiClient) -> Response[list[str]]: Returns: Response[list[str]] - """ + """ + - kwargs = _get_kwargs() + kwargs = _get_kwargs( + + ) - response = await client.arequest(**kwargs) + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, -async def asyncio(*, client: ApiClient) -> Optional[list[str]]: - """List Tags +) -> Optional[list[str]]: + """ List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -101,6 +142,10 @@ async def asyncio(*, client: ApiClient) -> Optional[list[str]]: Returns: list[str] - """ + """ + + + return (await asyncio_detailed( + client=client, - return (await asyncio_detailed(client=client)).parsed + )).parsed 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..1fce99d9d 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 @@ -1,21 +1,32 @@ 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.generated_scorer_validation_response import GeneratedScorerValidationResponse from ...models.http_validation_error import HTTPValidationError from ...models.manual_llm_validate_scorers_llm_validate_post_body import ManualLlmValidateScorersLlmValidatePostBody -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: ManualLlmValidateScorersLlmValidatePostBody, -def _get_kwargs(*, body: ManualLlmValidateScorersLlmValidatePostBody) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, @@ -25,6 +36,7 @@ def _get_kwargs(*, body: ManualLlmValidateScorersLlmValidatePostBody) -> dict[st _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +45,18 @@ def _get_kwargs(*, body: ManualLlmValidateScorersLlmValidatePostBody) -> dict[st return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: + if response.status_code == : response_200 = GeneratedScorerValidationResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: ManualLlmValidateScorersLlmValidatePostBody + *, + client: ApiClient, + body: ManualLlmValidateScorersLlmValidatePostBody, + ) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """Manual Llm Validate + """ Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -75,19 +90,27 @@ def sync_detailed( Returns: Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]] - """ + """ - 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: ManualLlmValidateScorersLlmValidatePostBody + *, + client: ApiClient, + body: ManualLlmValidateScorersLlmValidatePostBody, + ) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """Manual Llm Validate + """ Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -98,15 +121,22 @@ def sync( Returns: Union[GeneratedScorerValidationResponse, HTTPValidationError] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: ManualLlmValidateScorersLlmValidatePostBody + *, + client: ApiClient, + body: ManualLlmValidateScorersLlmValidatePostBody, + ) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """Manual Llm Validate + """ Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -117,19 +147,27 @@ async def asyncio_detailed( Returns: Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]] - """ + """ - 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: ManualLlmValidateScorersLlmValidatePostBody + *, + client: ApiClient, + body: ManualLlmValidateScorersLlmValidatePostBody, + ) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """Manual Llm Validate + """ Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -140,6 +178,11 @@ async def asyncio( Returns: Union[GeneratedScorerValidationResponse, HTTPValidationError] - """ + """ + + + 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/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..7da2a27fb 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 @@ -1,43 +1,57 @@ 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.base_scorer_version_response import BaseScorerVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, + version_number: int, -def _get_kwargs(scorer_id: str, version_number: int) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/scorers/{scorer_id}/versions/{version_number}/restore", + "path": "/scorers/{scorer_id}/versions/{version_number}/restore".format(scorer_id=scorer_id,version_number=version_number,), } + 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[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BaseScorerVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - scorer_id: str, version_number: int, *, client: ApiClient + scorer_id: str, + version_number: int, + *, + client: ApiClient, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Restore Scorer Version + """ Restore Scorer Version List all scorers. @@ -73,19 +89,29 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, version_number=version_number) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +version_number=version_number, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, version_number: int, *, client: ApiClient + scorer_id: str, + version_number: int, + *, + client: ApiClient, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Restore Scorer Version + """ Restore Scorer Version List all scorers. @@ -99,15 +125,24 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(scorer_id=scorer_id, version_number=version_number, client=client).parsed + return sync_detailed( + scorer_id=scorer_id, +version_number=version_number, +client=client, + ).parsed async def asyncio_detailed( - scorer_id: str, version_number: int, *, client: ApiClient + scorer_id: str, + version_number: int, + *, + client: ApiClient, + ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Restore Scorer Version + """ Restore Scorer Version List all scorers. @@ -121,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, version_number=version_number) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +version_number=version_number, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, version_number: int, *, client: ApiClient + scorer_id: str, + version_number: int, + *, + client: ApiClient, + ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """Restore Scorer Version + """ Restore Scorer Version List all scorers. @@ -147,6 +192,12 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +version_number=version_number, +client=client, - return (await asyncio_detailed(scorer_id=scorer_id, version_number=version_number, client=client)).parsed + )).parsed 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..ec697a259 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 @@ -1,30 +1,43 @@ 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.scorer_response import ScorerResponse from ...models.update_scorer_request import UpdateScorerRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + scorer_id: str, + *, + body: UpdateScorerRequest, -def _get_kwargs(scorer_id: str, *, body: UpdateScorerRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/scorers/{scorer_id}", + "path": "/scorers/{scorer_id}".format(scorer_id=scorer_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(scorer_id: str, *, body: UpdateScorerRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + if response.status_code == : response_200 = ScorerResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ScorerResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ScorerResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest + scorer_id: str, + *, + client: ApiClient, + body: UpdateScorerRequest, + ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """Update + """ Update Args: scorer_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest + scorer_id: str, + *, + client: ApiClient, + body: UpdateScorerRequest, + ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """Update + """ Update Args: scorer_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ + - return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed + return sync_detailed( + scorer_id=scorer_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest + scorer_id: str, + *, + client: ApiClient, + body: UpdateScorerRequest, + ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """Update + """ Update Args: scorer_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ - kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + scorer_id=scorer_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest + scorer_id: str, + *, + client: ApiClient, + body: UpdateScorerRequest, + ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """Update + """ Update Args: scorer_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ + + + return (await asyncio_detailed( + scorer_id=scorer_id, +client=client, +body=body, - return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed + )).parsed 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..03015ca14 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 @@ -1,21 +1,32 @@ 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.validate_llm_scorer_log_record_request import ValidateLLMScorerLogRecordRequest from ...models.validate_llm_scorer_log_record_response import ValidateLLMScorerLogRecordResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: ValidateLLMScorerLogRecordRequest, -def _get_kwargs(*, body: ValidateLLMScorerLogRecordRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, @@ -25,6 +36,7 @@ def _get_kwargs(*, body: ValidateLLMScorerLogRecordRequest) -> dict[str, Any]: _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +45,18 @@ def _get_kwargs(*, body: ValidateLLMScorerLogRecordRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: + if response.status_code == : response_200 = ValidateLLMScorerLogRecordResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: ValidateLLMScorerLogRecordRequest + *, + client: ApiClient, + body: ValidateLLMScorerLogRecordRequest, + ) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """Validate Llm Scorer Log Record + """ Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -78,19 +93,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]] - """ + """ - 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: ValidateLLMScorerLogRecordRequest + *, + client: ApiClient, + body: ValidateLLMScorerLogRecordRequest, + ) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """Validate Llm Scorer Log Record + """ Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -104,15 +127,22 @@ def sync( Returns: Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: ValidateLLMScorerLogRecordRequest + *, + client: ApiClient, + body: ValidateLLMScorerLogRecordRequest, + ) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """Validate Llm Scorer Log Record + """ Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -126,19 +156,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]] - """ + """ - 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: ValidateLLMScorerLogRecordRequest + *, + client: ApiClient, + body: ValidateLLMScorerLogRecordRequest, + ) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """Validate Llm Scorer Log Record + """ Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -152,6 +190,11 @@ async def asyncio( Returns: Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse] - """ + """ + + + 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/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..2db4082d9 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 @@ -1,20 +1,32 @@ 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.bulk_delete_datasets_request import BulkDeleteDatasetsRequest from ...models.bulk_delete_datasets_response import BulkDeleteDatasetsResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: BulkDeleteDatasetsRequest, -def _get_kwargs(*, body: BulkDeleteDatasetsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, @@ -24,22 +36,27 @@ def _get_kwargs(*, body: BulkDeleteDatasetsRequest) -> dict[str, Any]: _kwargs["json"] = body.to_dict() + 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[BulkDeleteDatasetsResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BulkDeleteDatasetsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -47,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -59,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: BulkDeleteDatasetsRequest + *, + client: ApiClient, + body: BulkDeleteDatasetsRequest, + ) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """Bulk Delete Datasets + """ Bulk Delete Datasets Delete multiple datasets in bulk. @@ -93,19 +111,27 @@ def sync_detailed( Returns: Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]] - """ + """ - 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: BulkDeleteDatasetsRequest + *, + client: ApiClient, + body: BulkDeleteDatasetsRequest, + ) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """Bulk Delete Datasets + """ Bulk Delete Datasets Delete multiple datasets in bulk. @@ -137,15 +163,22 @@ def sync( Returns: Union[BulkDeleteDatasetsResponse, HTTPValidationError] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: BulkDeleteDatasetsRequest + *, + client: ApiClient, + body: BulkDeleteDatasetsRequest, + ) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """Bulk Delete Datasets + """ Bulk Delete Datasets Delete multiple datasets in bulk. @@ -177,19 +210,27 @@ async def asyncio_detailed( Returns: Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]] - """ + """ - 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: BulkDeleteDatasetsRequest + *, + client: ApiClient, + body: BulkDeleteDatasetsRequest, + ) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """Bulk Delete Datasets + """ Bulk Delete Datasets Delete multiple datasets in bulk. @@ -221,6 +262,11 @@ async def asyncio( Returns: Union[BulkDeleteDatasetsResponse, HTTPValidationError] - """ + """ + + + 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/datasets/create_dataset_datasets_post.py b/src/galileo/resources/api/datasets/create_dataset_datasets_post.py index 88f6c6da3..d8ee6b632 100644 --- a/src/galileo/resources/api/datasets/create_dataset_datasets_post.py +++ b/src/galileo/resources/api/datasets/create_dataset_datasets_post.py @@ -1,18 +1,22 @@ 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.body_create_dataset_datasets_post import BodyCreateDatasetDatasetsPost from ...models.dataset_db import DatasetDB from ...models.dataset_format import DatasetFormat from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -20,8 +24,12 @@ def _get_kwargs( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -33,8 +41,10 @@ def _get_kwargs( params["hidden"] = hidden + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -44,6 +54,8 @@ def _get_kwargs( _kwargs["files"] = body.to_multipart() + + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -51,13 +63,17 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: - if response.status_code == 200: + if response.status_code == : response_200 = DatasetDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -80,8 +96,9 @@ def sync_detailed( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """Create Dataset + """ Create Dataset Creates a standalone dataset. @@ -96,14 +113,21 @@ def sync_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body, format_=format_, hidden=hidden) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, +format_=format_, +hidden=hidden, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( *, @@ -111,8 +135,9 @@ def sync( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """Create Dataset + """ Create Dataset Creates a standalone dataset. @@ -127,10 +152,16 @@ def sync( Returns: Union[DatasetDB, HTTPValidationError] - """ + """ + - return sync_detailed(client=client, body=body, format_=format_, hidden=hidden).parsed + return sync_detailed( + client=client, +body=body, +format_=format_, +hidden=hidden, + ).parsed async def asyncio_detailed( *, @@ -138,8 +169,9 @@ async def asyncio_detailed( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """Create Dataset + """ Create Dataset Creates a standalone dataset. @@ -154,14 +186,21 @@ async def asyncio_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body, format_=format_, hidden=hidden) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, +format_=format_, +hidden=hidden, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -169,8 +208,9 @@ async def asyncio( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """Create Dataset + """ Create Dataset Creates a standalone dataset. @@ -185,6 +225,13 @@ async def asyncio( Returns: Union[DatasetDB, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + client=client, +body=body, +format_=format_, +hidden=hidden, - return (await asyncio_detailed(client=client, body=body, format_=format_, hidden=hidden)).parsed + )).parsed 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..3ddbc9c44 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 @@ -1,26 +1,38 @@ 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.group_collaborator import GroupCollaborator from ...models.group_collaborator_create import GroupCollaboratorCreate from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, + *, + body: list['GroupCollaboratorCreate'], -def _get_kwargs(dataset_id: str, *, body: list["GroupCollaboratorCreate"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/groups", + "path": "/datasets/{dataset_id}/groups".format(dataset_id=dataset_id,), } _kwargs["json"] = [] @@ -28,6 +40,9 @@ def _get_kwargs(dataset_id: str, *, body: list["GroupCollaboratorCreate"]) -> di body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) + + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -36,21 +51,23 @@ def _get_kwargs(dataset_id: str, *, body: list["GroupCollaboratorCreate"]) -> di return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = GroupCollaborator.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,9 +85,13 @@ def _build_response( def sync_detailed( - dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Dataset Collaborators Share a dataset with groups. @@ -86,19 +105,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Dataset Collaborators Share a dataset with groups. @@ -112,15 +141,24 @@ def sync( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ + """ + - return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Dataset Collaborators Share a dataset with groups. @@ -134,19 +172,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Dataset Collaborators Share a dataset with groups. @@ -160,6 +208,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed + )).parsed 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..75946db45 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 @@ -1,26 +1,38 @@ 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.user_collaborator import UserCollaborator from ...models.user_collaborator_create import UserCollaboratorCreate -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, + *, + body: list['UserCollaboratorCreate'], -def _get_kwargs(dataset_id: str, *, body: list["UserCollaboratorCreate"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/users", + "path": "/datasets/{dataset_id}/users".format(dataset_id=dataset_id,), } _kwargs["json"] = [] @@ -28,6 +40,9 @@ def _get_kwargs(dataset_id: str, *, body: list["UserCollaboratorCreate"]) -> dic body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) + + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -36,21 +51,23 @@ def _get_kwargs(dataset_id: str, *, body: list["UserCollaboratorCreate"]) -> dic return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = UserCollaborator.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,9 +85,13 @@ def _build_response( def sync_detailed( - dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Dataset Collaborators Args: dataset_id (str): @@ -84,19 +103,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Dataset Collaborators Args: dataset_id (str): @@ -108,15 +137,24 @@ def sync( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ + """ + - return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Dataset Collaborators Args: dataset_id (str): @@ -128,19 +166,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Dataset Collaborators + dataset_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Dataset Collaborators Args: dataset_id (str): @@ -152,6 +200,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed + )).parsed 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..367e560c9 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 @@ -1,26 +1,37 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(dataset_id: str) -> dict[str, Any]: +def _get_kwargs( + dataset_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/datasets/{dataset_id}", + "path": "/datasets/{dataset_id}".format(dataset_id=dataset_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +39,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +63,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete Dataset +def sync_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete Dataset Args: dataset_id (str): @@ -62,17 +80,27 @@ def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + - kwargs = _get_kwargs(dataset_id=dataset_id) + kwargs = _get_kwargs( + dataset_id=dataset_id, - response = client.request(**kwargs) + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + dataset_id: str, + *, + client: ApiClient, -def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Dataset Args: dataset_id (str): @@ -83,13 +111,22 @@ def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValid Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, -async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete Dataset + ).parsed + +async def asyncio_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete Dataset Args: dataset_id (str): @@ -100,17 +137,27 @@ async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Un Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + dataset_id: str, + *, + client: ApiClient, -async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Dataset Args: dataset_id (str): @@ -121,6 +168,11 @@ async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, - return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed + )).parsed 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..1423dc8a8 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(dataset_id: str, group_id: str) -> dict[str, Any]: +def _get_kwargs( + dataset_id: str, + group_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/groups/{group_id}", + "path": "/datasets/{dataset_id}/groups/{group_id}".format(dataset_id=dataset_id,group_id=group_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(dataset_id: str, group_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete Group Dataset Collaborator +def sync_detailed( + dataset_id: str, + group_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -65,17 +85,29 @@ def sync_detailed(dataset_id: str, group_id: str, *, client: ApiClient) -> Respo Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +group_id=group_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + dataset_id: str, + group_id: str, + *, + client: ApiClient, -def sync(dataset_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Group Dataset Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -89,15 +121,24 @@ def sync(dataset_id: str, group_id: str, *, client: ApiClient) -> Optional[Union Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, group_id=group_id, client=client).parsed + return sync_detailed( + dataset_id=dataset_id, +group_id=group_id, +client=client, + + ).parsed async def asyncio_detailed( - dataset_id: str, group_id: str, *, client: ApiClient + dataset_id: str, + group_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Group Dataset Collaborator + """ Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -111,17 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +group_id=group_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + dataset_id: str, + group_id: str, + *, + client: ApiClient, -async def asyncio(dataset_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Group Dataset Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -135,6 +188,12 @@ async def asyncio(dataset_id: str, group_id: str, *, client: ApiClient) -> Optio Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +group_id=group_id, +client=client, - return (await asyncio_detailed(dataset_id=dataset_id, group_id=group_id, client=client)).parsed + )).parsed 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..6803e6d3e 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(project_id: str, dataset_id: str) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + dataset_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}/prompt_datasets/{dataset_id}", + "path": "/projects/{project_id}/prompt_datasets/{dataset_id}".format(project_id=project_id,dataset_id=dataset_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(project_id: str, dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete Prompt Dataset +def sync_detailed( + project_id: str, + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete Prompt Dataset Args: project_id (str): @@ -63,17 +83,29 @@ def sync_detailed(project_id: str, dataset_id: str, *, client: ApiClient) -> Res Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +dataset_id=dataset_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + dataset_id: str, + *, + client: ApiClient, -def sync(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Prompt Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Prompt Dataset Args: project_id (str): @@ -85,15 +117,24 @@ def sync(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Uni Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, dataset_id=dataset_id, client=client).parsed + return sync_detailed( + project_id=project_id, +dataset_id=dataset_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, dataset_id: str, *, client: ApiClient + project_id: str, + dataset_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Prompt Dataset + """ Delete Prompt Dataset Args: project_id (str): @@ -105,17 +146,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +dataset_id=dataset_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + project_id: str, + dataset_id: str, + *, + client: ApiClient, -async def asyncio(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Prompt Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Prompt Dataset Args: project_id (str): @@ -127,6 +180,12 @@ async def asyncio(project_id: str, dataset_id: str, *, client: ApiClient) -> Opt Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +dataset_id=dataset_id, +client=client, - return (await asyncio_detailed(project_id=project_id, dataset_id=dataset_id, client=client)).parsed + )).parsed 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..fd157083c 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(dataset_id: str, user_id: str) -> dict[str, Any]: +def _get_kwargs( + dataset_id: str, + user_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/users/{user_id}", + "path": "/datasets/{dataset_id}/users/{user_id}".format(dataset_id=dataset_id,user_id=user_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(dataset_id: str, user_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete User Dataset Collaborator +def sync_detailed( + dataset_id: str, + user_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -65,17 +85,29 @@ def sync_detailed(dataset_id: str, user_id: str, *, client: ApiClient) -> Respon Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +user_id=user_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + dataset_id: str, + user_id: str, + *, + client: ApiClient, -def sync(dataset_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete User Dataset Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -89,15 +121,24 @@ def sync(dataset_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[ Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, user_id=user_id, client=client).parsed + return sync_detailed( + dataset_id=dataset_id, +user_id=user_id, +client=client, + + ).parsed async def asyncio_detailed( - dataset_id: str, user_id: str, *, client: ApiClient + dataset_id: str, + user_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete User Dataset Collaborator + """ Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -111,17 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +user_id=user_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + dataset_id: str, + user_id: str, + *, + client: ApiClient, -async def asyncio(dataset_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete User Dataset Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -135,6 +188,12 @@ async def asyncio(dataset_id: str, user_id: str, *, client: ApiClient) -> Option Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +user_id=user_id, +client=client, - return (await asyncio_detailed(dataset_id=dataset_id, user_id=user_id, client=client)).parsed + )).parsed 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..5a0ae38be 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 @@ -1,26 +1,37 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(dataset_id: str) -> dict[str, Any]: +def _get_kwargs( + dataset_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/download", + "path": "/datasets/{dataset_id}/download".format(dataset_id=dataset_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +39,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +63,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Download Dataset +def sync_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Download Dataset Args: dataset_id (str): @@ -62,17 +80,27 @@ def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + - kwargs = _get_kwargs(dataset_id=dataset_id) + kwargs = _get_kwargs( + dataset_id=dataset_id, - response = client.request(**kwargs) + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + dataset_id: str, + *, + client: ApiClient, -def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Download Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Download Dataset Args: dataset_id (str): @@ -83,13 +111,22 @@ def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValid Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, -async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Download Dataset + ).parsed + +async def asyncio_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Download Dataset Args: dataset_id (str): @@ -100,17 +137,27 @@ async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Un Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + dataset_id: str, + *, + client: ApiClient, -async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Download Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Download Dataset Args: dataset_id (str): @@ -121,6 +168,11 @@ async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, - return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed + )).parsed 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..46d63207e 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 @@ -6,21 +6,33 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(project_id: str, dataset_id: str) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + dataset_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}/prompt_datasets/{dataset_id}", + "path": "/projects/{project_id}/prompt_datasets/{dataset_id}".format(project_id=project_id,dataset_id=dataset_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = cast(Any, None) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(project_id: str, dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Download Prompt Dataset +def sync_detailed( + project_id: str, + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Download Prompt Dataset Args: project_id (str): @@ -63,17 +83,29 @@ def sync_detailed(project_id: str, dataset_id: str, *, client: ApiClient) -> Res Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +dataset_id=dataset_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + dataset_id: str, + *, + client: ApiClient, -def sync(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Download Prompt Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Download Prompt Dataset Args: project_id (str): @@ -85,15 +117,24 @@ def sync(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Uni Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, dataset_id=dataset_id, client=client).parsed + return sync_detailed( + project_id=project_id, +dataset_id=dataset_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, dataset_id: str, *, client: ApiClient + project_id: str, + dataset_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Download Prompt Dataset + """ Download Prompt Dataset Args: project_id (str): @@ -105,17 +146,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +dataset_id=dataset_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + project_id: str, + dataset_id: str, + *, + client: ApiClient, -async def asyncio(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Download Prompt Dataset +) -> Optional[Union[Any, HTTPValidationError]]: + """ Download Prompt Dataset Args: project_id (str): @@ -127,6 +180,12 @@ async def asyncio(project_id: str, dataset_id: str, *, client: ApiClient) -> Opt Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +dataset_id=dataset_id, +client=client, - return (await asyncio_detailed(project_id=project_id, dataset_id=dataset_id, client=client)).parsed + )).parsed 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..8f73c3676 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 @@ -1,26 +1,42 @@ 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.synthetic_dataset_extension_request import SyntheticDatasetExtensionRequest from ...models.synthetic_dataset_extension_response import SyntheticDatasetExtensionResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: SyntheticDatasetExtensionRequest, -def _get_kwargs(*, body: SyntheticDatasetExtensionRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/datasets/extend"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/datasets/extend", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,16 +45,18 @@ def _get_kwargs(*, body: SyntheticDatasetExtensionRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: + if response.status_code == : response_200 = SyntheticDatasetExtensionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -46,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -58,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: SyntheticDatasetExtensionRequest + *, + client: ApiClient, + body: SyntheticDatasetExtensionRequest, + ) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """Extend Dataset Content + """ Extend Dataset Content Extends the dataset content @@ -73,19 +92,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]] - """ + """ - 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: SyntheticDatasetExtensionRequest + *, + client: ApiClient, + body: SyntheticDatasetExtensionRequest, + ) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """Extend Dataset Content + """ Extend Dataset Content Extends the dataset content @@ -98,15 +125,22 @@ def sync( Returns: Union[HTTPValidationError, SyntheticDatasetExtensionResponse] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: SyntheticDatasetExtensionRequest + *, + client: ApiClient, + body: SyntheticDatasetExtensionRequest, + ) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """Extend Dataset Content + """ Extend Dataset Content Extends the dataset content @@ -119,19 +153,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]] - """ + """ - 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: SyntheticDatasetExtensionRequest + *, + client: ApiClient, + body: SyntheticDatasetExtensionRequest, + ) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """Extend Dataset Content + """ Extend Dataset Content Extends the dataset content @@ -144,6 +186,11 @@ async def asyncio( Returns: Union[HTTPValidationError, SyntheticDatasetExtensionResponse] - """ + """ + + + 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/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..dec8ca58e 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 @@ -1,22 +1,33 @@ 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.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/content", + "path": "/datasets/{dataset_id}/content".format(dataset_id=dataset_id,), "params": params, } + 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[DatasetContent, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == : response_200 = DatasetContent.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DatasetContent, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Content + """ Get Dataset Content Args: dataset_id (str): @@ -83,19 +102,31 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Content + """ Get Dataset Content Args: dataset_id (str): @@ -108,15 +139,26 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Content + """ Get Dataset Content Args: dataset_id (str): @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Content + """ Get Dataset Content Args: dataset_id (str): @@ -154,8 +208,13 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..11a725868 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 @@ -1,27 +1,38 @@ 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.dataset_db import DatasetDB from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + -def _get_kwargs(dataset_id: str) -> dict[str, Any]: +def _get_kwargs( + dataset_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}", + "path": "/datasets/{dataset_id}".format(dataset_id=dataset_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -29,13 +40,17 @@ 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: + if response.status_code == : response_200 = DatasetDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -52,8 +67,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[DatasetDB, HTTPValidationError]]: - """Get Dataset +def sync_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[DatasetDB, HTTPValidationError]]: + """ Get Dataset Args: dataset_id (str): @@ -64,17 +84,27 @@ def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Datas Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + dataset_id: str, + *, + client: ApiClient, -def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """Get Dataset +) -> Optional[Union[DatasetDB, HTTPValidationError]]: + """ Get Dataset Args: dataset_id (str): @@ -85,13 +115,22 @@ def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[DatasetDB, HTT Returns: Union[DatasetDB, HTTPValidationError] - """ + """ + + + return sync_detailed( + dataset_id=dataset_id, +client=client, - return sync_detailed(dataset_id=dataset_id, client=client).parsed + ).parsed +async def asyncio_detailed( + dataset_id: str, + *, + client: ApiClient, -async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[DatasetDB, HTTPValidationError]]: - """Get Dataset +) -> Response[Union[DatasetDB, HTTPValidationError]]: + """ Get Dataset Args: dataset_id (str): @@ -102,17 +141,27 @@ async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Un Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + dataset_id: str, + *, + client: ApiClient, -async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """Get Dataset +) -> Optional[Union[DatasetDB, HTTPValidationError]]: + """ Get Dataset Args: dataset_id (str): @@ -123,6 +172,11 @@ async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Datas Returns: Union[DatasetDB, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, - return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed + )).parsed 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..1e896d439 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 @@ -1,43 +1,56 @@ 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.job_progress import JobProgress -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, -def _get_kwargs(dataset_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"/datasets/extend/{dataset_id}", + "path": "/datasets/extend/{dataset_id}".format(dataset_id=dataset_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, JobProgress]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, JobProgress]]: + if response.status_code == : response_200 = JobProgress.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, JobProgress]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, JobProgress]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -56,8 +67,13 @@ def _build_response( ) -def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobProgress]]: - """Get Dataset Synthetic Extend Status +def sync_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, JobProgress]]: + """ Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -68,17 +84,27 @@ def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[HTTPV Returns: Response[Union[HTTPValidationError, JobProgress]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + dataset_id: str, + *, + client: ApiClient, -def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobProgress]]: - """Get Dataset Synthetic Extend Status +) -> Optional[Union[HTTPValidationError, JobProgress]]: + """ Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -89,13 +115,22 @@ def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidation Returns: Union[HTTPValidationError, JobProgress] - """ + """ + - return sync_detailed(dataset_id=dataset_id, client=client).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, + ).parsed -async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobProgress]]: - """Get Dataset Synthetic Extend Status +async def asyncio_detailed( + dataset_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, JobProgress]]: + """ Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -106,17 +141,27 @@ async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Un Returns: Response[Union[HTTPValidationError, JobProgress]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + dataset_id: str, + *, + client: ApiClient, -async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobProgress]]: - """Get Dataset Synthetic Extend Status +) -> Optional[Union[HTTPValidationError, JobProgress]]: + """ Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -127,6 +172,11 @@ async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[HTTPV Returns: Union[HTTPValidationError, JobProgress] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, - return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed + )).parsed 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..f7c395336 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 @@ -1,22 +1,34 @@ 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.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - dataset_id: str, version_index: int, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + version_index: int, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +36,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/versions/{version_index}/content", + "path": "/datasets/{dataset_id}/versions/{version_index}/content".format(dataset_id=dataset_id,version_index=version_index,), "params": params, } + 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[DatasetContent, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == : response_200 = DatasetContent.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +73,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DatasetContent, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,8 +89,9 @@ def sync_detailed( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Version Content + """ Get Dataset Version Content Args: dataset_id (str): @@ -89,14 +105,22 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +version_index=version_index, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( dataset_id: str, @@ -105,8 +129,9 @@ def sync( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Version Content + """ Get Dataset Version Content Args: dataset_id (str): @@ -120,12 +145,17 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ + return sync_detailed( - dataset_id=dataset_id, version_index=version_index, client=client, starting_token=starting_token, limit=limit - ).parsed + dataset_id=dataset_id, +version_index=version_index, +client=client, +starting_token=starting_token, +limit=limit, + ).parsed async def asyncio_detailed( dataset_id: str, @@ -134,8 +164,9 @@ async def asyncio_detailed( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Version Content + """ Get Dataset Version Content Args: dataset_id (str): @@ -149,14 +180,22 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +version_index=version_index, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = await client.arequest( + **kwargs + ) + + return _build_response(client=client, response=response) async def asyncio( dataset_id: str, @@ -165,8 +204,9 @@ async def asyncio( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Get Dataset Version Content + """ Get Dataset Version Content Args: dataset_id (str): @@ -180,14 +220,14 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ - - return ( - await asyncio_detailed( - dataset_id=dataset_id, - version_index=version_index, - client=client, - starting_token=starting_token, - limit=limit, - ) - ).parsed + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +version_index=version_index, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..c56ada373 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 @@ -1,22 +1,33 @@ 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.list_dataset_projects_response import ListDatasetProjectsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/projects", + "path": "/datasets/{dataset_id}/projects".format(dataset_id=dataset_id,), "params": params, } + 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, ListDatasetProjectsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: + if response.status_code == : response_200 = ListDatasetProjectsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """List Dataset Projects + """ List Dataset Projects Args: dataset_id (str): @@ -83,19 +102,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetProjectsResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """List Dataset Projects + """ List Dataset Projects Args: dataset_id (str): @@ -108,15 +139,26 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetProjectsResponse] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """List Dataset Projects + """ List Dataset Projects Args: dataset_id (str): @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetProjectsResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """List Dataset Projects + """ List Dataset Projects Args: dataset_id (str): @@ -154,8 +208,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetProjectsResponse] - """ + """ - return ( - await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..01998c47d 100644 --- a/src/galileo/resources/api/datasets/list_datasets_datasets_get.py +++ b/src/galileo/resources/api/datasets/list_datasets_datasets_get.py @@ -1,17 +1,21 @@ 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.dataset_action import DatasetAction from ...models.http_validation_error import HTTPValidationError from ...models.list_dataset_response import ListDatasetResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -19,8 +23,12 @@ def _get_kwargs( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -31,14 +39,17 @@ def _get_kwargs( actions_item = actions_item_data.value json_actions.append(actions_item) + params["actions"] = json_actions params["starting_token"] = starting_token params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, @@ -46,22 +57,25 @@ def _get_kwargs( "params": params, } + 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, ListDatasetResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: + if response.status_code == : response_200 = ListDatasetResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -69,9 +83,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -86,8 +98,9 @@ def sync_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """List Datasets + """ List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -101,14 +114,21 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ - kwargs = _get_kwargs(actions=actions, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + actions=actions, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( *, @@ -116,8 +136,9 @@ def sync( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """List Datasets + """ List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -131,10 +152,16 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ + """ - return sync_detailed(client=client, actions=actions, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + client=client, +actions=actions, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( *, @@ -142,8 +169,9 @@ async def asyncio_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """List Datasets + """ List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -157,14 +185,21 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ - kwargs = _get_kwargs(actions=actions, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + actions=actions, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -172,8 +207,9 @@ async def asyncio( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """List Datasets + """ List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -187,6 +223,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ + """ + + + return (await asyncio_detailed( + client=client, +actions=actions, +starting_token=starting_token, +limit=limit, - return (await asyncio_detailed(client=client, actions=actions, starting_token=starting_token, limit=limit)).parsed + )).parsed 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..a7cee099b 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 @@ -1,22 +1,33 @@ 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.list_group_collaborators_response import ListGroupCollaboratorsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/groups", + "path": "/datasets/{dataset_id}/groups".format(dataset_id=dataset_id,), "params": params, } + 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, ListGroupCollaboratorsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: + if response.status_code == : response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Dataset Collaborators + """ List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Dataset Collaborators + """ List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -112,15 +143,26 @@ def sync( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Dataset Collaborators + """ List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Dataset Collaborators + """ List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ - return ( - await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..f3589352c 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 @@ -1,22 +1,33 @@ 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.list_prompt_dataset_response import ListPromptDatasetResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - project_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/prompt_datasets", + "path": "/projects/{project_id}/prompt_datasets".format(project_id=project_id,), "params": params, } + 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, ListPromptDatasetResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: + if response.status_code == : response_200 = ListPromptDatasetResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """List Prompt Datasets + """ List Prompt Datasets Args: project_id (str): @@ -83,19 +102,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListPromptDatasetResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """List Prompt Datasets + """ List Prompt Datasets Args: project_id (str): @@ -108,15 +139,26 @@ def sync( Returns: Union[HTTPValidationError, ListPromptDatasetResponse] - """ + """ - return sync_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + project_id=project_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """List Prompt Datasets + """ List Prompt Datasets Args: project_id (str): @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListPromptDatasetResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +starting_token=starting_token, +limit=limit, - 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, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """List Prompt Datasets + """ List Prompt Datasets Args: project_id (str): @@ -154,8 +208,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListPromptDatasetResponse] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..8e1b154f2 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 @@ -1,22 +1,33 @@ 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.list_user_collaborators_response import ListUserCollaboratorsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/users", + "path": "/datasets/{dataset_id}/users".format(dataset_id=dataset_id,), "params": params, } + 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, ListUserCollaboratorsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: + if response.status_code == : response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Dataset Collaborators + """ List User Dataset Collaborators List the users with which the dataset has been shared. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Dataset Collaborators + """ List User Dataset Collaborators List the users with which the dataset has been shared. @@ -112,15 +143,26 @@ def sync( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Dataset Collaborators + """ List User Dataset Collaborators List the users with which the dataset has been shared. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Dataset Collaborators + """ List User Dataset Collaborators List the users with which the dataset has been shared. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ - return ( - await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..f5ecdaa47 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 @@ -1,17 +1,21 @@ 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.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError from ...models.preview_dataset_request import PreviewDatasetRequest -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -20,8 +24,12 @@ def _get_kwargs( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -29,17 +37,20 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/preview", + "path": "/datasets/{dataset_id}/preview".format(dataset_id=dataset_id,), "params": params, } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -48,16 +59,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[DatasetContent, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == : response_200 = DatasetContent.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -65,9 +78,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DatasetContent, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -83,8 +94,9 @@ def sync_detailed( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Preview Dataset + """ Preview Dataset Args: dataset_id (str): @@ -98,14 +110,22 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( dataset_id: str, @@ -114,8 +134,9 @@ def sync( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Preview Dataset + """ Preview Dataset Args: dataset_id (str): @@ -129,12 +150,17 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ + return sync_detailed( - dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit - ).parsed + dataset_id=dataset_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + ).parsed async def asyncio_detailed( dataset_id: str, @@ -143,8 +169,9 @@ async def asyncio_detailed( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Preview Dataset + """ Preview Dataset Args: dataset_id (str): @@ -158,14 +185,22 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( dataset_id: str, @@ -174,8 +209,9 @@ async def asyncio( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Preview Dataset + """ Preview Dataset Args: dataset_id (str): @@ -189,10 +225,14 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed( - dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit - ) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + + )).parsed 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..2e2565f4d 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 @@ -1,23 +1,35 @@ 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.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError from ...models.query_dataset_params import QueryDatasetParams -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - dataset_id: str, *, body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + dataset_id: str, + *, + body: QueryDatasetParams, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -25,17 +37,20 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/content/query", + "path": "/datasets/{dataset_id}/content/query".format(dataset_id=dataset_id,), "params": params, } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -44,16 +59,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[DatasetContent, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == : response_200 = DatasetContent.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -61,9 +78,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DatasetContent, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -79,8 +94,9 @@ def sync_detailed( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Query Dataset Content + """ Query Dataset Content Args: dataset_id (str): @@ -94,14 +110,22 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( dataset_id: str, @@ -110,8 +134,9 @@ def sync( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Query Dataset Content + """ Query Dataset Content Args: dataset_id (str): @@ -125,12 +150,17 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ + return sync_detailed( - dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit - ).parsed + dataset_id=dataset_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + ).parsed async def asyncio_detailed( dataset_id: str, @@ -139,8 +169,9 @@ async def asyncio_detailed( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """Query Dataset Content + """ Query Dataset Content Args: dataset_id (str): @@ -154,14 +185,22 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = await client.arequest( + **kwargs + ) + + return _build_response(client=client, response=response) async def asyncio( dataset_id: str, @@ -170,8 +209,9 @@ async def asyncio( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """Query Dataset Content + """ Query Dataset Content Args: dataset_id (str): @@ -185,10 +225,14 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed( - dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit - ) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + + )).parsed 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..0390d8ebe 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 @@ -1,17 +1,21 @@ 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.list_dataset_version_params import ListDatasetVersionParams from ...models.list_dataset_version_response import ListDatasetVersionResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -20,8 +24,12 @@ def _get_kwargs( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -29,17 +37,20 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/versions/query", + "path": "/datasets/{dataset_id}/versions/query".format(dataset_id=dataset_id,), "params": params, } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -48,16 +59,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: + if response.status_code == : response_200 = ListDatasetVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -65,9 +78,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -83,8 +94,9 @@ def sync_detailed( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """Query Dataset Versions + """ Query Dataset Versions Args: dataset_id (str): @@ -98,14 +110,22 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetVersionResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( dataset_id: str, @@ -114,8 +134,9 @@ def sync( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """Query Dataset Versions + """ Query Dataset Versions Args: dataset_id (str): @@ -129,12 +150,17 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetVersionResponse] - """ + """ + return sync_detailed( - dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit - ).parsed + dataset_id=dataset_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + ).parsed async def asyncio_detailed( dataset_id: str, @@ -143,8 +169,9 @@ async def asyncio_detailed( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """Query Dataset Versions + """ Query Dataset Versions Args: dataset_id (str): @@ -158,14 +185,22 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetVersionResponse]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( dataset_id: str, @@ -174,8 +209,9 @@ async def asyncio( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """Query Dataset Versions + """ Query Dataset Versions Args: dataset_id (str): @@ -189,10 +225,14 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetVersionResponse] - """ + """ - return ( - await asyncio_detailed( - dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit - ) - ).parsed + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + + )).parsed 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..7edc9453e 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 @@ -1,18 +1,22 @@ 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.dataset_action import DatasetAction from ...models.http_validation_error import HTTPValidationError from ...models.list_dataset_params import ListDatasetParams from ...models.list_dataset_response import ListDatasetResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -21,8 +25,12 @@ def _get_kwargs( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -33,14 +41,17 @@ def _get_kwargs( actions_item = actions_item_data.value json_actions.append(actions_item) + params["actions"] = json_actions params["starting_token"] = starting_token params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -50,6 +61,7 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -58,16 +70,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: + if response.status_code == : response_200 = ListDatasetResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -75,9 +89,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -93,8 +105,9 @@ def sync_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """Query Datasets + """ Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -109,14 +122,22 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ - kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, +actions=actions, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( *, @@ -125,8 +146,9 @@ def sync( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """Query Datasets + """ Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -141,10 +163,17 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ + """ - return sync_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + client=client, +body=body, +actions=actions, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( *, @@ -153,8 +182,9 @@ async def asyncio_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """Query Datasets + """ Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -169,14 +199,22 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ - kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, +actions=actions, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -185,8 +223,9 @@ async def asyncio( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """Query Datasets + """ Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -201,8 +240,14 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ + """ - return ( - await asyncio_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + client=client, +body=body, +actions=actions, +starting_token=starting_token, +limit=limit, + + )).parsed 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..e5b605879 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 @@ -1,33 +1,49 @@ 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.update_dataset_content_request import UpdateDatasetContentRequest -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + def _get_kwargs( - dataset_id: str, *, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET + dataset_id: str, + *, + body: UpdateDatasetContentRequest, + if_match: Union[None, Unset, str] = UNSET, + ) -> dict[str, Any]: headers: dict[str, Any] = {} if not isinstance(if_match, Unset): headers["If-Match"] = if_match + + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/content", + "path": "/datasets/{dataset_id}/content".format(dataset_id=dataset_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -37,12 +53,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == 200: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -60,9 +78,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetContentRequest, + if_match: Union[None, Unset, str] = UNSET, + ) -> Response[Union[Any, HTTPValidationError]]: - """Update Dataset Content + """ Update Dataset Content Update the content of a dataset. @@ -97,19 +120,31 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, if_match=if_match) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +if_match=if_match, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetContentRequest, + if_match: Union[None, Unset, str] = UNSET, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Update Dataset Content + """ Update Dataset Content Update the content of a dataset. @@ -144,15 +179,26 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client, body=body, if_match=if_match).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +body=body, +if_match=if_match, + + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetContentRequest, + if_match: Union[None, Unset, str] = UNSET, + ) -> Response[Union[Any, HTTPValidationError]]: - """Update Dataset Content + """ Update Dataset Content Update the content of a dataset. @@ -187,19 +233,31 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body, if_match=if_match) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, +if_match=if_match, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetContentRequest, + if_match: Union[None, Unset, str] = UNSET, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Update Dataset Content + """ Update Dataset Content Update the content of a dataset. @@ -234,6 +292,13 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, +if_match=if_match, - return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body, if_match=if_match)).parsed + )).parsed 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..0556eccb4 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 @@ -1,30 +1,43 @@ 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.dataset_db import DatasetDB from ...models.http_validation_error import HTTPValidationError from ...models.update_dataset_request import UpdateDatasetRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, + *, + body: UpdateDatasetRequest, -def _get_kwargs(dataset_id: str, *, body: UpdateDatasetRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/datasets/{dataset_id}", + "path": "/datasets/{dataset_id}".format(dataset_id=dataset_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -34,13 +47,17 @@ 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: + if response.status_code == : response_200 = DatasetDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetRequest, + ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """Update Dataset + """ Update Dataset Args: dataset_id (str): @@ -72,19 +93,29 @@ def sync_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetRequest, + ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """Update Dataset + """ Update Dataset Args: dataset_id (str): @@ -96,15 +127,24 @@ def sync( Returns: Union[DatasetDB, HTTPValidationError] - """ + """ - return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +body=body, + + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetRequest, + ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """Update Dataset + """ Update Dataset Args: dataset_id (str): @@ -116,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest + dataset_id: str, + *, + client: ApiClient, + body: UpdateDatasetRequest, + ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """Update Dataset + """ Update Dataset Args: dataset_id (str): @@ -140,6 +190,12 @@ async def asyncio( Returns: Union[DatasetDB, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed + )).parsed 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..6ab815c16 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 @@ -1,30 +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.dataset_version_db import DatasetVersionDB from ...models.http_validation_error import HTTPValidationError from ...models.update_dataset_version_request import UpdateDatasetVersionRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, + version_index: int, + *, + body: UpdateDatasetVersionRequest, -def _get_kwargs(dataset_id: str, version_index: int, *, body: UpdateDatasetVersionRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/versions/{version_index}", + "path": "/datasets/{dataset_id}/versions/{version_index}".format(dataset_id=dataset_id,version_index=version_index,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(dataset_id: str, version_index: int, *, body: UpdateDatasetVersi return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: + if response.status_code == : response_200 = DatasetVersionDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest + dataset_id: str, + version_index: int, + *, + client: ApiClient, + body: UpdateDatasetVersionRequest, + ) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: - """Update Dataset Version + """ Update Dataset Version Args: dataset_id (str): @@ -77,19 +96,31 @@ def sync_detailed( Returns: Response[Union[DatasetVersionDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +version_index=version_index, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest + dataset_id: str, + version_index: int, + *, + client: ApiClient, + body: UpdateDatasetVersionRequest, + ) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: - """Update Dataset Version + """ Update Dataset Version Args: dataset_id (str): @@ -102,15 +133,26 @@ def sync( Returns: Union[DatasetVersionDB, HTTPValidationError] - """ + """ + - return sync_detailed(dataset_id=dataset_id, version_index=version_index, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +version_index=version_index, +client=client, +body=body, + ).parsed async def asyncio_detailed( - dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest + dataset_id: str, + version_index: int, + *, + client: ApiClient, + body: UpdateDatasetVersionRequest, + ) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: - """Update Dataset Version + """ Update Dataset Version Args: dataset_id (str): @@ -123,19 +165,31 @@ async def asyncio_detailed( Returns: Response[Union[DatasetVersionDB, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +version_index=version_index, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest + dataset_id: str, + version_index: int, + *, + client: ApiClient, + body: UpdateDatasetVersionRequest, + ) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: - """Update Dataset Version + """ Update Dataset Version Args: dataset_id (str): @@ -148,6 +202,13 @@ async def asyncio( Returns: Union[DatasetVersionDB, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +version_index=version_index, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, version_index=version_index, client=client, body=body)).parsed + )).parsed 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..be1a366e9 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 @@ -1,30 +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.collaborator_update import CollaboratorUpdate from ...models.group_collaborator import GroupCollaborator from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, + group_id: str, + *, + body: CollaboratorUpdate, -def _get_kwargs(dataset_id: str, group_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/groups/{group_id}", + "path": "/datasets/{dataset_id}/groups/{group_id}".format(dataset_id=dataset_id,group_id=group_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(dataset_id: str, group_id: str, *, body: CollaboratorUpdate) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: + if response.status_code == : response_200 = GroupCollaborator.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GroupCollaborator, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GroupCollaborator, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Dataset Collaborator + """ Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +group_id=group_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Dataset Collaborator + """ Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -106,15 +137,26 @@ def sync( Returns: Union[GroupCollaborator, HTTPValidationError] - """ + """ + - return sync_detailed(dataset_id=dataset_id, group_id=group_id, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +group_id=group_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Dataset Collaborator + """ Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +group_id=group_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Dataset Collaborator + """ Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[GroupCollaborator, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +group_id=group_id, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, group_id=group_id, client=client, body=body)).parsed + )).parsed 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..f02dad0ee 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 @@ -1,20 +1,23 @@ 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.body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put import ( - BodyUpdatePromptDatasetProjectsProjectIdPromptDatasetsDatasetIdPut, -) + +from ...models.body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put import BodyUpdatePromptDatasetProjectsProjectIdPromptDatasetsDatasetIdPut from ...models.dataset_format import DatasetFormat from ...models.http_validation_error import HTTPValidationError from ...models.prompt_dataset_db import PromptDatasetDB -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + def _get_kwargs( @@ -26,8 +29,12 @@ def _get_kwargs( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -53,33 +60,39 @@ def _get_kwargs( params["hidden"] = hidden + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}/prompt_datasets/{dataset_id}", + "path": "/projects/{project_id}/prompt_datasets/{dataset_id}".format(project_id=project_id,dataset_id=dataset_id,), "params": params, } _kwargs["files"] = body.to_multipart() + + 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, PromptDatasetDB]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: + if response.status_code == : response_200 = PromptDatasetDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -87,9 +100,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -108,8 +119,9 @@ def sync_detailed( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """Update Prompt Dataset + """ Update Prompt Dataset Args: project_id (str): @@ -126,23 +138,26 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ + """ + kwargs = _get_kwargs( project_id=project_id, - dataset_id=dataset_id, - body=body, - file_name=file_name, - num_rows=num_rows, - format_=format_, - hidden=hidden, +dataset_id=dataset_id, +body=body, +file_name=file_name, +num_rows=num_rows, +format_=format_, +hidden=hidden, + ) - response = client.request(**kwargs) + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) - def sync( project_id: str, dataset_id: str, @@ -153,8 +168,9 @@ def sync( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """Update Prompt Dataset + """ Update Prompt Dataset Args: project_id (str): @@ -171,19 +187,20 @@ def sync( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ + """ + return sync_detailed( project_id=project_id, - dataset_id=dataset_id, - client=client, - body=body, - file_name=file_name, - num_rows=num_rows, - format_=format_, - hidden=hidden, - ).parsed +dataset_id=dataset_id, +client=client, +body=body, +file_name=file_name, +num_rows=num_rows, +format_=format_, +hidden=hidden, + ).parsed async def asyncio_detailed( project_id: str, @@ -195,8 +212,9 @@ async def asyncio_detailed( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """Update Prompt Dataset + """ Update Prompt Dataset Args: project_id (str): @@ -213,23 +231,26 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ + """ + kwargs = _get_kwargs( project_id=project_id, - dataset_id=dataset_id, - body=body, - file_name=file_name, - num_rows=num_rows, - format_=format_, - hidden=hidden, +dataset_id=dataset_id, +body=body, +file_name=file_name, +num_rows=num_rows, +format_=format_, +hidden=hidden, + ) - response = await client.arequest(**kwargs) + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) - async def asyncio( project_id: str, dataset_id: str, @@ -240,8 +261,9 @@ async def asyncio( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """Update Prompt Dataset + """ Update Prompt Dataset Args: project_id (str): @@ -258,17 +280,17 @@ async def asyncio( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ - - return ( - await asyncio_detailed( - project_id=project_id, - dataset_id=dataset_id, - client=client, - body=body, - file_name=file_name, - num_rows=num_rows, - format_=format_, - hidden=hidden, - ) - ).parsed + """ + + + return (await asyncio_detailed( + project_id=project_id, +dataset_id=dataset_id, +client=client, +body=body, +file_name=file_name, +num_rows=num_rows, +format_=format_, +hidden=hidden, + + )).parsed 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..2b4e86ba7 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 @@ -1,30 +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.collaborator_update import CollaboratorUpdate from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator -from ...types import Response +from typing import cast + + +def _get_kwargs( + dataset_id: str, + user_id: str, + *, + body: CollaboratorUpdate, -def _get_kwargs(dataset_id: str, user_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/users/{user_id}", + "path": "/datasets/{dataset_id}/users/{user_id}".format(dataset_id=dataset_id,user_id=user_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(dataset_id: str, user_id: str, *, body: CollaboratorUpdate) -> d return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, UserCollaborator]]: + if response.status_code == : response_200 = UserCollaborator.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, UserCollaborator]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, UserCollaborator]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """Update User Dataset Collaborator + """ Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +user_id=user_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """Update User Dataset Collaborator + """ Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -106,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, UserCollaborator] - """ + """ + - return sync_detailed(dataset_id=dataset_id, user_id=user_id, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +user_id=user_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """Update User Dataset Collaborator + """ Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +user_id=user_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + dataset_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """Update User Dataset Collaborator + """ Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, UserCollaborator] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +user_id=user_id, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, user_id=user_id, client=client, body=body)).parsed + )).parsed 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..cb333b6ae 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 @@ -1,20 +1,22 @@ 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.body_upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post import ( - BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, -) + +from ...models.body_upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post import BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost from ...models.dataset_format import DatasetFormat from ...models.http_validation_error import HTTPValidationError from ...models.prompt_dataset_db import PromptDatasetDB -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -23,8 +25,12 @@ def _get_kwargs( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -36,33 +42,39 @@ def _get_kwargs( params["hidden"] = hidden + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/prompt_datasets", + "path": "/projects/{project_id}/prompt_datasets".format(project_id=project_id,), "params": params, } _kwargs["files"] = body.to_multipart() + + 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, PromptDatasetDB]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: + if response.status_code == : response_200 = PromptDatasetDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -70,9 +82,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -88,8 +98,9 @@ def sync_detailed( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """Upload Prompt Evaluation Dataset + """ Upload Prompt Evaluation Dataset Args: project_id (str): @@ -103,14 +114,22 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body, format_=format_, hidden=hidden) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, +format_=format_, +hidden=hidden, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( project_id: str, @@ -119,8 +138,9 @@ def sync( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """Upload Prompt Evaluation Dataset + """ Upload Prompt Evaluation Dataset Args: project_id (str): @@ -134,10 +154,17 @@ def sync( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ + """ - return sync_detailed(project_id=project_id, client=client, body=body, format_=format_, hidden=hidden).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, +format_=format_, +hidden=hidden, + + ).parsed async def asyncio_detailed( project_id: str, @@ -146,8 +173,9 @@ async def asyncio_detailed( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """Upload Prompt Evaluation Dataset + """ Upload Prompt Evaluation Dataset Args: project_id (str): @@ -161,14 +189,22 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body, format_=format_, hidden=hidden) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, +format_=format_, +hidden=hidden, - 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, @@ -177,8 +213,9 @@ async def asyncio( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, + ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """Upload Prompt Evaluation Dataset + """ Upload Prompt Evaluation Dataset Args: project_id (str): @@ -192,8 +229,14 @@ async def asyncio( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, client=client, body=body, format_=format_, hidden=hidden) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, +format_=format_, +hidden=hidden, + + )).parsed 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..06868019d 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 @@ -1,26 +1,39 @@ 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.rollback_request import RollbackRequest from ...models.upsert_dataset_content_request import UpsertDatasetContentRequest -from ...types import Response +from typing import cast +from typing import cast, Union + + +def _get_kwargs( + dataset_id: str, + *, + body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], -def _get_kwargs(dataset_id: str, *, body: Union["RollbackRequest", "UpsertDatasetContentRequest"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/datasets/{dataset_id}/content", + "path": "/datasets/{dataset_id}/content".format(dataset_id=dataset_id,), } _kwargs["json"]: dict[str, Any] @@ -29,6 +42,8 @@ def _get_kwargs(dataset_id: str, *, body: Union["RollbackRequest", "UpsertDatase else: _kwargs["json"] = body.to_dict() + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -38,12 +53,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -61,9 +78,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] + dataset_id: str, + *, + client: ApiClient, + body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], + ) -> Response[Union[Any, HTTPValidationError]]: - """Upsert Dataset Content + """ Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -77,19 +98,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] + dataset_id: str, + *, + client: ApiClient, + body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], + ) -> Optional[Union[Any, HTTPValidationError]]: - """Upsert Dataset Content + """ Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -103,15 +134,24 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed + return sync_detailed( + dataset_id=dataset_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] + dataset_id: str, + *, + client: ApiClient, + body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], + ) -> Response[Union[Any, HTTPValidationError]]: - """Upsert Dataset Content + """ Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -125,19 +165,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + dataset_id=dataset_id, +body=body, - return _build_response(client=client, response=response) + ) + response = await client.arequest( + **kwargs + ) + + return _build_response(client=client, response=response) async def asyncio( - dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] + dataset_id: str, + *, + client: ApiClient, + body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], + ) -> Optional[Union[Any, HTTPValidationError]]: - """Upsert Dataset Content + """ Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -151,6 +201,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + dataset_id=dataset_id, +client=client, +body=body, - return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed + )).parsed 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..e7c8e6d24 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 @@ -1,30 +1,43 @@ 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.experiment_create_request import ExperimentCreateRequest from ...models.experiment_response import ExperimentResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: ExperimentCreateRequest, -def _get_kwargs(project_id: str, *, body: ExperimentCreateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments", + "path": "/projects/{project_id}/experiments".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: ExperimentCreateRequest) -> dict[str, return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: + if response.status_code == : response_200 = ExperimentResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExperimentResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExperimentResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: ExperimentCreateRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentCreateRequest, + ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """Create Experiment + """ Create Experiment Create a new experiment for a project. @@ -78,19 +95,29 @@ def sync_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: ExperimentCreateRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentCreateRequest, + ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """Create Experiment + """ Create Experiment Create a new experiment for a project. @@ -104,15 +131,24 @@ def sync( Returns: Union[ExperimentResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: ExperimentCreateRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentCreateRequest, + ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """Create Experiment + """ Create Experiment Create a new experiment for a project. @@ -126,19 +162,29 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: ExperimentCreateRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentCreateRequest, + ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """Create Experiment + """ Create Experiment Create a new experiment for a project. @@ -152,6 +198,12 @@ async def asyncio( Returns: Union[ExperimentResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..d070da4ef 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 @@ -6,21 +6,33 @@ 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 ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, -def _get_kwargs(project_id: str, experiment_id: str) -> dict[str, Any]: +) -> 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}", + "path": "/projects/{project_id}/experiments/{experiment_id}".format(project_id=project_id,experiment_id=experiment_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_204 = cast(Any, None) return response_204 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -51,9 +65,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Experiment + """ Delete Experiment Delete a specific experiment. @@ -67,17 +85,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + experiment_id: str, + *, + client: ApiClient, -def sync(project_id: str, experiment_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Experiment +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Experiment Delete a specific experiment. @@ -91,15 +121,24 @@ def sync(project_id: str, experiment_id: str, *, client: ApiClient) -> Optional[ Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Experiment + """ Delete Experiment Delete a specific experiment. @@ -113,19 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_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, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Experiment + """ Delete Experiment Delete a specific experiment. @@ -139,6 +188,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, - return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client)).parsed + )).parsed 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..7cfb916a1 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 @@ -1,43 +1,56 @@ 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.experiments_available_columns_response import ExperimentsAvailableColumnsResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, -def _get_kwargs(project_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/available_columns", + "path": "/projects/{project_id}/experiments/available_columns".format(project_id=project_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[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: + if response.status_code == : response_200 = ExperimentsAvailableColumnsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +68,12 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient + project_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """Experiments Available Columns + """ Experiments Available Columns Procures the column information for experiments. @@ -72,19 +86,27 @@ def sync_detailed( Returns: Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient + project_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """Experiments Available Columns + """ Experiments Available Columns Procures the column information for experiments. @@ -97,15 +119,22 @@ def sync( Returns: Union[ExperimentsAvailableColumnsResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client).parsed + return sync_detailed( + project_id=project_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient + project_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """Experiments Available Columns + """ Experiments Available Columns Procures the column information for experiments. @@ -118,19 +147,27 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_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, *, client: ApiClient + project_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """Experiments Available Columns + """ Experiments Available Columns Procures the column information for experiments. @@ -143,6 +180,11 @@ async def asyncio( Returns: Union[ExperimentsAvailableColumnsResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, - return (await asyncio_detailed(project_id=project_id, client=client)).parsed + )).parsed 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..de3b8ed39 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 @@ -1,30 +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.experiment_metrics_request import ExperimentMetricsRequest from ...models.experiment_metrics_response import ExperimentMetricsResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, + *, + body: ExperimentMetricsRequest, -def _get_kwargs(project_id: str, experiment_id: str, *, body: ExperimentMetricsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/{experiment_id}/metrics", + "path": "/projects/{project_id}/experiments/{experiment_id}/metrics".format(project_id=project_id,experiment_id=experiment_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, experiment_id: str, *, body: ExperimentMetricsR return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: + if response.status_code == : response_200 = ExperimentMetricsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiment Metrics + """ Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiment Metrics + """ Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -106,15 +137,26 @@ def sync( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiment Metrics + """ Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +body=body, - 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, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiment Metrics + """ Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body)).parsed + )).parsed 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..199d2b652 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 @@ -1,43 +1,57 @@ 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.experiment_response import ExperimentResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, -def _get_kwargs(project_id: str, experiment_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}/experiments/{experiment_id}", + "path": "/projects/{project_id}/experiments/{experiment_id}".format(project_id=project_id,experiment_id=experiment_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[ExperimentResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: + if response.status_code == : response_200 = ExperimentResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExperimentResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExperimentResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """Get Experiment + """ Get Experiment Retrieve a specific experiment. @@ -73,19 +89,29 @@ def sync_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """Get Experiment + """ Get Experiment Retrieve a specific experiment. @@ -99,15 +125,24 @@ def sync( Returns: Union[ExperimentResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """Get Experiment + """ Get Experiment Retrieve a specific experiment. @@ -121,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_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, experiment_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """Get Experiment + """ Get Experiment Retrieve a specific experiment. @@ -147,6 +192,12 @@ async def asyncio( Returns: Union[ExperimentResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, - return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client)).parsed + )).parsed 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..b044f0b8d 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 @@ -1,30 +1,43 @@ 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.experiment_metrics_request import ExperimentMetricsRequest from ...models.experiment_metrics_response import ExperimentMetricsResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: ExperimentMetricsRequest, -def _get_kwargs(project_id: str, *, body: ExperimentMetricsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/metrics", + "path": "/projects/{project_id}/experiments/metrics".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: ExperimentMetricsRequest) -> dict[str, return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: + if response.status_code == : response_200 = ExperimentMetricsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiments Metrics + """ Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -78,19 +95,29 @@ def sync_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiments Metrics + """ Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -104,15 +131,24 @@ def sync( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiments Metrics + """ Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -126,19 +162,29 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: ExperimentMetricsRequest + project_id: str, + *, + client: ApiClient, + body: ExperimentMetricsRequest, + ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """Get Experiments Metrics + """ Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -152,6 +198,12 @@ async def asyncio( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..4f4b6ca01 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 @@ -1,48 +1,61 @@ 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.experiment_response import ExperimentResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, -def _get_kwargs(project_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}/experiments", + "path": "/projects/{project_id}/experiments".format(project_id=project_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, list["ExperimentResponse"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['ExperimentResponse']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = ExperimentResponse.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +63,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["ExperimentResponse"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['ExperimentResponse']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +73,12 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["ExperimentResponse"]]]: - """List Experiments + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['ExperimentResponse']]]: + """ List Experiments Retrieve all experiments for a project. @@ -77,17 +91,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['ExperimentResponse']]] - """ + """ + - kwargs = _get_kwargs(project_id=project_id) + kwargs = _get_kwargs( + project_id=project_id, - response = client.request(**kwargs) + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + *, + client: ApiClient, -def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, list["ExperimentResponse"]]]: - """List Experiments +) -> Optional[Union[HTTPValidationError, list['ExperimentResponse']]]: + """ List Experiments Retrieve all experiments for a project. @@ -100,15 +124,22 @@ def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidation Returns: Union[HTTPValidationError, list['ExperimentResponse']] - """ + """ + - return sync_detailed(project_id=project_id, client=client).parsed + return sync_detailed( + project_id=project_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["ExperimentResponse"]]]: - """List Experiments + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['ExperimentResponse']]]: + """ List Experiments Retrieve all experiments for a project. @@ -121,19 +152,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['ExperimentResponse']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_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, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, list["ExperimentResponse"]]]: - """List Experiments + project_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, list['ExperimentResponse']]]: + """ List Experiments Retrieve all experiments for a project. @@ -146,6 +185,11 @@ async def asyncio( Returns: Union[HTTPValidationError, list['ExperimentResponse']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, - return (await asyncio_detailed(project_id=project_id, client=client)).parsed + )).parsed 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..10aaa72b5 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 @@ -1,30 +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.experiment_response import ExperimentResponse from ...models.experiment_update_request import ExperimentUpdateRequest from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, + *, + body: ExperimentUpdateRequest, -def _get_kwargs(project_id: str, experiment_id: str, *, body: ExperimentUpdateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/{experiment_id}", + "path": "/projects/{project_id}/experiments/{experiment_id}".format(project_id=project_id,experiment_id=experiment_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, experiment_id: str, *, body: ExperimentUpdateRe return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: + if response.status_code == : response_200 = ExperimentResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExperimentResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExperimentResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentUpdateRequest, + ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """Update Experiment + """ Update Experiment Update a specific experiment. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentUpdateRequest, + ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """Update Experiment + """ Update Experiment Update a specific experiment. @@ -106,15 +137,26 @@ def sync( Returns: Union[ExperimentResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentUpdateRequest, + ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """Update Experiment + """ Update Experiment Update a specific experiment. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +body=body, - 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, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: ExperimentUpdateRequest, + ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """Update Experiment + """ Update Experiment Update a specific experiment. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[ExperimentResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body)).parsed + )).parsed 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..f087ae5a9 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 @@ -1,37 +1,58 @@ 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.delete_run_response import DeleteRunResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + + +def _get_kwargs( + project_id: str, + experiment_id: str, + tag_id: str, +) -> dict[str, Any]: + headers: dict[str, Any] = {} + + + + + -def _get_kwargs(project_id: str, experiment_id: str, tag_id: str) -> 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}", + "path": "/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}".format(project_id=project_id,experiment_id=experiment_id,tag_id=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[DeleteRunResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: + if response.status_code == : response_200 = DeleteRunResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -39,9 +60,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -51,9 +70,14 @@ def _build_response( def sync_detailed( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: - """Delete Experiment Tag + """ Delete Experiment Tag Args: project_id (str): @@ -66,19 +90,31 @@ def sync_detailed( Returns: Response[Union[DeleteRunResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: - """Delete Experiment Tag + """ Delete Experiment Tag Args: project_id (str): @@ -91,15 +127,26 @@ def sync( Returns: Union[DeleteRunResponse, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: - """Delete Experiment Tag + """ Delete Experiment Tag Args: project_id (str): @@ -112,19 +159,31 @@ async def asyncio_detailed( Returns: Response[Union[DeleteRunResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_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, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: - """Delete Experiment Tag + """ Delete Experiment Tag Args: project_id (str): @@ -137,8 +196,13 @@ async def asyncio( Returns: Union[DeleteRunResponse, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +client=client, + + )).parsed 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..44adafa5f 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 @@ -1,35 +1,58 @@ 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.run_tag_db import RunTagDB -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, + tag_id: str, + +) -> dict[str, Any]: + headers: dict[str, Any] = {} + + + + + -def _get_kwargs(project_id: str, experiment_id: str, tag_id: str) -> 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}", + "path": "/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}".format(project_id=project_id,experiment_id=experiment_id,tag_id=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: + if response.status_code == : response_200 = RunTagDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -47,9 +70,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """Get Experiment Tag + """ Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -64,19 +92,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """Get Experiment Tag + """ Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -91,15 +131,26 @@ def sync( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ - return sync_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """Get Experiment Tag + """ Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -114,19 +165,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_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, experiment_id: str, tag_id: str, *, client: ApiClient + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """Get Experiment Tag + """ Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -141,8 +204,13 @@ async def asyncio( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +client=client, + + )).parsed 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..4d8af7c95 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 @@ -1,42 +1,62 @@ 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.run_tag_db import RunTagDB -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, + +) -> dict[str, Any]: + headers: dict[str, Any] = {} + + + + + -def _get_kwargs(project_id: str, experiment_id: str) -> dict[str, Any]: _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/{experiment_id}/tags", + "path": "/projects/{project_id}/experiments/{experiment_id}/tags".format(project_id=project_id,experiment_id=experiment_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, list["RunTagDB"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['RunTagDB']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = RunTagDB.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -44,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["RunTagDB"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['RunTagDB']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -56,9 +74,13 @@ def _build_response( def sync_detailed( - project_id: str, experiment_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["RunTagDB"]]]: - """Get Experiment Tags + project_id: str, + experiment_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['RunTagDB']]]: + """ Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -72,19 +94,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['RunTagDB']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, list["RunTagDB"]]]: - """Get Experiment Tags + project_id: str, + experiment_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, list['RunTagDB']]]: + """ Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -98,15 +130,24 @@ def sync( Returns: Union[HTTPValidationError, list['RunTagDB']] - """ + """ + - return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["RunTagDB"]]]: - """Get Experiment Tags + project_id: str, + experiment_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['RunTagDB']]]: + """ Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -120,19 +161,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['RunTagDB']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_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, experiment_id: str, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, list["RunTagDB"]]]: - """Get Experiment Tags + project_id: str, + experiment_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, list['RunTagDB']]]: + """ Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -146,6 +197,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['RunTagDB']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, - return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client)).parsed + )).parsed 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..ea5d1f06a 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 @@ -1,43 +1,64 @@ 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.run_tag_create_request import RunTagCreateRequest from ...models.run_tag_db import RunTagDB -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, + *, + body: RunTagCreateRequest, -def _get_kwargs(project_id: str, experiment_id: str, *, body: RunTagCreateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/{experiment_id}/tags", + "path": "/projects/{project_id}/experiments/{experiment_id}/tags".format(project_id=project_id,experiment_id=experiment_id,), } _kwargs["json"] = body.to_dict() + 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: + if response.status_code == : response_200 = RunTagDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -55,9 +76,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """Set Tag For Experiment + """ Set Tag For Experiment Sets a tag for an experiment. @@ -72,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +body=body, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """Set Tag For Experiment + """ Set Tag For Experiment Sets a tag for an experiment. @@ -99,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ - return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, +body=body, + + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """Set Tag For Experiment + """ Set Tag For Experiment Sets a tag for an experiment. @@ -122,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +body=body, - 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, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """Set Tag For Experiment + """ Set Tag For Experiment Sets a tag for an experiment. @@ -149,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body)).parsed + )).parsed 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..3065b0113 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 @@ -1,43 +1,65 @@ 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.run_tag_create_request import RunTagCreateRequest from ...models.run_tag_db import RunTagDB -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + experiment_id: str, + tag_id: str, + *, + body: RunTagCreateRequest, -def _get_kwargs(project_id: str, experiment_id: str, tag_id: str, *, body: RunTagCreateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}", + "path": "/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}".format(project_id=project_id,experiment_id=experiment_id,tag_id=tag_id,), } _kwargs["json"] = body.to_dict() + 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: + if response.status_code == : response_200 = RunTagDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -55,9 +77,15 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """Update Tag For Experiment + """ Update Tag For Experiment Sets or updates a tag for an experiment. @@ -73,19 +101,33 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """Update Tag For Experiment + """ Update Tag For Experiment Sets or updates a tag for an experiment. @@ -101,17 +143,28 @@ def sync( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ + return sync_detailed( - project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client, body=body - ).parsed + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """Update Tag For Experiment + """ Update Tag For Experiment Sets or updates a tag for an experiment. @@ -127,19 +180,33 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +body=body, - 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, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest + project_id: str, + experiment_id: str, + tag_id: str, + *, + client: ApiClient, + body: RunTagCreateRequest, + ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """Update Tag For Experiment + """ Update Tag For Experiment Sets or updates a tag for an experiment. @@ -155,10 +222,14 @@ async def asyncio( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ - return ( - await asyncio_detailed( - project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client, body=body - ) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +experiment_id=experiment_id, +tag_id=tag_id, +client=client, +body=body, + + )).parsed diff --git a/src/galileo/resources/api/health/healthcheck_healthcheck_get.py b/src/galileo/resources/api/health/healthcheck_healthcheck_get.py index 2045b14aa..ba03cef06 100644 --- a/src/galileo/resources/api/health/healthcheck_healthcheck_get.py +++ b/src/galileo/resources/api/health/healthcheck_healthcheck_get.py @@ -1,21 +1,35 @@ from http import HTTPStatus -from typing import Any, Optional +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.healthcheck_response import HealthcheckResponse -from ...types import Response +from typing import cast + -def _get_kwargs() -> dict[str, Any]: +def _get_kwargs( + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.GET, + "return_raw_response": True, + "path": "/healthcheck", + } - _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": "/healthcheck"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -24,9 +38,11 @@ def _get_kwargs() -> dict[str, Any]: def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[HealthcheckResponse]: - if response.status_code == 200: + if response.status_code == : response_200 = HealthcheckResponse.from_dict(response.json()) + + return response_200 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -43,8 +59,12 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient) -> Response[HealthcheckResponse]: - """Healthcheck +def sync_detailed( + *, + client: ApiClient, + +) -> Response[HealthcheckResponse]: + """ Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -52,17 +72,25 @@ def sync_detailed(*, client: ApiClient) -> Response[HealthcheckResponse]: Returns: Response[HealthcheckResponse] - """ + """ - kwargs = _get_kwargs() - response = client.request(**kwargs) + kwargs = _get_kwargs( + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, -def sync(*, client: ApiClient) -> Optional[HealthcheckResponse]: - """Healthcheck +) -> Optional[HealthcheckResponse]: + """ Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -70,13 +98,20 @@ def sync(*, client: ApiClient) -> Optional[HealthcheckResponse]: Returns: HealthcheckResponse - """ + """ + - return sync_detailed(client=client).parsed + return sync_detailed( + client=client, + ).parsed -async def asyncio_detailed(*, client: ApiClient) -> Response[HealthcheckResponse]: - """Healthcheck +async def asyncio_detailed( + *, + client: ApiClient, + +) -> Response[HealthcheckResponse]: + """ Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -84,17 +119,25 @@ async def asyncio_detailed(*, client: ApiClient) -> Response[HealthcheckResponse Returns: Response[HealthcheckResponse] - """ + """ - kwargs = _get_kwargs() - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, -async def asyncio(*, client: ApiClient) -> Optional[HealthcheckResponse]: - """Healthcheck +) -> Optional[HealthcheckResponse]: + """ Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -102,6 +145,10 @@ async def asyncio(*, client: ApiClient) -> Optional[HealthcheckResponse]: Returns: HealthcheckResponse - """ + """ + + + return (await asyncio_detailed( + client=client, - return (await asyncio_detailed(client=client)).parsed + )).parsed 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..6474edc47 100644 --- a/src/galileo/resources/api/jobs/create_job_jobs_post.py +++ b/src/galileo/resources/api/jobs/create_job_jobs_post.py @@ -1,26 +1,42 @@ 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.create_job_request import CreateJobRequest from ...models.create_job_response import CreateJobResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: CreateJobRequest, -def _get_kwargs(*, body: CreateJobRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/jobs"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/jobs", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,16 +45,18 @@ def _get_kwargs(*, body: CreateJobRequest) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: + if response.status_code == : response_200 = CreateJobResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -46,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[CreateJobResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[CreateJobResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -58,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: CreateJobRequest + *, + client: ApiClient, + body: CreateJobRequest, + ) -> Response[Union[CreateJobResponse, HTTPValidationError]]: - """Create Job + """ Create Job Args: body (CreateJobRequest): @@ -71,17 +90,27 @@ def sync_detailed( Returns: Response[Union[CreateJobResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: CreateJobRequest, -def sync(*, client: ApiClient, body: CreateJobRequest) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: - """Create Job +) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: + """ Create Job Args: body (CreateJobRequest): @@ -92,15 +121,22 @@ def sync(*, client: ApiClient, body: CreateJobRequest) -> Optional[Union[CreateJ Returns: Union[CreateJobResponse, HTTPValidationError] - """ + """ - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: CreateJobRequest + *, + client: ApiClient, + body: CreateJobRequest, + ) -> Response[Union[CreateJobResponse, HTTPValidationError]]: - """Create Job + """ Create Job Args: body (CreateJobRequest): @@ -111,19 +147,27 @@ async def asyncio_detailed( Returns: Response[Union[CreateJobResponse, HTTPValidationError]] - """ + """ - 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: CreateJobRequest + *, + client: ApiClient, + body: CreateJobRequest, + ) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: - """Create Job + """ Create Job Args: body (CreateJobRequest): @@ -134,6 +178,11 @@ async def asyncio( Returns: Union[CreateJobResponse, HTTPValidationError] - """ + """ + + + 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/jobs/get_job_jobs_job_id_get.py b/src/galileo/resources/api/jobs/get_job_jobs_job_id_get.py index eea80f35c..80f618c90 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 @@ -1,22 +1,37 @@ 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.job_db import JobDB -from ...types import Response +from typing import cast + -def _get_kwargs(job_id: str) -> dict[str, Any]: +def _get_kwargs( + job_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.GET, + "return_raw_response": True, + "path": "/jobs/{job_id}".format(job_id=job_id,), + } - _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": f"/jobs/{job_id}"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -25,13 +40,17 @@ 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: + if response.status_code == : response_200 = JobDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -48,8 +67,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(job_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobDB]]: - """Get Job +def sync_detailed( + job_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, JobDB]]: + """ Get Job Get a job by id. @@ -62,17 +86,27 @@ def sync_detailed(job_id: str, *, client: ApiClient) -> Response[Union[HTTPValid Returns: Response[Union[HTTPValidationError, JobDB]] - """ + """ - kwargs = _get_kwargs(job_id=job_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + job_id=job_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + job_id: str, + *, + client: ApiClient, -def sync(job_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobDB]]: - """Get Job +) -> Optional[Union[HTTPValidationError, JobDB]]: + """ Get Job Get a job by id. @@ -85,13 +119,22 @@ def sync(job_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationErro Returns: Union[HTTPValidationError, JobDB] - """ + """ + + + return sync_detailed( + job_id=job_id, +client=client, - return sync_detailed(job_id=job_id, client=client).parsed + ).parsed +async def asyncio_detailed( + job_id: str, + *, + client: ApiClient, -async def asyncio_detailed(job_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobDB]]: - """Get Job +) -> Response[Union[HTTPValidationError, JobDB]]: + """ Get Job Get a job by id. @@ -104,17 +147,27 @@ async def asyncio_detailed(job_id: str, *, client: ApiClient) -> Response[Union[ Returns: Response[Union[HTTPValidationError, JobDB]] - """ + """ - kwargs = _get_kwargs(job_id=job_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + job_id=job_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + job_id: str, + *, + client: ApiClient, -async def asyncio(job_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobDB]]: - """Get Job +) -> Optional[Union[HTTPValidationError, JobDB]]: + """ Get Job Get a job by id. @@ -127,6 +180,11 @@ async def asyncio(job_id: str, *, client: ApiClient) -> Optional[Union[HTTPValid Returns: Union[HTTPValidationError, JobDB] - """ + """ + + + return (await asyncio_detailed( + job_id=job_id, +client=client, - return (await asyncio_detailed(job_id=job_id, client=client)).parsed + )).parsed 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..08ed08c09 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 @@ -1,20 +1,34 @@ 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.job_db import JobDB -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + + +def _get_kwargs( + project_id: str, + run_id: str, + *, + status: Union[None, Unset, str] = UNSET, -def _get_kwargs(project_id: str, run_id: str, *, status: Union[None, Unset, str] = UNSET) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -25,36 +39,41 @@ def _get_kwargs(project_id: str, run_id: str, *, status: Union[None, Unset, str] json_status = status params["status"] = json_status + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/runs/{run_id}/jobs", + "path": "/projects/{project_id}/runs/{run_id}/jobs".format(project_id=project_id,run_id=run_id,), "params": params, } + 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, list["JobDB"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['JobDB']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = JobDB.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -62,9 +81,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["JobDB"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['JobDB']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,9 +91,14 @@ def _build_response( def sync_detailed( - project_id: str, run_id: str, *, client: ApiClient, status: Union[None, Unset, str] = UNSET -) -> Response[Union[HTTPValidationError, list["JobDB"]]]: - """Get Jobs For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + status: Union[None, Unset, str] = UNSET, + +) -> Response[Union[HTTPValidationError, list['JobDB']]]: + """ Get Jobs For Project Run Get all jobs by for a project and run. @@ -93,19 +115,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['JobDB']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id, status=status) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, +status=status, - 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, status: Union[None, Unset, str] = UNSET -) -> Optional[Union[HTTPValidationError, list["JobDB"]]]: - """Get Jobs For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + status: Union[None, Unset, str] = UNSET, + +) -> Optional[Union[HTTPValidationError, list['JobDB']]]: + """ Get Jobs For Project Run Get all jobs by for a project and run. @@ -122,15 +156,26 @@ def sync( Returns: Union[HTTPValidationError, list['JobDB']] - """ + """ - return sync_detailed(project_id=project_id, run_id=run_id, client=client, status=status).parsed + return sync_detailed( + project_id=project_id, +run_id=run_id, +client=client, +status=status, + + ).parsed async def asyncio_detailed( - project_id: str, run_id: str, *, client: ApiClient, status: Union[None, Unset, str] = UNSET -) -> Response[Union[HTTPValidationError, list["JobDB"]]]: - """Get Jobs For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + status: Union[None, Unset, str] = UNSET, + +) -> Response[Union[HTTPValidationError, list['JobDB']]]: + """ Get Jobs For Project Run Get all jobs by for a project and run. @@ -147,19 +192,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['JobDB']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id, status=status) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, +status=status, - 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, status: Union[None, Unset, str] = UNSET -) -> Optional[Union[HTTPValidationError, list["JobDB"]]]: - """Get Jobs For Project Run + project_id: str, + run_id: str, + *, + client: ApiClient, + status: Union[None, Unset, str] = UNSET, + +) -> Optional[Union[HTTPValidationError, list['JobDB']]]: + """ Get Jobs For Project Run Get all jobs by for a project and run. @@ -176,6 +233,13 @@ async def asyncio( Returns: Union[HTTPValidationError, list['JobDB']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +run_id=run_id, +client=client, +status=status, - return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client, status=status)).parsed + )).parsed 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..36c7edf81 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]]]: - if response.status_code == 200: - - def _parse_response_200(data: object) -> Union["JobDB", None]: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['JobDB', None]]]: + if response.status_code == : + def _parse_response_200(data: object) -> Union['JobDB', None]: if data is None: return data try: @@ -41,17 +51,21 @@ 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()) return response_200 - if response.status_code == 422: + if response.status_code == : 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..3b163b3f7 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 @@ -1,30 +1,43 @@ 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.log_stream_create_request import LogStreamCreateRequest from ...models.log_stream_response import LogStreamResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogStreamCreateRequest, -def _get_kwargs(project_id: str, *, body: LogStreamCreateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/log_streams", + "path": "/projects/{project_id}/log_streams".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogStreamCreateRequest) -> dict[str, A return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: + if response.status_code == : response_200 = LogStreamResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogStreamResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogStreamResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogStreamCreateRequest + project_id: str, + *, + client: ApiClient, + body: LogStreamCreateRequest, + ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """Create Log Stream + """ Create Log Stream Create a new log stream for a project. @@ -78,19 +95,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogStreamCreateRequest + project_id: str, + *, + client: ApiClient, + body: LogStreamCreateRequest, + ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """Create Log Stream + """ Create Log Stream Create a new log stream for a project. @@ -104,15 +131,24 @@ def sync( Returns: Union[HTTPValidationError, LogStreamResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogStreamCreateRequest + project_id: str, + *, + client: ApiClient, + body: LogStreamCreateRequest, + ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """Create Log Stream + """ Create Log Stream Create a new log stream for a project. @@ -126,19 +162,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogStreamCreateRequest + project_id: str, + *, + client: ApiClient, + body: LogStreamCreateRequest, + ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """Create Log Stream + """ Create Log Stream Create a new log stream for a project. @@ -152,6 +198,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogStreamResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..f1d3bcea6 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 @@ -6,21 +6,33 @@ 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 ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + log_stream_id: str, -def _get_kwargs(project_id: str, log_stream_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/projects/{project_id}/log_streams/{log_stream_id}", + "path": "/projects/{project_id}/log_streams/{log_stream_id}".format(project_id=project_id,log_stream_id=log_stream_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_204 = cast(Any, None) return response_204 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -51,9 +65,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Log Stream + """ Delete Log Stream Delete a specific log stream. @@ -67,17 +85,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +log_stream_id=log_stream_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + log_stream_id: str, + *, + client: ApiClient, -def sync(project_id: str, log_stream_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Log Stream +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Log Stream Delete a specific log stream. @@ -91,15 +121,24 @@ def sync(project_id: str, log_stream_id: str, *, client: ApiClient) -> Optional[ Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client).parsed + return sync_detailed( + project_id=project_id, +log_stream_id=log_stream_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Log Stream + """ Delete Log Stream Delete a specific log stream. @@ -113,19 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +log_stream_id=log_stream_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, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Log Stream + """ Delete Log Stream Delete a specific log stream. @@ -139,6 +188,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +log_stream_id=log_stream_id, +client=client, - return (await asyncio_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client)).parsed + )).parsed 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..2df7143f9 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 @@ -1,43 +1,57 @@ 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.log_stream_response import LogStreamResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + log_stream_id: str, -def _get_kwargs(project_id: str, log_stream_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}/log_streams/{log_stream_id}", + "path": "/projects/{project_id}/log_streams/{log_stream_id}".format(project_id=project_id,log_stream_id=log_stream_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, LogStreamResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: + if response.status_code == : response_200 = LogStreamResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogStreamResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogStreamResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """Get Log Stream + """ Get Log Stream Retrieve a specific log stream. @@ -73,19 +89,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +log_stream_id=log_stream_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """Get Log Stream + """ Get Log Stream Retrieve a specific log stream. @@ -99,15 +125,24 @@ def sync( Returns: Union[HTTPValidationError, LogStreamResponse] - """ + """ + - return sync_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client).parsed + return sync_detailed( + project_id=project_id, +log_stream_id=log_stream_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """Get Log Stream + """ Get Log Stream Retrieve a specific log stream. @@ -121,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +log_stream_id=log_stream_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, log_stream_id: str, *, client: ApiClient + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """Get Log Stream + """ Get Log Stream Retrieve a specific log stream. @@ -147,6 +192,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogStreamResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +log_stream_id=log_stream_id, +client=client, - return (await asyncio_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client)).parsed + )).parsed 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..3841483ed 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 @@ -1,48 +1,61 @@ 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.log_stream_response import LogStreamResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, -def _get_kwargs(project_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}/log_streams", + "path": "/projects/{project_id}/log_streams".format(project_id=project_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, list["LogStreamResponse"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['LogStreamResponse']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = LogStreamResponse.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +63,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["LogStreamResponse"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['LogStreamResponse']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +73,12 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["LogStreamResponse"]]]: - """List Log Streams + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['LogStreamResponse']]]: + """ List Log Streams Retrieve all log streams for a project. @@ -77,17 +91,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['LogStreamResponse']]] - """ + """ + - kwargs = _get_kwargs(project_id=project_id) + kwargs = _get_kwargs( + project_id=project_id, - response = client.request(**kwargs) + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + *, + client: ApiClient, -def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, list["LogStreamResponse"]]]: - """List Log Streams +) -> Optional[Union[HTTPValidationError, list['LogStreamResponse']]]: + """ List Log Streams Retrieve all log streams for a project. @@ -100,15 +124,22 @@ def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidation Returns: Union[HTTPValidationError, list['LogStreamResponse']] - """ + """ + - return sync_detailed(project_id=project_id, client=client).parsed + return sync_detailed( + project_id=project_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["LogStreamResponse"]]]: - """List Log Streams + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['LogStreamResponse']]]: + """ List Log Streams Retrieve all log streams for a project. @@ -121,19 +152,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['LogStreamResponse']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_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, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, list["LogStreamResponse"]]]: - """List Log Streams + project_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, list['LogStreamResponse']]]: + """ List Log Streams Retrieve all log streams for a project. @@ -146,6 +185,11 @@ async def asyncio( Returns: Union[HTTPValidationError, list['LogStreamResponse']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, - return (await asyncio_detailed(project_id=project_id, client=client)).parsed + )).parsed 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..51c12e4d1 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 @@ -1,30 +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.log_stream_response import LogStreamResponse from ...models.log_stream_update_request import LogStreamUpdateRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + log_stream_id: str, + *, + body: LogStreamUpdateRequest, -def _get_kwargs(project_id: str, log_stream_id: str, *, body: LogStreamUpdateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}/log_streams/{log_stream_id}", + "path": "/projects/{project_id}/log_streams/{log_stream_id}".format(project_id=project_id,log_stream_id=log_stream_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, log_stream_id: str, *, body: LogStreamUpdateReq return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: + if response.status_code == : response_200 = LogStreamResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogStreamResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogStreamResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + body: LogStreamUpdateRequest, + ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """Update Log Stream + """ Update Log Stream Update a specific log stream. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +log_stream_id=log_stream_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + body: LogStreamUpdateRequest, + ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """Update Log Stream + """ Update Log Stream Update a specific log stream. @@ -106,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, LogStreamResponse] - """ + """ + - return sync_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +log_stream_id=log_stream_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + body: LogStreamUpdateRequest, + ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """Update Log Stream + """ Update Log Stream Update a specific log stream. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +log_stream_id=log_stream_id, +body=body, - 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, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest + project_id: str, + log_stream_id: str, + *, + client: ApiClient, + body: LogStreamUpdateRequest, + ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """Update Log Stream + """ Update Log Stream Update a specific log stream. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, LogStreamResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +log_stream_id=log_stream_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client, body=body)).parsed + )).parsed 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..e91e6bb0e 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 @@ -1,26 +1,38 @@ 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.group_collaborator import GroupCollaborator from ...models.group_collaborator_create import GroupCollaboratorCreate from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: list['GroupCollaboratorCreate'], -def _get_kwargs(project_id: str, *, body: list["GroupCollaboratorCreate"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/groups", + "path": "/projects/{project_id}/groups".format(project_id=project_id,), } _kwargs["json"] = [] @@ -28,6 +40,9 @@ def _get_kwargs(project_id: str, *, body: list["GroupCollaboratorCreate"]) -> di body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) + + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -36,21 +51,23 @@ def _get_kwargs(project_id: str, *, body: list["GroupCollaboratorCreate"]) -> di return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = GroupCollaborator.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,9 +85,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Project Collaborators Share a project with groups. @@ -86,19 +105,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Project Collaborators Share a project with groups. @@ -112,15 +141,24 @@ def sync( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Project Collaborators Share a project with groups. @@ -134,19 +172,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Project Collaborators Share a project with groups. @@ -160,6 +208,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..f85a9197b 100644 --- a/src/galileo/resources/api/projects/create_project_projects_post.py +++ b/src/galileo/resources/api/projects/create_project_projects_post.py @@ -1,26 +1,42 @@ 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.project_create import ProjectCreate from ...models.project_create_response import ProjectCreateResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: ProjectCreate, -def _get_kwargs(*, body: ProjectCreate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/projects"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/projects", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,16 +45,18 @@ def _get_kwargs(*, body: ProjectCreate) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: + if response.status_code == : response_200 = ProjectCreateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -46,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -58,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: ProjectCreate + *, + client: ApiClient, + body: ProjectCreate, + ) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: - """Create Project + """ Create Project Create a new project. @@ -73,17 +92,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ProjectCreateResponse]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: ProjectCreate, -def sync(*, client: ApiClient, body: ProjectCreate) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: - """Create Project +) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: + """ Create Project Create a new project. @@ -96,15 +125,22 @@ def sync(*, client: ApiClient, body: ProjectCreate) -> Optional[Union[HTTPValida Returns: Union[HTTPValidationError, ProjectCreateResponse] - """ + """ - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: ProjectCreate + *, + client: ApiClient, + body: ProjectCreate, + ) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: - """Create Project + """ Create Project Create a new project. @@ -117,19 +153,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectCreateResponse]] - """ + """ - 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: ProjectCreate + *, + client: ApiClient, + body: ProjectCreate, + ) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: - """Create Project + """ Create Project Create a new project. @@ -142,6 +186,11 @@ async def asyncio( Returns: Union[HTTPValidationError, ProjectCreateResponse] - """ + """ + + + 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/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..b78020a40 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 @@ -1,26 +1,38 @@ 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.user_collaborator import UserCollaborator from ...models.user_collaborator_create import UserCollaboratorCreate -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: list['UserCollaboratorCreate'], -def _get_kwargs(project_id: str, *, body: list["UserCollaboratorCreate"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/users", + "path": "/projects/{project_id}/users".format(project_id=project_id,), } _kwargs["json"] = [] @@ -28,6 +40,9 @@ def _get_kwargs(project_id: str, *, body: list["UserCollaboratorCreate"]) -> dic body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) + + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -36,21 +51,23 @@ def _get_kwargs(project_id: str, *, body: list["UserCollaboratorCreate"]) -> dic return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = UserCollaborator.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,9 +85,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Project Collaborators Share a project with users. @@ -86,19 +105,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Project Collaborators Share a project with users. @@ -112,15 +141,24 @@ def sync( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Project Collaborators Share a project with users. @@ -134,19 +172,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Project Collaborators + project_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Project Collaborators Share a project with users. @@ -160,6 +208,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..2be7f6cc9 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(project_id: str, group_id: str) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + group_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}/groups/{group_id}", + "path": "/projects/{project_id}/groups/{group_id}".format(project_id=project_id,group_id=group_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(project_id: str, group_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete Group Project Collaborator +def sync_detailed( + project_id: str, + group_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete Group Project Collaborator Remove a group's access to a project. @@ -65,17 +85,29 @@ def sync_detailed(project_id: str, group_id: str, *, client: ApiClient) -> Respo Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, group_id=group_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +group_id=group_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + group_id: str, + *, + client: ApiClient, -def sync(project_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Group Project Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Group Project Collaborator Remove a group's access to a project. @@ -89,15 +121,24 @@ def sync(project_id: str, group_id: str, *, client: ApiClient) -> Optional[Union Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, group_id=group_id, client=client).parsed + return sync_detailed( + project_id=project_id, +group_id=group_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, group_id: str, *, client: ApiClient + project_id: str, + group_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Group Project Collaborator + """ Delete Group Project Collaborator Remove a group's access to a project. @@ -111,17 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, group_id=group_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +group_id=group_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + project_id: str, + group_id: str, + *, + client: ApiClient, -async def asyncio(project_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Group Project Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Group Project Collaborator Remove a group's access to a project. @@ -135,6 +188,12 @@ async def asyncio(project_id: str, group_id: str, *, client: ApiClient) -> Optio Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +group_id=group_id, +client=client, - return (await asyncio_detailed(project_id=project_id, group_id=group_id, client=client)).parsed + )).parsed 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..6ded921fe 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 @@ -1,43 +1,56 @@ 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.project_delete_response import ProjectDeleteResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, -def _get_kwargs(project_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/projects/{project_id}", + "path": "/projects/{project_id}".format(project_id=project_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, ProjectDeleteResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: + if response.status_code == : response_200 = ProjectDeleteResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -56,8 +67,13 @@ def _build_response( ) -def sync_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: - """Delete Project +def sync_detailed( + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: + """ Delete Project Deletes a project and all associated runs and objects. @@ -73,17 +89,27 @@ def sync_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPV Returns: Response[Union[HTTPValidationError, ProjectDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + *, + client: ApiClient, -def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: - """Delete Project +) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: + """ Delete Project Deletes a project and all associated runs and objects. @@ -99,15 +125,22 @@ def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidation Returns: Union[HTTPValidationError, ProjectDeleteResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client).parsed + return sync_detailed( + project_id=project_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient + project_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: - """Delete Project + """ Delete Project Deletes a project and all associated runs and objects. @@ -123,17 +156,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + project_id: str, + *, + client: ApiClient, -async def asyncio(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: - """Delete Project +) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: + """ Delete Project Deletes a project and all associated runs and objects. @@ -149,6 +192,11 @@ async def asyncio(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPV Returns: Union[HTTPValidationError, ProjectDeleteResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, - return (await asyncio_detailed(project_id=project_id, client=client)).parsed + )).parsed 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..da69d7854 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(project_id: str, user_id: str) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + user_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}/users/{user_id}", + "path": "/projects/{project_id}/users/{user_id}".format(project_id=project_id,user_id=user_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(project_id: str, user_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete User Project Collaborator +def sync_detailed( + project_id: str, + user_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete User Project Collaborator Remove a user's access to a project. @@ -65,17 +85,29 @@ def sync_detailed(project_id: str, user_id: str, *, client: ApiClient) -> Respon Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, user_id=user_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +user_id=user_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + user_id: str, + *, + client: ApiClient, -def sync(project_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete User Project Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete User Project Collaborator Remove a user's access to a project. @@ -89,15 +121,24 @@ def sync(project_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[ Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, user_id=user_id, client=client).parsed + return sync_detailed( + project_id=project_id, +user_id=user_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, user_id: str, *, client: ApiClient + project_id: str, + user_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete User Project Collaborator + """ Delete User Project Collaborator Remove a user's access to a project. @@ -111,17 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, user_id=user_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +user_id=user_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + project_id: str, + user_id: str, + *, + client: ApiClient, -async def asyncio(project_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete User Project Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete User Project Collaborator Remove a user's access to a project. @@ -135,6 +188,12 @@ async def asyncio(project_id: str, user_id: str, *, client: ApiClient) -> Option Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +user_id=user_id, +client=client, - return (await asyncio_detailed(project_id=project_id, user_id=user_id, client=client)).parsed + )).parsed 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..685e6dcf2 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 @@ -1,21 +1,33 @@ 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.project_db_thin import ProjectDBThin from ...models.project_type import ProjectType -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + + +def _get_kwargs( + *, + type_: Union[None, ProjectType, Unset] = UNSET, -def _get_kwargs(*, type_: Union[None, ProjectType, Unset] = UNSET) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -28,8 +40,10 @@ def _get_kwargs(*, type_: Union[None, ProjectType, Unset] = UNSET) -> dict[str, json_type_ = type_ params["type"] = json_type_ + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, @@ -37,27 +51,30 @@ def _get_kwargs(*, type_: Union[None, ProjectType, Unset] = UNSET) -> dict[str, "params": params, } + 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, list["ProjectDBThin"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['ProjectDBThin']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = ProjectDBThin.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -65,9 +82,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["ProjectDBThin"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['ProjectDBThin']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -77,9 +92,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET -) -> Response[Union[HTTPValidationError, list["ProjectDBThin"]]]: - """Get All Projects + *, + client: ApiClient, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Response[Union[HTTPValidationError, list['ProjectDBThin']]]: + """ Get All Projects Gets all public projects and all private projects that the user has access to. @@ -96,19 +114,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDBThin']]] - """ + """ - kwargs = _get_kwargs(type_=type_) - response = client.request(**kwargs) + kwargs = _get_kwargs( + type_=type_, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET -) -> Optional[Union[HTTPValidationError, list["ProjectDBThin"]]]: - """Get All Projects + *, + client: ApiClient, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Optional[Union[HTTPValidationError, list['ProjectDBThin']]]: + """ Get All Projects Gets all public projects and all private projects that the user has access to. @@ -125,15 +151,22 @@ def sync( Returns: Union[HTTPValidationError, list['ProjectDBThin']] - """ + """ - return sync_detailed(client=client, type_=type_).parsed + return sync_detailed( + client=client, +type_=type_, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET -) -> Response[Union[HTTPValidationError, list["ProjectDBThin"]]]: - """Get All Projects + *, + client: ApiClient, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Response[Union[HTTPValidationError, list['ProjectDBThin']]]: + """ Get All Projects Gets all public projects and all private projects that the user has access to. @@ -150,19 +183,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDBThin']]] - """ + """ - kwargs = _get_kwargs(type_=type_) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + type_=type_, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET -) -> Optional[Union[HTTPValidationError, list["ProjectDBThin"]]]: - """Get All Projects + *, + client: ApiClient, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Optional[Union[HTTPValidationError, list['ProjectDBThin']]]: + """ Get All Projects Gets all public projects and all private projects that the user has access to. @@ -179,6 +220,11 @@ async def asyncio( Returns: Union[HTTPValidationError, list['ProjectDBThin']] - """ + """ + + + return (await asyncio_detailed( + client=client, +type_=type_, - return (await asyncio_detailed(client=client, type_=type_)).parsed + )).parsed 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..34c7ed936 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 @@ -1,21 +1,35 @@ from http import HTTPStatus -from typing import Any, Optional +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.collaborator_role_info import CollaboratorRoleInfo -from ...types import Response +from typing import cast + -def _get_kwargs() -> dict[str, Any]: +def _get_kwargs( + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.GET, + "return_raw_response": True, + "path": "/collaborator_roles", + } - _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": "/collaborator_roles"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -23,13 +37,15 @@ def _get_kwargs() -> dict[str, Any]: return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list["CollaboratorRoleInfo"]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list['CollaboratorRoleInfo']]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = CollaboratorRoleInfo.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 @@ -39,7 +55,7 @@ def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[ return None -def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[list["CollaboratorRoleInfo"]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[list['CollaboratorRoleInfo']]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -48,8 +64,12 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient) -> Response[list["CollaboratorRoleInfo"]]: - """Get Collaborator Roles +def sync_detailed( + *, + client: ApiClient, + +) -> Response[list['CollaboratorRoleInfo']]: + """ Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -57,17 +77,25 @@ def sync_detailed(*, client: ApiClient) -> Response[list["CollaboratorRoleInfo"] Returns: Response[list['CollaboratorRoleInfo']] - """ + """ - kwargs = _get_kwargs() - response = client.request(**kwargs) + kwargs = _get_kwargs( + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, -def sync(*, client: ApiClient) -> Optional[list["CollaboratorRoleInfo"]]: - """Get Collaborator Roles +) -> Optional[list['CollaboratorRoleInfo']]: + """ Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -75,13 +103,20 @@ def sync(*, client: ApiClient) -> Optional[list["CollaboratorRoleInfo"]]: Returns: list['CollaboratorRoleInfo'] - """ + """ + - return sync_detailed(client=client).parsed + return sync_detailed( + client=client, + ).parsed -async def asyncio_detailed(*, client: ApiClient) -> Response[list["CollaboratorRoleInfo"]]: - """Get Collaborator Roles +async def asyncio_detailed( + *, + client: ApiClient, + +) -> Response[list['CollaboratorRoleInfo']]: + """ Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -89,17 +124,25 @@ async def asyncio_detailed(*, client: ApiClient) -> Response[list["CollaboratorR Returns: Response[list['CollaboratorRoleInfo']] - """ + """ - kwargs = _get_kwargs() - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, -async def asyncio(*, client: ApiClient) -> Optional[list["CollaboratorRoleInfo"]]: - """Get Collaborator Roles +) -> Optional[list['CollaboratorRoleInfo']]: + """ Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -107,6 +150,10 @@ async def asyncio(*, client: ApiClient) -> Optional[list["CollaboratorRoleInfo"] Returns: list['CollaboratorRoleInfo'] - """ + """ + + + return (await asyncio_detailed( + client=client, - return (await asyncio_detailed(client=client)).parsed + )).parsed 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..b803f4ed0 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 @@ -1,27 +1,38 @@ 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.project_db import ProjectDB -from ...types import Response +from typing import cast + -def _get_kwargs(project_id: str) -> dict[str, Any]: +def _get_kwargs( + project_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}", + "path": "/projects/{project_id}".format(project_id=project_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -29,13 +40,17 @@ 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: + if response.status_code == : response_200 = ProjectDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -52,8 +67,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ProjectDB]]: - """Get Project +def sync_detailed( + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, ProjectDB]]: + """ Get Project Args: project_id (str): @@ -64,17 +84,27 @@ def sync_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPV Returns: Response[Union[HTTPValidationError, ProjectDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + project_id: str, + *, + client: ApiClient, -def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDB]]: - """Get Project +) -> Optional[Union[HTTPValidationError, ProjectDB]]: + """ Get Project Args: project_id (str): @@ -85,13 +115,22 @@ def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidation Returns: Union[HTTPValidationError, ProjectDB] - """ + """ + + + return sync_detailed( + project_id=project_id, +client=client, - return sync_detailed(project_id=project_id, client=client).parsed + ).parsed +async def asyncio_detailed( + project_id: str, + *, + client: ApiClient, -async def asyncio_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ProjectDB]]: - """Get Project +) -> Response[Union[HTTPValidationError, ProjectDB]]: + """ Get Project Args: project_id (str): @@ -102,17 +141,27 @@ async def asyncio_detailed(project_id: str, *, client: ApiClient) -> Response[Un Returns: Response[Union[HTTPValidationError, ProjectDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + project_id: str, + *, + client: ApiClient, -async def asyncio(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDB]]: - """Get Project +) -> Optional[Union[HTTPValidationError, ProjectDB]]: + """ Get Project Args: project_id (str): @@ -123,6 +172,11 @@ async def asyncio(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPV Returns: Union[HTTPValidationError, ProjectDB] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, - return (await asyncio_detailed(project_id=project_id, client=client)).parsed + )).parsed 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..157bed32a 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 @@ -6,20 +6,36 @@ 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.project_collection_params import ProjectCollectionParams -from ...types import Response +from typing import cast + -def _get_kwargs(*, body: ProjectCollectionParams) -> dict[str, Any]: +def _get_kwargs( + *, + body: ProjectCollectionParams, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/projects/count"} + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/projects/count", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,12 +45,14 @@ 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: + if response.status_code == : response_200 = cast(int, response.json()) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -51,8 +69,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient, body: ProjectCollectionParams) -> Response[Union[HTTPValidationError, int]]: - """Get Projects Count +def sync_detailed( + *, + client: ApiClient, + body: ProjectCollectionParams, + +) -> Response[Union[HTTPValidationError, int]]: + """ Get Projects Count Gets total count of projects for a user with applied filters. @@ -65,17 +88,27 @@ def sync_detailed(*, client: ApiClient, body: ProjectCollectionParams) -> Respon Returns: Response[Union[HTTPValidationError, int]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: ProjectCollectionParams, -def sync(*, client: ApiClient, body: ProjectCollectionParams) -> Optional[Union[HTTPValidationError, int]]: - """Get Projects Count +) -> Optional[Union[HTTPValidationError, int]]: + """ Get Projects Count Gets total count of projects for a user with applied filters. @@ -88,15 +121,22 @@ def sync(*, client: ApiClient, body: ProjectCollectionParams) -> Optional[Union[ Returns: Union[HTTPValidationError, int] - """ + """ - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: ProjectCollectionParams + *, + client: ApiClient, + body: ProjectCollectionParams, + ) -> Response[Union[HTTPValidationError, int]]: - """Get Projects Count + """ Get Projects Count Gets total count of projects for a user with applied filters. @@ -109,17 +149,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, int]] - """ + """ - kwargs = _get_kwargs(body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, + body: ProjectCollectionParams, -async def asyncio(*, client: ApiClient, body: ProjectCollectionParams) -> Optional[Union[HTTPValidationError, int]]: - """Get Projects Count +) -> Optional[Union[HTTPValidationError, int]]: + """ Get Projects Count Gets total count of projects for a user with applied filters. @@ -132,6 +182,11 @@ async def asyncio(*, client: ApiClient, body: ProjectCollectionParams) -> Option Returns: Union[HTTPValidationError, int] - """ + """ + + + 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/projects/get_projects_paginated_projects_paginated_post.py b/src/galileo/resources/api/projects/get_projects_paginated_projects_paginated_post.py index 98ce9da47..75292c50c 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 @@ -1,18 +1,22 @@ 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.get_projects_paginated_response import GetProjectsPaginatedResponse from ...models.http_validation_error import HTTPValidationError from ...models.project_action import ProjectAction from ...models.project_collection_params import ProjectCollectionParams -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -21,8 +25,12 @@ def _get_kwargs( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -33,14 +41,17 @@ def _get_kwargs( actions_item = actions_item_data.value json_actions.append(actions_item) + params["actions"] = json_actions params["starting_token"] = starting_token params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -50,6 +61,7 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -58,16 +70,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: + if response.status_code == : response_200 = GetProjectsPaginatedResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -75,9 +89,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -93,8 +105,9 @@ def sync_detailed( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """Get Projects Paginated + """ Get Projects Paginated Gets projects for a user with pagination. @@ -113,14 +126,22 @@ def sync_detailed( Returns: Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, +actions=actions, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( *, @@ -129,8 +150,9 @@ def sync( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """Get Projects Paginated + """ Get Projects Paginated Gets projects for a user with pagination. @@ -149,10 +171,17 @@ def sync( Returns: Union[GetProjectsPaginatedResponse, HTTPValidationError] - """ + """ - return sync_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + client=client, +body=body, +actions=actions, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( *, @@ -161,8 +190,9 @@ async def asyncio_detailed( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """Get Projects Paginated + """ Get Projects Paginated Gets projects for a user with pagination. @@ -181,14 +211,22 @@ async def asyncio_detailed( Returns: Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, +actions=actions, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -197,8 +235,9 @@ async def asyncio( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """Get Projects Paginated + """ Get Projects Paginated Gets projects for a user with pagination. @@ -217,8 +256,14 @@ async def asyncio( Returns: Union[GetProjectsPaginatedResponse, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + client=client, +body=body, +actions=actions, +starting_token=starting_token, +limit=limit, + + )).parsed 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..82308a179 100644 --- a/src/galileo/resources/api/projects/get_projects_projects_get.py +++ b/src/galileo/resources/api/projects/get_projects_projects_get.py @@ -1,23 +1,34 @@ 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.project_db import ProjectDB from ...models.project_type import ProjectType -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + def _get_kwargs( - *, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET + *, + project_name: Union[None, Unset, str] = UNSET, + type_: Union[None, ProjectType, Unset] = UNSET, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -37,8 +48,10 @@ def _get_kwargs( json_type_ = type_ params["type"] = json_type_ + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, @@ -46,27 +59,30 @@ def _get_kwargs( "params": params, } + 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, list["ProjectDB"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['ProjectDB']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = ProjectDB.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -74,9 +90,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["ProjectDB"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['ProjectDB']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -86,9 +100,13 @@ def _build_response( def sync_detailed( - *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET -) -> Response[Union[HTTPValidationError, list["ProjectDB"]]]: - """Get Projects + *, + client: ApiClient, + project_name: Union[None, Unset, str] = UNSET, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Response[Union[HTTPValidationError, list['ProjectDB']]]: + """ Get Projects Gets projects for a user. @@ -106,19 +124,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDB']]] - """ + """ - kwargs = _get_kwargs(project_name=project_name, type_=type_) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_name=project_name, +type_=type_, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET -) -> Optional[Union[HTTPValidationError, list["ProjectDB"]]]: - """Get Projects + *, + client: ApiClient, + project_name: Union[None, Unset, str] = UNSET, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Optional[Union[HTTPValidationError, list['ProjectDB']]]: + """ Get Projects Gets projects for a user. @@ -136,15 +164,24 @@ def sync( Returns: Union[HTTPValidationError, list['ProjectDB']] - """ + """ - return sync_detailed(client=client, project_name=project_name, type_=type_).parsed + return sync_detailed( + client=client, +project_name=project_name, +type_=type_, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET -) -> Response[Union[HTTPValidationError, list["ProjectDB"]]]: - """Get Projects + *, + client: ApiClient, + project_name: Union[None, Unset, str] = UNSET, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Response[Union[HTTPValidationError, list['ProjectDB']]]: + """ Get Projects Gets projects for a user. @@ -162,19 +199,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDB']]] - """ + """ - kwargs = _get_kwargs(project_name=project_name, type_=type_) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_name=project_name, +type_=type_, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET -) -> Optional[Union[HTTPValidationError, list["ProjectDB"]]]: - """Get Projects + *, + client: ApiClient, + project_name: Union[None, Unset, str] = UNSET, + type_: Union[None, ProjectType, Unset] = UNSET, + +) -> Optional[Union[HTTPValidationError, list['ProjectDB']]]: + """ Get Projects Gets projects for a user. @@ -192,6 +239,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['ProjectDB']] - """ + """ + + + return (await asyncio_detailed( + client=client, +project_name=project_name, +type_=type_, - return (await asyncio_detailed(client=client, project_name=project_name, type_=type_)).parsed + )).parsed 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..fa1be6082 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 @@ -1,22 +1,33 @@ 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.list_group_collaborators_response import ListGroupCollaboratorsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - project_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/groups", + "path": "/projects/{project_id}/groups".format(project_id=project_id,), "params": params, } + 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, ListGroupCollaboratorsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: + if response.status_code == : response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Project Collaborators + """ List Group Project Collaborators List the groups with which the project has been shared. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Project Collaborators + """ List Group Project Collaborators List the groups with which the project has been shared. @@ -112,15 +143,26 @@ def sync( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ - return sync_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + project_id=project_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Project Collaborators + """ List Group Project Collaborators List the groups with which the project has been shared. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +starting_token=starting_token, +limit=limit, - 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, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Project Collaborators + """ List Group Project Collaborators List the groups with which the project has been shared. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..adc2fc7a1 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 @@ -1,22 +1,33 @@ 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.list_user_collaborators_response import ListUserCollaboratorsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - project_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/users", + "path": "/projects/{project_id}/users".format(project_id=project_id,), "params": params, } + 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, ListUserCollaboratorsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: + if response.status_code == : response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Project Collaborators + """ List User Project Collaborators List the users with which the project has been shared. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Project Collaborators + """ List User Project Collaborators List the users with which the project has been shared. @@ -112,15 +143,26 @@ def sync( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ - return sync_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + project_id=project_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Project Collaborators + """ List User Project Collaborators List the users with which the project has been shared. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +starting_token=starting_token, +limit=limit, - 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, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + project_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Project Collaborators + """ List User Project Collaborators List the users with which the project has been shared. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..8aa2b02fb 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 @@ -1,30 +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.collaborator_update import CollaboratorUpdate from ...models.group_collaborator import GroupCollaborator from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + group_id: str, + *, + body: CollaboratorUpdate, -def _get_kwargs(project_id: str, group_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/projects/{project_id}/groups/{group_id}", + "path": "/projects/{project_id}/groups/{group_id}".format(project_id=project_id,group_id=group_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, group_id: str, *, body: CollaboratorUpdate) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: + if response.status_code == : response_200 = GroupCollaborator.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GroupCollaborator, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GroupCollaborator, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Project Collaborator + """ Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, group_id=group_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +group_id=group_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Project Collaborator + """ Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -106,15 +137,26 @@ def sync( Returns: Union[GroupCollaborator, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, group_id=group_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +group_id=group_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Project Collaborator + """ Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, group_id=group_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +group_id=group_id, +body=body, - 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, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Project Collaborator + """ Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[GroupCollaborator, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +group_id=group_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, group_id=group_id, client=client, body=body)).parsed + )).parsed 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..94df3dc7a 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 @@ -1,30 +1,43 @@ 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.project_update import ProjectUpdate from ...models.project_update_response import ProjectUpdateResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: ProjectUpdate, -def _get_kwargs(project_id: str, *, body: ProjectUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}", + "path": "/projects/{project_id}".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: ProjectUpdate) -> dict[str, Any]: return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: + if response.status_code == : response_200 = ProjectUpdateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: ProjectUpdate + project_id: str, + *, + client: ApiClient, + body: ProjectUpdate, + ) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: - """Update Project + """ Update Project Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ProjectUpdateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: ProjectUpdate + project_id: str, + *, + client: ApiClient, + body: ProjectUpdate, + ) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: - """Update Project + """ Update Project Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, ProjectUpdateResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: ProjectUpdate + project_id: str, + *, + client: ApiClient, + body: ProjectUpdate, + ) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: - """Update Project + """ Update Project Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectUpdateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: ProjectUpdate + project_id: str, + *, + client: ApiClient, + body: ProjectUpdate, + ) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: - """Update Project + """ Update Project Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, ProjectUpdateResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..a5e9b80e6 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 @@ -1,30 +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.collaborator_update import CollaboratorUpdate from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + user_id: str, + *, + body: CollaboratorUpdate, -def _get_kwargs(project_id: str, user_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/projects/{project_id}/users/{user_id}", + "path": "/projects/{project_id}/users/{user_id}".format(project_id=project_id,user_id=user_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, user_id: str, *, body: CollaboratorUpdate) -> d return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, UserCollaborator]]: + if response.status_code == : response_200 = UserCollaborator.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, UserCollaborator]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, UserCollaborator]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """Update User Project Collaborator + """ Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, user_id=user_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +user_id=user_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """Update User Project Collaborator + """ Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -106,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, UserCollaborator] - """ + """ + - return sync_detailed(project_id=project_id, user_id=user_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +user_id=user_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """Update User Project Collaborator + """ Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, user_id=user_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +user_id=user_id, +body=body, - 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, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + project_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """Update User Project Collaborator + """ Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, UserCollaborator] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +user_id=user_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, user_id=user_id, client=client, body=body)).parsed + )).parsed 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..756bc5ade 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 @@ -1,31 +1,43 @@ 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.body_upload_file_projects_project_id_upload_file_post import ( - BodyUploadFileProjectsProjectIdUploadFilePost, -) + +from ...models.body_upload_file_projects_project_id_upload_file_post import BodyUploadFileProjectsProjectIdUploadFilePost from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: BodyUploadFileProjectsProjectIdUploadFilePost, -def _get_kwargs(project_id: str, *, body: BodyUploadFileProjectsProjectIdUploadFilePost) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/upload_file", + "path": "/projects/{project_id}/upload_file".format(project_id=project_id,), } _kwargs["files"] = body.to_multipart() + + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -33,12 +45,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +70,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost + project_id: str, + *, + client: ApiClient, + body: BodyUploadFileProjectsProjectIdUploadFilePost, + ) -> Response[Union[Any, HTTPValidationError]]: - """Upload File + """ Upload File Args: project_id (str): @@ -70,19 +88,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost + project_id: str, + *, + client: ApiClient, + body: BodyUploadFileProjectsProjectIdUploadFilePost, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Upload File + """ Upload File Args: project_id (str): @@ -94,15 +122,24 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost + project_id: str, + *, + client: ApiClient, + body: BodyUploadFileProjectsProjectIdUploadFilePost, + ) -> Response[Union[Any, HTTPValidationError]]: - """Upload File + """ Upload File Args: project_id (str): @@ -114,19 +151,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost + project_id: str, + *, + client: ApiClient, + body: BodyUploadFileProjectsProjectIdUploadFilePost, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Upload File + """ Upload File Args: project_id (str): @@ -138,6 +185,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..b9fc6a3e8 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 @@ -1,19 +1,31 @@ from http import HTTPStatus -from typing import Any, Optional +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.bulk_delete_prompt_templates_request import BulkDeletePromptTemplatesRequest from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + -def _get_kwargs(*, body: BulkDeletePromptTemplatesRequest) -> dict[str, Any]: +def _get_kwargs( + *, + body: BulkDeletePromptTemplatesRequest, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, @@ -23,16 +35,21 @@ def _get_kwargs(*, body: BulkDeletePromptTemplatesRequest) -> dict[str, Any]: _kwargs["json"] = body.to_dict() + 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: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -49,8 +66,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Response[HTTPValidationError]: - """Bulk Delete Global Templates +def sync_detailed( + *, + client: ApiClient, + body: BulkDeletePromptTemplatesRequest, + +) -> Response[HTTPValidationError]: + """ Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -82,17 +104,27 @@ def sync_detailed(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) Returns: Response[HTTPValidationError] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: BulkDeletePromptTemplatesRequest, -def sync(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Optional[HTTPValidationError]: - """Bulk Delete Global Templates +) -> Optional[HTTPValidationError]: + """ Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -124,15 +156,22 @@ def sync(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Option Returns: HTTPValidationError - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: BulkDeletePromptTemplatesRequest + *, + client: ApiClient, + body: BulkDeletePromptTemplatesRequest, + ) -> Response[HTTPValidationError]: - """Bulk Delete Global Templates + """ Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -164,17 +203,27 @@ async def asyncio_detailed( Returns: Response[HTTPValidationError] - """ + """ - kwargs = _get_kwargs(body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, + body: BulkDeletePromptTemplatesRequest, -async def asyncio(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Optional[HTTPValidationError]: - """Bulk Delete Global Templates +) -> Optional[HTTPValidationError]: + """ Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -206,6 +255,11 @@ async def asyncio(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) Returns: HTTPValidationError - """ + """ + + + 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/prompts/create_global_prompt_template_templates_post.py b/src/galileo/resources/api/prompts/create_global_prompt_template_templates_post.py index b1acf22ad..330c24203 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 @@ -1,26 +1,42 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.create_prompt_template_with_version_request_body import CreatePromptTemplateWithVersionRequestBody from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + *, + body: CreatePromptTemplateWithVersionRequestBody, -def _get_kwargs(*, body: CreatePromptTemplateWithVersionRequestBody) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/templates"} + + + + + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/templates", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,16 +45,18 @@ def _get_kwargs(*, body: CreatePromptTemplateWithVersionRequestBody) -> dict[str return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -46,9 +64,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -58,9 +74,12 @@ def _build_response( def sync_detailed( - *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Global Prompt Template + """ Create Global Prompt Template Create a global prompt template. @@ -90,19 +109,27 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - 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: CreatePromptTemplateWithVersionRequestBody + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Global Prompt Template + """ Create Global Prompt Template Create a global prompt template. @@ -132,15 +159,22 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Global Prompt Template + """ Create Global Prompt Template Create a global prompt template. @@ -170,19 +204,27 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - 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: CreatePromptTemplateWithVersionRequestBody + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Global Prompt Template + """ Create Global Prompt Template Create a global prompt template. @@ -212,6 +254,11 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + + + 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/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..7784e3de0 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 @@ -1,30 +1,43 @@ 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.base_prompt_template_version import BasePromptTemplateVersion from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + *, + body: BasePromptTemplateVersion, -def _get_kwargs(template_id: str, *, body: BasePromptTemplateVersion) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/templates/{template_id}/versions", + "path": "/templates/{template_id}/versions".format(template_id=template_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(template_id: str, *, body: BasePromptTemplateVersion) -> dict[st return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Global Prompt Template Version + """ Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -92,19 +109,29 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Global Prompt Template Version + """ Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -132,15 +159,24 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(template_id=template_id, client=client, body=body).parsed + return sync_detailed( + template_id=template_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Global Prompt Template Version + """ Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -168,19 +204,29 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Global Prompt Template Version + """ Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -208,6 +254,12 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +client=client, +body=body, - return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed + )).parsed 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..ba9de9807 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 @@ -1,26 +1,38 @@ 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.group_collaborator import GroupCollaborator from ...models.group_collaborator_create import GroupCollaboratorCreate from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + *, + body: list['GroupCollaboratorCreate'], -def _get_kwargs(template_id: str, *, body: list["GroupCollaboratorCreate"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/templates/{template_id}/groups", + "path": "/templates/{template_id}/groups".format(template_id=template_id,), } _kwargs["json"] = [] @@ -28,6 +40,9 @@ def _get_kwargs(template_id: str, *, body: list["GroupCollaboratorCreate"]) -> d body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) + + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -36,21 +51,23 @@ def _get_kwargs(template_id: str, *, body: list["GroupCollaboratorCreate"]) -> d return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = GroupCollaborator.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,9 +85,13 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -86,19 +105,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -112,15 +141,24 @@ def sync( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ + """ + - return sync_detailed(template_id=template_id, client=client, body=body).parsed + return sync_detailed( + template_id=template_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -134,19 +172,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: - """Create Group Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['GroupCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: + """ Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -160,6 +208,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +client=client, +body=body, - return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed + )).parsed 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..eb512cfff 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 @@ -1,30 +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.base_prompt_template_version import BasePromptTemplateVersion from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + template_id: str, + *, + body: BasePromptTemplateVersion, -def _get_kwargs(project_id: str, template_id: str, *, body: BasePromptTemplateVersion) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/templates/{template_id}/versions", + "path": "/projects/{project_id}/templates/{template_id}/versions".format(project_id=project_id,template_id=template_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, template_id: str, *, body: BasePromptTemplateVe return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + project_id: str, + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Prompt Template Version + """ Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -97,19 +116,31 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + project_id: str, + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Prompt Template Version + """ Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -142,15 +173,26 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, template_id=template_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +template_id=template_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + project_id: str, + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Prompt Template Version + """ Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -183,19 +225,31 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, +body=body, - 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, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion + project_id: str, + template_id: str, + *, + client: ApiClient, + body: BasePromptTemplateVersion, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Create Prompt Template Version + """ Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -228,6 +282,13 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +template_id=template_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, template_id=template_id, client=client, body=body)).parsed + )).parsed 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..dd2baf895 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 @@ -1,30 +1,43 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.create_prompt_template_with_version_request_body import CreatePromptTemplateWithVersionRequestBody from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: CreatePromptTemplateWithVersionRequestBody, -def _get_kwargs(project_id: str, *, body: CreatePromptTemplateWithVersionRequestBody) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/templates", + "path": "/projects/{project_id}/templates".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: CreatePromptTemplateWithVersionRequest return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody + project_id: str, + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Prompt Template With Version + """ Create Prompt Template With Version For a given project, create a prompt template. @@ -101,19 +118,29 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody + project_id: str, + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Prompt Template With Version + """ Create Prompt Template With Version For a given project, create a prompt template. @@ -150,15 +177,24 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody + project_id: str, + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Prompt Template With Version + """ Create Prompt Template With Version For a given project, create a prompt template. @@ -195,19 +231,29 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody + project_id: str, + *, + client: ApiClient, + body: CreatePromptTemplateWithVersionRequestBody, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Create Prompt Template With Version + """ Create Prompt Template With Version For a given project, create a prompt template. @@ -244,6 +290,12 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..a36c32e40 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 @@ -1,26 +1,38 @@ 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.user_collaborator import UserCollaborator from ...models.user_collaborator_create import UserCollaboratorCreate -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + *, + body: list['UserCollaboratorCreate'], -def _get_kwargs(template_id: str, *, body: list["UserCollaboratorCreate"]) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/templates/{template_id}/users", + "path": "/templates/{template_id}/users".format(template_id=template_id,), } _kwargs["json"] = [] @@ -28,6 +40,9 @@ def _get_kwargs(template_id: str, *, body: list["UserCollaboratorCreate"]) -> di body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) + + + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -36,21 +51,23 @@ def _get_kwargs(template_id: str, *, body: list["UserCollaboratorCreate"]) -> di return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = UserCollaborator.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,9 +85,13 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Prompt Template Collaborators Args: template_id (str): @@ -84,19 +103,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Prompt Template Collaborators Args: template_id (str): @@ -108,15 +137,24 @@ def sync( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ + """ + - return sync_detailed(template_id=template_id, client=client, body=body).parsed + return sync_detailed( + template_id=template_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Prompt Template Collaborators Args: template_id (str): @@ -128,19 +166,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] -) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: - """Create User Prompt Template Collaborators + template_id: str, + *, + client: ApiClient, + body: list['UserCollaboratorCreate'], + +) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: + """ Create User Prompt Template Collaborators Args: template_id (str): @@ -152,6 +200,12 @@ async def asyncio( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +client=client, +body=body, - return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed + )).parsed 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..7462b009d 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 @@ -1,43 +1,56 @@ 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.delete_prompt_response import DeletePromptResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, -def _get_kwargs(template_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/templates/{template_id}", + "path": "/templates/{template_id}".format(template_id=template_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[DeletePromptResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + if response.status_code == : response_200 = DeletePromptResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -56,8 +67,13 @@ def _build_response( ) -def sync_detailed(template_id: str, *, client: ApiClient) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Global Template +def sync_detailed( + template_id: str, + *, + client: ApiClient, + +) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: + """ Delete Global Template Delete a global prompt template given a template ID. @@ -82,17 +98,27 @@ def sync_detailed(template_id: str, *, client: ApiClient) -> Response[Union[Dele Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + template_id: str, + *, + client: ApiClient, -def sync(template_id: str, *, client: ApiClient) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Global Template +) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + """ Delete Global Template Delete a global prompt template given a template ID. @@ -117,15 +143,22 @@ def sync(template_id: str, *, client: ApiClient) -> Optional[Union[DeletePromptR Returns: Union[DeletePromptResponse, HTTPValidationError] - """ + """ + - return sync_detailed(template_id=template_id, client=client).parsed + return sync_detailed( + template_id=template_id, +client=client, + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Global Template + """ Delete Global Template Delete a global prompt template given a template ID. @@ -150,17 +183,27 @@ async def asyncio_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + template_id: str, + *, + client: ApiClient, -async def asyncio(template_id: str, *, client: ApiClient) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Global Template +) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + """ Delete Global Template Delete a global prompt template given a template ID. @@ -185,6 +228,11 @@ async def asyncio(template_id: str, *, client: ApiClient) -> Optional[Union[Dele Returns: Union[DeletePromptResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +client=client, - return (await asyncio_detailed(template_id=template_id, client=client)).parsed + )).parsed 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..58c833e33 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(template_id: str, group_id: str) -> dict[str, Any]: +def _get_kwargs( + template_id: str, + group_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/templates/{template_id}/groups/{group_id}", + "path": "/templates/{template_id}/groups/{group_id}".format(template_id=template_id,group_id=group_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(template_id: str, group_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete Group Prompt Template Collaborator +def sync_detailed( + template_id: str, + group_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -65,17 +85,29 @@ def sync_detailed(template_id: str, group_id: str, *, client: ApiClient) -> Resp Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, group_id=group_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +group_id=group_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + template_id: str, + group_id: str, + *, + client: ApiClient, -def sync(template_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Group Prompt Template Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -89,15 +121,24 @@ def sync(template_id: str, group_id: str, *, client: ApiClient) -> Optional[Unio Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(template_id=template_id, group_id=group_id, client=client).parsed + return sync_detailed( + template_id=template_id, +group_id=group_id, +client=client, + + ).parsed async def asyncio_detailed( - template_id: str, group_id: str, *, client: ApiClient + template_id: str, + group_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete Group Prompt Template Collaborator + """ Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -111,17 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, group_id=group_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +group_id=group_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + template_id: str, + group_id: str, + *, + client: ApiClient, -async def asyncio(template_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete Group Prompt Template Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -135,6 +188,12 @@ async def asyncio(template_id: str, group_id: str, *, client: ApiClient) -> Opti Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +group_id=group_id, +client=client, - return (await asyncio_detailed(template_id=template_id, group_id=group_id, client=client)).parsed + )).parsed 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..4e97c746c 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 @@ -1,43 +1,57 @@ 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.delete_prompt_response import DeletePromptResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + template_id: str, -def _get_kwargs(project_id: str, template_id: str) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/projects/{project_id}/templates/{template_id}", + "path": "/projects/{project_id}/templates/{template_id}".format(project_id=project_id,template_id=template_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[DeletePromptResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + if response.status_code == : response_200 = DeletePromptResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Template + """ Delete Template Args: project_id (str): @@ -71,19 +87,29 @@ def sync_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Template + """ Delete Template Args: project_id (str): @@ -95,15 +121,24 @@ def sync( Returns: Union[DeletePromptResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, template_id=template_id, client=client).parsed + return sync_detailed( + project_id=project_id, +template_id=template_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Template + """ Delete Template Args: project_id (str): @@ -115,19 +150,29 @@ async def asyncio_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_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, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """Delete Template + """ Delete Template Args: project_id (str): @@ -139,6 +184,12 @@ async def asyncio( Returns: Union[DeletePromptResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +template_id=template_id, +client=client, - return (await asyncio_detailed(project_id=project_id, template_id=template_id, client=client)).parsed + )).parsed 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..3286c8d9c 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 @@ -1,26 +1,38 @@ 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 ...types import Response +from typing import cast + -def _get_kwargs(template_id: str, user_id: str) -> dict[str, Any]: +def _get_kwargs( + template_id: str, + user_id: str, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": f"/templates/{template_id}/users/{user_id}", + "path": "/templates/{template_id}/users/{user_id}".format(template_id=template_id,user_id=user_id,), } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -28,12 +40,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,8 +64,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(template_id: str, user_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: - """Delete User Prompt Template Collaborator +def sync_detailed( + template_id: str, + user_id: str, + *, + client: ApiClient, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -65,17 +85,29 @@ def sync_detailed(template_id: str, user_id: str, *, client: ApiClient) -> Respo Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, user_id=user_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +user_id=user_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + template_id: str, + user_id: str, + *, + client: ApiClient, -def sync(template_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete User Prompt Template Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -89,15 +121,24 @@ def sync(template_id: str, user_id: str, *, client: ApiClient) -> Optional[Union Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(template_id=template_id, user_id=user_id, client=client).parsed + return sync_detailed( + template_id=template_id, +user_id=user_id, +client=client, + + ).parsed async def asyncio_detailed( - template_id: str, user_id: str, *, client: ApiClient + template_id: str, + user_id: str, + *, + client: ApiClient, + ) -> Response[Union[Any, HTTPValidationError]]: - """Delete User Prompt Template Collaborator + """ Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -111,17 +152,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, user_id=user_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +user_id=user_id, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + template_id: str, + user_id: str, + *, + client: ApiClient, -async def asyncio(template_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: - """Delete User Prompt Template Collaborator +) -> Optional[Union[Any, HTTPValidationError]]: + """ Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -135,6 +188,12 @@ async def asyncio(template_id: str, user_id: str, *, client: ApiClient) -> Optio Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +user_id=user_id, +client=client, - return (await asyncio_detailed(template_id=template_id, user_id=user_id, client=client)).parsed + )).parsed 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..6af0238e1 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 @@ -1,25 +1,41 @@ 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.template_stub_request import TemplateStubRequest -from ...types import Response +from typing import cast + -def _get_kwargs(*, body: TemplateStubRequest) -> dict[str, Any]: +def _get_kwargs( + *, + body: TemplateStubRequest, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + - _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/input_stub"} + _kwargs: dict[str, Any] = { + "method": RequestMethod.POST, + "return_raw_response": True, + "path": "/input_stub", + } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -29,12 +45,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -51,8 +69,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed(*, client: ApiClient, body: TemplateStubRequest) -> Response[Union[Any, HTTPValidationError]]: - """Generate Template Input Stub +def sync_detailed( + *, + client: ApiClient, + body: TemplateStubRequest, + +) -> Response[Union[Any, HTTPValidationError]]: + """ Generate Template Input Stub Args: body (TemplateStubRequest): @@ -63,17 +86,27 @@ def sync_detailed(*, client: ApiClient, body: TemplateStubRequest) -> Response[U Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + *, + client: ApiClient, + body: TemplateStubRequest, -def sync(*, client: ApiClient, body: TemplateStubRequest) -> Optional[Union[Any, HTTPValidationError]]: - """Generate Template Input Stub +) -> Optional[Union[Any, HTTPValidationError]]: + """ Generate Template Input Stub Args: body (TemplateStubRequest): @@ -84,15 +117,22 @@ def sync(*, client: ApiClient, body: TemplateStubRequest) -> Optional[Union[Any, Returns: Union[Any, HTTPValidationError] - """ + """ - return sync_detailed(client=client, body=body).parsed + return sync_detailed( + client=client, +body=body, + + ).parsed async def asyncio_detailed( - *, client: ApiClient, body: TemplateStubRequest + *, + client: ApiClient, + body: TemplateStubRequest, + ) -> Response[Union[Any, HTTPValidationError]]: - """Generate Template Input Stub + """ Generate Template Input Stub Args: body (TemplateStubRequest): @@ -103,17 +143,27 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, + + ) + + response = await client.arequest( + **kwargs + ) return _build_response(client=client, response=response) +async def asyncio( + *, + client: ApiClient, + body: TemplateStubRequest, -async def asyncio(*, client: ApiClient, body: TemplateStubRequest) -> Optional[Union[Any, HTTPValidationError]]: - """Generate Template Input Stub +) -> Optional[Union[Any, HTTPValidationError]]: + """ Generate Template Input Stub Args: body (TemplateStubRequest): @@ -124,6 +174,11 @@ async def asyncio(*, client: ApiClient, body: TemplateStubRequest) -> Optional[U Returns: Union[Any, HTTPValidationError] - """ + """ + + + 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/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..2089ad632 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 @@ -1,43 +1,56 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, -def _get_kwargs(template_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"/templates/{template_id}", + "path": "/templates/{template_id}".format(template_id=template_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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +58,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +68,12 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Global Template + """ Get Global Template Get a global prompt template given a template ID. @@ -86,17 +100,27 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, + + ) + + response = client.request( + **kwargs, + ) return _build_response(client=client, response=response) +def sync( + template_id: str, + *, + client: ApiClient, -def sync(template_id: str, *, client: ApiClient) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Global Template +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + """ Get Global Template Get a global prompt template given a template ID. @@ -123,15 +147,22 @@ def sync(template_id: str, *, client: ApiClient) -> Optional[Union[BasePromptTem Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ - return sync_detailed(template_id=template_id, client=client).parsed + return sync_detailed( + template_id=template_id, +client=client, + + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Global Template + """ Get Global Template Get a global prompt template given a template ID. @@ -158,19 +189,27 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient + template_id: str, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Global Template + """ Get Global Template Get a global prompt template given a template ID. @@ -197,6 +236,11 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +client=client, - return (await asyncio_detailed(template_id=template_id, client=client)).parsed + )).parsed 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..b7a9588ba 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 @@ -1,43 +1,57 @@ 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.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + version: int, -def _get_kwargs(template_id: str, version: int) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/templates/{template_id}/versions/{version}", + "path": "/templates/{template_id}/versions/{version}".format(template_id=template_id,version=version,), } + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Global Template Version + """ Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -87,19 +103,29 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +version=version, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Global Template Version + """ Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -127,15 +153,24 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + - return sync_detailed(template_id=template_id, version=version, client=client).parsed + return sync_detailed( + template_id=template_id, +version=version, +client=client, + ).parsed async def asyncio_detailed( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Global Template Version + """ Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -163,19 +198,29 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +version=version, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Global Template Version + """ Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -203,6 +248,12 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +version=version, +client=client, - return (await asyncio_detailed(template_id=template_id, version=version, client=client)).parsed + )).parsed 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..561c4b825 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 @@ -1,48 +1,61 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, -def _get_kwargs(project_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}/templates", + "path": "/projects/{project_id}/templates".format(project_id=project_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, list["BasePromptTemplateResponse"]]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: + if response.status_code == : response_200 = [] _response_200 = response.json() - for response_200_item_data in _response_200: + for response_200_item_data in (_response_200): response_200_item = BasePromptTemplateResponse.from_dict(response_200_item_data) + + response_200.append(response_200_item) return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +63,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +73,12 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: - """Get Project Templates + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: + """ Get Project Templates Get all prompt templates for a project. @@ -89,19 +103,27 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: - """Get Project Templates + project_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: + """ Get Project Templates Get all prompt templates for a project. @@ -126,15 +148,22 @@ def sync( Returns: Union[HTTPValidationError, list['BasePromptTemplateResponse']] - """ + """ - return sync_detailed(project_id=project_id, client=client).parsed + return sync_detailed( + project_id=project_id, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient -) -> Response[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: - """Get Project Templates + project_id: str, + *, + client: ApiClient, + +) -> Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: + """ Get Project Templates Get all prompt templates for a project. @@ -159,19 +188,27 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]] - """ + """ - kwargs = _get_kwargs(project_id=project_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_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, *, client: ApiClient -) -> Optional[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: - """Get Project Templates + project_id: str, + *, + client: ApiClient, + +) -> Optional[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: + """ Get Project Templates Get all prompt templates for a project. @@ -196,6 +233,11 @@ async def asyncio( Returns: Union[HTTPValidationError, list['BasePromptTemplateResponse']] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, - return (await asyncio_detailed(project_id=project_id, client=client)).parsed + )).parsed 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..c2504fe03 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 @@ -1,43 +1,57 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + template_id: str, -def _get_kwargs(project_id: str, template_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}/templates/{template_id}", + "path": "/projects/{project_id}/templates/{template_id}".format(project_id=project_id,template_id=template_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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Template From Project + """ Get Template From Project Get a prompt template from a project. @@ -87,19 +103,29 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Template From Project + """ Get Template From Project Get a prompt template from a project. @@ -127,15 +153,24 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, template_id=template_id, client=client).parsed + return sync_detailed( + project_id=project_id, +template_id=template_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Template From Project + """ Get Template From Project Get a prompt template from a project. @@ -163,19 +198,29 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_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, template_id: str, *, client: ApiClient + project_id: str, + template_id: str, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Get Template From Project + """ Get Template From Project Get a prompt template from a project. @@ -203,6 +248,12 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +template_id=template_id, +client=client, - return (await asyncio_detailed(project_id=project_id, template_id=template_id, client=client)).parsed + )).parsed 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..9ce56fe2a 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 @@ -1,20 +1,34 @@ 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.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + -def _get_kwargs(project_id: str, *, template_name: str, version: Union[None, Unset, int] = UNSET) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + *, + template_name: str, + version: Union[None, Unset, int] = UNSET, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -27,31 +41,36 @@ def _get_kwargs(project_id: str, *, template_name: str, version: Union[None, Uns json_version = version params["version"] = json_version + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/templates/versions", + "path": "/projects/{project_id}/templates/versions".format(project_id=project_id,), "params": params, } + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : 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 +78,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -71,9 +88,14 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET + project_id: str, + *, + client: ApiClient, + template_name: str, + version: Union[None, Unset, int] = UNSET, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version By Name + """ Get Template Version By Name Get a prompt template from a project. @@ -105,19 +127,31 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_name=template_name, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_name=template_name, +version=version, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET + project_id: str, + *, + client: ApiClient, + template_name: str, + version: Union[None, Unset, int] = UNSET, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version By Name + """ Get Template Version By Name Get a prompt template from a project. @@ -149,15 +183,26 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, client=client, template_name=template_name, version=version).parsed + return sync_detailed( + project_id=project_id, +client=client, +template_name=template_name, +version=version, + + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET + project_id: str, + *, + client: ApiClient, + template_name: str, + version: Union[None, Unset, int] = UNSET, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version By Name + """ Get Template Version By Name Get a prompt template from a project. @@ -189,19 +234,31 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_name=template_name, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_name=template_name, +version=version, - 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, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET + project_id: str, + *, + client: ApiClient, + template_name: str, + version: Union[None, Unset, int] = UNSET, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version By Name + """ Get Template Version By Name Get a prompt template from a project. @@ -233,8 +290,13 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, client=client, template_name=template_name, version=version) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +client=client, +template_name=template_name, +version=version, + + )).parsed 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..1cf88fc49 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 @@ -1,43 +1,58 @@ 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.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + template_id: str, + version: int, -def _get_kwargs(project_id: str, template_id: str, version: int) -> 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}/templates/{template_id}/versions/{version}", + "path": "/projects/{project_id}/templates/{template_id}/versions/{version}".format(project_id=project_id,template_id=template_id,version=version,), } + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +60,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +70,14 @@ def _build_response( def sync_detailed( - project_id: str, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version + """ Get Template Version Get a specific version of a prompt template. @@ -88,19 +106,31 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, +version=version, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version + """ Get Template Version Get a specific version of a prompt template. @@ -129,15 +159,26 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, template_id=template_id, version=version, client=client).parsed + return sync_detailed( + project_id=project_id, +template_id=template_id, +version=version, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version + """ Get Template Version Get a specific version of a prompt template. @@ -166,19 +207,31 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, +version=version, - 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, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """Get Template Version + """ Get Template Version Get a specific version of a prompt template. @@ -207,8 +260,13 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, template_id=template_id, version=version, client=client) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +template_id=template_id, +version=version, +client=client, + + )).parsed 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..365325487 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 @@ -1,22 +1,33 @@ 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.list_group_collaborators_response import ListGroupCollaboratorsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - template_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/templates/{template_id}/groups", + "path": "/templates/{template_id}/groups".format(template_id=template_id,), "params": params, } + 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, ListGroupCollaboratorsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: + if response.status_code == : response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Prompt Template Collaborators + """ List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Prompt Template Collaborators + """ List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -112,15 +143,26 @@ def sync( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ - return sync_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + template_id=template_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Prompt Template Collaborators + """ List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """List Group Prompt Template Collaborators + """ List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ - return ( - await asyncio_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + template_id=template_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..4a654e629 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 @@ -1,22 +1,33 @@ 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.list_user_collaborators_response import ListUserCollaboratorsResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - template_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -24,31 +35,36 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/templates/{template_id}/users", + "path": "/templates/{template_id}/users".format(template_id=template_id,), "params": params, } + 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, ListUserCollaboratorsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: + if response.status_code == : response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +72,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,9 +82,14 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Prompt Template Collaborators + """ List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -85,19 +104,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Prompt Template Collaborators + """ List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -112,15 +143,26 @@ def sync( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ - return sync_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + template_id=template_id, +client=client, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Prompt Template Collaborators + """ List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -135,19 +177,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + template_id: str, + *, + client: ApiClient, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """List User Prompt Template Collaborators + """ List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -162,8 +216,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ - return ( - await asyncio_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit) - ).parsed + + return (await asyncio_detailed( + template_id=template_id, +client=client, +starting_token=starting_token, +limit=limit, + + )).parsed 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..84178103e 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 @@ -1,17 +1,21 @@ 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.list_prompt_template_version_params import ListPromptTemplateVersionParams from ...models.list_prompt_template_version_response import ListPromptTemplateVersionResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( @@ -20,8 +24,12 @@ def _get_kwargs( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -29,17 +37,20 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/templates/{template_id}/versions/query", + "path": "/templates/{template_id}/versions/query".format(template_id=template_id,), "params": params, } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -48,16 +59,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: + if response.status_code == : response_200 = ListPromptTemplateVersionResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -65,9 +78,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -83,8 +94,9 @@ def sync_detailed( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """Query Template Versions + """ Query Template Versions Query versions of a specific prompt template. @@ -116,14 +128,22 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( template_id: str, @@ -132,8 +152,9 @@ def sync( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """Query Template Versions + """ Query Template Versions Query versions of a specific prompt template. @@ -165,12 +186,17 @@ def sync( Returns: Union[HTTPValidationError, ListPromptTemplateVersionResponse] - """ + """ + return sync_detailed( - template_id=template_id, client=client, body=body, starting_token=starting_token, limit=limit - ).parsed + template_id=template_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + ).parsed async def asyncio_detailed( template_id: str, @@ -179,8 +205,9 @@ async def asyncio_detailed( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """Query Template Versions + """ Query Template Versions Query versions of a specific prompt template. @@ -212,14 +239,22 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( template_id: str, @@ -228,8 +263,9 @@ async def asyncio( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """Query Template Versions + """ Query Template Versions Query versions of a specific prompt template. @@ -261,10 +297,14 @@ async def asyncio( Returns: Union[HTTPValidationError, ListPromptTemplateVersionResponse] - """ + """ - return ( - await asyncio_detailed( - template_id=template_id, client=client, body=body, starting_token=starting_token, limit=limit - ) - ).parsed + + return (await asyncio_detailed( + template_id=template_id, +client=client, +body=body, +starting_token=starting_token, +limit=limit, + + )).parsed 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..dfd7b71b7 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 @@ -1,23 +1,34 @@ 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.list_prompt_template_params import ListPromptTemplateParams from ...models.list_prompt_template_response import ListPromptTemplateResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - *, body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + *, + body: ListPromptTemplateParams, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -25,8 +36,10 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -36,6 +49,7 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -44,16 +58,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: + if response.status_code == : response_200 = ListPromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -61,9 +77,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -78,8 +92,9 @@ def sync_detailed( body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """Query Templates + """ Query Templates Query prompt templates the user has access to. @@ -108,14 +123,21 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateResponse]] - """ + """ - kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( *, @@ -123,8 +145,9 @@ def sync( body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """Query Templates + """ Query Templates Query prompt templates the user has access to. @@ -153,10 +176,16 @@ def sync( Returns: Union[HTTPValidationError, ListPromptTemplateResponse] - """ + """ - return sync_detailed(client=client, body=body, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + client=client, +body=body, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( *, @@ -164,8 +193,9 @@ async def asyncio_detailed( body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """Query Templates + """ Query Templates Query prompt templates the user has access to. @@ -194,14 +224,21 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateResponse]] - """ + """ - kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -209,8 +246,9 @@ async def asyncio( body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """Query Templates + """ Query Templates Query prompt templates the user has access to. @@ -239,6 +277,13 @@ async def asyncio( Returns: Union[HTTPValidationError, ListPromptTemplateResponse] - """ + """ + + + return (await asyncio_detailed( + client=client, +body=body, +starting_token=starting_token, +limit=limit, - return (await asyncio_detailed(client=client, body=body, starting_token=starting_token, limit=limit)).parsed + )).parsed 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..1bd6625e7 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 @@ -1,23 +1,34 @@ 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.render_template_request import RenderTemplateRequest from ...models.render_template_response import RenderTemplateResponse -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + def _get_kwargs( - *, body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 + *, + body: RenderTemplateRequest, + starting_token: Union[Unset, int] = 0, + limit: Union[Unset, int] = 100, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -25,8 +36,10 @@ def _get_kwargs( params["limit"] = limit + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -36,6 +49,7 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -44,16 +58,18 @@ def _get_kwargs( return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: + if response.status_code == : response_200 = RenderTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -61,9 +77,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -78,8 +92,9 @@ def sync_detailed( body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: - """Render Template + """ Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -92,14 +107,21 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RenderTemplateResponse]] - """ + """ - kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - response = client.request(**kwargs) + kwargs = _get_kwargs( + body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( *, @@ -107,8 +129,9 @@ def sync( body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: - """Render Template + """ Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -121,10 +144,16 @@ def sync( Returns: Union[HTTPValidationError, RenderTemplateResponse] - """ + """ - return sync_detailed(client=client, body=body, starting_token=starting_token, limit=limit).parsed + return sync_detailed( + client=client, +body=body, +starting_token=starting_token, +limit=limit, + + ).parsed async def asyncio_detailed( *, @@ -132,8 +161,9 @@ async def asyncio_detailed( body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: - """Render Template + """ Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -146,14 +176,21 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RenderTemplateResponse]] - """ + """ - kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + body=body, +starting_token=starting_token, +limit=limit, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( *, @@ -161,8 +198,9 @@ async def asyncio( body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, + ) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: - """Render Template + """ Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -175,6 +213,13 @@ async def asyncio( Returns: Union[HTTPValidationError, RenderTemplateResponse] - """ + """ + + + return (await asyncio_detailed( + client=client, +body=body, +starting_token=starting_token, +limit=limit, - return (await asyncio_detailed(client=client, body=body, starting_token=starting_token, limit=limit)).parsed + )).parsed 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..072924a09 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 @@ -1,43 +1,57 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + version: int, -def _get_kwargs(template_id: str, version: int) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/templates/{template_id}/versions/{version}", + "path": "/templates/{template_id}/versions/{version}".format(template_id=template_id,version=version,), } + 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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Global Template Version + """ Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -87,19 +103,29 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +version=version, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Global Template Version + """ Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -127,15 +153,24 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + - return sync_detailed(template_id=template_id, version=version, client=client).parsed + return sync_detailed( + template_id=template_id, +version=version, +client=client, + ).parsed async def asyncio_detailed( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Global Template Version + """ Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -163,19 +198,29 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +version=version, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, version: int, *, client: ApiClient + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Global Template Version + """ Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -203,6 +248,12 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +version=version, +client=client, - return (await asyncio_detailed(template_id=template_id, version=version, client=client)).parsed + )).parsed 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..b54f89ac8 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 @@ -1,43 +1,58 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + template_id: str, + version: int, -def _get_kwargs(project_id: str, template_id: str, version: int) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}/templates/{template_id}/versions/{version}", + "path": "/projects/{project_id}/templates/{template_id}/versions/{version}".format(project_id=project_id,template_id=template_id,version=version,), } + 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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +60,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +70,14 @@ def _build_response( def sync_detailed( - project_id: str, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Template Version + """ Set Selected Template Version Args: project_id (str): @@ -72,19 +90,31 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, +version=version, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Template Version + """ Set Selected Template Version Args: project_id (str): @@ -97,15 +127,26 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ - return sync_detailed(project_id=project_id, template_id=template_id, version=version, client=client).parsed + return sync_detailed( + project_id=project_id, +template_id=template_id, +version=version, +client=client, + + ).parsed async def asyncio_detailed( - project_id: str, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Template Version + """ Set Selected Template Version Args: project_id (str): @@ -118,19 +159,31 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +template_id=template_id, +version=version, - 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, template_id: str, version: int, *, client: ApiClient + project_id: str, + template_id: str, + version: int, + *, + client: ApiClient, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Set Selected Template Version + """ Set Selected Template Version Args: project_id (str): @@ -143,8 +196,13 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, template_id=template_id, version=version, client=client) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +template_id=template_id, +version=version, +client=client, + + )).parsed 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..23e9dea12 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 @@ -1,30 +1,43 @@ 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.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError from ...models.update_prompt_template_request import UpdatePromptTemplateRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + *, + body: UpdatePromptTemplateRequest, -def _get_kwargs(template_id: str, *, body: UpdatePromptTemplateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/templates/{template_id}", + "path": "/templates/{template_id}".format(template_id=template_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(template_id: str, *, body: UpdatePromptTemplateRequest) -> dict[ return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == : response_200 = BasePromptTemplateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest + template_id: str, + *, + client: ApiClient, + body: UpdatePromptTemplateRequest, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Update Global Template + """ Update Global Template Update a global prompt template. @@ -94,19 +111,29 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest + template_id: str, + *, + client: ApiClient, + body: UpdatePromptTemplateRequest, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Update Global Template + """ Update Global Template Update a global prompt template. @@ -136,15 +163,24 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + - return sync_detailed(template_id=template_id, client=client, body=body).parsed + return sync_detailed( + template_id=template_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest + template_id: str, + *, + client: ApiClient, + body: UpdatePromptTemplateRequest, + ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Update Global Template + """ Update Global Template Update a global prompt template. @@ -174,19 +210,29 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest + template_id: str, + *, + client: ApiClient, + body: UpdatePromptTemplateRequest, + ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """Update Global Template + """ Update Global Template Update a global prompt template. @@ -216,6 +262,12 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +client=client, +body=body, - return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed + )).parsed 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..a0e378608 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 @@ -1,30 +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.collaborator_update import CollaboratorUpdate from ...models.group_collaborator import GroupCollaborator from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + group_id: str, + *, + body: CollaboratorUpdate, -def _get_kwargs(template_id: str, group_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/templates/{template_id}/groups/{group_id}", + "path": "/templates/{template_id}/groups/{group_id}".format(template_id=template_id,group_id=group_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(template_id: str, group_id: str, *, body: CollaboratorUpdate) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: + if response.status_code == : response_200 = GroupCollaborator.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[GroupCollaborator, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[GroupCollaborator, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Prompt Template Collaborator + """ Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, group_id=group_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +group_id=group_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Prompt Template Collaborator + """ Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -106,15 +137,26 @@ def sync( Returns: Union[GroupCollaborator, HTTPValidationError] - """ + """ + - return sync_detailed(template_id=template_id, group_id=group_id, client=client, body=body).parsed + return sync_detailed( + template_id=template_id, +group_id=group_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Prompt Template Collaborator + """ Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, group_id=group_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +group_id=group_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + group_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """Update Group Prompt Template Collaborator + """ Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[GroupCollaborator, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +group_id=group_id, +client=client, +body=body, - return (await asyncio_detailed(template_id=template_id, group_id=group_id, client=client, body=body)).parsed + )).parsed 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..b9109a08e 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 @@ -1,30 +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.collaborator_update import CollaboratorUpdate from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator -from ...types import Response +from typing import cast + + +def _get_kwargs( + template_id: str, + user_id: str, + *, + body: CollaboratorUpdate, -def _get_kwargs(template_id: str, user_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/templates/{template_id}/users/{user_id}", + "path": "/templates/{template_id}/users/{user_id}".format(template_id=template_id,user_id=user_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(template_id: str, user_id: str, *, body: CollaboratorUpdate) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, UserCollaborator]]: + if response.status_code == : response_200 = UserCollaborator.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, UserCollaborator]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, UserCollaborator]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """Update User Prompt Template Collaborator + """ Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, user_id=user_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +user_id=user_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """Update User Prompt Template Collaborator + """ Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -106,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, UserCollaborator] - """ + """ + - return sync_detailed(template_id=template_id, user_id=user_id, client=client, body=body).parsed + return sync_detailed( + template_id=template_id, +user_id=user_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """Update User Prompt Template Collaborator + """ Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ - kwargs = _get_kwargs(template_id=template_id, user_id=user_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + template_id=template_id, +user_id=user_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = await client.arequest( + **kwargs + ) + return _build_response(client=client, response=response) async def asyncio( - template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate + template_id: str, + user_id: str, + *, + client: ApiClient, + body: CollaboratorUpdate, + ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """Update User Prompt Template Collaborator + """ Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, UserCollaborator] - """ + """ + + + return (await asyncio_detailed( + template_id=template_id, +user_id=user_id, +client=client, +body=body, - return (await asyncio_detailed(template_id=template_id, user_id=user_id, client=client, body=body)).parsed + )).parsed 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..c67203b60 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 @@ -1,30 +1,43 @@ 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.stage_db import StageDB from ...models.stage_with_rulesets import StageWithRulesets -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: StageWithRulesets, -def _get_kwargs(project_id: str, *, body: StageWithRulesets) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/stages", + "path": "/projects/{project_id}/stages".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -34,13 +47,17 @@ 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: + if response.status_code == : response_200 = StageDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +75,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, *, client: ApiClient, body: StageWithRulesets + project_id: str, + *, + client: ApiClient, + body: StageWithRulesets, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Create Stage + """ Create Stage Args: project_id (str): @@ -72,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: StageWithRulesets + project_id: str, + *, + client: ApiClient, + body: StageWithRulesets, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Create Stage + """ Create Stage Args: project_id (str): @@ -96,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: StageWithRulesets + project_id: str, + *, + client: ApiClient, + body: StageWithRulesets, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Create Stage + """ Create Stage Args: project_id (str): @@ -116,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: StageWithRulesets + project_id: str, + *, + client: ApiClient, + body: StageWithRulesets, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Create Stage + """ Create Stage Args: project_id (str): @@ -140,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..df56595c3 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 @@ -1,22 +1,34 @@ 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.stage_db import StageDB -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import cast, Union +from typing import Union + def _get_kwargs( - project_id: str, *, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET + project_id: str, + *, + stage_name: Union[None, Unset, str] = UNSET, + stage_id: Union[None, Unset, str] = UNSET, + ) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} @@ -34,15 +46,18 @@ def _get_kwargs( json_stage_id = 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} + _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": f"/projects/{project_id}/stages", + "path": "/projects/{project_id}/stages".format(project_id=project_id,), "params": params, } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -50,13 +65,17 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: - if response.status_code == 200: + if response.status_code == : response_200 = StageDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -79,8 +98,9 @@ def sync_detailed( client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Get Stage + """ Get Stage Args: project_id (str): @@ -93,14 +113,21 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, stage_name=stage_name, stage_id=stage_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +stage_name=stage_name, +stage_id=stage_id, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( project_id: str, @@ -108,8 +135,9 @@ def sync( client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Get Stage + """ Get Stage Args: project_id (str): @@ -122,10 +150,16 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ - return sync_detailed(project_id=project_id, client=client, stage_name=stage_name, stage_id=stage_id).parsed + return sync_detailed( + project_id=project_id, +client=client, +stage_name=stage_name, +stage_id=stage_id, + + ).parsed async def asyncio_detailed( project_id: str, @@ -133,8 +167,9 @@ async def asyncio_detailed( client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Get Stage + """ Get Stage Args: project_id (str): @@ -147,14 +182,21 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, stage_name=stage_name, stage_id=stage_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +stage_name=stage_name, +stage_id=stage_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, @@ -162,8 +204,9 @@ async def asyncio( client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Get Stage + """ Get Stage Args: project_id (str): @@ -176,8 +219,13 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ + """ - return ( - await asyncio_detailed(project_id=project_id, client=client, stage_name=stage_name, stage_id=stage_id) - ).parsed + + return (await asyncio_detailed( + project_id=project_id, +client=client, +stage_name=stage_name, +stage_id=stage_id, + + )).parsed 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..06b9b5108 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,32 +47,35 @@ def _get_kwargs(*, body: ProtectRequest) -> dict[str, Any]: return _kwargs -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(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: + if response.status_code == : + 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()) return response_200 - if response.status_code == 422: + if response.status_code == : 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..2bb1fe9a7 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 @@ -1,34 +1,50 @@ 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.stage_db import StageDB -from ...types import UNSET, Response, Unset +from ...types import UNSET, Unset +from typing import cast +from typing import Union + + +def _get_kwargs( + project_id: str, + stage_id: str, + *, + pause: Union[Unset, bool] = False, -def _get_kwargs(project_id: str, stage_id: str, *, pause: Union[Unset, bool] = False) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + params: dict[str, Any] = {} params["pause"] = pause + params = {k: v for k, v in params.items() if v is not UNSET and v is not None} + _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": f"/projects/{project_id}/stages/{stage_id}", + "path": "/projects/{project_id}/stages/{stage_id}".format(project_id=project_id,stage_id=stage_id,), "params": params, } + headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -36,13 +52,17 @@ 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: + if response.status_code == : response_200 = StageDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -60,9 +80,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False + project_id: str, + stage_id: str, + *, + client: ApiClient, + pause: Union[Unset, bool] = False, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Pause Stage + """ Pause Stage Args: project_id (str): @@ -75,19 +100,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, pause=pause) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +stage_id=stage_id, +pause=pause, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False + project_id: str, + stage_id: str, + *, + client: ApiClient, + pause: Union[Unset, bool] = False, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Pause Stage + """ Pause Stage Args: project_id (str): @@ -100,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ - return sync_detailed(project_id=project_id, stage_id=stage_id, client=client, pause=pause).parsed + return sync_detailed( + project_id=project_id, +stage_id=stage_id, +client=client, +pause=pause, + + ).parsed async def asyncio_detailed( - project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False + project_id: str, + stage_id: str, + *, + client: ApiClient, + pause: Union[Unset, bool] = False, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Pause Stage + """ Pause Stage Args: project_id (str): @@ -121,19 +169,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, pause=pause) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +stage_id=stage_id, +pause=pause, - 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, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False + project_id: str, + stage_id: str, + *, + client: ApiClient, + pause: Union[Unset, bool] = False, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Pause Stage + """ Pause Stage Args: project_id (str): @@ -146,6 +206,13 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +stage_id=stage_id, +client=client, +pause=pause, - return (await asyncio_detailed(project_id=project_id, stage_id=stage_id, client=client, pause=pause)).parsed + )).parsed 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..9933208b2 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 @@ -1,30 +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.rulesets_mixin import RulesetsMixin from ...models.stage_db import StageDB -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + stage_id: str, + *, + body: RulesetsMixin, -def _get_kwargs(project_id: str, stage_id: str, *, body: RulesetsMixin) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/stages/{stage_id}", + "path": "/projects/{project_id}/stages/{stage_id}".format(project_id=project_id,stage_id=stage_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -34,13 +48,17 @@ 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: + if response.status_code == : response_200 = StageDB.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -58,9 +76,14 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin + project_id: str, + stage_id: str, + *, + client: ApiClient, + body: RulesetsMixin, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Update Stage + """ Update Stage Args: project_id (str): @@ -73,19 +96,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +stage_id=stage_id, +body=body, - return _build_response(client=client, response=response) + ) + response = client.request( + **kwargs, + ) + + return _build_response(client=client, response=response) def sync( - project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin + project_id: str, + stage_id: str, + *, + client: ApiClient, + body: RulesetsMixin, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Update Stage + """ Update Stage Args: project_id (str): @@ -98,15 +133,26 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ - return sync_detailed(project_id=project_id, stage_id=stage_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +stage_id=stage_id, +client=client, +body=body, + + ).parsed async def asyncio_detailed( - project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin + project_id: str, + stage_id: str, + *, + client: ApiClient, + body: RulesetsMixin, + ) -> Response[Union[HTTPValidationError, StageDB]]: - """Update Stage + """ Update Stage Args: project_id (str): @@ -119,19 +165,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +stage_id=stage_id, +body=body, - 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, stage_id: str, *, client: ApiClient, body: RulesetsMixin + project_id: str, + stage_id: str, + *, + client: ApiClient, + body: RulesetsMixin, + ) -> Optional[Union[HTTPValidationError, StageDB]]: - """Update Stage + """ Update Stage Args: project_id (str): @@ -144,6 +202,13 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +stage_id=stage_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, stage_id=stage_id, client=client, body=body)).parsed + )).parsed 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..e0f1474a2 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 @@ -1,43 +1,57 @@ 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.run_scorer_settings_response import RunScorerSettingsResponse -from ...types import Response +from typing import cast + + +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}/scorer-settings", + "path": "/projects/{project_id}/runs/{run_id}/scorer-settings".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, RunScorerSettingsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: + if response.status_code == : response_200 = RunScorerSettingsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, run_id: str, *, client: ApiClient + project_id: str, + run_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Get Settings + """ Get Settings Args: project_id (str): @@ -71,19 +87,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ - 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 + project_id: str, + run_id: str, + *, + client: ApiClient, + ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Get Settings + """ Get Settings Args: project_id (str): @@ -95,15 +121,24 @@ def sync( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ + """ + - 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 + project_id: str, + run_id: str, + *, + client: ApiClient, + ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Get Settings + """ Get Settings Args: project_id (str): @@ -115,19 +150,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ - 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 + project_id: str, + run_id: str, + *, + client: ApiClient, + ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Get Settings + """ Get Settings Args: project_id (str): @@ -139,6 +184,12 @@ async def asyncio( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ + """ + + + 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/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..8da1ba9ca 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 @@ -1,30 +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.run_scorer_settings_patch_request import RunScorerSettingsPatchRequest from ...models.run_scorer_settings_response import RunScorerSettingsResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + run_id: str, + *, + body: RunScorerSettingsPatchRequest, -def _get_kwargs(project_id: str, run_id: str, *, body: RunScorerSettingsPatchRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/projects/{project_id}/runs/{run_id}/scorer-settings", + "path": "/projects/{project_id}/runs/{run_id}/scorer-settings".format(project_id=project_id,run_id=run_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, run_id: str, *, body: RunScorerSettingsPatchReq return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: + if response.status_code == : response_200 = RunScorerSettingsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -77,19 +96,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, +body=body, - 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, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -102,15 +133,26 @@ def sync( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ + """ + - return sync_detailed(project_id=project_id, run_id=run_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +run_id=run_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -123,19 +165,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, +body=body, - 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, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -148,6 +202,13 @@ async def asyncio( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +run_id=run_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client, body=body)).parsed + )).parsed 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..38f3d24d9 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 @@ -1,30 +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.run_scorer_settings_patch_request import RunScorerSettingsPatchRequest from ...models.run_scorer_settings_response import RunScorerSettingsResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + run_id: str, + *, + body: RunScorerSettingsPatchRequest, -def _get_kwargs(project_id: str, run_id: str, *, body: RunScorerSettingsPatchRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/runs/{run_id}/scorer-settings", + "path": "/projects/{project_id}/runs/{run_id}/scorer-settings".format(project_id=project_id,run_id=run_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, run_id: str, *, body: RunScorerSettingsPatchReq return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: + if response.status_code == : response_200 = RunScorerSettingsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -77,19 +96,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, +body=body, - 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, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -102,15 +133,26 @@ def sync( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ + """ + - return sync_detailed(project_id=project_id, run_id=run_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +run_id=run_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -123,19 +165,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +run_id=run_id, +body=body, - 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, body: RunScorerSettingsPatchRequest + project_id: str, + run_id: str, + *, + client: ApiClient, + body: RunScorerSettingsPatchRequest, + ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """Upsert Scorers Config + """ Upsert Scorers Config Args: project_id (str): @@ -148,6 +202,13 @@ async def asyncio( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +run_id=run_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client, body=body)).parsed + )).parsed 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..9b50e5aac 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 @@ -1,30 +1,43 @@ 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.session_create_request import SessionCreateRequest from ...models.session_create_response import SessionCreateResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: SessionCreateRequest, -def _get_kwargs(project_id: str, *, body: SessionCreateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/sessions", + "path": "/projects/{project_id}/sessions".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: SessionCreateRequest) -> dict[str, Any return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: + if response.status_code == : response_200 = SessionCreateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: SessionCreateRequest + project_id: str, + *, + client: ApiClient, + body: SessionCreateRequest, + ) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: - """Create Session + """ Create Session Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, SessionCreateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: SessionCreateRequest + project_id: str, + *, + client: ApiClient, + body: SessionCreateRequest, + ) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: - """Create Session + """ Create Session Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, SessionCreateResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: SessionCreateRequest + project_id: str, + *, + client: ApiClient, + body: SessionCreateRequest, + ) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: - """Create Session + """ Create Session Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, SessionCreateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: SessionCreateRequest + project_id: str, + *, + client: ApiClient, + body: SessionCreateRequest, + ) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: - """Create Session + """ Create Session Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, SessionCreateResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..3019db0b2 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 @@ -1,45 +1,63 @@ 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.log_records_delete_request import LogRecordsDeleteRequest from ...models.log_records_delete_response import LogRecordsDeleteResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsDeleteRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/sessions/delete", + "path": "/projects/{project_id}/sessions/delete".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + 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, LogRecordsDeleteResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: + if response.status_code == : response_200 = LogRecordsDeleteResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -47,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -59,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Sessions + """ Delete Sessions Delete all session records that match the provided filters. @@ -77,19 +97,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Sessions + """ Delete Sessions Delete all session records that match the provided filters. @@ -105,15 +135,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Sessions + """ Delete Sessions Delete all session records that match the provided filters. @@ -129,19 +168,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Sessions + """ Delete Sessions Delete all session records that match the provided filters. @@ -157,6 +206,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..9c2f9dbd4 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 @@ -1,45 +1,63 @@ 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.log_records_delete_request import LogRecordsDeleteRequest from ...models.log_records_delete_response import LogRecordsDeleteResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsDeleteRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/spans/delete", + "path": "/projects/{project_id}/spans/delete".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + 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, LogRecordsDeleteResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: + if response.status_code == : response_200 = LogRecordsDeleteResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -47,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -59,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Spans + """ Delete Spans Delete all span records that match the provided filters. @@ -77,19 +97,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Spans + """ Delete Spans Delete all span records that match the provided filters. @@ -105,15 +135,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Spans + """ Delete Spans Delete all span records that match the provided filters. @@ -129,19 +168,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Spans + """ Delete Spans Delete all span records that match the provided filters. @@ -157,6 +206,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..2635e7c0f 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 @@ -1,45 +1,63 @@ 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.log_records_delete_request import LogRecordsDeleteRequest from ...models.log_records_delete_response import LogRecordsDeleteResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsDeleteRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/traces/delete", + "path": "/projects/{project_id}/traces/delete".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + 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, LogRecordsDeleteResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: + if response.status_code == : response_200 = LogRecordsDeleteResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -47,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -59,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Traces + """ Delete Traces Delete all trace records that match the provided filters. @@ -77,19 +97,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Traces + """ Delete Traces Delete all trace records that match the provided filters. @@ -105,15 +135,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Traces + """ Delete Traces Delete all trace records that match the provided filters. @@ -129,19 +168,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsDeleteRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsDeleteRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """Delete Traces + """ Delete Traces Delete all trace records that match the provided filters. @@ -157,6 +206,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..8d31867c3 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 @@ -1,29 +1,42 @@ 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.log_records_export_request import LogRecordsExportRequest -from ...types import Response +from typing import cast + -def _get_kwargs(project_id: str, *, body: LogRecordsExportRequest) -> dict[str, Any]: +def _get_kwargs( + project_id: str, + *, + body: LogRecordsExportRequest, + +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/export_records", + "path": "/projects/{project_id}/export_records".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,12 +46,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -54,11 +69,33 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ parsed=_parse_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_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsExportRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsExportRequest, + ) -> Response[Union[Any, HTTPValidationError]]: - """Export Records + """ Export Records Args: project_id (str): @@ -71,27 +108,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - - kwargs = _get_kwargs(project_id=project_id, body=body) + """ - response = client.request(**kwargs) - 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) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - response = client.request(**kwargs) + ) - return response + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsExportRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsExportRequest, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Export Records + """ Export Records Args: project_id (str): @@ -104,15 +143,24 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsExportRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsExportRequest, + ) -> Response[Union[Any, HTTPValidationError]]: - """Export Records + """ Export Records Args: project_id (str): @@ -125,19 +173,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsExportRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsExportRequest, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Export Records + """ Export Records Args: project_id (str): @@ -150,6 +208,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..0bcdf8382 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 @@ -1,30 +1,43 @@ 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.aggregated_trace_view_request import AggregatedTraceViewRequest from ...models.aggregated_trace_view_response import AggregatedTraceViewResponse from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: AggregatedTraceViewRequest, -def _get_kwargs(project_id: str, *, body: AggregatedTraceViewRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/traces/aggregated", + "path": "/projects/{project_id}/traces/aggregated".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: AggregatedTraceViewRequest) -> dict[st return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: + if response.status_code == : response_200 = AggregatedTraceViewResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest + project_id: str, + *, + client: ApiClient, + body: AggregatedTraceViewRequest, + ) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """Get Aggregated Trace View + """ Get Aggregated Trace View Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[AggregatedTraceViewResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest + project_id: str, + *, + client: ApiClient, + body: AggregatedTraceViewRequest, + ) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """Get Aggregated Trace View + """ Get Aggregated Trace View Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[AggregatedTraceViewResponse, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest + project_id: str, + *, + client: ApiClient, + body: AggregatedTraceViewRequest, + ) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """Get Aggregated Trace View + """ Get Aggregated Trace View Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[AggregatedTraceViewResponse, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: AggregatedTraceViewRequest + project_id: str, + *, + client: ApiClient, + body: AggregatedTraceViewRequest, + ) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """Get Aggregated Trace View + """ Get Aggregated Trace View Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[AggregatedTraceViewResponse, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..ad0ca8939 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 @@ -1,43 +1,57 @@ 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_session_record_with_children import ExtendedSessionRecordWithChildren from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + session_id: str, -def _get_kwargs(project_id: str, session_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}/sessions/{session_id}", + "path": "/projects/{project_id}/sessions/{session_id}".format(project_id=project_id,session_id=session_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[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: + if response.status_code == : response_200 = ExtendedSessionRecordWithChildren.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, session_id: str, *, client: ApiClient + project_id: str, + session_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """Get Session + """ Get Session Args: project_id (str): @@ -71,19 +87,29 @@ def sync_detailed( Returns: Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, session_id=session_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +session_id=session_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, session_id: str, *, client: ApiClient + project_id: str, + session_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """Get Session + """ Get Session Args: project_id (str): @@ -95,15 +121,24 @@ def sync( Returns: Union[ExtendedSessionRecordWithChildren, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, session_id=session_id, client=client).parsed + return sync_detailed( + project_id=project_id, +session_id=session_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, session_id: str, *, client: ApiClient + project_id: str, + session_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """Get Session + """ Get Session Args: project_id (str): @@ -115,19 +150,29 @@ async def asyncio_detailed( Returns: Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, session_id=session_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +session_id=session_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, session_id: str, *, client: ApiClient + project_id: str, + session_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """Get Session + """ Get Session Args: project_id (str): @@ -139,6 +184,12 @@ async def asyncio( Returns: Union[ExtendedSessionRecordWithChildren, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +session_id=session_id, +client=client, - return (await asyncio_detailed(project_id=project_id, session_id=session_id, client=client)).parsed + )).parsed 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..85fb7b6c2 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,106 +1,109 @@ 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", - ], - ] -]: - if response.status_code == 200: - - def _parse_response_200( - data: object, - ) -> 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 == : + 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()) return response_200 - if response.status_code == 422: + if response.status_code == : 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..3b7afd3dd 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 @@ -1,43 +1,57 @@ 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_trace_record_with_children import ExtendedTraceRecordWithChildren from ...models.http_validation_error import HTTPValidationError -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + trace_id: str, -def _get_kwargs(project_id: str, trace_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}/traces/{trace_id}", + "path": "/projects/{project_id}/traces/{trace_id}".format(project_id=project_id,trace_id=trace_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[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: + if response.status_code == : response_200 = ExtendedTraceRecordWithChildren.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -45,9 +59,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -57,9 +69,13 @@ def _build_response( def sync_detailed( - project_id: str, trace_id: str, *, client: ApiClient + project_id: str, + trace_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """Get Trace + """ Get Trace Args: project_id (str): @@ -71,19 +87,29 @@ def sync_detailed( Returns: Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +trace_id=trace_id, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, trace_id: str, *, client: ApiClient + project_id: str, + trace_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """Get Trace + """ Get Trace Args: project_id (str): @@ -95,15 +121,24 @@ def sync( Returns: Union[ExtendedTraceRecordWithChildren, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, trace_id=trace_id, client=client).parsed + return sync_detailed( + project_id=project_id, +trace_id=trace_id, +client=client, + ).parsed async def asyncio_detailed( - project_id: str, trace_id: str, *, client: ApiClient + project_id: str, + trace_id: str, + *, + client: ApiClient, + ) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """Get Trace + """ Get Trace Args: project_id (str): @@ -115,19 +150,29 @@ async def asyncio_detailed( Returns: Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +trace_id=trace_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, trace_id: str, *, client: ApiClient + project_id: str, + trace_id: str, + *, + client: ApiClient, + ) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """Get Trace + """ Get Trace Args: project_id (str): @@ -139,6 +184,12 @@ async def asyncio( Returns: Union[ExtendedTraceRecordWithChildren, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +trace_id=trace_id, +client=client, - return (await asyncio_detailed(project_id=project_id, trace_id=trace_id, client=client)).parsed + )).parsed 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..4e3235e10 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 @@ -1,30 +1,43 @@ 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.log_spans_ingest_request import LogSpansIngestRequest from ...models.log_spans_ingest_response import LogSpansIngestResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogSpansIngestRequest, -def _get_kwargs(project_id: str, *, body: LogSpansIngestRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/spans", + "path": "/projects/{project_id}/spans".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogSpansIngestRequest) -> dict[str, An return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: + if response.status_code == : response_200 = LogSpansIngestResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogSpansIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogSpansIngestRequest, + ) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: - """Log Spans + """ Log Spans Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogSpansIngestResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogSpansIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogSpansIngestRequest, + ) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: - """Log Spans + """ Log Spans Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogSpansIngestResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogSpansIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogSpansIngestRequest, + ) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: - """Log Spans + """ Log Spans Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogSpansIngestResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogSpansIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogSpansIngestRequest, + ) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: - """Log Spans + """ Log Spans Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogSpansIngestResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..bbc44989e 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 @@ -1,30 +1,43 @@ 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.log_traces_ingest_request import LogTracesIngestRequest from ...models.log_traces_ingest_response import LogTracesIngestResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogTracesIngestRequest, -def _get_kwargs(project_id: str, *, body: LogTracesIngestRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/traces", + "path": "/projects/{project_id}/traces".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogTracesIngestRequest) -> dict[str, A return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: + if response.status_code == : response_200 = LogTracesIngestResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogTracesIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogTracesIngestRequest, + ) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: - """Log Traces + """ Log Traces Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogTracesIngestResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogTracesIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogTracesIngestRequest, + ) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: - """Log Traces + """ Log Traces Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogTracesIngestResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogTracesIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogTracesIngestRequest, + ) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: - """Log Traces + """ Log Traces Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogTracesIngestResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogTracesIngestRequest + project_id: str, + *, + client: ApiClient, + body: LogTracesIngestRequest, + ) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: - """Log Traces + """ Log Traces Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogTracesIngestResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..e7d601d14 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 @@ -1,30 +1,43 @@ 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.log_records_available_columns_response import LogRecordsAvailableColumnsResponse from ...models.metrics_testing_available_columns_request import MetricsTestingAvailableColumnsRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: MetricsTestingAvailableColumnsRequest, -def _get_kwargs(project_id: str, *, body: MetricsTestingAvailableColumnsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/metrics-testing/available_columns", + "path": "/projects/{project_id}/metrics-testing/available_columns".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: MetricsTestingAvailableColumnsRequest) return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == : response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: MetricsTestingAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Metrics Testing Available Columns + """ Metrics Testing Available Columns Args: project_id (str): @@ -77,19 +94,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: MetricsTestingAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Metrics Testing Available Columns + """ Metrics Testing Available Columns Args: project_id (str): @@ -102,15 +129,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: MetricsTestingAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Metrics Testing Available Columns + """ Metrics Testing Available Columns Args: project_id (str): @@ -123,19 +159,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: MetricsTestingAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Metrics Testing Available Columns + """ Metrics Testing Available Columns Args: project_id (str): @@ -148,6 +194,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..255c0d49b 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 @@ -1,30 +1,43 @@ 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.log_records_metrics_query_request import LogRecordsMetricsQueryRequest from ...models.log_records_metrics_response import LogRecordsMetricsResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsMetricsQueryRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsMetricsQueryRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/metrics/search", + "path": "/projects/{project_id}/metrics/search".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsMetricsQueryRequest) -> dict return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: + if response.status_code == : response_200 = LogRecordsMetricsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsMetricsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """Query Metrics + """ Query Metrics Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsMetricsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsMetricsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """Query Metrics + """ Query Metrics Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsMetricsResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsMetricsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """Query Metrics + """ Query Metrics Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsMetricsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsMetricsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsMetricsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """Query Metrics + """ Query Metrics Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsMetricsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..841852aaa 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 @@ -1,30 +1,43 @@ 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.log_records_query_request import LogRecordsQueryRequest from ...models.log_records_query_response import LogRecordsQueryResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsQueryRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/sessions/search", + "path": "/projects/{project_id}/sessions/search".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, A return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: + if response.status_code == : response_200 = LogRecordsQueryResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Sessions + """ Query Sessions Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Sessions + """ Query Sessions Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Sessions + """ Query Sessions Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Sessions + """ Query Sessions Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..5c1b382f5 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 @@ -1,30 +1,43 @@ 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.log_records_query_request import LogRecordsQueryRequest from ...models.log_records_query_response import LogRecordsQueryResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsQueryRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/spans/search", + "path": "/projects/{project_id}/spans/search".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, A return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: + if response.status_code == : response_200 = LogRecordsQueryResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Spans + """ Query Spans Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Spans + """ Query Spans Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Spans + """ Query Spans Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Spans + """ Query Spans Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..c7cf74ce8 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 @@ -1,30 +1,43 @@ 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.log_records_query_request import LogRecordsQueryRequest from ...models.log_records_query_response import LogRecordsQueryResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsQueryRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/traces/search", + "path": "/projects/{project_id}/traces/search".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, A return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: + if response.status_code == : response_200 = LogRecordsQueryResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Traces + """ Query Traces Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Traces + """ Query Traces Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Traces + """ Query Traces Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsQueryRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsQueryRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """Query Traces + """ Query Traces Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..d167180c2 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 @@ -1,29 +1,42 @@ 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.recompute_log_records_metrics_request import RecomputeLogRecordsMetricsRequest -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: RecomputeLogRecordsMetricsRequest, -def _get_kwargs(project_id: str, *, body: RecomputeLogRecordsMetricsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/recompute-metrics", + "path": "/projects/{project_id}/recompute-metrics".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,12 +46,14 @@ 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: + if response.status_code == : response_200 = response.json() return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -56,9 +71,13 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest + project_id: str, + *, + client: ApiClient, + body: RecomputeLogRecordsMetricsRequest, + ) -> Response[Union[Any, HTTPValidationError]]: - """Recompute Metrics + """ Recompute Metrics Args: project_id (str): @@ -73,19 +92,29 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest + project_id: str, + *, + client: ApiClient, + body: RecomputeLogRecordsMetricsRequest, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Recompute Metrics + """ Recompute Metrics Args: project_id (str): @@ -100,15 +129,24 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest + project_id: str, + *, + client: ApiClient, + body: RecomputeLogRecordsMetricsRequest, + ) -> Response[Union[Any, HTTPValidationError]]: - """Recompute Metrics + """ Recompute Metrics Args: project_id (str): @@ -123,19 +161,29 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest + project_id: str, + *, + client: ApiClient, + body: RecomputeLogRecordsMetricsRequest, + ) -> Optional[Union[Any, HTTPValidationError]]: - """Recompute Metrics + """ Recompute Metrics Args: project_id (str): @@ -150,6 +198,12 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..a3084e251 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 @@ -1,30 +1,43 @@ 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.log_records_available_columns_request import LogRecordsAvailableColumnsRequest from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsAvailableColumnsRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/sessions/available_columns", + "path": "/projects/{project_id}/sessions/available_columns".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == : response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Sessions Available Columns + """ Sessions Available Columns Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Sessions Available Columns + """ Sessions Available Columns Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Sessions Available Columns + """ Sessions Available Columns Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Sessions Available Columns + """ Sessions Available Columns Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..2ab4be41c 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 @@ -1,30 +1,43 @@ 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.log_records_available_columns_request import LogRecordsAvailableColumnsRequest from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsAvailableColumnsRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/spans/available_columns", + "path": "/projects/{project_id}/spans/available_columns".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == : response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Spans Available Columns + """ Spans Available Columns Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Spans Available Columns + """ Spans Available Columns Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Spans Available Columns + """ Spans Available Columns Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Spans Available Columns + """ Spans Available Columns Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..8eb9a6e9a 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 @@ -1,30 +1,43 @@ 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.log_records_available_columns_request import LogRecordsAvailableColumnsRequest from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + *, + body: LogRecordsAvailableColumnsRequest, -def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": f"/projects/{project_id}/traces/available_columns", + "path": "/projects/{project_id}/traces/available_columns".format(project_id=project_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +46,18 @@ def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == : response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +65,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +75,13 @@ def _build_response( def sync_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Traces Available Columns + """ Traces Available Columns Args: project_id (str): @@ -76,19 +93,29 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Traces Available Columns + """ Traces Available Columns Args: project_id (str): @@ -100,15 +127,24 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + - return sync_detailed(project_id=project_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Traces Available Columns + """ Traces Available Columns Args: project_id (str): @@ -120,19 +156,29 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +body=body, - 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, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest + project_id: str, + *, + client: ApiClient, + body: LogRecordsAvailableColumnsRequest, + ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """Traces Available Columns + """ Traces Available Columns Args: project_id (str): @@ -144,6 +190,12 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed + )).parsed 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..b7f9130d4 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 @@ -1,30 +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.log_span_update_request import LogSpanUpdateRequest from ...models.log_span_update_response import LogSpanUpdateResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + span_id: str, + *, + body: LogSpanUpdateRequest, -def _get_kwargs(project_id: str, span_id: str, *, body: LogSpanUpdateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "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,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, span_id: str, *, body: LogSpanUpdateRequest) -> return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: + if response.status_code == : response_200 = LogSpanUpdateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, span_id: str, *, client: ApiClient, body: LogSpanUpdateRequest + project_id: str, + span_id: str, + *, + client: ApiClient, + body: LogSpanUpdateRequest, + ) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """Update Span + """ Update Span Update a span with the given ID. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogSpanUpdateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, span_id=span_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +span_id=span_id, +body=body, - 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, body: LogSpanUpdateRequest + project_id: str, + span_id: str, + *, + client: ApiClient, + body: LogSpanUpdateRequest, + ) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """Update Span + """ Update Span Update a span with the given ID. @@ -106,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, LogSpanUpdateResponse] - """ + """ + - return sync_detailed(project_id=project_id, span_id=span_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +span_id=span_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, span_id: str, *, client: ApiClient, body: LogSpanUpdateRequest + project_id: str, + span_id: str, + *, + client: ApiClient, + body: LogSpanUpdateRequest, + ) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """Update Span + """ Update Span Update a span with the given ID. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogSpanUpdateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, span_id=span_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +span_id=span_id, +body=body, - 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, body: LogSpanUpdateRequest + project_id: str, + span_id: str, + *, + client: ApiClient, + body: LogSpanUpdateRequest, + ) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """Update Span + """ Update Span Update a span with the given ID. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, LogSpanUpdateResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +span_id=span_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, span_id=span_id, client=client, body=body)).parsed + )).parsed 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..ad60c958e 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 @@ -1,30 +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.log_trace_update_request import LogTraceUpdateRequest from ...models.log_trace_update_response import LogTraceUpdateResponse -from ...types import Response +from typing import cast + + +def _get_kwargs( + project_id: str, + trace_id: str, + *, + body: LogTraceUpdateRequest, -def _get_kwargs(project_id: str, trace_id: str, *, body: LogTraceUpdateRequest) -> dict[str, Any]: +) -> dict[str, Any]: headers: dict[str, Any] = {} + + + + + _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": f"/projects/{project_id}/traces/{trace_id}", + "path": "/projects/{project_id}/traces/{trace_id}".format(project_id=project_id,trace_id=trace_id,), } _kwargs["json"] = body.to_dict() + headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -33,16 +47,18 @@ def _get_kwargs(project_id: str, trace_id: str, *, body: LogTraceUpdateRequest) return _kwargs -def _parse_response( - *, client: ApiClient, response: httpx.Response -) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: - if response.status_code == 200: +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: + if response.status_code == : response_200 = LogTraceUpdateResponse.from_dict(response.json()) + + return response_200 - if response.status_code == 422: + if response.status_code == : response_422 = HTTPValidationError.from_dict(response.json()) + + return response_422 if client.raise_on_unexpected_status: raise errors.UnexpectedStatus(response.status_code, response.content) @@ -50,9 +66,7 @@ def _parse_response( return None -def _build_response( - *, client: ApiClient, response: httpx.Response -) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -62,9 +76,14 @@ def _build_response( def sync_detailed( - project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest + project_id: str, + trace_id: str, + *, + client: ApiClient, + body: LogTraceUpdateRequest, + ) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """Update Trace + """ Update Trace Update a trace with the given ID. @@ -79,19 +98,31 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogTraceUpdateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id, body=body) - response = client.request(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +trace_id=trace_id, +body=body, - return _build_response(client=client, response=response) + ) + + response = client.request( + **kwargs, + ) + return _build_response(client=client, response=response) def sync( - project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest + project_id: str, + trace_id: str, + *, + client: ApiClient, + body: LogTraceUpdateRequest, + ) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """Update Trace + """ Update Trace Update a trace with the given ID. @@ -106,15 +137,26 @@ def sync( Returns: Union[HTTPValidationError, LogTraceUpdateResponse] - """ + """ + - return sync_detailed(project_id=project_id, trace_id=trace_id, client=client, body=body).parsed + return sync_detailed( + project_id=project_id, +trace_id=trace_id, +client=client, +body=body, + ).parsed async def asyncio_detailed( - project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest + project_id: str, + trace_id: str, + *, + client: ApiClient, + body: LogTraceUpdateRequest, + ) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """Update Trace + """ Update Trace Update a trace with the given ID. @@ -129,19 +171,31 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogTraceUpdateResponse]] - """ + """ - kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id, body=body) - response = await client.arequest(**kwargs) + kwargs = _get_kwargs( + project_id=project_id, +trace_id=trace_id, +body=body, - 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, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest + project_id: str, + trace_id: str, + *, + client: ApiClient, + body: LogTraceUpdateRequest, + ) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """Update Trace + """ Update Trace Update a trace with the given ID. @@ -156,6 +210,13 @@ async def asyncio( Returns: Union[HTTPValidationError, LogTraceUpdateResponse] - """ + """ + + + return (await asyncio_detailed( + project_id=project_id, +trace_id=trace_id, +client=client, +body=body, - return (await asyncio_detailed(project_id=project_id, trace_id=trace_id, client=client, body=body)).parsed + )).parsed 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) From b28902d821d7c623817efdef2d96c344de33556e Mon Sep 17 00:00:00 2001 From: Nachiket Paranjape Date: Wed, 24 Sep 2025 17:41:08 -0700 Subject: [PATCH 2/3] patch http code --- openapi-client-config.yaml | 2 + scripts/patch_status_codes.py | 84 +++++++++ .../auth/login_api_key_login_api_key_post.py | 116 +++---------- .../api/auth/login_email_login_post.py | 115 +++---------- ...gen_llm_scorer_scorers_llm_autogen_post.py | 116 ++++--------- ...ion_scorers_scorer_id_version_code_post.py | 132 +++++--------- ...sion_scorers_scorer_id_version_llm_post.py | 127 ++++---------- ...ion_scorers_scorer_id_version_luna_post.py | 127 ++++---------- ...n_scorers_scorer_id_version_preset_post.py | 127 ++++---------- .../resources/api/data/create_scorers_post.py | 122 ++++--------- .../delete_scorer_scorers_scorer_id_delete.py | 119 ++++--------- .../data/get_scorer_scorers_scorer_id_get.py | 119 ++++--------- ...code_scorers_scorer_id_version_code_get.py | 125 ++++---------- ...or_latest_scorers_scorer_id_version_get.py | 129 ++++---------- ...r_scorer_scorers_scorer_id_versions_get.py | 138 +++++---------- ...er_route_scorers_scorer_id_projects_get.py | 138 +++++---------- ...versions_scorer_version_id_projects_get.py | 125 +++++--------- ..._scorers_with_filters_scorers_list_post.py | 110 ++++-------- .../api/data/list_tags_scorers_tags_get.py | 99 +++-------- ..._llm_validate_scorers_llm_validate_post.py | 116 ++++--------- ...id_versions_version_number_restore_post.py | 126 ++++---------- .../data/update_scorers_scorer_id_patch.py | 127 ++++---------- ...rd_scorers_llm_validate_log_record_post.py | 116 ++++--------- ...te_datasets_datasets_bulk_delete_delete.py | 116 ++++--------- .../datasets/create_dataset_datasets_post.py | 98 +++-------- ...orators_datasets_dataset_id_groups_post.py | 136 +++++---------- ...borators_datasets_dataset_id_users_post.py | 136 +++++---------- ...lete_dataset_datasets_dataset_id_delete.py | 111 +++--------- ...asets_dataset_id_groups_group_id_delete.py | 118 +++---------- ...ct_id_prompt_datasets_dataset_id_delete.py | 118 +++---------- ...atasets_dataset_id_users_user_id_delete.py | 118 +++---------- ...ataset_datasets_dataset_id_download_get.py | 111 +++--------- ...oject_id_prompt_datasets_dataset_id_get.py | 116 +++---------- ...nd_dataset_content_datasets_extend_post.py | 122 ++++--------- ...content_datasets_dataset_id_content_get.py | 138 +++++---------- .../get_dataset_datasets_dataset_id_get.py | 113 +++--------- ...d_status_datasets_extend_dataset_id_get.py | 121 ++++--------- ...t_id_versions_version_index_content_get.py | 127 +++++--------- ...ojects_datasets_dataset_id_projects_get.py | 138 +++++---------- .../datasets/list_datasets_datasets_get.py | 106 ++++-------- ...borators_datasets_dataset_id_groups_get.py | 138 +++++---------- ...projects_project_id_prompt_datasets_get.py | 138 +++++---------- ...aborators_datasets_dataset_id_users_get.py | 138 +++++---------- ...ataset_datasets_dataset_id_preview_post.py | 115 ++++--------- ..._datasets_dataset_id_content_query_post.py | 121 ++++--------- ...datasets_dataset_id_versions_query_post.py | 115 ++++--------- .../query_datasets_datasets_query_post.py | 112 ++++-------- ...ntent_datasets_dataset_id_content_patch.py | 128 ++++---------- ...pdate_dataset_datasets_dataset_id_patch.py | 119 ++++--------- ...dataset_id_versions_version_index_patch.py | 136 ++++----------- ...tasets_dataset_id_groups_group_id_patch.py | 136 ++++----------- ...oject_id_prompt_datasets_dataset_id_put.py | 161 +++++++----------- ...datasets_dataset_id_users_user_id_patch.py | 136 ++++----------- ...rojects_project_id_prompt_datasets_post.py | 118 ++++--------- ...content_datasets_dataset_id_content_put.py | 119 ++++--------- ...nt_projects_project_id_experiments_post.py | 127 ++++---------- ...ect_id_experiments_experiment_id_delete.py | 114 +++---------- ...t_id_experiments_available_columns_post.py | 117 ++++--------- ..._experiments_experiment_id_metrics_post.py | 136 ++++----------- ...roject_id_experiments_experiment_id_get.py | 126 ++++---------- ...cts_project_id_experiments_metrics_post.py | 127 ++++---------- ...nts_projects_project_id_experiments_get.py | 122 +++++-------- ...roject_id_experiments_experiment_id_put.py | 136 ++++----------- ...iments_experiment_id_tags_tag_id_delete.py | 137 +++++---------- ...periments_experiment_id_tags_tag_id_get.py | 129 ++++---------- ...t_id_experiments_experiment_id_tags_get.py | 133 +++++---------- ..._id_experiments_experiment_id_tags_post.py | 128 ++++---------- ...periments_experiment_id_tags_tag_id_put.py | 141 ++++----------- .../api/health/healthcheck_healthcheck_get.py | 101 +++-------- .../api/jobs/create_job_jobs_post.py | 122 ++++--------- .../api/jobs/get_job_jobs_job_id_get.py | 117 +++---------- ...rojects_project_id_runs_run_id_jobs_get.py | 151 +++++----------- ..._project_id_runs_run_id_jobs_latest_get.py | 2 +- ...am_projects_project_id_log_streams_post.py | 127 ++++---------- ...ect_id_log_streams_log_stream_id_delete.py | 114 +++---------- ...roject_id_log_streams_log_stream_id_get.py | 126 ++++---------- ...ams_projects_project_id_log_streams_get.py | 122 +++++-------- ...roject_id_log_streams_log_stream_id_put.py | 136 ++++----------- ...orators_projects_project_id_groups_post.py | 136 +++++---------- .../projects/create_project_projects_post.py | 122 ++++--------- ...borators_projects_project_id_users_post.py | 136 +++++---------- ...jects_project_id_groups_group_id_delete.py | 118 +++---------- ...lete_project_projects_project_id_delete.py | 119 ++++--------- ...rojects_project_id_users_user_id_delete.py | 118 +++---------- .../get_all_projects_projects_all_get.py | 126 +++++--------- ...llaborator_roles_collaborator_roles_get.py | 104 +++-------- .../get_project_projects_project_id_get.py | 113 +++--------- .../get_projects_count_projects_count_post.py | 112 +++--------- ...jects_paginated_projects_paginated_post.py | 112 ++++-------- .../api/projects/get_projects_projects_get.py | 138 +++++---------- ...borators_projects_project_id_groups_get.py | 138 +++++---------- ...aborators_projects_project_id_users_get.py | 138 +++++---------- ...ojects_project_id_groups_group_id_patch.py | 136 ++++----------- .../update_project_projects_project_id_put.py | 127 ++++---------- ...projects_project_id_users_user_id_patch.py | 136 ++++----------- ...le_projects_project_id_upload_file_post.py | 122 ++++--------- ..._templates_templates_bulk_delete_delete.py | 103 +++-------- ...e_global_prompt_template_templates_post.py | 122 ++++--------- ...ion_templates_template_id_versions_post.py | 127 ++++---------- ...ators_templates_template_id_groups_post.py | 136 +++++---------- ..._id_templates_template_id_versions_post.py | 136 ++++----------- ...sion_projects_project_id_templates_post.py | 127 ++++---------- ...rators_templates_template_id_users_post.py | 136 +++++---------- ...l_template_templates_template_id_delete.py | 119 ++++--------- ...ates_template_id_groups_group_id_delete.py | 118 +++---------- ...project_id_templates_template_id_delete.py | 126 ++++---------- ...plates_template_id_users_user_id_delete.py | 118 +++---------- ...ate_template_input_stub_input_stub_post.py | 114 +++---------- ...obal_template_templates_template_id_get.py | 117 ++++--------- ...plates_template_id_versions_version_get.py | 126 ++++---------- ...lates_projects_project_id_templates_get.py | 124 +++++--------- ...ts_project_id_templates_template_id_get.py | 126 ++++---------- ...jects_project_id_templates_versions_get.py | 146 +++++----------- ...plates_template_id_versions_version_get.py | 137 +++++---------- ...rators_templates_template_id_groups_get.py | 138 +++++---------- ...orators_templates_template_id_users_get.py | 138 +++++---------- ...mplates_template_id_versions_query_post.py | 115 ++++--------- .../query_templates_templates_query_post.py | 110 ++++-------- .../render_template_render_template_post.py | 110 ++++-------- ...plates_template_id_versions_version_put.py | 126 ++++---------- ...plates_template_id_versions_version_put.py | 137 +++++---------- ...al_template_templates_template_id_patch.py | 127 ++++---------- ...lates_template_id_groups_group_id_patch.py | 136 ++++----------- ...mplates_template_id_users_user_id_patch.py | 136 ++++----------- ...e_stage_projects_project_id_stages_post.py | 119 ++++--------- ...et_stage_projects_project_id_stages_get.py | 117 ++++--------- .../api/protect/invoke_protect_invoke_post.py | 2 +- ...projects_project_id_stages_stage_id_put.py | 130 ++++---------- ...rojects_project_id_stages_stage_id_post.py | 128 ++++---------- ...ject_id_runs_run_id_scorer_settings_get.py | 126 ++++---------- ...ct_id_runs_run_id_scorer_settings_patch.py | 136 ++++----------- ...ect_id_runs_run_id_scorer_settings_post.py | 136 ++++----------- ...ssion_projects_project_id_sessions_post.py | 127 ++++---------- ...rojects_project_id_sessions_delete_post.py | 127 ++++---------- ...s_projects_project_id_spans_delete_post.py | 127 ++++---------- ..._projects_project_id_traces_delete_post.py | 127 ++++---------- ...projects_project_id_export_records_post.py | 135 ++++----------- ...jects_project_id_traces_aggregated_post.py | 127 ++++---------- ...ects_project_id_sessions_session_id_get.py | 126 ++++---------- ...n_projects_project_id_spans_span_id_get.py | 2 +- ...projects_project_id_traces_trace_id_get.py | 126 ++++---------- ...og_spans_projects_project_id_spans_post.py | 127 ++++---------- ..._traces_projects_project_id_traces_post.py | 127 ++++---------- ..._metrics_testing_available_columns_post.py | 127 ++++---------- ...projects_project_id_metrics_search_post.py | 127 ++++---------- ...rojects_project_id_sessions_search_post.py | 127 ++++---------- ...s_projects_project_id_spans_search_post.py | 127 ++++---------- ..._projects_project_id_traces_search_post.py | 127 ++++---------- ...jects_project_id_recompute_metrics_post.py | 117 ++++--------- ...ject_id_sessions_available_columns_post.py | 127 ++++---------- ...project_id_spans_available_columns_post.py | 127 ++++---------- ...roject_id_traces_available_columns_post.py | 127 ++++---------- ...projects_project_id_spans_span_id_patch.py | 136 ++++----------- ...ojects_project_id_traces_trace_id_patch.py | 136 ++++----------- 154 files changed, 5258 insertions(+), 13520 deletions(-) create mode 100755 scripts/patch_status_codes.py 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/scripts/patch_status_codes.py b/scripts/patch_status_codes.py new file mode 100755 index 000000000..fff2a371e --- /dev/null +++ b/scripts/patch_status_codes.py @@ -0,0 +1,84 @@ +#!/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 + + # Pattern to match: 'if response.status_code == :' followed by 'response_XXX' + # This captures the status code from the response variable name + pattern = r'if response\.status_code == :\s*\n(\s*)response_(\d+)' + + def replace_status_code(match): + indent = match.group(1) + status_code = match.group(2) + return f'if response.status_code == {status_code}:\n{indent}response_{status_code}' + + content = re.sub(pattern, replace_status_code, content, flags=re.MULTILINE) + + 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 622262229..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 @@ -1,42 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.api_key_login_request import ApiKeyLoginRequest from ...models.http_validation_error import HTTPValidationError from ...models.token import Token -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: ApiKeyLoginRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: ApiKeyLoginRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/login/api_key", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/login/api_key"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,22 +30,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Token]]: - if response.status_code == : - response_200 = Token.from_dict(response.json()) + if response.status_code == 200: + return Token.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -73,13 +50,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - body: ApiKeyLoginRequest, - -) -> Response[Union[HTTPValidationError, Token]]: - """ Login Api Key +def sync_detailed(*, client: ApiClient, body: ApiKeyLoginRequest) -> Response[Union[HTTPValidationError, Token]]: + """Login Api Key Args: body (ApiKeyLoginRequest): @@ -90,27 +62,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, Token]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: ApiKeyLoginRequest, -) -> Optional[Union[HTTPValidationError, Token]]: - """ Login Api Key +def sync(*, client: ApiClient, body: ApiKeyLoginRequest) -> Optional[Union[HTTPValidationError, Token]]: + """Login Api Key Args: body (ApiKeyLoginRequest): @@ -121,22 +83,15 @@ def sync( Returns: Union[HTTPValidationError, Token] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: ApiKeyLoginRequest, - + *, client: ApiClient, body: ApiKeyLoginRequest ) -> Response[Union[HTTPValidationError, Token]]: - """ Login Api Key + """Login Api Key Args: body (ApiKeyLoginRequest): @@ -147,27 +102,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Token]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: ApiKeyLoginRequest, -) -> Optional[Union[HTTPValidationError, Token]]: - """ Login Api Key +async def asyncio(*, client: ApiClient, body: ApiKeyLoginRequest) -> Optional[Union[HTTPValidationError, Token]]: + """Login Api Key Args: body (ApiKeyLoginRequest): @@ -178,11 +123,6 @@ async def asyncio( Returns: Union[HTTPValidationError, Token] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 5cac8f64c..881357f3f 100644 --- a/src/galileo/resources/api/auth/login_email_login_post.py +++ b/src/galileo/resources/api/auth/login_email_login_post.py @@ -1,38 +1,23 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.body_login_email_login_post import BodyLoginEmailLoginPost from ...models.http_validation_error import HTTPValidationError from ...models.token import Token -from typing import cast - +from ...types import Response -def _get_kwargs( - *, - body: BodyLoginEmailLoginPost, - -) -> dict[str, Any]: +def _get_kwargs(*, body: BodyLoginEmailLoginPost) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/login", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/login"} _kwargs["data"] = body.to_dict() @@ -45,22 +30,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Token]]: - if response.status_code == : - response_200 = Token.from_dict(response.json()) + if response.status_code == 200: + return Token.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -72,13 +50,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - body: BodyLoginEmailLoginPost, - -) -> Response[Union[HTTPValidationError, Token]]: - """ Login Email +def sync_detailed(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Response[Union[HTTPValidationError, Token]]: + """Login Email Args: body (BodyLoginEmailLoginPost): @@ -89,27 +62,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, Token]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: BodyLoginEmailLoginPost, -) -> Optional[Union[HTTPValidationError, Token]]: - """ Login Email +def sync(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Optional[Union[HTTPValidationError, Token]]: + """Login Email Args: body (BodyLoginEmailLoginPost): @@ -120,22 +83,15 @@ def sync( Returns: Union[HTTPValidationError, Token] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: BodyLoginEmailLoginPost, - + *, client: ApiClient, body: BodyLoginEmailLoginPost ) -> Response[Union[HTTPValidationError, Token]]: - """ Login Email + """Login Email Args: body (BodyLoginEmailLoginPost): @@ -146,27 +102,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, Token]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: BodyLoginEmailLoginPost, -) -> Optional[Union[HTTPValidationError, Token]]: - """ Login Email +async def asyncio(*, client: ApiClient, body: BodyLoginEmailLoginPost) -> Optional[Union[HTTPValidationError, Token]]: + """Login Email Args: body (BodyLoginEmailLoginPost): @@ -177,11 +123,6 @@ async def asyncio( Returns: Union[HTTPValidationError, Token] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 c4d3f4c45..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 @@ -1,32 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.create_llm_scorer_autogen_request import CreateLLMScorerAutogenRequest from ...models.generation_response import GenerationResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: CreateLLMScorerAutogenRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: CreateLLMScorerAutogenRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, @@ -36,7 +25,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GenerationResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = GenerationResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GenerationResponse, HTTPValidationError]]: + if response.status_code == 200: + return GenerationResponse.from_dict(response.json()) + if response.status_code == 422: + 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[GenerationResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GenerationResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: CreateLLMScorerAutogenRequest, - + *, client: ApiClient, body: CreateLLMScorerAutogenRequest ) -> Response[Union[GenerationResponse, HTTPValidationError]]: - """ Autogen Llm Scorer + """Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -94,27 +76,19 @@ def sync_detailed( Returns: Response[Union[GenerationResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: CreateLLMScorerAutogenRequest, +def sync( + *, client: ApiClient, body: CreateLLMScorerAutogenRequest ) -> Optional[Union[GenerationResponse, HTTPValidationError]]: - """ Autogen Llm Scorer + """Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -129,22 +103,15 @@ def sync( Returns: Union[GenerationResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: CreateLLMScorerAutogenRequest, - + *, client: ApiClient, body: CreateLLMScorerAutogenRequest ) -> Response[Union[GenerationResponse, HTTPValidationError]]: - """ Autogen Llm Scorer + """Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -159,27 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[GenerationResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: CreateLLMScorerAutogenRequest, +async def asyncio( + *, client: ApiClient, body: CreateLLMScorerAutogenRequest ) -> Optional[Union[GenerationResponse, HTTPValidationError]]: - """ Autogen Llm Scorer + """Autogen Llm Scorer Autogenerate an LLM scorer configuration. @@ -194,11 +153,6 @@ async def asyncio( Returns: Union[GenerationResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 1932de98e..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 @@ -1,70 +1,55 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_scorer_version_response import BaseScorerVersionResponse -from ...models.body_create_code_scorer_version_scorers_scorer_id_version_code_post import BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost +from ...models.body_create_code_scorer_version_scorers_scorer_id_version_code_post import ( + BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, +) from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, - *, - body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/scorers/{scorer_id}/version/code".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/version/code", } _kwargs["files"] = body.to_multipart() - - 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[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BaseScorerVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BaseScorerVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, - + scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Code Scorer Version + """Create Code Scorer Version Args: scorer_id (str): @@ -92,29 +73,19 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, +def sync( + scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Code Scorer Version + """Create Code Scorer Version Args: scorer_id (str): @@ -126,24 +97,15 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -body=body, + return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, - + scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Code Scorer Version + """Create Code Scorer Version Args: scorer_id (str): @@ -155,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost, +async def asyncio( + scorer_id: str, *, client: ApiClient, body: BodyCreateCodeScorerVersionScorersScorerIdVersionCodePost ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Code Scorer Version + """Create Code Scorer Version Args: scorer_id (str): @@ -189,12 +141,6 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed 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 2af939226..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_scorer_version_response import BaseScorerVersionResponse from ...models.create_llm_scorer_version_request import CreateLLMScorerVersionRequest from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, - *, - body: CreateLLMScorerVersionRequest, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, body: CreateLLMScorerVersionRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/scorers/{scorer_id}/version/llm".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/version/llm", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BaseScorerVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BaseScorerVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - body: CreateLLMScorerVersionRequest, - + scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Llm Scorer Version + """Create Llm Scorer Version Args: scorer_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - body: CreateLLMScorerVersionRequest, +def sync( + scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Llm Scorer Version + """Create Llm Scorer Version Args: scorer_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -body=body, + return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - body: CreateLLMScorerVersionRequest, - + scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Llm Scorer Version + """Create Llm Scorer Version Args: scorer_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - body: CreateLLMScorerVersionRequest, +async def asyncio( + scorer_id: str, *, client: ApiClient, body: CreateLLMScorerVersionRequest ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Llm Scorer Version + """Create Llm Scorer Version Args: scorer_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed 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 18937b003..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_scorer_version_response import BaseScorerVersionResponse from ...models.create_custom_luna_scorer_version_request import CreateCustomLunaScorerVersionRequest from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, - *, - body: CreateCustomLunaScorerVersionRequest, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, body: CreateCustomLunaScorerVersionRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/scorers/{scorer_id}/version/luna".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/version/luna", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BaseScorerVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BaseScorerVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - body: CreateCustomLunaScorerVersionRequest, - + scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Luna Scorer Version + """Create Luna Scorer Version Args: scorer_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - body: CreateCustomLunaScorerVersionRequest, +def sync( + scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Luna Scorer Version + """Create Luna Scorer Version Args: scorer_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -body=body, + return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - body: CreateCustomLunaScorerVersionRequest, - + scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Luna Scorer Version + """Create Luna Scorer Version Args: scorer_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - body: CreateCustomLunaScorerVersionRequest, +async def asyncio( + scorer_id: str, *, client: ApiClient, body: CreateCustomLunaScorerVersionRequest ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Luna Scorer Version + """Create Luna Scorer Version Args: scorer_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed 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 317568496..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_scorer_version_response import BaseScorerVersionResponse from ...models.create_scorer_version_request import CreateScorerVersionRequest from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, - *, - body: CreateScorerVersionRequest, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, body: CreateScorerVersionRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/scorers/{scorer_id}/version/preset".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/version/preset", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BaseScorerVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BaseScorerVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - body: CreateScorerVersionRequest, - + scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Preset Scorer Version + """Create Preset Scorer Version Create a preset scorer version. @@ -95,29 +75,19 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - body: CreateScorerVersionRequest, +def sync( + scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Preset Scorer Version + """Create Preset Scorer Version Create a preset scorer version. @@ -131,24 +101,15 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -body=body, + return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - body: CreateScorerVersionRequest, - + scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Preset Scorer Version + """Create Preset Scorer Version Create a preset scorer version. @@ -162,29 +123,19 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - body: CreateScorerVersionRequest, +async def asyncio( + scorer_id: str, *, client: ApiClient, body: CreateScorerVersionRequest ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Create Preset Scorer Version + """Create Preset Scorer Version Create a preset scorer version. @@ -198,12 +149,6 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed diff --git a/src/galileo/resources/api/data/create_scorers_post.py b/src/galileo/resources/api/data/create_scorers_post.py index 365764895..7bc84c7ce 100644 --- a/src/galileo/resources/api/data/create_scorers_post.py +++ b/src/galileo/resources/api/data/create_scorers_post.py @@ -1,42 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.create_scorer_request import CreateScorerRequest from ...models.http_validation_error import HTTPValidationError from ...models.scorer_response import ScorerResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: CreateScorerRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: CreateScorerRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/scorers", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/scorers"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +29,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - if response.status_code == : - response_200 = ScorerResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + if response.status_code == 200: + return ScorerResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ScorerResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ScorerResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: CreateScorerRequest, - + *, client: ApiClient, body: CreateScorerRequest ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """ Create + """Create Args: body (CreateScorerRequest): @@ -90,27 +68,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: CreateScorerRequest, -) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """ Create +def sync(*, client: ApiClient, body: CreateScorerRequest) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + """Create Args: body (CreateScorerRequest): @@ -121,22 +89,15 @@ def sync( Returns: Union[HTTPValidationError, ScorerResponse] - """ + """ + return sync_detailed(client=client, body=body).parsed - return sync_detailed( - client=client, -body=body, - - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: CreateScorerRequest, - + *, client: ApiClient, body: CreateScorerRequest ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """ Create + """Create Args: body (CreateScorerRequest): @@ -147,27 +108,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: CreateScorerRequest, +async def asyncio( + *, client: ApiClient, body: CreateScorerRequest ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """ Create + """Create Args: body (CreateScorerRequest): @@ -178,11 +131,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ScorerResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 cbeb656df..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.delete_scorer_response import DeleteScorerResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/scorers/{scorer_id}".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_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[DeleteScorerResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = DeleteScorerResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: + if response.status_code == 200: + return DeleteScorerResponse.from_dict(response.json()) + if response.status_code == 422: + 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[DeleteScorerResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -67,13 +53,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - -) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: - """ Delete Scorer +def sync_detailed(scorer_id: str, *, client: ApiClient) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: + """Delete Scorer Args: scorer_id (str): @@ -84,27 +65,17 @@ def sync_detailed( Returns: Response[Union[DeleteScorerResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id) - kwargs = _get_kwargs( - scorer_id=scorer_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, -) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: - """ Delete Scorer +def sync(scorer_id: str, *, client: ApiClient) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: + """Delete Scorer Args: scorer_id (str): @@ -115,22 +86,15 @@ def sync( Returns: Union[DeleteScorerResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, + return sync_detailed(scorer_id=scorer_id, client=client).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - + scorer_id: str, *, client: ApiClient ) -> Response[Union[DeleteScorerResponse, HTTPValidationError]]: - """ Delete Scorer + """Delete Scorer Args: scorer_id (str): @@ -141,27 +105,17 @@ async def asyncio_detailed( Returns: Response[Union[DeleteScorerResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id) - kwargs = _get_kwargs( - scorer_id=scorer_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, -) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: - """ Delete Scorer +async def asyncio(scorer_id: str, *, client: ApiClient) -> Optional[Union[DeleteScorerResponse, HTTPValidationError]]: + """Delete Scorer Args: scorer_id (str): @@ -172,11 +126,6 @@ async def asyncio( Returns: Union[DeleteScorerResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client)).parsed 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 a38e004fb..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.scorer_response import ScorerResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/scorers/{scorer_id}".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_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, ScorerResponse]]: - if response.status_code == : - response_200 = ScorerResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + if response.status_code == 200: + return ScorerResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ScorerResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ScorerResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -67,13 +53,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """ Get Scorer +def sync_detailed(scorer_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ScorerResponse]]: + """Get Scorer Args: scorer_id (str): @@ -84,27 +65,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id) - kwargs = _get_kwargs( - scorer_id=scorer_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """ Get Scorer +def sync(scorer_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + """Get Scorer Args: scorer_id (str): @@ -115,22 +86,15 @@ def sync( Returns: Union[HTTPValidationError, ScorerResponse] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, + return sync_detailed(scorer_id=scorer_id, client=client).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - + scorer_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """ Get Scorer + """Get Scorer Args: scorer_id (str): @@ -141,27 +105,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id) - kwargs = _get_kwargs( - scorer_id=scorer_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """ Get Scorer +async def asyncio(scorer_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + """Get Scorer Args: scorer_id (str): @@ -172,11 +126,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ScorerResponse] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client)).parsed 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 a2a9d7a71..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 @@ -1,54 +1,35 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - - +from ...types import UNSET, Response, Unset -def _get_kwargs( - scorer_id: str, - *, - version: Union[None, Unset, int] = UNSET, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, version: Union[None, Unset, int] = UNSET) -> dict[str, Any]: headers: dict[str, Any] = {} - - - 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} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/scorers/{scorer_id}/version/code".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/version/code", "params": params, } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -56,19 +37,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -81,13 +57,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - version: Union[None, Unset, int] = UNSET, - + scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET ) -> Response[Union[Any, HTTPValidationError]]: - """ Get Scorer Version Code + """Get Scorer Version Code Args: scorer_id (str): @@ -99,29 +71,19 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - kwargs = _get_kwargs( - scorer_id=scorer_id, -version=version, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - version: Union[None, Unset, int] = UNSET, +def sync( + scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET ) -> Optional[Union[Any, HTTPValidationError]]: - """ Get Scorer Version Code + """Get Scorer Version Code Args: scorer_id (str): @@ -133,24 +95,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -version=version, + return sync_detailed(scorer_id=scorer_id, client=client, version=version).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - version: Union[None, Unset, int] = UNSET, - + scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET ) -> Response[Union[Any, HTTPValidationError]]: - """ Get Scorer Version Code + """Get Scorer Version Code Args: scorer_id (str): @@ -162,29 +115,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -version=version, + kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - version: Union[None, Unset, int] = UNSET, +async def asyncio( + scorer_id: str, *, client: ApiClient, version: Union[None, Unset, int] = UNSET ) -> Optional[Union[Any, HTTPValidationError]]: - """ Get Scorer Version Code + """Get Scorer Version Code Args: scorer_id (str): @@ -196,12 +139,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -version=version, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, version=version)).parsed 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 45d153cb6..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 @@ -1,75 +1,57 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_scorer_version_response import BaseScorerVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import Union - - +from ...types import UNSET, Response, Unset -def _get_kwargs( - scorer_id: str, - *, - version: Union[Unset, int] = UNSET, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, version: Union[Unset, int] = UNSET) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} params["version"] = version - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/scorers/{scorer_id}/version".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/version", "params": params, } - 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[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BaseScorerVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BaseScorerVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -79,13 +61,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - version: Union[Unset, int] = UNSET, - + scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Get Scorer Version Or Latest + """Get Scorer Version Or Latest Args: scorer_id (str): @@ -97,29 +75,19 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - kwargs = _get_kwargs( - scorer_id=scorer_id, -version=version, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - version: Union[Unset, int] = UNSET, +def sync( + scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Get Scorer Version Or Latest + """Get Scorer Version Or Latest Args: scorer_id (str): @@ -131,24 +99,15 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -version=version, + return sync_detailed(scorer_id=scorer_id, client=client, version=version).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - version: Union[Unset, int] = UNSET, - + scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Get Scorer Version Or Latest + """Get Scorer Version Or Latest Args: scorer_id (str): @@ -160,29 +119,19 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, version=version) - kwargs = _get_kwargs( - scorer_id=scorer_id, -version=version, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - version: Union[Unset, int] = UNSET, +async def asyncio( + scorer_id: str, *, client: ApiClient, version: Union[Unset, int] = UNSET ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Get Scorer Version Or Latest + """Get Scorer Version Or Latest Args: scorer_id (str): @@ -194,12 +143,6 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -version=version, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, version=version)).parsed 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 a21e5bd30..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_scorer_versions_response import ListScorerVersionsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - scorer_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/scorers/{scorer_id}/versions".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/versions", "params": params, } - 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, ListScorerVersionsResponse]]: - if response.status_code == : - response_200 = ListScorerVersionsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: + if response.status_code == 200: + return ListScorerVersionsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListScorerVersionsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """ List All Versions For Scorer + """List All Versions For Scorer Args: scorer_id (str): @@ -102,31 +80,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListScorerVersionsResponse]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """ List All Versions For Scorer + """List All Versions For Scorer Args: scorer_id (str): @@ -139,26 +105,15 @@ def sync( Returns: Union[HTTPValidationError, ListScorerVersionsResponse] - """ + """ + return sync_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - scorer_id=scorer_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """ List All Versions For Scorer + """List All Versions For Scorer Args: scorer_id (str): @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListScorerVersionsResponse]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListScorerVersionsResponse]]: - """ List All Versions For Scorer + """List All Versions For Scorer Args: scorer_id (str): @@ -208,13 +151,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListScorerVersionsResponse] - """ - + """ - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 38cf5f125..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.get_projects_paginated_response_v2 import GetProjectsPaginatedResponseV2 from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - scorer_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/scorers/{scorer_id}/projects".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}/projects", "params": params, } - 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[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - if response.status_code == : - response_200 = GetProjectsPaginatedResponseV2.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: + if response.status_code == 200: + return GetProjectsPaginatedResponseV2.from_dict(response.json()) + if response.status_code == 422: + 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[GetProjectsPaginatedResponseV2, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Route + """List Projects For Scorer Route List all projects associated with a specific scorer. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Route + """List Projects For Scorer Route List all projects associated with a specific scorer. @@ -143,26 +109,15 @@ def sync( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ + """ + return sync_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - scorer_id=scorer_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Route + """List Projects For Scorer Route List all projects associated with a specific scorer. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(scorer_id=scorer_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + scorer_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Route + """List Projects For Scorer Route List all projects associated with a specific scorer. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(scorer_id=scorer_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 cce7208e9..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 @@ -1,34 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.get_projects_paginated_response_v2 import GetProjectsPaginatedResponseV2 from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - scorer_version_id: str, - *, - scorer_id: str, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + scorer_version_id: str, *, scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -38,44 +26,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/scorers/versions/{scorer_version_id}/projects".format(scorer_version_id=scorer_version_id,), + "path": f"/scorers/versions/{scorer_version_id}/projects", "params": params, } - 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[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - if response.status_code == : - response_200 = GetProjectsPaginatedResponseV2.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: + if response.status_code == 200: + return GetProjectsPaginatedResponseV2.from_dict(response.json()) + if response.status_code == 422: + 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[GetProjectsPaginatedResponseV2, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -91,9 +73,8 @@ def sync_detailed( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Version Route + """List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -109,23 +90,17 @@ def sync_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ - + """ kwargs = _get_kwargs( - scorer_version_id=scorer_version_id, -scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, - + scorer_version_id=scorer_version_id, scorer_id=scorer_id, starting_token=starting_token, limit=limit ) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( scorer_version_id: str, *, @@ -133,9 +108,8 @@ def sync( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Version Route + """List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -151,18 +125,17 @@ def sync( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ - + """ return sync_detailed( scorer_version_id=scorer_version_id, -client=client, -scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, - + client=client, + scorer_id=scorer_id, + starting_token=starting_token, + limit=limit, ).parsed + async def asyncio_detailed( scorer_version_id: str, *, @@ -170,9 +143,8 @@ async def asyncio_detailed( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Version Route + """List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -188,23 +160,17 @@ async def asyncio_detailed( Returns: Response[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]] - """ - + """ kwargs = _get_kwargs( - scorer_version_id=scorer_version_id, -scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, - + scorer_version_id=scorer_version_id, scorer_id=scorer_id, starting_token=starting_token, limit=limit ) - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( scorer_version_id: str, *, @@ -212,9 +178,8 @@ async def asyncio( scorer_id: str, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[GetProjectsPaginatedResponseV2, HTTPValidationError]]: - """ List Projects For Scorer Version Route + """List Projects For Scorer Version Route List all projects associated with a specific scorer version. @@ -230,14 +195,14 @@ async def asyncio( Returns: Union[GetProjectsPaginatedResponseV2, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_version_id=scorer_version_id, -client=client, -scorer_id=scorer_id, -starting_token=starting_token, -limit=limit, - - )).parsed + """ + + return ( + await asyncio_detailed( + scorer_version_id=scorer_version_id, + client=client, + scorer_id=scorer_id, + starting_token=starting_token, + limit=limit, + ) + ).parsed 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 f6cae9772..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 @@ -1,34 +1,23 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_scorers_request import ListScorersRequest from ...models.list_scorers_response import ListScorersResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - *, - body: ListScorersRequest, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + *, body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -36,10 +25,8 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -49,7 +36,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -58,26 +44,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: - if response.status_code == : - response_200 = ListScorersResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: + if response.status_code == 200: + return ListScorersResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListScorersResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListScorersResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -92,9 +75,8 @@ def sync_detailed( body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListScorersResponse]]: - """ List Scorers With Filters + """List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -107,31 +89,23 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListScorersResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: - """ List Scorers With Filters + """List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -144,16 +118,10 @@ def sync( Returns: Union[HTTPValidationError, ListScorersResponse] - """ + """ + return sync_detailed(client=client, body=body, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - client=client, -body=body, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( *, @@ -161,9 +129,8 @@ async def asyncio_detailed( body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListScorersResponse]]: - """ List Scorers With Filters + """List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -176,31 +143,23 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListScorersResponse]] - """ + """ + kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, body: ListScorersRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListScorersResponse]]: - """ List Scorers With Filters + """List Scorers With Filters Args: starting_token (Union[Unset, int]): Default: 0. @@ -213,13 +172,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ListScorersResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, -starting_token=starting_token, -limit=limit, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body, starting_token=starting_token, limit=limit)).parsed 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 9106bfade..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 @@ -1,34 +1,20 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, 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 typing import cast +from ... import errors +from ...types import Response -def _get_kwargs( - -) -> dict[str, Any]: +def _get_kwargs() -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.GET, - "return_raw_response": True, - "path": "/scorers/tags", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": "/scorers/tags"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -37,14 +23,12 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list[str]]: - if response.status_code == : - response_200 = cast(list[str], response.json()) + if response.status_code == 200: + 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]]: @@ -56,12 +40,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - -) -> Response[list[str]]: - """ List Tags +def sync_detailed(*, client: ApiClient) -> Response[list[str]]: + """List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -69,25 +49,17 @@ def sync_detailed( Returns: Response[list[str]] - """ + """ + kwargs = _get_kwargs() - kwargs = _get_kwargs( - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, -) -> Optional[list[str]]: - """ List Tags +def sync(*, client: ApiClient) -> Optional[list[str]]: + """List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -95,20 +67,13 @@ def sync( Returns: list[str] - """ - + """ - return sync_detailed( - client=client, + return sync_detailed(client=client).parsed - ).parsed -async def asyncio_detailed( - *, - client: ApiClient, - -) -> Response[list[str]]: - """ List Tags +async def asyncio_detailed(*, client: ApiClient) -> Response[list[str]]: + """List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -116,25 +81,17 @@ async def asyncio_detailed( Returns: Response[list[str]] - """ - + """ - kwargs = _get_kwargs( - - ) + kwargs = _get_kwargs() - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, -) -> Optional[list[str]]: - """ List Tags +async def asyncio(*, client: ApiClient) -> Optional[list[str]]: + """List Tags Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -142,10 +99,6 @@ async def asyncio( Returns: list[str] - """ - - - return (await asyncio_detailed( - client=client, + """ - )).parsed + return (await asyncio_detailed(client=client)).parsed 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 1fce99d9d..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 @@ -1,32 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.generated_scorer_validation_response import GeneratedScorerValidationResponse from ...models.http_validation_error import HTTPValidationError from ...models.manual_llm_validate_scorers_llm_validate_post_body import ManualLlmValidateScorersLlmValidatePostBody -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: ManualLlmValidateScorersLlmValidatePostBody, -) -> dict[str, Any]: +def _get_kwargs(*, body: ManualLlmValidateScorersLlmValidatePostBody) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, @@ -36,7 +25,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = GeneratedScorerValidationResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: + if response.status_code == 200: + return GeneratedScorerValidationResponse.from_dict(response.json()) + if response.status_code == 422: + 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[GeneratedScorerValidationResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: ManualLlmValidateScorersLlmValidatePostBody, - + *, client: ApiClient, body: ManualLlmValidateScorersLlmValidatePostBody ) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """ Manual Llm Validate + """Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -90,27 +72,19 @@ def sync_detailed( Returns: Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: ManualLlmValidateScorersLlmValidatePostBody, +def sync( + *, client: ApiClient, body: ManualLlmValidateScorersLlmValidatePostBody ) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """ Manual Llm Validate + """Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -121,22 +95,15 @@ def sync( Returns: Union[GeneratedScorerValidationResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: ManualLlmValidateScorersLlmValidatePostBody, - + *, client: ApiClient, body: ManualLlmValidateScorersLlmValidatePostBody ) -> Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """ Manual Llm Validate + """Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -147,27 +114,19 @@ async def asyncio_detailed( Returns: Response[Union[GeneratedScorerValidationResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: ManualLlmValidateScorersLlmValidatePostBody, +async def asyncio( + *, client: ApiClient, body: ManualLlmValidateScorersLlmValidatePostBody ) -> Optional[Union[GeneratedScorerValidationResponse, HTTPValidationError]]: - """ Manual Llm Validate + """Manual Llm Validate Args: body (ManualLlmValidateScorersLlmValidatePostBody): @@ -178,11 +137,6 @@ async def asyncio( Returns: Union[GeneratedScorerValidationResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 7da2a27fb..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_scorer_version_response import BaseScorerVersionResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, - version_number: int, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, version_number: int) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/scorers/{scorer_id}/versions/{version_number}/restore".format(scorer_id=scorer_id,version_number=version_number,), + "path": f"/scorers/{scorer_id}/versions/{version_number}/restore", } - 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[BaseScorerVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BaseScorerVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BaseScorerVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BaseScorerVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - version_number: int, - *, - client: ApiClient, - + scorer_id: str, version_number: int, *, client: ApiClient ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Restore Scorer Version + """Restore Scorer Version List all scorers. @@ -89,29 +70,19 @@ def sync_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -version_number=version_number, + kwargs = _get_kwargs(scorer_id=scorer_id, version_number=version_number) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - version_number: int, - *, - client: ApiClient, +def sync( + scorer_id: str, version_number: int, *, client: ApiClient ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Restore Scorer Version + """Restore Scorer Version List all scorers. @@ -125,24 +96,15 @@ def sync( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -version_number=version_number, -client=client, + return sync_detailed(scorer_id=scorer_id, version_number=version_number, client=client).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - version_number: int, - *, - client: ApiClient, - + scorer_id: str, version_number: int, *, client: ApiClient ) -> Response[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Restore Scorer Version + """Restore Scorer Version List all scorers. @@ -156,29 +118,19 @@ async def asyncio_detailed( Returns: Response[Union[BaseScorerVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, version_number=version_number) - kwargs = _get_kwargs( - scorer_id=scorer_id, -version_number=version_number, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - version_number: int, - *, - client: ApiClient, +async def asyncio( + scorer_id: str, version_number: int, *, client: ApiClient ) -> Optional[Union[BaseScorerVersionResponse, HTTPValidationError]]: - """ Restore Scorer Version + """Restore Scorer Version List all scorers. @@ -192,12 +144,6 @@ async def asyncio( Returns: Union[BaseScorerVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -version_number=version_number, -client=client, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, version_number=version_number, client=client)).parsed 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 ec697a259..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.scorer_response import ScorerResponse from ...models.update_scorer_request import UpdateScorerRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - scorer_id: str, - *, - body: UpdateScorerRequest, -) -> dict[str, Any]: +def _get_kwargs(scorer_id: str, *, body: UpdateScorerRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/scorers/{scorer_id}".format(scorer_id=scorer_id,), + "path": f"/scorers/{scorer_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - if response.status_code == : - response_200 = ScorerResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ScorerResponse]]: + if response.status_code == 200: + return ScorerResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ScorerResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ScorerResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - scorer_id: str, - *, - client: ApiClient, - body: UpdateScorerRequest, - + scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """ Update + """Update Args: scorer_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ - + """ - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - scorer_id: str, - *, - client: ApiClient, - body: UpdateScorerRequest, +def sync( + scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """ Update + """Update Args: scorer_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, ScorerResponse] - """ - + """ - return sync_detailed( - scorer_id=scorer_id, -client=client, -body=body, + return sync_detailed(scorer_id=scorer_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - scorer_id: str, - *, - client: ApiClient, - body: UpdateScorerRequest, - + scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest ) -> Response[Union[HTTPValidationError, ScorerResponse]]: - """ Update + """Update Args: scorer_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ScorerResponse]] - """ + """ + kwargs = _get_kwargs(scorer_id=scorer_id, body=body) - kwargs = _get_kwargs( - scorer_id=scorer_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - scorer_id: str, - *, - client: ApiClient, - body: UpdateScorerRequest, +async def asyncio( + scorer_id: str, *, client: ApiClient, body: UpdateScorerRequest ) -> Optional[Union[HTTPValidationError, ScorerResponse]]: - """ Update + """Update Args: scorer_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ScorerResponse] - """ - - - return (await asyncio_detailed( - scorer_id=scorer_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(scorer_id=scorer_id, client=client, body=body)).parsed 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 03015ca14..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 @@ -1,32 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.validate_llm_scorer_log_record_request import ValidateLLMScorerLogRecordRequest from ...models.validate_llm_scorer_log_record_response import ValidateLLMScorerLogRecordResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: ValidateLLMScorerLogRecordRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: ValidateLLMScorerLogRecordRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, @@ -36,7 +25,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - if response.status_code == : - response_200 = ValidateLLMScorerLogRecordResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: + if response.status_code == 200: + return ValidateLLMScorerLogRecordResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ValidateLLMScorerLogRecordResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: ValidateLLMScorerLogRecordRequest, - + *, client: ApiClient, body: ValidateLLMScorerLogRecordRequest ) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """ Validate Llm Scorer Log Record + """Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -93,27 +75,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: ValidateLLMScorerLogRecordRequest, +def sync( + *, client: ApiClient, body: ValidateLLMScorerLogRecordRequest ) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """ Validate Llm Scorer Log Record + """Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -127,22 +101,15 @@ def sync( Returns: Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: ValidateLLMScorerLogRecordRequest, - + *, client: ApiClient, body: ValidateLLMScorerLogRecordRequest ) -> Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """ Validate Llm Scorer Log Record + """Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -156,27 +123,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: ValidateLLMScorerLogRecordRequest, +async def asyncio( + *, client: ApiClient, body: ValidateLLMScorerLogRecordRequest ) -> Optional[Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse]]: - """ Validate Llm Scorer Log Record + """Validate Llm Scorer Log Record Args: body (ValidateLLMScorerLogRecordRequest): Request to validate a new LLM scorer based on a @@ -190,11 +149,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ValidateLLMScorerLogRecordResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 2db4082d9..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 @@ -1,32 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.bulk_delete_datasets_request import BulkDeleteDatasetsRequest from ...models.bulk_delete_datasets_response import BulkDeleteDatasetsResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: BulkDeleteDatasetsRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: BulkDeleteDatasetsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, @@ -36,7 +25,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BulkDeleteDatasetsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: + if response.status_code == 200: + return BulkDeleteDatasetsResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BulkDeleteDatasetsResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: BulkDeleteDatasetsRequest, - + *, client: ApiClient, body: BulkDeleteDatasetsRequest ) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """ Bulk Delete Datasets + """Bulk Delete Datasets Delete multiple datasets in bulk. @@ -111,27 +93,19 @@ def sync_detailed( Returns: Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: BulkDeleteDatasetsRequest, +def sync( + *, client: ApiClient, body: BulkDeleteDatasetsRequest ) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """ Bulk Delete Datasets + """Bulk Delete Datasets Delete multiple datasets in bulk. @@ -163,22 +137,15 @@ def sync( Returns: Union[BulkDeleteDatasetsResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: BulkDeleteDatasetsRequest, - + *, client: ApiClient, body: BulkDeleteDatasetsRequest ) -> Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """ Bulk Delete Datasets + """Bulk Delete Datasets Delete multiple datasets in bulk. @@ -210,27 +177,19 @@ async def asyncio_detailed( Returns: Response[Union[BulkDeleteDatasetsResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: BulkDeleteDatasetsRequest, +async def asyncio( + *, client: ApiClient, body: BulkDeleteDatasetsRequest ) -> Optional[Union[BulkDeleteDatasetsResponse, HTTPValidationError]]: - """ Bulk Delete Datasets + """Bulk Delete Datasets Delete multiple datasets in bulk. @@ -262,11 +221,6 @@ async def asyncio( Returns: Union[BulkDeleteDatasetsResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 d8ee6b632..ad73a929d 100644 --- a/src/galileo/resources/api/datasets/create_dataset_datasets_post.py +++ b/src/galileo/resources/api/datasets/create_dataset_datasets_post.py @@ -1,22 +1,18 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.body_create_dataset_datasets_post import BodyCreateDatasetDatasetsPost from ...models.dataset_db import DatasetDB from ...models.dataset_format import DatasetFormat from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -24,12 +20,8 @@ def _get_kwargs( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -41,10 +33,8 @@ def _get_kwargs( params["hidden"] = hidden - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -54,8 +44,6 @@ def _get_kwargs( _kwargs["files"] = body.to_multipart() - - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -63,22 +51,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetDB.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 200: + return DatasetDB.from_dict(response.json()) + if response.status_code == 422: + 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]]: @@ -96,9 +77,8 @@ def sync_detailed( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """ Create Dataset + """Create Dataset Creates a standalone dataset. @@ -113,31 +93,23 @@ def sync_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, -format_=format_, -hidden=hidden, + kwargs = _get_kwargs(body=body, format_=format_, hidden=hidden) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """ Create Dataset + """Create Dataset Creates a standalone dataset. @@ -152,16 +124,10 @@ def sync( Returns: Union[DatasetDB, HTTPValidationError] - """ - + """ - return sync_detailed( - client=client, -body=body, -format_=format_, -hidden=hidden, + return sync_detailed(client=client, body=body, format_=format_, hidden=hidden).parsed - ).parsed async def asyncio_detailed( *, @@ -169,9 +135,8 @@ async def asyncio_detailed( body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """ Create Dataset + """Create Dataset Creates a standalone dataset. @@ -186,31 +151,23 @@ async def asyncio_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body, format_=format_, hidden=hidden) - kwargs = _get_kwargs( - body=body, -format_=format_, -hidden=hidden, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, body: BodyCreateDatasetDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """ Create Dataset + """Create Dataset Creates a standalone dataset. @@ -225,13 +182,6 @@ async def asyncio( Returns: Union[DatasetDB, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, -format_=format_, -hidden=hidden, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body, format_=format_, hidden=hidden)).parsed 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 3ddbc9c44..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.group_collaborator import GroupCollaborator from ...models.group_collaborator_create import GroupCollaboratorCreate from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - *, - body: list['GroupCollaboratorCreate'], -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, *, body: list["GroupCollaboratorCreate"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/datasets/{dataset_id}/groups".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/groups", } _kwargs["json"] = [] @@ -40,9 +28,6 @@ def _get_kwargs( body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) - - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -51,31 +36,29 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = GroupCollaborator.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['GroupCollaborator']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -85,13 +68,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Dataset Collaborators + dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Dataset Collaborators Share a dataset with groups. @@ -105,29 +84,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Dataset Collaborators +def sync( + dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Dataset Collaborators Share a dataset with groups. @@ -141,24 +110,15 @@ def sync( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ - + """ - return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, + return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Dataset Collaborators + dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Dataset Collaborators Share a dataset with groups. @@ -172,29 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Dataset Collaborators +async def asyncio( + dataset_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Dataset Collaborators Share a dataset with groups. @@ -208,12 +158,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed 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 75946db45..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator from ...models.user_collaborator_create import UserCollaboratorCreate -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - *, - body: list['UserCollaboratorCreate'], -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, *, body: list["UserCollaboratorCreate"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/datasets/{dataset_id}/users".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/users", } _kwargs["json"] = [] @@ -40,9 +28,6 @@ def _get_kwargs( body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) - - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -51,31 +36,29 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = UserCollaborator.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['UserCollaborator']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -85,13 +68,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Dataset Collaborators + dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Dataset Collaborators Args: dataset_id (str): @@ -103,29 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Dataset Collaborators +def sync( + dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Dataset Collaborators Args: dataset_id (str): @@ -137,24 +106,15 @@ def sync( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ - + """ - return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, + return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Dataset Collaborators + dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Dataset Collaborators Args: dataset_id (str): @@ -166,29 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Dataset Collaborators +async def asyncio( + dataset_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Dataset Collaborators Args: dataset_id (str): @@ -200,12 +150,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed 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 367e560c9..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 @@ -1,37 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - dataset_id: str, - -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/datasets/{dataset_id}".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -39,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -63,13 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Dataset +def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete Dataset Args: dataset_id (str): @@ -80,27 +59,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, + kwargs = _get_kwargs(dataset_id=dataset_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Dataset +def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Dataset Args: dataset_id (str): @@ -111,22 +80,13 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, - ).parsed - -async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Dataset +async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete Dataset Args: dataset_id (str): @@ -137,27 +97,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Dataset +async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Dataset Args: dataset_id (str): @@ -168,11 +118,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed 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 1423dc8a8..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - dataset_id: str, - group_id: str, - -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, group_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/datasets/{dataset_id}/groups/{group_id}".format(dataset_id=dataset_id,group_id=group_id,), + "path": f"/datasets/{dataset_id}/groups/{group_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - dataset_id: str, - group_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Group Dataset Collaborator +def sync_detailed(dataset_id: str, group_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -85,29 +62,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, -group_id=group_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - group_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Group Dataset Collaborator +def sync(dataset_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -121,24 +86,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, group_id=group_id, client=client).parsed - return sync_detailed( - dataset_id=dataset_id, -group_id=group_id, -client=client, - - ).parsed async def asyncio_detailed( - dataset_id: str, - group_id: str, - *, - client: ApiClient, - + dataset_id: str, group_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Group Dataset Collaborator + """Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -152,29 +108,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, -group_id=group_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - group_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Group Dataset Collaborator +async def asyncio(dataset_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Group Dataset Collaborator Remove a group's access to a dataset. @@ -188,12 +132,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -group_id=group_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, group_id=group_id, client=client)).parsed 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 6803e6d3e..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - project_id: str, - dataset_id: str, - -) -> dict[str, Any]: +def _get_kwargs(project_id: str, dataset_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}/prompt_datasets/{dataset_id}".format(project_id=project_id,dataset_id=dataset_id,), + "path": f"/projects/{project_id}/prompt_datasets/{dataset_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - project_id: str, - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Prompt Dataset +def sync_detailed(project_id: str, dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete Prompt Dataset Args: project_id (str): @@ -83,29 +60,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - kwargs = _get_kwargs( - project_id=project_id, -dataset_id=dataset_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Prompt Dataset +def sync(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Prompt Dataset Args: project_id (str): @@ -117,24 +82,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, dataset_id=dataset_id, client=client).parsed - return sync_detailed( - project_id=project_id, -dataset_id=dataset_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - dataset_id: str, - *, - client: ApiClient, - + project_id: str, dataset_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Prompt Dataset + """Delete Prompt Dataset Args: project_id (str): @@ -146,29 +102,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - kwargs = _get_kwargs( - project_id=project_id, -dataset_id=dataset_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Prompt Dataset +async def asyncio(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Prompt Dataset Args: project_id (str): @@ -180,12 +124,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -dataset_id=dataset_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, dataset_id=dataset_id, client=client)).parsed 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 fd157083c..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - dataset_id: str, - user_id: str, - -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, user_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/datasets/{dataset_id}/users/{user_id}".format(dataset_id=dataset_id,user_id=user_id,), + "path": f"/datasets/{dataset_id}/users/{user_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - dataset_id: str, - user_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete User Dataset Collaborator +def sync_detailed(dataset_id: str, user_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -85,29 +62,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, -user_id=user_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - user_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete User Dataset Collaborator +def sync(dataset_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -121,24 +86,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, user_id=user_id, client=client).parsed - return sync_detailed( - dataset_id=dataset_id, -user_id=user_id, -client=client, - - ).parsed async def asyncio_detailed( - dataset_id: str, - user_id: str, - *, - client: ApiClient, - + dataset_id: str, user_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete User Dataset Collaborator + """Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -152,29 +108,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, -user_id=user_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - user_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete User Dataset Collaborator +async def asyncio(dataset_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete User Dataset Collaborator Remove a user's access to a dataset. @@ -188,12 +132,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -user_id=user_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, user_id=user_id, client=client)).parsed 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 5a0ae38be..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 @@ -1,37 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - dataset_id: str, - -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}/download".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/download", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -39,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -63,13 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Download Dataset +def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Download Dataset Args: dataset_id (str): @@ -80,27 +59,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, + kwargs = _get_kwargs(dataset_id=dataset_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Download Dataset +def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Download Dataset Args: dataset_id (str): @@ -111,22 +80,13 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, - ).parsed - -async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Download Dataset +async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Download Dataset Args: dataset_id (str): @@ -137,27 +97,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Download Dataset +async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Download Dataset Args: dataset_id (str): @@ -168,11 +118,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed 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 46d63207e..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 @@ -6,33 +6,21 @@ 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - project_id: str, - dataset_id: str, - -) -> dict[str, Any]: +def _get_kwargs(project_id: str, dataset_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/prompt_datasets/{dataset_id}".format(project_id=project_id,dataset_id=dataset_id,), + "path": f"/projects/{project_id}/prompt_datasets/{dataset_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = cast(Any, None) - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return cast(Any, None) + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - project_id: str, - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Download Prompt Dataset +def sync_detailed(project_id: str, dataset_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Download Prompt Dataset Args: project_id (str): @@ -83,29 +60,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - kwargs = _get_kwargs( - project_id=project_id, -dataset_id=dataset_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Download Prompt Dataset +def sync(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Download Prompt Dataset Args: project_id (str): @@ -117,24 +82,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, dataset_id=dataset_id, client=client).parsed - return sync_detailed( - project_id=project_id, -dataset_id=dataset_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - dataset_id: str, - *, - client: ApiClient, - + project_id: str, dataset_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Download Prompt Dataset + """Download Prompt Dataset Args: project_id (str): @@ -146,29 +102,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, dataset_id=dataset_id) - kwargs = _get_kwargs( - project_id=project_id, -dataset_id=dataset_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Download Prompt Dataset +async def asyncio(project_id: str, dataset_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Download Prompt Dataset Args: project_id (str): @@ -180,12 +124,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -dataset_id=dataset_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, dataset_id=dataset_id, client=client)).parsed 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 8f73c3676..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 @@ -1,42 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.synthetic_dataset_extension_request import SyntheticDatasetExtensionRequest from ...models.synthetic_dataset_extension_response import SyntheticDatasetExtensionResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: SyntheticDatasetExtensionRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: SyntheticDatasetExtensionRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/datasets/extend", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/datasets/extend"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +29,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - if response.status_code == : - response_200 = SyntheticDatasetExtensionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: + if response.status_code == 200: + return SyntheticDatasetExtensionResponse.from_dict(response.json()) + if response.status_code == 422: + 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, SyntheticDatasetExtensionResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: SyntheticDatasetExtensionRequest, - + *, client: ApiClient, body: SyntheticDatasetExtensionRequest ) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """ Extend Dataset Content + """Extend Dataset Content Extends the dataset content @@ -92,27 +70,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: SyntheticDatasetExtensionRequest, +def sync( + *, client: ApiClient, body: SyntheticDatasetExtensionRequest ) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """ Extend Dataset Content + """Extend Dataset Content Extends the dataset content @@ -125,22 +95,15 @@ def sync( Returns: Union[HTTPValidationError, SyntheticDatasetExtensionResponse] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: SyntheticDatasetExtensionRequest, - + *, client: ApiClient, body: SyntheticDatasetExtensionRequest ) -> Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """ Extend Dataset Content + """Extend Dataset Content Extends the dataset content @@ -153,27 +116,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: SyntheticDatasetExtensionRequest, +async def asyncio( + *, client: ApiClient, body: SyntheticDatasetExtensionRequest ) -> Optional[Union[HTTPValidationError, SyntheticDatasetExtensionResponse]]: - """ Extend Dataset Content + """Extend Dataset Content Extends the dataset content @@ -186,11 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, SyntheticDatasetExtensionResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 dec8ca58e..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}/content".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/content", "params": params, } - 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[DatasetContent, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetContent.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == 200: + return DatasetContent.from_dict(response.json()) + if response.status_code == 422: + 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[DatasetContent, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Content + """Get Dataset Content Args: dataset_id (str): @@ -102,31 +80,19 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Content + """Get Dataset Content Args: dataset_id (str): @@ -139,26 +105,15 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Content + """Get Dataset Content Args: dataset_id (str): @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Content + """Get Dataset Content Args: dataset_id (str): @@ -208,13 +151,8 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 11a725868..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 @@ -1,38 +1,27 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_db import DatasetDB from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - dataset_id: str, - -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,22 +29,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetDB.from_dict(response.json()) - - + if response.status_code == 200: + return DatasetDB.from_dict(response.json()) - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 422: + 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]]: @@ -67,13 +49,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[DatasetDB, HTTPValidationError]]: - """ Get Dataset +def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[DatasetDB, HTTPValidationError]]: + """Get Dataset Args: dataset_id (str): @@ -84,27 +61,17 @@ def sync_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """ Get Dataset +def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[DatasetDB, HTTPValidationError]]: + """Get Dataset Args: dataset_id (str): @@ -115,22 +82,13 @@ def sync( Returns: Union[DatasetDB, HTTPValidationError] - """ - - - return sync_detailed( - dataset_id=dataset_id, -client=client, + """ - ).parsed + return sync_detailed(dataset_id=dataset_id, client=client).parsed -async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, -) -> Response[Union[DatasetDB, HTTPValidationError]]: - """ Get Dataset +async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[DatasetDB, HTTPValidationError]]: + """Get Dataset Args: dataset_id (str): @@ -141,27 +99,17 @@ async def asyncio_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """ Get Dataset +async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[DatasetDB, HTTPValidationError]]: + """Get Dataset Args: dataset_id (str): @@ -172,11 +120,6 @@ async def asyncio( Returns: Union[DatasetDB, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed 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 1e896d439..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.job_progress import JobProgress -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/extend/{dataset_id}".format(dataset_id=dataset_id,), + "path": f"/datasets/extend/{dataset_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, JobProgress]]: - if response.status_code == : - response_200 = JobProgress.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, JobProgress]]: + if response.status_code == 200: + return JobProgress.from_dict(response.json()) + if response.status_code == 422: + 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, JobProgress]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, JobProgress]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -67,13 +53,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, JobProgress]]: - """ Get Dataset Synthetic Extend Status +def sync_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobProgress]]: + """Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -84,27 +65,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, JobProgress]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, JobProgress]]: - """ Get Dataset Synthetic Extend Status +def sync(dataset_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobProgress]]: + """Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -115,22 +86,13 @@ def sync( Returns: Union[HTTPValidationError, JobProgress] - """ - + """ - return sync_detailed( - dataset_id=dataset_id, -client=client, + return sync_detailed(dataset_id=dataset_id, client=client).parsed - ).parsed -async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, JobProgress]]: - """ Get Dataset Synthetic Extend Status +async def asyncio_detailed(dataset_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobProgress]]: + """Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -141,27 +103,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, JobProgress]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id) - kwargs = _get_kwargs( - dataset_id=dataset_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, JobProgress]]: - """ Get Dataset Synthetic Extend Status +async def asyncio(dataset_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobProgress]]: + """Get Dataset Synthetic Extend Status Args: dataset_id (str): @@ -172,11 +124,6 @@ async def asyncio( Returns: Union[HTTPValidationError, JobProgress] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client)).parsed 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 f7c395336..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 @@ -1,34 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - version_index: int, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, version_index: int, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -36,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}/versions/{version_index}/content".format(dataset_id=dataset_id,version_index=version_index,), + "path": f"/datasets/{dataset_id}/versions/{version_index}/content", "params": params, } - 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[DatasetContent, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetContent.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == 200: + return DatasetContent.from_dict(response.json()) + if response.status_code == 422: + 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[DatasetContent, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -89,9 +71,8 @@ def sync_detailed( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Version Content + """Get Dataset Version Content Args: dataset_id (str): @@ -105,23 +86,15 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -version_index=version_index, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( dataset_id: str, version_index: int, @@ -129,9 +102,8 @@ def sync( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Version Content + """Get Dataset Version Content Args: dataset_id (str): @@ -145,18 +117,13 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ - + """ return sync_detailed( - dataset_id=dataset_id, -version_index=version_index, -client=client, -starting_token=starting_token, -limit=limit, - + dataset_id=dataset_id, version_index=version_index, client=client, starting_token=starting_token, limit=limit ).parsed + async def asyncio_detailed( dataset_id: str, version_index: int, @@ -164,9 +131,8 @@ async def asyncio_detailed( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Version Content + """Get Dataset Version Content Args: dataset_id (str): @@ -180,23 +146,15 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -version_index=version_index, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( dataset_id: str, version_index: int, @@ -204,9 +162,8 @@ async def asyncio( client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Get Dataset Version Content + """Get Dataset Version Content Args: dataset_id (str): @@ -220,14 +177,14 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -version_index=version_index, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + """ + + return ( + await asyncio_detailed( + dataset_id=dataset_id, + version_index=version_index, + client=client, + starting_token=starting_token, + limit=limit, + ) + ).parsed 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 c56ada373..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_dataset_projects_response import ListDatasetProjectsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}/projects".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/projects", "params": params, } - 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, ListDatasetProjectsResponse]]: - if response.status_code == : - response_200 = ListDatasetProjectsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: + if response.status_code == 200: + return ListDatasetProjectsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListDatasetProjectsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """ List Dataset Projects + """List Dataset Projects Args: dataset_id (str): @@ -102,31 +80,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetProjectsResponse]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """ List Dataset Projects + """List Dataset Projects Args: dataset_id (str): @@ -139,26 +105,15 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetProjectsResponse] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """ List Dataset Projects + """List Dataset Projects Args: dataset_id (str): @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetProjectsResponse]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListDatasetProjectsResponse]]: - """ List Dataset Projects + """List Dataset Projects Args: dataset_id (str): @@ -208,13 +151,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetProjectsResponse] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 01998c47d..9df57d02d 100644 --- a/src/galileo/resources/api/datasets/list_datasets_datasets_get.py +++ b/src/galileo/resources/api/datasets/list_datasets_datasets_get.py @@ -1,21 +1,17 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_action import DatasetAction from ...models.http_validation_error import HTTPValidationError from ...models.list_dataset_response import ListDatasetResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -23,12 +19,8 @@ def _get_kwargs( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -39,17 +31,14 @@ def _get_kwargs( actions_item = actions_item_data.value json_actions.append(actions_item) - params["actions"] = json_actions params["starting_token"] = starting_token params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, @@ -57,33 +46,29 @@ def _get_kwargs( "params": params, } - 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, ListDatasetResponse]]: - if response.status_code == : - response_200 = ListDatasetResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: + if response.status_code == 200: + return ListDatasetResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListDatasetResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -98,9 +83,8 @@ def sync_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """ List Datasets + """List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -114,31 +98,23 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ + kwargs = _get_kwargs(actions=actions, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - actions=actions, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """ List Datasets + """List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -152,16 +128,10 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ + """ + return sync_detailed(client=client, actions=actions, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - client=client, -actions=actions, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( *, @@ -169,9 +139,8 @@ async def asyncio_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """ List Datasets + """List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -185,31 +154,23 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ + kwargs = _get_kwargs(actions=actions, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - actions=actions, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """ List Datasets + """List Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -223,13 +184,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ - - - return (await asyncio_detailed( - client=client, -actions=actions, -starting_token=starting_token, -limit=limit, + """ - )).parsed + return (await asyncio_detailed(client=client, actions=actions, starting_token=starting_token, limit=limit)).parsed 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 a7cee099b..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_group_collaborators_response import ListGroupCollaboratorsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}/groups".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/groups", "params": params, } - 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, ListGroupCollaboratorsResponse]]: - if response.status_code == : - response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: + if response.status_code == 200: + return ListGroupCollaboratorsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListGroupCollaboratorsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Dataset Collaborators + """List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Dataset Collaborators + """List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -143,26 +109,15 @@ def sync( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Dataset Collaborators + """List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Dataset Collaborators + """List Group Dataset Collaborators List the groups with which the dataset has been shared. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 f3589352c..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_prompt_dataset_response import ListPromptDatasetResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - project_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/prompt_datasets".format(project_id=project_id,), + "path": f"/projects/{project_id}/prompt_datasets", "params": params, } - 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, ListPromptDatasetResponse]]: - if response.status_code == : - response_200 = ListPromptDatasetResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: + if response.status_code == 200: + return ListPromptDatasetResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListPromptDatasetResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """ List Prompt Datasets + """List Prompt Datasets Args: project_id (str): @@ -102,31 +80,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListPromptDatasetResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - project_id=project_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """ List Prompt Datasets + """List Prompt Datasets Args: project_id (str): @@ -139,26 +105,15 @@ def sync( Returns: Union[HTTPValidationError, ListPromptDatasetResponse] - """ + """ + return sync_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - project_id=project_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """ List Prompt Datasets + """List Prompt Datasets Args: project_id (str): @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListPromptDatasetResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListPromptDatasetResponse]]: - """ List Prompt Datasets + """List Prompt Datasets Args: project_id (str): @@ -208,13 +151,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListPromptDatasetResponse] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 8e1b154f2..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_user_collaborators_response import ListUserCollaboratorsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/datasets/{dataset_id}/users".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/users", "params": params, } - 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, ListUserCollaboratorsResponse]]: - if response.status_code == : - response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: + if response.status_code == 200: + return ListUserCollaboratorsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListUserCollaboratorsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Dataset Collaborators + """List User Dataset Collaborators List the users with which the dataset has been shared. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Dataset Collaborators + """List User Dataset Collaborators List the users with which the dataset has been shared. @@ -143,26 +109,15 @@ def sync( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Dataset Collaborators + """List User Dataset Collaborators List the users with which the dataset has been shared. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(dataset_id=dataset_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + dataset_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Dataset Collaborators + """List User Dataset Collaborators List the users with which the dataset has been shared. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(dataset_id=dataset_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 f5ecdaa47..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 @@ -1,21 +1,17 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError from ...models.preview_dataset_request import PreviewDatasetRequest -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -24,12 +20,8 @@ def _get_kwargs( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -37,20 +29,17 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/datasets/{dataset_id}/preview".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/preview", "params": params, } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -59,26 +48,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetContent, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetContent.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == 200: + return DatasetContent.from_dict(response.json()) + if response.status_code == 422: + 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[DatasetContent, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -94,9 +80,8 @@ def sync_detailed( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Preview Dataset + """Preview Dataset Args: dataset_id (str): @@ -110,23 +95,15 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -starting_token=starting_token, -limit=limit, + """ - ) + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( dataset_id: str, *, @@ -134,9 +111,8 @@ def sync( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Preview Dataset + """Preview Dataset Args: dataset_id (str): @@ -150,18 +126,13 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ - + """ return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - + dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit ).parsed + async def asyncio_detailed( dataset_id: str, *, @@ -169,9 +140,8 @@ async def asyncio_detailed( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Preview Dataset + """Preview Dataset Args: dataset_id (str): @@ -185,23 +155,15 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( dataset_id: str, *, @@ -209,9 +171,8 @@ async def asyncio( body: PreviewDatasetRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Preview Dataset + """Preview Dataset Args: dataset_id (str): @@ -225,14 +186,10 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed( + dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit + ) + ).parsed 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 2e2565f4d..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 @@ -1,35 +1,23 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_content import DatasetContent from ...models.http_validation_error import HTTPValidationError from ...models.query_dataset_params import QueryDatasetParams -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - *, - body: QueryDatasetParams, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + dataset_id: str, *, body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -37,20 +25,17 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/datasets/{dataset_id}/content/query".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/content/query", "params": params, } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -59,26 +44,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetContent, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetContent.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DatasetContent, HTTPValidationError]]: + if response.status_code == 200: + return DatasetContent.from_dict(response.json()) + if response.status_code == 422: + 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[DatasetContent, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DatasetContent, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -94,9 +76,8 @@ def sync_detailed( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Query Dataset Content + """Query Dataset Content Args: dataset_id (str): @@ -110,23 +91,15 @@ def sync_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( dataset_id: str, *, @@ -134,9 +107,8 @@ def sync( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Query Dataset Content + """Query Dataset Content Args: dataset_id (str): @@ -150,18 +122,13 @@ def sync( Returns: Union[DatasetContent, HTTPValidationError] - """ - + """ return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - + dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit ).parsed + async def asyncio_detailed( dataset_id: str, *, @@ -169,9 +136,8 @@ async def asyncio_detailed( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[DatasetContent, HTTPValidationError]]: - """ Query Dataset Content + """Query Dataset Content Args: dataset_id (str): @@ -185,23 +151,15 @@ async def asyncio_detailed( Returns: Response[Union[DatasetContent, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( dataset_id: str, *, @@ -209,9 +167,8 @@ async def asyncio( body: QueryDatasetParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[DatasetContent, HTTPValidationError]]: - """ Query Dataset Content + """Query Dataset Content Args: dataset_id (str): @@ -225,14 +182,10 @@ async def asyncio( Returns: Union[DatasetContent, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed( + dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit + ) + ).parsed 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 0390d8ebe..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 @@ -1,21 +1,17 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_dataset_version_params import ListDatasetVersionParams from ...models.list_dataset_version_response import ListDatasetVersionResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -24,12 +20,8 @@ def _get_kwargs( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -37,20 +29,17 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/datasets/{dataset_id}/versions/query".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/versions/query", "params": params, } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -59,26 +48,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: - if response.status_code == : - response_200 = ListDatasetVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: + if response.status_code == 200: + return ListDatasetVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListDatasetVersionResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -94,9 +80,8 @@ def sync_detailed( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """ Query Dataset Versions + """Query Dataset Versions Args: dataset_id (str): @@ -110,23 +95,15 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetVersionResponse]] - """ - - - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -starting_token=starting_token, -limit=limit, + """ - ) + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( dataset_id: str, *, @@ -134,9 +111,8 @@ def sync( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """ Query Dataset Versions + """Query Dataset Versions Args: dataset_id (str): @@ -150,18 +126,13 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetVersionResponse] - """ - + """ return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - + dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit ).parsed + async def asyncio_detailed( dataset_id: str, *, @@ -169,9 +140,8 @@ async def asyncio_detailed( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """ Query Dataset Versions + """Query Dataset Versions Args: dataset_id (str): @@ -185,23 +155,15 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetVersionResponse]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( dataset_id: str, *, @@ -209,9 +171,8 @@ async def asyncio( body: ListDatasetVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListDatasetVersionResponse]]: - """ Query Dataset Versions + """Query Dataset Versions Args: dataset_id (str): @@ -225,14 +186,10 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetVersionResponse] - """ - + """ - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed( + dataset_id=dataset_id, client=client, body=body, starting_token=starting_token, limit=limit + ) + ).parsed 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 7edc9453e..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 @@ -1,22 +1,18 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_action import DatasetAction from ...models.http_validation_error import HTTPValidationError from ...models.list_dataset_params import ListDatasetParams from ...models.list_dataset_response import ListDatasetResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -25,12 +21,8 @@ def _get_kwargs( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -41,17 +33,14 @@ def _get_kwargs( actions_item = actions_item_data.value json_actions.append(actions_item) - params["actions"] = json_actions params["starting_token"] = starting_token params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -61,7 +50,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -70,26 +58,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - if response.status_code == : - response_200 = ListDatasetResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: + if response.status_code == 200: + return ListDatasetResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListDatasetResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -105,9 +90,8 @@ def sync_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """ Query Datasets + """Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -122,23 +106,15 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, -actions=actions, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, @@ -146,9 +122,8 @@ def sync( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """ Query Datasets + """Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -163,17 +138,10 @@ def sync( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ + """ + return sync_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - client=client, -body=body, -actions=actions, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( *, @@ -182,9 +150,8 @@ async def asyncio_detailed( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListDatasetResponse]]: - """ Query Datasets + """Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -199,23 +166,15 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListDatasetResponse]] - """ + """ + kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - body=body, -actions=actions, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, @@ -223,9 +182,8 @@ async def asyncio( actions: Union[Unset, list[DatasetAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListDatasetResponse]]: - """ Query Datasets + """Query Datasets Args: actions (Union[Unset, list[DatasetAction]]): Actions to include in the 'permissions' @@ -240,14 +198,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListDatasetResponse] - """ - + """ - return (await asyncio_detailed( - client=client, -body=body, -actions=actions, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit) + ).parsed 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 e5b605879..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 @@ -1,49 +1,33 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.update_dataset_content_request import UpdateDatasetContentRequest -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - dataset_id: str, - *, - body: UpdateDatasetContentRequest, - if_match: Union[None, Unset, str] = UNSET, - + dataset_id: str, *, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET ) -> dict[str, Any]: headers: dict[str, Any] = {} if not isinstance(if_match, Unset): headers["If-Match"] = if_match - - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/datasets/{dataset_id}/content".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/content", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -53,19 +37,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -78,14 +57,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetContentRequest, - if_match: Union[None, Unset, str] = UNSET, - + dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET ) -> Response[Union[Any, HTTPValidationError]]: - """ Update Dataset Content + """Update Dataset Content Update the content of a dataset. @@ -120,31 +94,19 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -if_match=if_match, + """ - ) + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, if_match=if_match) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetContentRequest, - if_match: Union[None, Unset, str] = UNSET, +def sync( + dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET ) -> Optional[Union[Any, HTTPValidationError]]: - """ Update Dataset Content + """Update Dataset Content Update the content of a dataset. @@ -179,26 +141,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, body=body, if_match=if_match).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, -if_match=if_match, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetContentRequest, - if_match: Union[None, Unset, str] = UNSET, - + dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET ) -> Response[Union[Any, HTTPValidationError]]: - """ Update Dataset Content + """Update Dataset Content Update the content of a dataset. @@ -233,31 +184,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, -if_match=if_match, + kwargs = _get_kwargs(dataset_id=dataset_id, body=body, if_match=if_match) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetContentRequest, - if_match: Union[None, Unset, str] = UNSET, +async def asyncio( + dataset_id: str, *, client: ApiClient, body: UpdateDatasetContentRequest, if_match: Union[None, Unset, str] = UNSET ) -> Optional[Union[Any, HTTPValidationError]]: - """ Update Dataset Content + """Update Dataset Content Update the content of a dataset. @@ -292,13 +231,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, -if_match=if_match, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body, if_match=if_match)).parsed 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 0556eccb4..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_db import DatasetDB from ...models.http_validation_error import HTTPValidationError from ...models.update_dataset_request import UpdateDatasetRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - *, - body: UpdateDatasetRequest, -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, *, body: UpdateDatasetRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/datasets/{dataset_id}".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,22 +34,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetDB, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetDB.from_dict(response.json()) + if response.status_code == 200: + return DatasetDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -75,13 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetRequest, - + dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """ Update Dataset + """Update Dataset Args: dataset_id (str): @@ -93,29 +69,19 @@ def sync_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, + """ - ) + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetRequest, +def sync( + dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """ Update Dataset + """Update Dataset Args: dataset_id (str): @@ -127,24 +93,15 @@ def sync( Returns: Union[DatasetDB, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetRequest, - + dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest ) -> Response[Union[DatasetDB, HTTPValidationError]]: - """ Update Dataset + """Update Dataset Args: dataset_id (str): @@ -156,29 +113,19 @@ async def asyncio_detailed( Returns: Response[Union[DatasetDB, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - body: UpdateDatasetRequest, +async def asyncio( + dataset_id: str, *, client: ApiClient, body: UpdateDatasetRequest ) -> Optional[Union[DatasetDB, HTTPValidationError]]: - """ Update Dataset + """Update Dataset Args: dataset_id (str): @@ -190,12 +137,6 @@ async def asyncio( Returns: Union[DatasetDB, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed 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 6ab815c16..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.dataset_version_db import DatasetVersionDB from ...models.http_validation_error import HTTPValidationError from ...models.update_dataset_version_request import UpdateDatasetVersionRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - version_index: int, - *, - body: UpdateDatasetVersionRequest, -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, version_index: int, *, body: UpdateDatasetVersionRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/datasets/{dataset_id}/versions/{version_index}".format(dataset_id=dataset_id,version_index=version_index,), + "path": f"/datasets/{dataset_id}/versions/{version_index}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: - if response.status_code == : - response_200 = DatasetVersionDB.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: + if response.status_code == 200: + return DatasetVersionDB.from_dict(response.json()) + if response.status_code == 422: + 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[DatasetVersionDB, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - version_index: int, - *, - client: ApiClient, - body: UpdateDatasetVersionRequest, - + dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest ) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: - """ Update Dataset Version + """Update Dataset Version Args: dataset_id (str): @@ -96,31 +74,19 @@ def sync_detailed( Returns: Response[Union[DatasetVersionDB, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -version_index=version_index, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - version_index: int, - *, - client: ApiClient, - body: UpdateDatasetVersionRequest, +def sync( + dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest ) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: - """ Update Dataset Version + """Update Dataset Version Args: dataset_id (str): @@ -133,26 +99,15 @@ def sync( Returns: Union[DatasetVersionDB, HTTPValidationError] - """ - + """ - return sync_detailed( - dataset_id=dataset_id, -version_index=version_index, -client=client, -body=body, + return sync_detailed(dataset_id=dataset_id, version_index=version_index, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - dataset_id: str, - version_index: int, - *, - client: ApiClient, - body: UpdateDatasetVersionRequest, - + dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest ) -> Response[Union[DatasetVersionDB, HTTPValidationError]]: - """ Update Dataset Version + """Update Dataset Version Args: dataset_id (str): @@ -165,31 +120,19 @@ async def asyncio_detailed( Returns: Response[Union[DatasetVersionDB, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, version_index=version_index, body=body) - kwargs = _get_kwargs( - dataset_id=dataset_id, -version_index=version_index, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - version_index: int, - *, - client: ApiClient, - body: UpdateDatasetVersionRequest, +async def asyncio( + dataset_id: str, version_index: int, *, client: ApiClient, body: UpdateDatasetVersionRequest ) -> Optional[Union[DatasetVersionDB, HTTPValidationError]]: - """ Update Dataset Version + """Update Dataset Version Args: dataset_id (str): @@ -202,13 +145,6 @@ async def asyncio( Returns: Union[DatasetVersionDB, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -version_index=version_index, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, version_index=version_index, client=client, body=body)).parsed 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 be1a366e9..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.collaborator_update import CollaboratorUpdate from ...models.group_collaborator import GroupCollaborator from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - group_id: str, - *, - body: CollaboratorUpdate, -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, group_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/datasets/{dataset_id}/groups/{group_id}".format(dataset_id=dataset_id,group_id=group_id,), + "path": f"/datasets/{dataset_id}/groups/{group_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - if response.status_code == : - response_200 = GroupCollaborator.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: + if response.status_code == 200: + return GroupCollaborator.from_dict(response.json()) + if response.status_code == 422: + 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[GroupCollaborator, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GroupCollaborator, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Dataset Collaborator + """Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -group_id=group_id, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +def sync( + dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Dataset Collaborator + """Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -137,26 +103,15 @@ def sync( Returns: Union[GroupCollaborator, HTTPValidationError] - """ - + """ - return sync_detailed( - dataset_id=dataset_id, -group_id=group_id, -client=client, -body=body, + return sync_detailed(dataset_id=dataset_id, group_id=group_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - dataset_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Dataset Collaborator + """Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, group_id=group_id, body=body) - kwargs = _get_kwargs( - dataset_id=dataset_id, -group_id=group_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +async def asyncio( + dataset_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Dataset Collaborator + """Update Group Dataset Collaborator Update the sharing permissions of a group on a dataset. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[GroupCollaborator, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -group_id=group_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, group_id=group_id, client=client, body=body)).parsed 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 f02dad0ee..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 @@ -1,23 +1,20 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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.body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put import BodyUpdatePromptDatasetProjectsProjectIdPromptDatasetsDatasetIdPut +from ... import errors +from ...models.body_update_prompt_dataset_projects_project_id_prompt_datasets_dataset_id_put import ( + BodyUpdatePromptDatasetProjectsProjectIdPromptDatasetsDatasetIdPut, +) from ...models.dataset_format import DatasetFormat from ...models.http_validation_error import HTTPValidationError from ...models.prompt_dataset_db import PromptDatasetDB -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -29,27 +26,17 @@ def _get_kwargs( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - 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 @@ -60,47 +47,40 @@ def _get_kwargs( params["hidden"] = hidden - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}/prompt_datasets/{dataset_id}".format(project_id=project_id,dataset_id=dataset_id,), + "path": f"/projects/{project_id}/prompt_datasets/{dataset_id}", "params": params, } _kwargs["files"] = body.to_multipart() - - 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, PromptDatasetDB]]: - if response.status_code == : - response_200 = PromptDatasetDB.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: + if response.status_code == 200: + return PromptDatasetDB.from_dict(response.json()) + if response.status_code == 422: + 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, PromptDatasetDB]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -119,9 +99,8 @@ def sync_detailed( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """ Update Prompt Dataset + """Update Prompt Dataset Args: project_id (str): @@ -138,26 +117,23 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ - + """ kwargs = _get_kwargs( project_id=project_id, -dataset_id=dataset_id, -body=body, -file_name=file_name, -num_rows=num_rows, -format_=format_, -hidden=hidden, - + dataset_id=dataset_id, + body=body, + file_name=file_name, + num_rows=num_rows, + format_=format_, + hidden=hidden, ) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( project_id: str, dataset_id: str, @@ -168,9 +144,8 @@ def sync( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """ Update Prompt Dataset + """Update Prompt Dataset Args: project_id (str): @@ -187,21 +162,20 @@ def sync( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ - + """ return sync_detailed( project_id=project_id, -dataset_id=dataset_id, -client=client, -body=body, -file_name=file_name, -num_rows=num_rows, -format_=format_, -hidden=hidden, - + dataset_id=dataset_id, + client=client, + body=body, + file_name=file_name, + num_rows=num_rows, + format_=format_, + hidden=hidden, ).parsed + async def asyncio_detailed( project_id: str, dataset_id: str, @@ -212,9 +186,8 @@ async def asyncio_detailed( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """ Update Prompt Dataset + """Update Prompt Dataset Args: project_id (str): @@ -231,26 +204,23 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ - + """ kwargs = _get_kwargs( project_id=project_id, -dataset_id=dataset_id, -body=body, -file_name=file_name, -num_rows=num_rows, -format_=format_, -hidden=hidden, - + dataset_id=dataset_id, + body=body, + file_name=file_name, + num_rows=num_rows, + format_=format_, + hidden=hidden, ) - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( project_id: str, dataset_id: str, @@ -261,9 +231,8 @@ async def asyncio( num_rows: Union[None, Unset, int] = UNSET, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """ Update Prompt Dataset + """Update Prompt Dataset Args: project_id (str): @@ -280,17 +249,17 @@ async def asyncio( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ - - - return (await asyncio_detailed( - project_id=project_id, -dataset_id=dataset_id, -client=client, -body=body, -file_name=file_name, -num_rows=num_rows, -format_=format_, -hidden=hidden, - - )).parsed + """ + + return ( + await asyncio_detailed( + project_id=project_id, + dataset_id=dataset_id, + client=client, + body=body, + file_name=file_name, + num_rows=num_rows, + format_=format_, + hidden=hidden, + ) + ).parsed 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 2b4e86ba7..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.collaborator_update import CollaboratorUpdate from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator -from typing import cast - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - user_id: str, - *, - body: CollaboratorUpdate, -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, user_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/datasets/{dataset_id}/users/{user_id}".format(dataset_id=dataset_id,user_id=user_id,), + "path": f"/datasets/{dataset_id}/users/{user_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - if response.status_code == : - response_200 = UserCollaborator.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, UserCollaborator]]: + if response.status_code == 200: + return UserCollaborator.from_dict(response.json()) + if response.status_code == 422: + 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, UserCollaborator]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, UserCollaborator]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Dataset Collaborator + """Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -user_id=user_id, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +def sync( + dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Dataset Collaborator + """Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -137,26 +103,15 @@ def sync( Returns: Union[HTTPValidationError, UserCollaborator] - """ - + """ - return sync_detailed( - dataset_id=dataset_id, -user_id=user_id, -client=client, -body=body, + return sync_detailed(dataset_id=dataset_id, user_id=user_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - dataset_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Dataset Collaborator + """Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ + kwargs = _get_kwargs(dataset_id=dataset_id, user_id=user_id, body=body) - kwargs = _get_kwargs( - dataset_id=dataset_id, -user_id=user_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +async def asyncio( + dataset_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Dataset Collaborator + """Update User Dataset Collaborator Update the sharing permissions of a user on a dataset. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[HTTPValidationError, UserCollaborator] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -user_id=user_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, user_id=user_id, client=client, body=body)).parsed 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 cb333b6ae..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 @@ -1,22 +1,20 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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.body_upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post import BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost +from ... import errors +from ...models.body_upload_prompt_evaluation_dataset_projects_project_id_prompt_datasets_post import ( + BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, +) from ...models.dataset_format import DatasetFormat from ...models.http_validation_error import HTTPValidationError from ...models.prompt_dataset_db import PromptDatasetDB -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -25,12 +23,8 @@ def _get_kwargs( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -42,47 +36,40 @@ def _get_kwargs( params["hidden"] = hidden - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/prompt_datasets".format(project_id=project_id,), + "path": f"/projects/{project_id}/prompt_datasets", "params": params, } _kwargs["files"] = body.to_multipart() - - 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, PromptDatasetDB]]: - if response.status_code == : - response_200 = PromptDatasetDB.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: + if response.status_code == 200: + return PromptDatasetDB.from_dict(response.json()) + if response.status_code == 422: + 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, PromptDatasetDB]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -98,9 +85,8 @@ def sync_detailed( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """ Upload Prompt Evaluation Dataset + """Upload Prompt Evaluation Dataset Args: project_id (str): @@ -114,23 +100,15 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, -format_=format_, -hidden=hidden, + kwargs = _get_kwargs(project_id=project_id, body=body, format_=format_, hidden=hidden) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( project_id: str, *, @@ -138,9 +116,8 @@ def sync( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """ Upload Prompt Evaluation Dataset + """Upload Prompt Evaluation Dataset Args: project_id (str): @@ -154,17 +131,10 @@ def sync( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ + """ + return sync_detailed(project_id=project_id, client=client, body=body, format_=format_, hidden=hidden).parsed - return sync_detailed( - project_id=project_id, -client=client, -body=body, -format_=format_, -hidden=hidden, - - ).parsed async def asyncio_detailed( project_id: str, @@ -173,9 +143,8 @@ async def asyncio_detailed( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Response[Union[HTTPValidationError, PromptDatasetDB]]: - """ Upload Prompt Evaluation Dataset + """Upload Prompt Evaluation Dataset Args: project_id (str): @@ -189,23 +158,15 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, PromptDatasetDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body, format_=format_, hidden=hidden) - kwargs = _get_kwargs( - project_id=project_id, -body=body, -format_=format_, -hidden=hidden, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( project_id: str, *, @@ -213,9 +174,8 @@ async def asyncio( body: BodyUploadPromptEvaluationDatasetProjectsProjectIdPromptDatasetsPost, format_: Union[Unset, DatasetFormat] = UNSET, hidden: Union[Unset, bool] = False, - ) -> Optional[Union[HTTPValidationError, PromptDatasetDB]]: - """ Upload Prompt Evaluation Dataset + """Upload Prompt Evaluation Dataset Args: project_id (str): @@ -229,14 +189,8 @@ async def asyncio( Returns: Union[HTTPValidationError, PromptDatasetDB] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, -format_=format_, -hidden=hidden, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, client=client, body=body, format_=format_, hidden=hidden) + ).parsed 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 06868019d..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 @@ -1,39 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.rollback_request import RollbackRequest from ...models.upsert_dataset_content_request import UpsertDatasetContentRequest -from typing import cast -from typing import cast, Union - - +from ...types import Response -def _get_kwargs( - dataset_id: str, - *, - body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], -) -> dict[str, Any]: +def _get_kwargs(dataset_id: str, *, body: Union["RollbackRequest", "UpsertDatasetContentRequest"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/datasets/{dataset_id}/content".format(dataset_id=dataset_id,), + "path": f"/datasets/{dataset_id}/content", } _kwargs["json"]: dict[str, Any] @@ -42,8 +29,6 @@ def _get_kwargs( else: _kwargs["json"] = body.to_dict() - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -53,19 +38,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -78,13 +58,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - dataset_id: str, - *, - client: ApiClient, - body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], - + dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] ) -> Response[Union[Any, HTTPValidationError]]: - """ Upsert Dataset Content + """Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -98,29 +74,19 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, + """ - ) + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - dataset_id: str, - *, - client: ApiClient, - body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], +def sync( + dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] ) -> Optional[Union[Any, HTTPValidationError]]: - """ Upsert Dataset Content + """Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -134,24 +100,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(dataset_id=dataset_id, client=client, body=body).parsed - return sync_detailed( - dataset_id=dataset_id, -client=client, -body=body, - - ).parsed async def asyncio_detailed( - dataset_id: str, - *, - client: ApiClient, - body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], - + dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] ) -> Response[Union[Any, HTTPValidationError]]: - """ Upsert Dataset Content + """Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -165,29 +122,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - dataset_id=dataset_id, -body=body, + kwargs = _get_kwargs(dataset_id=dataset_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - dataset_id: str, - *, - client: ApiClient, - body: Union['RollbackRequest', 'UpsertDatasetContentRequest'], +async def asyncio( + dataset_id: str, *, client: ApiClient, body: Union["RollbackRequest", "UpsertDatasetContentRequest"] ) -> Optional[Union[Any, HTTPValidationError]]: - """ Upsert Dataset Content + """Upsert Dataset Content Rollback the content of a dataset to a previous version. @@ -201,12 +148,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - dataset_id=dataset_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(dataset_id=dataset_id, client=client, body=body)).parsed 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 e7c8e6d24..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiment_create_request import ExperimentCreateRequest from ...models.experiment_response import ExperimentResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: ExperimentCreateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: ExperimentCreateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/experiments".format(project_id=project_id,), + "path": f"/projects/{project_id}/experiments", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = ExperimentResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: + if response.status_code == 200: + return ExperimentResponse.from_dict(response.json()) + if response.status_code == 422: + 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[ExperimentResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExperimentResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: ExperimentCreateRequest, - + project_id: str, *, client: ApiClient, body: ExperimentCreateRequest ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """ Create Experiment + """Create Experiment Create a new experiment for a project. @@ -95,29 +75,19 @@ def sync_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: ExperimentCreateRequest, +def sync( + project_id: str, *, client: ApiClient, body: ExperimentCreateRequest ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """ Create Experiment + """Create Experiment Create a new experiment for a project. @@ -131,24 +101,15 @@ def sync( Returns: Union[ExperimentResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: ExperimentCreateRequest, - + project_id: str, *, client: ApiClient, body: ExperimentCreateRequest ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """ Create Experiment + """Create Experiment Create a new experiment for a project. @@ -162,29 +123,19 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: ExperimentCreateRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: ExperimentCreateRequest ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """ Create Experiment + """Create Experiment Create a new experiment for a project. @@ -198,12 +149,6 @@ async def asyncio( Returns: Union[ExperimentResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 d070da4ef..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 @@ -6,33 +6,21 @@ 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, experiment_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/{experiment_id}".format(project_id=project_id,experiment_id=experiment_id,), + "path": f"/projects/{project_id}/experiments/{experiment_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_204 = cast(Any, None) - return response_204 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 204: + return cast(Any, None) + if response.status_code == 422: + 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]]: @@ -65,13 +48,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Experiment + """Delete Experiment Delete a specific experiment. @@ -85,29 +64,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Experiment +def sync(project_id: str, experiment_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Experiment Delete a specific experiment. @@ -121,24 +88,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, + return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Experiment + """Delete Experiment Delete a specific experiment. @@ -152,29 +110,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, + """ - ) + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, experiment_id: str, *, client: ApiClient ) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Experiment + """Delete Experiment Delete a specific experiment. @@ -188,12 +136,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client)).parsed 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 7cfb916a1..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiments_available_columns_response import ExperimentsAvailableColumnsResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/available_columns".format(project_id=project_id,), + "path": f"/projects/{project_id}/experiments/available_columns", } - 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[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = ExperimentsAvailableColumnsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: + if response.status_code == 200: + return ExperimentsAvailableColumnsResponse.from_dict(response.json()) + if response.status_code == 422: + 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[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,12 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - + project_id: str, *, client: ApiClient ) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """ Experiments Available Columns + """Experiments Available Columns Procures the column information for experiments. @@ -86,27 +69,19 @@ def sync_detailed( Returns: Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, + kwargs = _get_kwargs(project_id=project_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, +def sync( + project_id: str, *, client: ApiClient ) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """ Experiments Available Columns + """Experiments Available Columns Procures the column information for experiments. @@ -119,22 +94,15 @@ def sync( Returns: Union[ExperimentsAvailableColumnsResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, + return sync_detailed(project_id=project_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - + project_id: str, *, client: ApiClient ) -> Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """ Experiments Available Columns + """Experiments Available Columns Procures the column information for experiments. @@ -147,27 +115,19 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id) - kwargs = _get_kwargs( - project_id=project_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, *, client: ApiClient ) -> Optional[Union[ExperimentsAvailableColumnsResponse, HTTPValidationError]]: - """ Experiments Available Columns + """Experiments Available Columns Procures the column information for experiments. @@ -180,11 +140,6 @@ async def asyncio( Returns: Union[ExperimentsAvailableColumnsResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client)).parsed 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 de3b8ed39..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiment_metrics_request import ExperimentMetricsRequest from ...models.experiment_metrics_response import ExperimentMetricsResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, - *, - body: ExperimentMetricsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, experiment_id: str, *, body: ExperimentMetricsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/{experiment_id}/metrics".format(project_id=project_id,experiment_id=experiment_id,), + "path": f"/projects/{project_id}/experiments/{experiment_id}/metrics", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = ExperimentMetricsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: + if response.status_code == 200: + return ExperimentMetricsResponse.from_dict(response.json()) + if response.status_code == 422: + 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[ExperimentMetricsResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, - + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiment Metrics + """Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, +def sync( + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiment Metrics + """Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -137,26 +103,15 @@ def sync( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, - + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiment Metrics + """Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, +async def asyncio( + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiment Metrics + """Get Experiment Metrics Retrieve metrics for a specific experiment. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body)).parsed 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 199d2b652..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiment_response import ExperimentResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, -) -> dict[str, Any]: +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": "/projects/{project_id}/experiments/{experiment_id}".format(project_id=project_id,experiment_id=experiment_id,), + "path": f"/projects/{project_id}/experiments/{experiment_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[ExperimentResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = ExperimentResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: + if response.status_code == 200: + return ExperimentResponse.from_dict(response.json()) + if response.status_code == 422: + 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[ExperimentResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExperimentResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, *, client: ApiClient ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """ Get Experiment + """Get Experiment Retrieve a specific experiment. @@ -89,29 +70,19 @@ def sync_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - *, - client: ApiClient, +def sync( + project_id: str, experiment_id: str, *, client: ApiClient ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """ Get Experiment + """Get Experiment Retrieve a specific experiment. @@ -125,24 +96,15 @@ def sync( Returns: Union[ExperimentResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, + return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, *, client: ApiClient ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """ Get Experiment + """Get Experiment Retrieve a specific experiment. @@ -156,29 +118,19 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, experiment_id: str, *, client: ApiClient ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """ Get Experiment + """Get Experiment Retrieve a specific experiment. @@ -192,12 +144,6 @@ async def asyncio( Returns: Union[ExperimentResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client)).parsed 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 b044f0b8d..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiment_metrics_request import ExperimentMetricsRequest from ...models.experiment_metrics_response import ExperimentMetricsResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: ExperimentMetricsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: ExperimentMetricsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/metrics".format(project_id=project_id,), + "path": f"/projects/{project_id}/experiments/metrics", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = ExperimentMetricsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: + if response.status_code == 200: + return ExperimentMetricsResponse.from_dict(response.json()) + if response.status_code == 422: + 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[ExperimentMetricsResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, - + project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiments Metrics + """Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -95,29 +75,19 @@ def sync_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, +def sync( + project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiments Metrics + """Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -131,24 +101,15 @@ def sync( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, - + project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Response[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiments Metrics + """Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -162,29 +123,19 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentMetricsResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: ExperimentMetricsRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: ExperimentMetricsRequest ) -> Optional[Union[ExperimentMetricsResponse, HTTPValidationError]]: - """ Get Experiments Metrics + """Get Experiments Metrics Retrieve metrics for all experiments in a project. @@ -198,12 +149,6 @@ async def asyncio( Returns: Union[ExperimentMetricsResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 4f4b6ca01..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 @@ -1,69 +1,56 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiment_response import ExperimentResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/experiments".format(project_id=project_id,), + "path": f"/projects/{project_id}/experiments", } - 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, list['ExperimentResponse']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["ExperimentResponse"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = ExperimentResponse.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['ExperimentResponse']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["ExperimentResponse"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,12 +60,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['ExperimentResponse']]]: - """ List Experiments + project_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["ExperimentResponse"]]]: + """List Experiments Retrieve all experiments for a project. @@ -91,27 +75,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['ExperimentResponse']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, + kwargs = _get_kwargs(project_id=project_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['ExperimentResponse']]]: - """ List Experiments +def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, list["ExperimentResponse"]]]: + """List Experiments Retrieve all experiments for a project. @@ -124,22 +98,15 @@ def sync( Returns: Union[HTTPValidationError, list['ExperimentResponse']] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, + return sync_detailed(project_id=project_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['ExperimentResponse']]]: - """ List Experiments + project_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["ExperimentResponse"]]]: + """List Experiments Retrieve all experiments for a project. @@ -152,27 +119,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['ExperimentResponse']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, + kwargs = _get_kwargs(project_id=project_id) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['ExperimentResponse']]]: - """ List Experiments +async def asyncio( + project_id: str, *, client: ApiClient +) -> Optional[Union[HTTPValidationError, list["ExperimentResponse"]]]: + """List Experiments Retrieve all experiments for a project. @@ -185,11 +144,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['ExperimentResponse']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client)).parsed 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 10aaa72b5..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.experiment_response import ExperimentResponse from ...models.experiment_update_request import ExperimentUpdateRequest from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, - *, - body: ExperimentUpdateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, experiment_id: str, *, body: ExperimentUpdateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/{experiment_id}".format(project_id=project_id,experiment_id=experiment_id,), + "path": f"/projects/{project_id}/experiments/{experiment_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = ExperimentResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: + if response.status_code == 200: + return ExperimentResponse.from_dict(response.json()) + if response.status_code == 422: + 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[ExperimentResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExperimentResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentUpdateRequest, - + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """ Update Experiment + """Update Experiment Update a specific experiment. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentUpdateRequest, +def sync( + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """ Update Experiment + """Update Experiment Update a specific experiment. @@ -137,26 +103,15 @@ def sync( Returns: Union[ExperimentResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentUpdateRequest, - + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest ) -> Response[Union[ExperimentResponse, HTTPValidationError]]: - """ Update Experiment + """Update Experiment Update a specific experiment. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[ExperimentResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: ExperimentUpdateRequest, +async def asyncio( + project_id: str, experiment_id: str, *, client: ApiClient, body: ExperimentUpdateRequest ) -> Optional[Union[ExperimentResponse, HTTPValidationError]]: - """ Update Experiment + """Update Experiment Update a specific experiment. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[ExperimentResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body)).parsed 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 f087ae5a9..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 @@ -1,66 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.delete_run_response import DeleteRunResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, - tag_id: str, -) -> dict[str, Any]: +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": "/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}".format(project_id=project_id,experiment_id=experiment_id,tag_id=tag_id,), + "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[DeleteRunResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = DeleteRunResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: + if response.status_code == 200: + return DeleteRunResponse.from_dict(response.json()) + if response.status_code == 422: + 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[DeleteRunResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,14 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: - """ Delete Experiment Tag + """Delete Experiment Tag Args: project_id (str): @@ -90,31 +69,19 @@ def sync_detailed( Returns: Response[Union[DeleteRunResponse, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, + """ - ) + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, +def sync( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: - """ Delete Experiment Tag + """Delete Experiment Tag Args: project_id (str): @@ -127,26 +94,15 @@ def sync( Returns: Union[DeleteRunResponse, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client).parsed - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Response[Union[DeleteRunResponse, HTTPValidationError]]: - """ Delete Experiment Tag + """Delete Experiment Tag Args: project_id (str): @@ -159,31 +115,19 @@ async def asyncio_detailed( Returns: Response[Union[DeleteRunResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Optional[Union[DeleteRunResponse, HTTPValidationError]]: - """ Delete Experiment Tag + """Delete Experiment Tag Args: project_id (str): @@ -196,13 +140,8 @@ async def asyncio( Returns: Union[DeleteRunResponse, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -client=client, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client) + ).parsed 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 44adafa5f..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 @@ -1,40 +1,27 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_tag_db import RunTagDB -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, - tag_id: str, -) -> dict[str, Any]: +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": "/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}".format(project_id=project_id,experiment_id=experiment_id,tag_id=tag_id,), + "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 @@ -42,22 +29,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunTagDB]]: - if response.status_code == : - response_200 = RunTagDB.from_dict(response.json()) - + if response.status_code == 200: + return RunTagDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -70,14 +50,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """ Get Experiment Tag + """Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -92,31 +67,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, +def sync( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """ Get Experiment Tag + """Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -131,26 +94,15 @@ def sync( Returns: Union[HTTPValidationError, RunTagDB] - """ - + """ - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -client=client, + return sync_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """ Get Experiment Tag + """Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -165,31 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """ Get Experiment Tag + """Get Experiment Tag Gets a tag for a given project_id/experiment_id. @@ -204,13 +144,8 @@ async def asyncio( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -client=client, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client) + ).parsed 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 4d8af7c95..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 @@ -1,70 +1,56 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_tag_db import RunTagDB -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, -) -> dict[str, Any]: +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": "/projects/{project_id}/experiments/{experiment_id}/tags".format(project_id=project_id,experiment_id=experiment_id,), + "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 -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['RunTagDB']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["RunTagDB"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = RunTagDB.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 422: + 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, list['RunTagDB']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["RunTagDB"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,13 +60,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['RunTagDB']]]: - """ Get Experiment Tags + project_id: str, experiment_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["RunTagDB"]]]: + """Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -94,29 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['RunTagDB']]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, + """ - ) + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['RunTagDB']]]: - """ Get Experiment Tags +def sync( + project_id: str, experiment_id: str, *, client: ApiClient +) -> Optional[Union[HTTPValidationError, list["RunTagDB"]]]: + """Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -130,24 +102,15 @@ def sync( Returns: Union[HTTPValidationError, list['RunTagDB']] - """ + """ + return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client).parsed - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['RunTagDB']]]: - """ Get Experiment Tags + project_id: str, experiment_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["RunTagDB"]]]: + """Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -161,29 +124,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['RunTagDB']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['RunTagDB']]]: - """ Get Experiment Tags +async def asyncio( + project_id: str, experiment_id: str, *, client: ApiClient +) -> Optional[Union[HTTPValidationError, list["RunTagDB"]]]: + """Get Experiment Tags Gets tags for a given project_id/experiment_id. @@ -197,12 +150,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['RunTagDB']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client)).parsed 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 ea5d1f06a..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_tag_create_request import RunTagCreateRequest from ...models.run_tag_db import RunTagDB -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, - *, - body: RunTagCreateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, experiment_id: str, *, body: RunTagCreateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/{experiment_id}/tags".format(project_id=project_id,experiment_id=experiment_id,), + "path": f"/projects/{project_id}/experiments/{experiment_id}/tags", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -48,22 +34,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunTagDB]]: - if response.status_code == : - response_200 = RunTagDB.from_dict(response.json()) + if response.status_code == 200: + return RunTagDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -76,14 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, - + project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """ Set Tag For Experiment + """Set Tag For Experiment Sets a tag for an experiment. @@ -98,31 +72,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -body=body, + """ - ) + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, +def sync( + project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """ Set Tag For Experiment + """Set Tag For Experiment Sets a tag for an experiment. @@ -137,26 +99,15 @@ def sync( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ + return sync_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body).parsed - return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, -body=body, - - ).parsed async def asyncio_detailed( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, - + project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """ Set Tag For Experiment + """Set Tag For Experiment Sets a tag for an experiment. @@ -171,31 +122,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, +async def asyncio( + project_id: str, experiment_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """ Set Tag For Experiment + """Set Tag For Experiment Sets a tag for an experiment. @@ -210,13 +149,6 @@ async def asyncio( Returns: Union[HTTPValidationError, RunTagDB] - """ - - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, experiment_id=experiment_id, client=client, body=body)).parsed 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 3065b0113..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 @@ -1,45 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_tag_create_request import RunTagCreateRequest from ...models.run_tag_db import RunTagDB -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - experiment_id: str, - tag_id: str, - *, - body: RunTagCreateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, experiment_id: str, tag_id: str, *, body: RunTagCreateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}".format(project_id=project_id,experiment_id=experiment_id,tag_id=tag_id,), + "path": f"/projects/{project_id}/experiments/{experiment_id}/tags/{tag_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -49,22 +34,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunTagDB]]: - if response.status_code == : - response_200 = RunTagDB.from_dict(response.json()) - + if response.status_code == 200: + return RunTagDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -77,15 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, - + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """ Update Tag For Experiment + """Update Tag For Experiment Sets or updates a tag for an experiment. @@ -101,33 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, +def sync( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """ Update Tag For Experiment + """Update Tag For Experiment Sets or updates a tag for an experiment. @@ -143,28 +101,17 @@ def sync( Returns: Union[HTTPValidationError, RunTagDB] - """ - + """ return sync_detailed( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -client=client, -body=body, - + project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client, body=body ).parsed -async def asyncio_detailed( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, +async def asyncio_detailed( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Response[Union[HTTPValidationError, RunTagDB]]: - """ Update Tag For Experiment + """Update Tag For Experiment Sets or updates a tag for an experiment. @@ -180,33 +127,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunTagDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - experiment_id: str, - tag_id: str, - *, - client: ApiClient, - body: RunTagCreateRequest, +async def asyncio( + project_id: str, experiment_id: str, tag_id: str, *, client: ApiClient, body: RunTagCreateRequest ) -> Optional[Union[HTTPValidationError, RunTagDB]]: - """ Update Tag For Experiment + """Update Tag For Experiment Sets or updates a tag for an experiment. @@ -222,14 +155,10 @@ async def asyncio( Returns: Union[HTTPValidationError, RunTagDB] - """ + """ - - return (await asyncio_detailed( - project_id=project_id, -experiment_id=experiment_id, -tag_id=tag_id, -client=client, -body=body, - - )).parsed + return ( + await asyncio_detailed( + project_id=project_id, experiment_id=experiment_id, tag_id=tag_id, client=client, body=body + ) + ).parsed diff --git a/src/galileo/resources/api/health/healthcheck_healthcheck_get.py b/src/galileo/resources/api/health/healthcheck_healthcheck_get.py index ba03cef06..59d64e220 100644 --- a/src/galileo/resources/api/health/healthcheck_healthcheck_get.py +++ b/src/galileo/resources/api/health/healthcheck_healthcheck_get.py @@ -1,35 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional 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 ... import errors from ...models.healthcheck_response import HealthcheckResponse -from typing import cast - +from ...types import Response -def _get_kwargs( - -) -> dict[str, Any]: +def _get_kwargs() -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.GET, - "return_raw_response": True, - "path": "/healthcheck", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": "/healthcheck"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -38,16 +24,12 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[HealthcheckResponse]: - if response.status_code == : - response_200 = HealthcheckResponse.from_dict(response.json()) - - + if response.status_code == 200: + 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]: @@ -59,12 +41,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - -) -> Response[HealthcheckResponse]: - """ Healthcheck +def sync_detailed(*, client: ApiClient) -> Response[HealthcheckResponse]: + """Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -72,25 +50,17 @@ def sync_detailed( Returns: Response[HealthcheckResponse] - """ + """ + kwargs = _get_kwargs() - kwargs = _get_kwargs( - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, -) -> Optional[HealthcheckResponse]: - """ Healthcheck +def sync(*, client: ApiClient) -> Optional[HealthcheckResponse]: + """Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -98,20 +68,13 @@ def sync( Returns: HealthcheckResponse - """ - + """ - return sync_detailed( - client=client, + return sync_detailed(client=client).parsed - ).parsed -async def asyncio_detailed( - *, - client: ApiClient, - -) -> Response[HealthcheckResponse]: - """ Healthcheck +async def asyncio_detailed(*, client: ApiClient) -> Response[HealthcheckResponse]: + """Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -119,25 +82,17 @@ async def asyncio_detailed( Returns: Response[HealthcheckResponse] - """ + """ + kwargs = _get_kwargs() - kwargs = _get_kwargs( - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, -) -> Optional[HealthcheckResponse]: - """ Healthcheck +async def asyncio(*, client: ApiClient) -> Optional[HealthcheckResponse]: + """Healthcheck Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -145,10 +100,6 @@ async def asyncio( Returns: HealthcheckResponse - """ - - - return (await asyncio_detailed( - client=client, + """ - )).parsed + return (await asyncio_detailed(client=client)).parsed 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 6474edc47..df317fbf6 100644 --- a/src/galileo/resources/api/jobs/create_job_jobs_post.py +++ b/src/galileo/resources/api/jobs/create_job_jobs_post.py @@ -1,42 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.create_job_request import CreateJobRequest from ...models.create_job_response import CreateJobResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: CreateJobRequest, -) -> dict[str, Any]: +def _get_kwargs(*, body: CreateJobRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/jobs", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/jobs"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +29,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = CreateJobResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: + if response.status_code == 200: + return CreateJobResponse.from_dict(response.json()) + if response.status_code == 422: + 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[CreateJobResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[CreateJobResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: CreateJobRequest, - + *, client: ApiClient, body: CreateJobRequest ) -> Response[Union[CreateJobResponse, HTTPValidationError]]: - """ Create Job + """Create Job Args: body (CreateJobRequest): @@ -90,27 +68,17 @@ def sync_detailed( Returns: Response[Union[CreateJobResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: CreateJobRequest, -) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: - """ Create Job +def sync(*, client: ApiClient, body: CreateJobRequest) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: + """Create Job Args: body (CreateJobRequest): @@ -121,22 +89,15 @@ def sync( Returns: Union[CreateJobResponse, HTTPValidationError] - """ + """ + return sync_detailed(client=client, body=body).parsed - return sync_detailed( - client=client, -body=body, - - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: CreateJobRequest, - + *, client: ApiClient, body: CreateJobRequest ) -> Response[Union[CreateJobResponse, HTTPValidationError]]: - """ Create Job + """Create Job Args: body (CreateJobRequest): @@ -147,27 +108,19 @@ async def asyncio_detailed( Returns: Response[Union[CreateJobResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: CreateJobRequest, +async def asyncio( + *, client: ApiClient, body: CreateJobRequest ) -> Optional[Union[CreateJobResponse, HTTPValidationError]]: - """ Create Job + """Create Job Args: body (CreateJobRequest): @@ -178,11 +131,6 @@ async def asyncio( Returns: Union[CreateJobResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 80f618c90..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 @@ -1,37 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.job_db import JobDB -from typing import cast - +from ...types import Response -def _get_kwargs( - job_id: str, - -) -> dict[str, Any]: +def _get_kwargs(job_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.GET, - "return_raw_response": True, - "path": "/jobs/{job_id}".format(job_id=job_id,), - } + _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": f"/jobs/{job_id}"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -40,22 +25,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, JobDB]]: - if response.status_code == : - response_200 = JobDB.from_dict(response.json()) - - + if response.status_code == 200: + return JobDB.from_dict(response.json()) - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 422: + 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]]: @@ -67,13 +45,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - job_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, JobDB]]: - """ Get Job +def sync_detailed(job_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobDB]]: + """Get Job Get a job by id. @@ -86,27 +59,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, JobDB]] - """ + """ + kwargs = _get_kwargs(job_id=job_id) - kwargs = _get_kwargs( - job_id=job_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - job_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, JobDB]]: - """ Get Job +def sync(job_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobDB]]: + """Get Job Get a job by id. @@ -119,22 +82,13 @@ def sync( Returns: Union[HTTPValidationError, JobDB] - """ - - - return sync_detailed( - job_id=job_id, -client=client, + """ - ).parsed + return sync_detailed(job_id=job_id, client=client).parsed -async def asyncio_detailed( - job_id: str, - *, - client: ApiClient, -) -> Response[Union[HTTPValidationError, JobDB]]: - """ Get Job +async def asyncio_detailed(job_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, JobDB]]: + """Get Job Get a job by id. @@ -147,27 +101,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, JobDB]] - """ + """ + kwargs = _get_kwargs(job_id=job_id) - kwargs = _get_kwargs( - job_id=job_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - job_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, JobDB]]: - """ Get Job +async def asyncio(job_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, JobDB]]: + """Get Job Get a job by id. @@ -180,11 +124,6 @@ async def asyncio( Returns: Union[HTTPValidationError, JobDB] - """ - - - return (await asyncio_detailed( - job_id=job_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(job_id=job_id, client=client)).parsed 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 08ed08c09..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 @@ -1,87 +1,65 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.job_db import JobDB -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - - +from ...types import UNSET, Response, Unset -def _get_kwargs( - project_id: str, - run_id: str, - *, - status: Union[None, Unset, str] = UNSET, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, run_id: str, *, status: Union[None, Unset, str] = UNSET) -> dict[str, Any]: headers: dict[str, Any] = {} - - - 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} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/runs/{run_id}/jobs".format(project_id=project_id,run_id=run_id,), + "path": f"/projects/{project_id}/runs/{run_id}/jobs", "params": params, } - 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, list['JobDB']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["JobDB"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = JobDB.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 422: + 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, list['JobDB']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["JobDB"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -91,14 +69,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - status: Union[None, Unset, str] = UNSET, - -) -> Response[Union[HTTPValidationError, list['JobDB']]]: - """ Get Jobs For Project Run + project_id: str, run_id: str, *, client: ApiClient, status: Union[None, Unset, str] = UNSET +) -> Response[Union[HTTPValidationError, list["JobDB"]]]: + """Get Jobs For Project Run Get all jobs by for a project and run. @@ -115,31 +88,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['JobDB']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, -status=status, + kwargs = _get_kwargs(project_id=project_id, run_id=run_id, status=status) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - run_id: str, - *, - client: ApiClient, - status: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, list['JobDB']]]: - """ Get Jobs For Project Run +def sync( + project_id: str, run_id: str, *, client: ApiClient, status: Union[None, Unset, str] = UNSET +) -> Optional[Union[HTTPValidationError, list["JobDB"]]]: + """Get Jobs For Project Run Get all jobs by for a project and run. @@ -156,26 +117,15 @@ def sync( Returns: Union[HTTPValidationError, list['JobDB']] - """ + """ + return sync_detailed(project_id=project_id, run_id=run_id, client=client, status=status).parsed - return sync_detailed( - project_id=project_id, -run_id=run_id, -client=client, -status=status, - - ).parsed async def asyncio_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - status: Union[None, Unset, str] = UNSET, - -) -> Response[Union[HTTPValidationError, list['JobDB']]]: - """ Get Jobs For Project Run + project_id: str, run_id: str, *, client: ApiClient, status: Union[None, Unset, str] = UNSET +) -> Response[Union[HTTPValidationError, list["JobDB"]]]: + """Get Jobs For Project Run Get all jobs by for a project and run. @@ -192,31 +142,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['JobDB']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, -status=status, + kwargs = _get_kwargs(project_id=project_id, run_id=run_id, status=status) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - run_id: str, - *, - client: ApiClient, - status: Union[None, Unset, str] = UNSET, -) -> Optional[Union[HTTPValidationError, list['JobDB']]]: - """ Get Jobs For Project Run +async def asyncio( + project_id: str, run_id: str, *, client: ApiClient, status: Union[None, Unset, str] = UNSET +) -> Optional[Union[HTTPValidationError, list["JobDB"]]]: + """Get Jobs For Project Run Get all jobs by for a project and run. @@ -233,13 +171,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['JobDB']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -run_id=run_id, -client=client, -status=status, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client, status=status)).parsed 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 36c7edf81..a867c9824 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 @@ -61,7 +61,7 @@ def _parse_response_200(data: object) -> Union['JobDB', None]: response_200 = _parse_response_200(response.json()) return response_200 - if response.status_code == : + if response.status_code == 422: response_422 = HTTPValidationError.from_dict(response.json()) 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 3b163b3f7..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_stream_create_request import LogStreamCreateRequest from ...models.log_stream_response import LogStreamResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogStreamCreateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogStreamCreateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/log_streams".format(project_id=project_id,), + "path": f"/projects/{project_id}/log_streams", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - if response.status_code == : - response_200 = LogStreamResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: + if response.status_code == 200: + return LogStreamResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogStreamResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogStreamResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogStreamCreateRequest, - + project_id: str, *, client: ApiClient, body: LogStreamCreateRequest ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """ Create Log Stream + """Create Log Stream Create a new log stream for a project. @@ -95,29 +75,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogStreamCreateRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogStreamCreateRequest ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """ Create Log Stream + """Create Log Stream Create a new log stream for a project. @@ -131,24 +101,15 @@ def sync( Returns: Union[HTTPValidationError, LogStreamResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogStreamCreateRequest, - + project_id: str, *, client: ApiClient, body: LogStreamCreateRequest ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """ Create Log Stream + """Create Log Stream Create a new log stream for a project. @@ -162,29 +123,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogStreamCreateRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogStreamCreateRequest ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """ Create Log Stream + """Create Log Stream Create a new log stream for a project. @@ -198,12 +149,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogStreamResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 f1d3bcea6..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 @@ -6,33 +6,21 @@ 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - log_stream_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, log_stream_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}/log_streams/{log_stream_id}".format(project_id=project_id,log_stream_id=log_stream_id,), + "path": f"/projects/{project_id}/log_streams/{log_stream_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_204 = cast(Any, None) - return response_204 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 204: + return cast(Any, None) + if response.status_code == 422: + 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]]: @@ -65,13 +48,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Log Stream + """Delete Log Stream Delete a specific log stream. @@ -85,29 +64,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - kwargs = _get_kwargs( - project_id=project_id, -log_stream_id=log_stream_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Log Stream +def sync(project_id: str, log_stream_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Log Stream Delete a specific log stream. @@ -121,24 +88,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -log_stream_id=log_stream_id, -client=client, + return sync_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Log Stream + """Delete Log Stream Delete a specific log stream. @@ -152,29 +110,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -log_stream_id=log_stream_id, + """ - ) + kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Log Stream + """Delete Log Stream Delete a specific log stream. @@ -188,12 +136,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -log_stream_id=log_stream_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client)).parsed 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 2df7143f9..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_stream_response import LogStreamResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - log_stream_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, log_stream_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/log_streams/{log_stream_id}".format(project_id=project_id,log_stream_id=log_stream_id,), + "path": f"/projects/{project_id}/log_streams/{log_stream_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, LogStreamResponse]]: - if response.status_code == : - response_200 = LogStreamResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: + if response.status_code == 200: + return LogStreamResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogStreamResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogStreamResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """ Get Log Stream + """Get Log Stream Retrieve a specific log stream. @@ -89,29 +70,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -log_stream_id=log_stream_id, + kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, +def sync( + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """ Get Log Stream + """Get Log Stream Retrieve a specific log stream. @@ -125,24 +96,15 @@ def sync( Returns: Union[HTTPValidationError, LogStreamResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -log_stream_id=log_stream_id, -client=client, + return sync_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """ Get Log Stream + """Get Log Stream Retrieve a specific log stream. @@ -156,29 +118,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id) - kwargs = _get_kwargs( - project_id=project_id, -log_stream_id=log_stream_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, log_stream_id: str, *, client: ApiClient ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """ Get Log Stream + """Get Log Stream Retrieve a specific log stream. @@ -192,12 +144,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogStreamResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -log_stream_id=log_stream_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client)).parsed 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 3841483ed..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 @@ -1,69 +1,56 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_stream_response import LogStreamResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/log_streams".format(project_id=project_id,), + "path": f"/projects/{project_id}/log_streams", } - 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, list['LogStreamResponse']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["LogStreamResponse"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = LogStreamResponse.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['LogStreamResponse']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["LogStreamResponse"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,12 +60,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['LogStreamResponse']]]: - """ List Log Streams + project_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["LogStreamResponse"]]]: + """List Log Streams Retrieve all log streams for a project. @@ -91,27 +75,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['LogStreamResponse']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, + kwargs = _get_kwargs(project_id=project_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['LogStreamResponse']]]: - """ List Log Streams +def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, list["LogStreamResponse"]]]: + """List Log Streams Retrieve all log streams for a project. @@ -124,22 +98,15 @@ def sync( Returns: Union[HTTPValidationError, list['LogStreamResponse']] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, + return sync_detailed(project_id=project_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['LogStreamResponse']]]: - """ List Log Streams + project_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["LogStreamResponse"]]]: + """List Log Streams Retrieve all log streams for a project. @@ -152,27 +119,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['LogStreamResponse']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, + kwargs = _get_kwargs(project_id=project_id) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['LogStreamResponse']]]: - """ List Log Streams +async def asyncio( + project_id: str, *, client: ApiClient +) -> Optional[Union[HTTPValidationError, list["LogStreamResponse"]]]: + """List Log Streams Retrieve all log streams for a project. @@ -185,11 +144,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['LogStreamResponse']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client)).parsed 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 51c12e4d1..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_stream_response import LogStreamResponse from ...models.log_stream_update_request import LogStreamUpdateRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - log_stream_id: str, - *, - body: LogStreamUpdateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, log_stream_id: str, *, body: LogStreamUpdateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}/log_streams/{log_stream_id}".format(project_id=project_id,log_stream_id=log_stream_id,), + "path": f"/projects/{project_id}/log_streams/{log_stream_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - if response.status_code == : - response_200 = LogStreamResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: + if response.status_code == 200: + return LogStreamResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogStreamResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogStreamResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - body: LogStreamUpdateRequest, - + project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """ Update Log Stream + """Update Log Stream Update a specific log stream. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -log_stream_id=log_stream_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - body: LogStreamUpdateRequest, +def sync( + project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """ Update Log Stream + """Update Log Stream Update a specific log stream. @@ -137,26 +103,15 @@ def sync( Returns: Union[HTTPValidationError, LogStreamResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -log_stream_id=log_stream_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - body: LogStreamUpdateRequest, - + project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest ) -> Response[Union[HTTPValidationError, LogStreamResponse]]: - """ Update Log Stream + """Update Log Stream Update a specific log stream. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogStreamResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, log_stream_id=log_stream_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -log_stream_id=log_stream_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - log_stream_id: str, - *, - client: ApiClient, - body: LogStreamUpdateRequest, +async def asyncio( + project_id: str, log_stream_id: str, *, client: ApiClient, body: LogStreamUpdateRequest ) -> Optional[Union[HTTPValidationError, LogStreamResponse]]: - """ Update Log Stream + """Update Log Stream Update a specific log stream. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogStreamResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -log_stream_id=log_stream_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, log_stream_id=log_stream_id, client=client, body=body)).parsed 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 e91e6bb0e..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.group_collaborator import GroupCollaborator from ...models.group_collaborator_create import GroupCollaboratorCreate from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: list['GroupCollaboratorCreate'], -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: list["GroupCollaboratorCreate"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/groups".format(project_id=project_id,), + "path": f"/projects/{project_id}/groups", } _kwargs["json"] = [] @@ -40,9 +28,6 @@ def _get_kwargs( body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) - - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -51,31 +36,29 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = GroupCollaborator.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['GroupCollaborator']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -85,13 +68,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Project Collaborators + project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Project Collaborators Share a project with groups. @@ -105,29 +84,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Project Collaborators +def sync( + project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Project Collaborators Share a project with groups. @@ -141,24 +110,15 @@ def sync( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Project Collaborators + project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Project Collaborators Share a project with groups. @@ -172,29 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Project Collaborators +async def asyncio( + project_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Project Collaborators Share a project with groups. @@ -208,12 +158,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 f85a9197b..12f150ed8 100644 --- a/src/galileo/resources/api/projects/create_project_projects_post.py +++ b/src/galileo/resources/api/projects/create_project_projects_post.py @@ -1,42 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_create import ProjectCreate from ...models.project_create_response import ProjectCreateResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: ProjectCreate, -) -> dict[str, Any]: +def _get_kwargs(*, body: ProjectCreate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/projects", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/projects"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +29,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: - if response.status_code == : - response_200 = ProjectCreateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: + if response.status_code == 200: + return ProjectCreateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ProjectCreateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: ProjectCreate, - + *, client: ApiClient, body: ProjectCreate ) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: - """ Create Project + """Create Project Create a new project. @@ -92,27 +70,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ProjectCreateResponse]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: ProjectCreate, -) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: - """ Create Project +def sync(*, client: ApiClient, body: ProjectCreate) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: + """Create Project Create a new project. @@ -125,22 +93,15 @@ def sync( Returns: Union[HTTPValidationError, ProjectCreateResponse] - """ + """ + return sync_detailed(client=client, body=body).parsed - return sync_detailed( - client=client, -body=body, - - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: ProjectCreate, - + *, client: ApiClient, body: ProjectCreate ) -> Response[Union[HTTPValidationError, ProjectCreateResponse]]: - """ Create Project + """Create Project Create a new project. @@ -153,27 +114,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectCreateResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: ProjectCreate, +async def asyncio( + *, client: ApiClient, body: ProjectCreate ) -> Optional[Union[HTTPValidationError, ProjectCreateResponse]]: - """ Create Project + """Create Project Create a new project. @@ -186,11 +139,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ProjectCreateResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 b78020a40..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator from ...models.user_collaborator_create import UserCollaboratorCreate -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: list['UserCollaboratorCreate'], -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: list["UserCollaboratorCreate"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/users".format(project_id=project_id,), + "path": f"/projects/{project_id}/users", } _kwargs["json"] = [] @@ -40,9 +28,6 @@ def _get_kwargs( body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) - - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -51,31 +36,29 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = UserCollaborator.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['UserCollaborator']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -85,13 +68,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Project Collaborators + project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Project Collaborators Share a project with users. @@ -105,29 +84,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Project Collaborators +def sync( + project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Project Collaborators Share a project with users. @@ -141,24 +110,15 @@ def sync( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Project Collaborators + project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Project Collaborators Share a project with users. @@ -172,29 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Project Collaborators +async def asyncio( + project_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Project Collaborators Share a project with users. @@ -208,12 +158,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 2be7f6cc9..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - project_id: str, - group_id: str, - -) -> dict[str, Any]: +def _get_kwargs(project_id: str, group_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}/groups/{group_id}".format(project_id=project_id,group_id=group_id,), + "path": f"/projects/{project_id}/groups/{group_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - project_id: str, - group_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Group Project Collaborator +def sync_detailed(project_id: str, group_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete Group Project Collaborator Remove a group's access to a project. @@ -85,29 +62,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, group_id=group_id) - kwargs = _get_kwargs( - project_id=project_id, -group_id=group_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - group_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Group Project Collaborator +def sync(project_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Group Project Collaborator Remove a group's access to a project. @@ -121,24 +86,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, group_id=group_id, client=client).parsed - return sync_detailed( - project_id=project_id, -group_id=group_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - group_id: str, - *, - client: ApiClient, - + project_id: str, group_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Group Project Collaborator + """Delete Group Project Collaborator Remove a group's access to a project. @@ -152,29 +108,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, group_id=group_id) - kwargs = _get_kwargs( - project_id=project_id, -group_id=group_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - group_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Group Project Collaborator +async def asyncio(project_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Group Project Collaborator Remove a group's access to a project. @@ -188,12 +132,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -group_id=group_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, group_id=group_id, client=client)).parsed 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 6ded921fe..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_delete_response import ProjectDeleteResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}".format(project_id=project_id,), + "path": f"/projects/{project_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, ProjectDeleteResponse]]: - if response.status_code == : - response_200 = ProjectDeleteResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: + if response.status_code == 200: + return ProjectDeleteResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ProjectDeleteResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -67,13 +53,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: - """ Delete Project +def sync_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: + """Delete Project Deletes a project and all associated runs and objects. @@ -89,27 +70,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ProjectDeleteResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id) - kwargs = _get_kwargs( - project_id=project_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: - """ Delete Project +def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: + """Delete Project Deletes a project and all associated runs and objects. @@ -125,22 +96,15 @@ def sync( Returns: Union[HTTPValidationError, ProjectDeleteResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, + return sync_detailed(project_id=project_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - + project_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, ProjectDeleteResponse]]: - """ Delete Project + """Delete Project Deletes a project and all associated runs and objects. @@ -156,27 +120,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectDeleteResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id) - kwargs = _get_kwargs( - project_id=project_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: - """ Delete Project +async def asyncio(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDeleteResponse]]: + """Delete Project Deletes a project and all associated runs and objects. @@ -192,11 +146,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ProjectDeleteResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client)).parsed 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 da69d7854..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - project_id: str, - user_id: str, - -) -> dict[str, Any]: +def _get_kwargs(project_id: str, user_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}/users/{user_id}".format(project_id=project_id,user_id=user_id,), + "path": f"/projects/{project_id}/users/{user_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - project_id: str, - user_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete User Project Collaborator +def sync_detailed(project_id: str, user_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete User Project Collaborator Remove a user's access to a project. @@ -85,29 +62,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, user_id=user_id) - kwargs = _get_kwargs( - project_id=project_id, -user_id=user_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - user_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete User Project Collaborator +def sync(project_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete User Project Collaborator Remove a user's access to a project. @@ -121,24 +86,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, user_id=user_id, client=client).parsed - return sync_detailed( - project_id=project_id, -user_id=user_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - user_id: str, - *, - client: ApiClient, - + project_id: str, user_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete User Project Collaborator + """Delete User Project Collaborator Remove a user's access to a project. @@ -152,29 +108,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, user_id=user_id) - kwargs = _get_kwargs( - project_id=project_id, -user_id=user_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - user_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete User Project Collaborator +async def asyncio(project_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete User Project Collaborator Remove a user's access to a project. @@ -188,12 +132,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -user_id=user_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, user_id=user_id, client=client)).parsed 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 685e6dcf2..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 @@ -1,33 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_db_thin import ProjectDBThin from ...models.project_type import ProjectType -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - - +from ...types import UNSET, Response, Unset -def _get_kwargs( - *, - type_: Union[None, ProjectType, Unset] = UNSET, -) -> dict[str, Any]: +def _get_kwargs(*, type_: Union[None, ProjectType, Unset] = UNSET) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -40,10 +28,8 @@ def _get_kwargs( json_type_ = type_ params["type"] = json_type_ - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, @@ -51,38 +37,35 @@ def _get_kwargs( "params": params, } - 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, list['ProjectDBThin']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["ProjectDBThin"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = ProjectDBThin.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 422: + 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, list['ProjectDBThin']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["ProjectDBThin"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -92,12 +75,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - type_: Union[None, ProjectType, Unset] = UNSET, - -) -> Response[Union[HTTPValidationError, list['ProjectDBThin']]]: - """ Get All Projects + *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET +) -> Response[Union[HTTPValidationError, list["ProjectDBThin"]]]: + """Get All Projects Gets all public projects and all private projects that the user has access to. @@ -114,27 +94,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDBThin']]] - """ - + """ - kwargs = _get_kwargs( - type_=type_, + kwargs = _get_kwargs(type_=type_) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - type_: Union[None, ProjectType, Unset] = UNSET, -) -> Optional[Union[HTTPValidationError, list['ProjectDBThin']]]: - """ Get All Projects +def sync( + *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET +) -> Optional[Union[HTTPValidationError, list["ProjectDBThin"]]]: + """Get All Projects Gets all public projects and all private projects that the user has access to. @@ -151,22 +123,15 @@ def sync( Returns: Union[HTTPValidationError, list['ProjectDBThin']] - """ - + """ - return sync_detailed( - client=client, -type_=type_, + return sync_detailed(client=client, type_=type_).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - type_: Union[None, ProjectType, Unset] = UNSET, - -) -> Response[Union[HTTPValidationError, list['ProjectDBThin']]]: - """ Get All Projects + *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET +) -> Response[Union[HTTPValidationError, list["ProjectDBThin"]]]: + """Get All Projects Gets all public projects and all private projects that the user has access to. @@ -183,27 +148,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDBThin']]] - """ + """ + kwargs = _get_kwargs(type_=type_) - kwargs = _get_kwargs( - type_=type_, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - type_: Union[None, ProjectType, Unset] = UNSET, -) -> Optional[Union[HTTPValidationError, list['ProjectDBThin']]]: - """ Get All Projects +async def asyncio( + *, client: ApiClient, type_: Union[None, ProjectType, Unset] = UNSET +) -> Optional[Union[HTTPValidationError, list["ProjectDBThin"]]]: + """Get All Projects Gets all public projects and all private projects that the user has access to. @@ -220,11 +177,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['ProjectDBThin']] - """ - - - return (await asyncio_detailed( - client=client, -type_=type_, + """ - )).parsed + return (await asyncio_detailed(client=client, type_=type_)).parsed 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 34c7ed936..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 @@ -1,35 +1,21 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional 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 ... import errors from ...models.collaborator_role_info import CollaboratorRoleInfo -from typing import cast - +from ...types import Response -def _get_kwargs( - -) -> dict[str, Any]: +def _get_kwargs() -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.GET, - "return_raw_response": True, - "path": "/collaborator_roles", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.GET, "return_raw_response": True, "path": "/collaborator_roles"} headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -37,25 +23,22 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list['CollaboratorRoleInfo']]: - if response.status_code == : +def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[list["CollaboratorRoleInfo"]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = CollaboratorRoleInfo.from_dict(response_200_item_data) - - response_200.append(response_200_item) 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']]: +def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[list["CollaboratorRoleInfo"]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -64,12 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - -) -> Response[list['CollaboratorRoleInfo']]: - """ Get Collaborator Roles +def sync_detailed(*, client: ApiClient) -> Response[list["CollaboratorRoleInfo"]]: + """Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -77,25 +56,17 @@ def sync_detailed( Returns: Response[list['CollaboratorRoleInfo']] - """ + """ + kwargs = _get_kwargs() - kwargs = _get_kwargs( - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, -) -> Optional[list['CollaboratorRoleInfo']]: - """ Get Collaborator Roles +def sync(*, client: ApiClient) -> Optional[list["CollaboratorRoleInfo"]]: + """Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -103,20 +74,13 @@ def sync( Returns: list['CollaboratorRoleInfo'] - """ - + """ - return sync_detailed( - client=client, + return sync_detailed(client=client).parsed - ).parsed -async def asyncio_detailed( - *, - client: ApiClient, - -) -> Response[list['CollaboratorRoleInfo']]: - """ Get Collaborator Roles +async def asyncio_detailed(*, client: ApiClient) -> Response[list["CollaboratorRoleInfo"]]: + """Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -124,25 +88,17 @@ async def asyncio_detailed( Returns: Response[list['CollaboratorRoleInfo']] - """ + """ + kwargs = _get_kwargs() - kwargs = _get_kwargs( - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, -) -> Optional[list['CollaboratorRoleInfo']]: - """ Get Collaborator Roles +async def asyncio(*, client: ApiClient) -> Optional[list["CollaboratorRoleInfo"]]: + """Get Collaborator Roles Raises: errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True. @@ -150,10 +106,6 @@ async def asyncio( Returns: list['CollaboratorRoleInfo'] - """ - - - return (await asyncio_detailed( - client=client, + """ - )).parsed + return (await asyncio_detailed(client=client)).parsed 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 b803f4ed0..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 @@ -1,38 +1,27 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_db import ProjectDB -from typing import cast - +from ...types import Response -def _get_kwargs( - project_id: str, - -) -> dict[str, Any]: +def _get_kwargs(project_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}".format(project_id=project_id,), + "path": f"/projects/{project_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,22 +29,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectDB]]: - if response.status_code == : - response_200 = ProjectDB.from_dict(response.json()) - - + if response.status_code == 200: + return ProjectDB.from_dict(response.json()) - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 422: + 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]]: @@ -67,13 +49,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, ProjectDB]]: - """ Get Project +def sync_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ProjectDB]]: + """Get Project Args: project_id (str): @@ -84,27 +61,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ProjectDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id) - kwargs = _get_kwargs( - project_id=project_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, ProjectDB]]: - """ Get Project +def sync(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDB]]: + """Get Project Args: project_id (str): @@ -115,22 +82,13 @@ def sync( Returns: Union[HTTPValidationError, ProjectDB] - """ - - - return sync_detailed( - project_id=project_id, -client=client, + """ - ).parsed + return sync_detailed(project_id=project_id, client=client).parsed -async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, -) -> Response[Union[HTTPValidationError, ProjectDB]]: - """ Get Project +async def asyncio_detailed(project_id: str, *, client: ApiClient) -> Response[Union[HTTPValidationError, ProjectDB]]: + """Get Project Args: project_id (str): @@ -141,27 +99,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id) - kwargs = _get_kwargs( - project_id=project_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, ProjectDB]]: - """ Get Project +async def asyncio(project_id: str, *, client: ApiClient) -> Optional[Union[HTTPValidationError, ProjectDB]]: + """Get Project Args: project_id (str): @@ -172,11 +120,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ProjectDB] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client)).parsed 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 157bed32a..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 @@ -6,36 +6,20 @@ 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_collection_params import ProjectCollectionParams -from typing import cast - +from ...types import Response -def _get_kwargs( - *, - body: ProjectCollectionParams, - -) -> dict[str, Any]: +def _get_kwargs(*, body: ProjectCollectionParams) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/projects/count", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/projects/count"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,19 +29,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, int]]: - if response.status_code == : - response_200 = cast(int, response.json()) - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return cast(int, response.json()) + if response.status_code == 422: + 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]]: @@ -69,13 +48,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - body: ProjectCollectionParams, - -) -> Response[Union[HTTPValidationError, int]]: - """ Get Projects Count +def sync_detailed(*, client: ApiClient, body: ProjectCollectionParams) -> Response[Union[HTTPValidationError, int]]: + """Get Projects Count Gets total count of projects for a user with applied filters. @@ -88,27 +62,17 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, int]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: ProjectCollectionParams, -) -> Optional[Union[HTTPValidationError, int]]: - """ Get Projects Count +def sync(*, client: ApiClient, body: ProjectCollectionParams) -> Optional[Union[HTTPValidationError, int]]: + """Get Projects Count Gets total count of projects for a user with applied filters. @@ -121,22 +85,15 @@ def sync( Returns: Union[HTTPValidationError, int] - """ + """ + return sync_detailed(client=client, body=body).parsed - return sync_detailed( - client=client, -body=body, - - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: ProjectCollectionParams, - + *, client: ApiClient, body: ProjectCollectionParams ) -> Response[Union[HTTPValidationError, int]]: - """ Get Projects Count + """Get Projects Count Gets total count of projects for a user with applied filters. @@ -149,27 +106,17 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, int]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: ProjectCollectionParams, -) -> Optional[Union[HTTPValidationError, int]]: - """ Get Projects Count +async def asyncio(*, client: ApiClient, body: ProjectCollectionParams) -> Optional[Union[HTTPValidationError, int]]: + """Get Projects Count Gets total count of projects for a user with applied filters. @@ -182,11 +129,6 @@ async def asyncio( Returns: Union[HTTPValidationError, int] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 75292c50c..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 @@ -1,22 +1,18 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.get_projects_paginated_response import GetProjectsPaginatedResponse from ...models.http_validation_error import HTTPValidationError from ...models.project_action import ProjectAction from ...models.project_collection_params import ProjectCollectionParams -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -25,12 +21,8 @@ def _get_kwargs( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -41,17 +33,14 @@ def _get_kwargs( actions_item = actions_item_data.value json_actions.append(actions_item) - params["actions"] = json_actions params["starting_token"] = starting_token params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -61,7 +50,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -70,26 +58,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = GetProjectsPaginatedResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: + if response.status_code == 200: + return GetProjectsPaginatedResponse.from_dict(response.json()) + if response.status_code == 422: + 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[GetProjectsPaginatedResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -105,9 +90,8 @@ def sync_detailed( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """ Get Projects Paginated + """Get Projects Paginated Gets projects for a user with pagination. @@ -126,23 +110,15 @@ def sync_detailed( Returns: Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, -actions=actions, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, @@ -150,9 +126,8 @@ def sync( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """ Get Projects Paginated + """Get Projects Paginated Gets projects for a user with pagination. @@ -171,17 +146,10 @@ def sync( Returns: Union[GetProjectsPaginatedResponse, HTTPValidationError] - """ + """ + return sync_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - client=client, -body=body, -actions=actions, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( *, @@ -190,9 +158,8 @@ async def asyncio_detailed( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """ Get Projects Paginated + """Get Projects Paginated Gets projects for a user with pagination. @@ -211,23 +178,15 @@ async def asyncio_detailed( Returns: Response[Union[GetProjectsPaginatedResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body, actions=actions, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - body=body, -actions=actions, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, @@ -235,9 +194,8 @@ async def asyncio( actions: Union[Unset, list[ProjectAction]] = UNSET, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[GetProjectsPaginatedResponse, HTTPValidationError]]: - """ Get Projects Paginated + """Get Projects Paginated Gets projects for a user with pagination. @@ -256,14 +214,8 @@ async def asyncio( Returns: Union[GetProjectsPaginatedResponse, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - client=client, -body=body, -actions=actions, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(client=client, body=body, actions=actions, starting_token=starting_token, limit=limit) + ).parsed 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 82308a179..0f25aab75 100644 --- a/src/galileo/resources/api/projects/get_projects_projects_get.py +++ b/src/galileo/resources/api/projects/get_projects_projects_get.py @@ -1,42 +1,28 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_db import ProjectDB from ...models.project_type import ProjectType -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - *, - project_name: Union[None, Unset, str] = UNSET, - type_: Union[None, ProjectType, Unset] = UNSET, - + *, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - 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] @@ -48,10 +34,8 @@ def _get_kwargs( json_type_ = type_ params["type"] = json_type_ - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, @@ -59,38 +43,35 @@ def _get_kwargs( "params": params, } - 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, list['ProjectDB']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["ProjectDB"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = ProjectDB.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['ProjectDB']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["ProjectDB"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -100,13 +81,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - project_name: Union[None, Unset, str] = UNSET, - type_: Union[None, ProjectType, Unset] = UNSET, - -) -> Response[Union[HTTPValidationError, list['ProjectDB']]]: - """ Get Projects + *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET +) -> Response[Union[HTTPValidationError, list["ProjectDB"]]]: + """Get Projects Gets projects for a user. @@ -124,29 +101,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDB']]] - """ + """ + kwargs = _get_kwargs(project_name=project_name, type_=type_) - kwargs = _get_kwargs( - project_name=project_name, -type_=type_, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - project_name: Union[None, Unset, str] = UNSET, - type_: Union[None, ProjectType, Unset] = UNSET, -) -> Optional[Union[HTTPValidationError, list['ProjectDB']]]: - """ Get Projects +def sync( + *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET +) -> Optional[Union[HTTPValidationError, list["ProjectDB"]]]: + """Get Projects Gets projects for a user. @@ -164,24 +131,15 @@ def sync( Returns: Union[HTTPValidationError, list['ProjectDB']] - """ - + """ - return sync_detailed( - client=client, -project_name=project_name, -type_=type_, + return sync_detailed(client=client, project_name=project_name, type_=type_).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - project_name: Union[None, Unset, str] = UNSET, - type_: Union[None, ProjectType, Unset] = UNSET, - -) -> Response[Union[HTTPValidationError, list['ProjectDB']]]: - """ Get Projects + *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET +) -> Response[Union[HTTPValidationError, list["ProjectDB"]]]: + """Get Projects Gets projects for a user. @@ -199,29 +157,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['ProjectDB']]] - """ + """ + kwargs = _get_kwargs(project_name=project_name, type_=type_) - kwargs = _get_kwargs( - project_name=project_name, -type_=type_, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - project_name: Union[None, Unset, str] = UNSET, - type_: Union[None, ProjectType, Unset] = UNSET, -) -> Optional[Union[HTTPValidationError, list['ProjectDB']]]: - """ Get Projects +async def asyncio( + *, client: ApiClient, project_name: Union[None, Unset, str] = UNSET, type_: Union[None, ProjectType, Unset] = UNSET +) -> Optional[Union[HTTPValidationError, list["ProjectDB"]]]: + """Get Projects Gets projects for a user. @@ -239,12 +187,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['ProjectDB']] - """ - - - return (await asyncio_detailed( - client=client, -project_name=project_name, -type_=type_, + """ - )).parsed + return (await asyncio_detailed(client=client, project_name=project_name, type_=type_)).parsed 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 fa1be6082..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_group_collaborators_response import ListGroupCollaboratorsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - project_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/groups".format(project_id=project_id,), + "path": f"/projects/{project_id}/groups", "params": params, } - 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, ListGroupCollaboratorsResponse]]: - if response.status_code == : - response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: + if response.status_code == 200: + return ListGroupCollaboratorsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListGroupCollaboratorsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Project Collaborators + """List Group Project Collaborators List the groups with which the project has been shared. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - project_id=project_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Project Collaborators + """List Group Project Collaborators List the groups with which the project has been shared. @@ -143,26 +109,15 @@ def sync( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ + return sync_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - project_id=project_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Project Collaborators + """List Group Project Collaborators List the groups with which the project has been shared. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Project Collaborators + """List Group Project Collaborators List the groups with which the project has been shared. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 adc2fc7a1..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_user_collaborators_response import ListUserCollaboratorsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - project_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/users".format(project_id=project_id,), + "path": f"/projects/{project_id}/users", "params": params, } - 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, ListUserCollaboratorsResponse]]: - if response.status_code == : - response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: + if response.status_code == 200: + return ListUserCollaboratorsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListUserCollaboratorsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Project Collaborators + """List User Project Collaborators List the users with which the project has been shared. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - project_id=project_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Project Collaborators + """List User Project Collaborators List the users with which the project has been shared. @@ -143,26 +109,15 @@ def sync( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ + return sync_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - project_id=project_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Project Collaborators + """List User Project Collaborators List the users with which the project has been shared. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(project_id=project_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + project_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Project Collaborators + """List User Project Collaborators List the users with which the project has been shared. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 8aa2b02fb..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.collaborator_update import CollaboratorUpdate from ...models.group_collaborator import GroupCollaborator from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - group_id: str, - *, - body: CollaboratorUpdate, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, group_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/projects/{project_id}/groups/{group_id}".format(project_id=project_id,group_id=group_id,), + "path": f"/projects/{project_id}/groups/{group_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - if response.status_code == : - response_200 = GroupCollaborator.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: + if response.status_code == 200: + return GroupCollaborator.from_dict(response.json()) + if response.status_code == 422: + 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[GroupCollaborator, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GroupCollaborator, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Project Collaborator + """Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -group_id=group_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, group_id=group_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +def sync( + project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Project Collaborator + """Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -137,26 +103,15 @@ def sync( Returns: Union[GroupCollaborator, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -group_id=group_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, group_id=group_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Project Collaborator + """Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, group_id=group_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -group_id=group_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +async def asyncio( + project_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Project Collaborator + """Update Group Project Collaborator Update the sharing permissions of a group on a project. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[GroupCollaborator, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -group_id=group_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, group_id=group_id, client=client, body=body)).parsed 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 94df3dc7a..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.project_update import ProjectUpdate from ...models.project_update_response import ProjectUpdateResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: ProjectUpdate, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: ProjectUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}".format(project_id=project_id,), + "path": f"/projects/{project_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: - if response.status_code == : - response_200 = ProjectUpdateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: + if response.status_code == 200: + return ProjectUpdateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ProjectUpdateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: ProjectUpdate, - + project_id: str, *, client: ApiClient, body: ProjectUpdate ) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: - """ Update Project + """Update Project Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ProjectUpdateResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: ProjectUpdate, +def sync( + project_id: str, *, client: ApiClient, body: ProjectUpdate ) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: - """ Update Project + """Update Project Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, ProjectUpdateResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: ProjectUpdate, - + project_id: str, *, client: ApiClient, body: ProjectUpdate ) -> Response[Union[HTTPValidationError, ProjectUpdateResponse]]: - """ Update Project + """Update Project Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ProjectUpdateResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: ProjectUpdate, +async def asyncio( + project_id: str, *, client: ApiClient, body: ProjectUpdate ) -> Optional[Union[HTTPValidationError, ProjectUpdateResponse]]: - """ Update Project + """Update Project Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ProjectUpdateResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 a5e9b80e6..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.collaborator_update import CollaboratorUpdate from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - user_id: str, - *, - body: CollaboratorUpdate, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, user_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/projects/{project_id}/users/{user_id}".format(project_id=project_id,user_id=user_id,), + "path": f"/projects/{project_id}/users/{user_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - if response.status_code == : - response_200 = UserCollaborator.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, UserCollaborator]]: + if response.status_code == 200: + return UserCollaborator.from_dict(response.json()) + if response.status_code == 422: + 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, UserCollaborator]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, UserCollaborator]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Project Collaborator + """Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -user_id=user_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, user_id=user_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +def sync( + project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Project Collaborator + """Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -137,26 +103,15 @@ def sync( Returns: Union[HTTPValidationError, UserCollaborator] - """ - + """ - return sync_detailed( - project_id=project_id, -user_id=user_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, user_id=user_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Project Collaborator + """Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, user_id=user_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -user_id=user_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +async def asyncio( + project_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Project Collaborator + """Update User Project Collaborator Update the sharing permissions of a user on a project. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[HTTPValidationError, UserCollaborator] - """ - - - return (await asyncio_detailed( - project_id=project_id, -user_id=user_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, user_id=user_id, client=client, body=body)).parsed 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 756bc5ade..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 @@ -1,43 +1,31 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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.body_upload_file_projects_project_id_upload_file_post import BodyUploadFileProjectsProjectIdUploadFilePost +from ... import errors +from ...models.body_upload_file_projects_project_id_upload_file_post import ( + BodyUploadFileProjectsProjectIdUploadFilePost, +) from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: BodyUploadFileProjectsProjectIdUploadFilePost, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: BodyUploadFileProjectsProjectIdUploadFilePost) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/upload_file".format(project_id=project_id,), + "path": f"/projects/{project_id}/upload_file", } _kwargs["files"] = body.to_multipart() - - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -45,19 +33,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -70,13 +53,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: BodyUploadFileProjectsProjectIdUploadFilePost, - + project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost ) -> Response[Union[Any, HTTPValidationError]]: - """ Upload File + """Upload File Args: project_id (str): @@ -88,29 +67,19 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: BodyUploadFileProjectsProjectIdUploadFilePost, +def sync( + project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost ) -> Optional[Union[Any, HTTPValidationError]]: - """ Upload File + """Upload File Args: project_id (str): @@ -122,24 +91,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: BodyUploadFileProjectsProjectIdUploadFilePost, - + project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost ) -> Response[Union[Any, HTTPValidationError]]: - """ Upload File + """Upload File Args: project_id (str): @@ -151,29 +111,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: BodyUploadFileProjectsProjectIdUploadFilePost, +async def asyncio( + project_id: str, *, client: ApiClient, body: BodyUploadFileProjectsProjectIdUploadFilePost ) -> Optional[Union[Any, HTTPValidationError]]: - """ Upload File + """Upload File Args: project_id (str): @@ -185,12 +135,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 b9fc6a3e8..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 @@ -1,31 +1,20 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional 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 ... import errors from ...models.bulk_delete_prompt_templates_request import BulkDeletePromptTemplatesRequest from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - *, - body: BulkDeletePromptTemplatesRequest, - -) -> dict[str, Any]: +def _get_kwargs(*, body: BulkDeletePromptTemplatesRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, @@ -35,7 +24,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,16 +33,12 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[HTTPValidationError]: - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 422: + 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]: @@ -66,13 +50,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - body: BulkDeletePromptTemplatesRequest, - -) -> Response[HTTPValidationError]: - """ Bulk Delete Global Templates +def sync_detailed(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Response[HTTPValidationError]: + """Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -104,27 +83,17 @@ def sync_detailed( Returns: Response[HTTPValidationError] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: BulkDeletePromptTemplatesRequest, -) -> Optional[HTTPValidationError]: - """ Bulk Delete Global Templates +def sync(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Optional[HTTPValidationError]: + """Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -156,22 +125,15 @@ def sync( Returns: HTTPValidationError - """ + """ + return sync_detailed(client=client, body=body).parsed - return sync_detailed( - client=client, -body=body, - - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: BulkDeletePromptTemplatesRequest, - + *, client: ApiClient, body: BulkDeletePromptTemplatesRequest ) -> Response[HTTPValidationError]: - """ Bulk Delete Global Templates + """Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -203,27 +165,17 @@ async def asyncio_detailed( Returns: Response[HTTPValidationError] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: BulkDeletePromptTemplatesRequest, -) -> Optional[HTTPValidationError]: - """ Bulk Delete Global Templates +async def asyncio(*, client: ApiClient, body: BulkDeletePromptTemplatesRequest) -> Optional[HTTPValidationError]: + """Bulk Delete Global Templates Delete multiple global prompt templates in bulk. @@ -255,11 +207,6 @@ async def asyncio( Returns: HTTPValidationError - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 330c24203..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 @@ -1,42 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.create_prompt_template_with_version_request_body import CreatePromptTemplateWithVersionRequestBody from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - *, - body: CreatePromptTemplateWithVersionRequestBody, -) -> dict[str, Any]: +def _get_kwargs(*, body: CreatePromptTemplateWithVersionRequestBody) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/templates", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/templates"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,26 +29,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -74,12 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, - + *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Global Prompt Template + """Create Global Prompt Template Create a global prompt template. @@ -109,27 +87,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - body=body, + kwargs = _get_kwargs(body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, +def sync( + *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Global Prompt Template + """Create Global Prompt Template Create a global prompt template. @@ -159,22 +129,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - client=client, -body=body, + return sync_detailed(client=client, body=body).parsed - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, - + *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Global Prompt Template + """Create Global Prompt Template Create a global prompt template. @@ -204,27 +167,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, +async def asyncio( + *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Global Prompt Template + """Create Global Prompt Template Create a global prompt template. @@ -254,11 +209,6 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 7784e3de0..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_version import BasePromptTemplateVersion from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - *, - body: BasePromptTemplateVersion, -) -> dict[str, Any]: +def _get_kwargs(template_id: str, *, body: BasePromptTemplateVersion) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/templates/{template_id}/versions".format(template_id=template_id,), + "path": f"/templates/{template_id}/versions", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, - + template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Global Prompt Template Version + """Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -109,29 +89,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -body=body, + kwargs = _get_kwargs(template_id=template_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, +def sync( + template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Global Prompt Template Version + """Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -159,24 +129,15 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - template_id=template_id, -client=client, -body=body, + return sync_detailed(template_id=template_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, - + template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Global Prompt Template Version + """Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -204,29 +165,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, body=body) - kwargs = _get_kwargs( - template_id=template_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, +async def asyncio( + template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Global Prompt Template Version + """Create Global Prompt Template Version Create a prompt template version for a given prompt template. @@ -254,12 +205,6 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed 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 ba9de9807..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.group_collaborator import GroupCollaborator from ...models.group_collaborator_create import GroupCollaboratorCreate from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - *, - body: list['GroupCollaboratorCreate'], -) -> dict[str, Any]: +def _get_kwargs(template_id: str, *, body: list["GroupCollaboratorCreate"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/templates/{template_id}/groups".format(template_id=template_id,), + "path": f"/templates/{template_id}/groups", } _kwargs["json"] = [] @@ -40,9 +28,6 @@ def _get_kwargs( body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) - - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -51,31 +36,29 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = GroupCollaborator.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['GroupCollaborator']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -85,13 +68,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Prompt Template Collaborators + template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -105,29 +84,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -body=body, + kwargs = _get_kwargs(template_id=template_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Prompt Template Collaborators +def sync( + template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -141,24 +110,15 @@ def sync( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ - + """ - return sync_detailed( - template_id=template_id, -client=client, -body=body, + return sync_detailed(template_id=template_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Prompt Template Collaborators + template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -172,29 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['GroupCollaborator']]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, body=body) - kwargs = _get_kwargs( - template_id=template_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, - body: list['GroupCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['GroupCollaborator']]]: - """ Create Group Prompt Template Collaborators +async def asyncio( + template_id: str, *, client: ApiClient, body: list["GroupCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["GroupCollaborator"]]]: + """Create Group Prompt Template Collaborators Share a prompt template with groups. @@ -208,12 +158,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['GroupCollaborator']] - """ - - - return (await asyncio_detailed( - template_id=template_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed 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 eb512cfff..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_version import BasePromptTemplateVersion from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - template_id: str, - *, - body: BasePromptTemplateVersion, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, template_id: str, *, body: BasePromptTemplateVersion) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/templates/{template_id}/versions".format(project_id=project_id,template_id=template_id,), + "path": f"/projects/{project_id}/templates/{template_id}/versions", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, - + project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Prompt Template Version + """Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -116,31 +94,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, template_id=template_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, +def sync( + project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Prompt Template Version + """Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -173,26 +139,15 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -template_id=template_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, template_id=template_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, - + project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Prompt Template Version + """Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -225,31 +180,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, template_id=template_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - template_id: str, - *, - client: ApiClient, - body: BasePromptTemplateVersion, +async def asyncio( + project_id: str, template_id: str, *, client: ApiClient, body: BasePromptTemplateVersion ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Create Prompt Template Version + """Create Prompt Template Version Create a prompt template version for a given prompt template. @@ -282,13 +225,6 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -template_id=template_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, template_id=template_id, client=client, body=body)).parsed 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 dd2baf895..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.create_prompt_template_with_version_request_body import CreatePromptTemplateWithVersionRequestBody from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: CreatePromptTemplateWithVersionRequestBody, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: CreatePromptTemplateWithVersionRequestBody) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/templates".format(project_id=project_id,), + "path": f"/projects/{project_id}/templates", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, - + project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Prompt Template With Version + """Create Prompt Template With Version For a given project, create a prompt template. @@ -118,29 +98,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, +def sync( + project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Prompt Template With Version + """Create Prompt Template With Version For a given project, create a prompt template. @@ -177,24 +147,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, - + project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Prompt Template With Version + """Create Prompt Template With Version For a given project, create a prompt template. @@ -231,29 +192,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: CreatePromptTemplateWithVersionRequestBody, +async def asyncio( + project_id: str, *, client: ApiClient, body: CreatePromptTemplateWithVersionRequestBody ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Create Prompt Template With Version + """Create Prompt Template With Version For a given project, create a prompt template. @@ -290,12 +241,6 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 a36c32e40..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator from ...models.user_collaborator_create import UserCollaboratorCreate -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - *, - body: list['UserCollaboratorCreate'], -) -> dict[str, Any]: +def _get_kwargs(template_id: str, *, body: list["UserCollaboratorCreate"]) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/templates/{template_id}/users".format(template_id=template_id,), + "path": f"/templates/{template_id}/users", } _kwargs["json"] = [] @@ -40,9 +28,6 @@ def _get_kwargs( body_item = body_item_data.to_dict() _kwargs["json"].append(body_item) - - - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -51,31 +36,29 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = UserCollaborator.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 422: + 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, list['UserCollaborator']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -85,13 +68,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Prompt Template Collaborators + template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Prompt Template Collaborators Args: template_id (str): @@ -103,29 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -body=body, + kwargs = _get_kwargs(template_id=template_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Prompt Template Collaborators +def sync( + template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Prompt Template Collaborators Args: template_id (str): @@ -137,24 +106,15 @@ def sync( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ - + """ - return sync_detailed( - template_id=template_id, -client=client, -body=body, + return sync_detailed(template_id=template_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], - -) -> Response[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Prompt Template Collaborators + template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Response[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Prompt Template Collaborators Args: template_id (str): @@ -166,29 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['UserCollaborator']]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, body=body) - kwargs = _get_kwargs( - template_id=template_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, - body: list['UserCollaboratorCreate'], -) -> Optional[Union[HTTPValidationError, list['UserCollaborator']]]: - """ Create User Prompt Template Collaborators +async def asyncio( + template_id: str, *, client: ApiClient, body: list["UserCollaboratorCreate"] +) -> Optional[Union[HTTPValidationError, list["UserCollaborator"]]]: + """Create User Prompt Template Collaborators Args: template_id (str): @@ -200,12 +150,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['UserCollaborator']] - """ - - - return (await asyncio_detailed( - template_id=template_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed 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 7462b009d..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.delete_prompt_response import DeletePromptResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, -) -> dict[str, Any]: +def _get_kwargs(template_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/templates/{template_id}".format(template_id=template_id,), + "path": f"/templates/{template_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[DeletePromptResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = DeletePromptResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + if response.status_code == 200: + return DeletePromptResponse.from_dict(response.json()) + if response.status_code == 422: + 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[DeletePromptResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -67,13 +53,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - template_id: str, - *, - client: ApiClient, - -) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Global Template +def sync_detailed(template_id: str, *, client: ApiClient) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: + """Delete Global Template Delete a global prompt template given a template ID. @@ -98,27 +79,17 @@ def sync_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id) - kwargs = _get_kwargs( - template_id=template_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, -) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Global Template +def sync(template_id: str, *, client: ApiClient) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + """Delete Global Template Delete a global prompt template given a template ID. @@ -143,22 +114,15 @@ def sync( Returns: Union[DeletePromptResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - template_id=template_id, -client=client, + return sync_detailed(template_id=template_id, client=client).parsed - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - + template_id: str, *, client: ApiClient ) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Global Template + """Delete Global Template Delete a global prompt template given a template ID. @@ -183,27 +147,17 @@ async def asyncio_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id) - kwargs = _get_kwargs( - template_id=template_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, -) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Global Template +async def asyncio(template_id: str, *, client: ApiClient) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + """Delete Global Template Delete a global prompt template given a template ID. @@ -228,11 +182,6 @@ async def asyncio( Returns: Union[DeletePromptResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, client=client)).parsed 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 58c833e33..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - template_id: str, - group_id: str, - -) -> dict[str, Any]: +def _get_kwargs(template_id: str, group_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/templates/{template_id}/groups/{group_id}".format(template_id=template_id,group_id=group_id,), + "path": f"/templates/{template_id}/groups/{group_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - template_id: str, - group_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Group Prompt Template Collaborator +def sync_detailed(template_id: str, group_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -85,29 +62,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, group_id=group_id) - kwargs = _get_kwargs( - template_id=template_id, -group_id=group_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - group_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Group Prompt Template Collaborator +def sync(template_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -121,24 +86,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(template_id=template_id, group_id=group_id, client=client).parsed - return sync_detailed( - template_id=template_id, -group_id=group_id, -client=client, - - ).parsed async def asyncio_detailed( - template_id: str, - group_id: str, - *, - client: ApiClient, - + template_id: str, group_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete Group Prompt Template Collaborator + """Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -152,29 +108,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, group_id=group_id) - kwargs = _get_kwargs( - template_id=template_id, -group_id=group_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - group_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete Group Prompt Template Collaborator +async def asyncio(template_id: str, group_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete Group Prompt Template Collaborator Remove a group's access to a prompt template. @@ -188,12 +132,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -group_id=group_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, group_id=group_id, client=client)).parsed 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 4e97c746c..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.delete_prompt_response import DeletePromptResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - template_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, template_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/projects/{project_id}/templates/{template_id}".format(project_id=project_id,template_id=template_id,), + "path": f"/projects/{project_id}/templates/{template_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[DeletePromptResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = DeletePromptResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: + if response.status_code == 200: + return DeletePromptResponse.from_dict(response.json()) + if response.status_code == 422: + 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[DeletePromptResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - template_id: str, - *, - client: ApiClient, - + project_id: str, template_id: str, *, client: ApiClient ) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Template + """Delete Template Args: project_id (str): @@ -87,29 +68,19 @@ def sync_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, + kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - template_id: str, - *, - client: ApiClient, +def sync( + project_id: str, template_id: str, *, client: ApiClient ) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Template + """Delete Template Args: project_id (str): @@ -121,24 +92,15 @@ def sync( Returns: Union[DeletePromptResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -template_id=template_id, -client=client, + return sync_detailed(project_id=project_id, template_id=template_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - template_id: str, - *, - client: ApiClient, - + project_id: str, template_id: str, *, client: ApiClient ) -> Response[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Template + """Delete Template Args: project_id (str): @@ -150,29 +112,19 @@ async def asyncio_detailed( Returns: Response[Union[DeletePromptResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - template_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, template_id: str, *, client: ApiClient ) -> Optional[Union[DeletePromptResponse, HTTPValidationError]]: - """ Delete Template + """Delete Template Args: project_id (str): @@ -184,12 +136,6 @@ async def asyncio( Returns: Union[DeletePromptResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -template_id=template_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, template_id=template_id, client=client)).parsed 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 3286c8d9c..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 @@ -1,38 +1,26 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError -from typing import cast - +from ...types import Response -def _get_kwargs( - template_id: str, - user_id: str, - -) -> dict[str, Any]: +def _get_kwargs(template_id: str, user_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.DELETE, "return_raw_response": True, - "path": "/templates/{template_id}/users/{user_id}".format(template_id=template_id,user_id=user_id,), + "path": f"/templates/{template_id}/users/{user_id}", } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -40,19 +28,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -64,14 +47,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - template_id: str, - user_id: str, - *, - client: ApiClient, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Delete User Prompt Template Collaborator +def sync_detailed(template_id: str, user_id: str, *, client: ApiClient) -> Response[Union[Any, HTTPValidationError]]: + """Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -85,29 +62,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, user_id=user_id) - kwargs = _get_kwargs( - template_id=template_id, -user_id=user_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - user_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete User Prompt Template Collaborator +def sync(template_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -121,24 +86,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(template_id=template_id, user_id=user_id, client=client).parsed - return sync_detailed( - template_id=template_id, -user_id=user_id, -client=client, - - ).parsed async def asyncio_detailed( - template_id: str, - user_id: str, - *, - client: ApiClient, - + template_id: str, user_id: str, *, client: ApiClient ) -> Response[Union[Any, HTTPValidationError]]: - """ Delete User Prompt Template Collaborator + """Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -152,29 +108,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, user_id=user_id) - kwargs = _get_kwargs( - template_id=template_id, -user_id=user_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - user_id: str, - *, - client: ApiClient, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Delete User Prompt Template Collaborator +async def asyncio(template_id: str, user_id: str, *, client: ApiClient) -> Optional[Union[Any, HTTPValidationError]]: + """Delete User Prompt Template Collaborator Remove a user's access to a prompt template. @@ -188,12 +132,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -user_id=user_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, user_id=user_id, client=client)).parsed 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 6af0238e1..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 @@ -1,41 +1,25 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.template_stub_request import TemplateStubRequest -from typing import cast - +from ...types import Response -def _get_kwargs( - *, - body: TemplateStubRequest, - -) -> dict[str, Any]: +def _get_kwargs(*, body: TemplateStubRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - - _kwargs: dict[str, Any] = { - "method": RequestMethod.POST, - "return_raw_response": True, - "path": "/input_stub", - } + _kwargs: dict[str, Any] = {"method": RequestMethod.POST, "return_raw_response": True, "path": "/input_stub"} _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -45,19 +29,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -69,13 +48,8 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ ) -def sync_detailed( - *, - client: ApiClient, - body: TemplateStubRequest, - -) -> Response[Union[Any, HTTPValidationError]]: - """ Generate Template Input Stub +def sync_detailed(*, client: ApiClient, body: TemplateStubRequest) -> Response[Union[Any, HTTPValidationError]]: + """Generate Template Input Stub Args: body (TemplateStubRequest): @@ -86,27 +60,17 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - *, - client: ApiClient, - body: TemplateStubRequest, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Generate Template Input Stub +def sync(*, client: ApiClient, body: TemplateStubRequest) -> Optional[Union[Any, HTTPValidationError]]: + """Generate Template Input Stub Args: body (TemplateStubRequest): @@ -117,22 +81,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(client=client, body=body).parsed - return sync_detailed( - client=client, -body=body, - - ).parsed async def asyncio_detailed( - *, - client: ApiClient, - body: TemplateStubRequest, - + *, client: ApiClient, body: TemplateStubRequest ) -> Response[Union[Any, HTTPValidationError]]: - """ Generate Template Input Stub + """Generate Template Input Stub Args: body (TemplateStubRequest): @@ -143,27 +100,17 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(body=body) - kwargs = _get_kwargs( - body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - *, - client: ApiClient, - body: TemplateStubRequest, -) -> Optional[Union[Any, HTTPValidationError]]: - """ Generate Template Input Stub +async def asyncio(*, client: ApiClient, body: TemplateStubRequest) -> Optional[Union[Any, HTTPValidationError]]: + """Generate Template Input Stub Args: body (TemplateStubRequest): @@ -174,11 +121,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body)).parsed 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 2089ad632..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 @@ -1,64 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, -) -> dict[str, Any]: +def _get_kwargs(template_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/templates/{template_id}".format(template_id=template_id,), + "path": f"/templates/{template_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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -68,12 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - + template_id: str, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Global Template + """Get Global Template Get a global prompt template given a template ID. @@ -100,27 +83,17 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id) - kwargs = _get_kwargs( - template_id=template_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, -) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Global Template +def sync(template_id: str, *, client: ApiClient) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + """Get Global Template Get a global prompt template given a template ID. @@ -147,22 +120,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + return sync_detailed(template_id=template_id, client=client).parsed - return sync_detailed( - template_id=template_id, -client=client, - - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - + template_id: str, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Global Template + """Get Global Template Get a global prompt template given a template ID. @@ -189,27 +155,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, + kwargs = _get_kwargs(template_id=template_id) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, +async def asyncio( + template_id: str, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Global Template + """Get Global Template Get a global prompt template given a template ID. @@ -236,11 +194,6 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, client=client)).parsed 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 b7a9588ba..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - version: int, -) -> dict[str, Any]: +def _get_kwargs(template_id: str, version: int) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/templates/{template_id}/versions/{version}".format(template_id=template_id,version=version,), + "path": f"/templates/{template_id}/versions/{version}", } - 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - version: int, - *, - client: ApiClient, - + template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Global Template Version + """Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -103,29 +84,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -version=version, + kwargs = _get_kwargs(template_id=template_id, version=version) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - version: int, - *, - client: ApiClient, +def sync( + template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Global Template Version + """Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -153,24 +124,15 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - template_id=template_id, -version=version, -client=client, + return sync_detailed(template_id=template_id, version=version, client=client).parsed - ).parsed async def asyncio_detailed( - template_id: str, - version: int, - *, - client: ApiClient, - + template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Global Template Version + """Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -198,29 +160,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, version=version) - kwargs = _get_kwargs( - template_id=template_id, -version=version, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - version: int, - *, - client: ApiClient, +async def asyncio( + template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Global Template Version + """Get Global Template Version Get a global prompt template version given a template ID and version number. @@ -248,12 +200,6 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -version=version, -client=client, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, version=version, client=client)).parsed 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 561c4b825..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 @@ -1,69 +1,56 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/templates".format(project_id=project_id,), + "path": f"/projects/{project_id}/templates", } - 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, list['BasePromptTemplateResponse']]]: - if response.status_code == : +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: + if response.status_code == 200: response_200 = [] _response_200 = response.json() - for response_200_item_data in (_response_200): + for response_200_item_data in _response_200: response_200_item = BasePromptTemplateResponse.from_dict(response_200_item_data) - - response_200.append(response_200_item) return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - - + if response.status_code == 422: + 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, list['BasePromptTemplateResponse']]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -73,12 +60,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: - """ Get Project Templates + project_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: + """Get Project Templates Get all prompt templates for a project. @@ -103,27 +87,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, + """ - ) + kwargs = _get_kwargs(project_id=project_id) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: - """ Get Project Templates +def sync( + project_id: str, *, client: ApiClient +) -> Optional[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: + """Get Project Templates Get all prompt templates for a project. @@ -148,22 +124,15 @@ def sync( Returns: Union[HTTPValidationError, list['BasePromptTemplateResponse']] - """ + """ + return sync_detailed(project_id=project_id, client=client).parsed - return sync_detailed( - project_id=project_id, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - -) -> Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: - """ Get Project Templates + project_id: str, *, client: ApiClient +) -> Response[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: + """Get Project Templates Get all prompt templates for a project. @@ -188,27 +157,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, list['BasePromptTemplateResponse']]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, + kwargs = _get_kwargs(project_id=project_id) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, -) -> Optional[Union[HTTPValidationError, list['BasePromptTemplateResponse']]]: - """ Get Project Templates +async def asyncio( + project_id: str, *, client: ApiClient +) -> Optional[Union[HTTPValidationError, list["BasePromptTemplateResponse"]]]: + """Get Project Templates Get all prompt templates for a project. @@ -233,11 +194,6 @@ async def asyncio( Returns: Union[HTTPValidationError, list['BasePromptTemplateResponse']] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client)).parsed 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 c2504fe03..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - template_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, template_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/templates/{template_id}".format(project_id=project_id,template_id=template_id,), + "path": f"/projects/{project_id}/templates/{template_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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - template_id: str, - *, - client: ApiClient, - + project_id: str, template_id: str, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Template From Project + """Get Template From Project Get a prompt template from a project. @@ -103,29 +84,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, + kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - template_id: str, - *, - client: ApiClient, +def sync( + project_id: str, template_id: str, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Template From Project + """Get Template From Project Get a prompt template from a project. @@ -153,24 +124,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -template_id=template_id, -client=client, + return sync_detailed(project_id=project_id, template_id=template_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - template_id: str, - *, - client: ApiClient, - + project_id: str, template_id: str, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Template From Project + """Get Template From Project Get a prompt template from a project. @@ -198,29 +160,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, template_id=template_id) - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - template_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, template_id: str, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Get Template From Project + """Get Template From Project Get a prompt template from a project. @@ -248,12 +200,6 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -template_id=template_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, template_id=template_id, client=client)).parsed 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 9ce56fe2a..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 @@ -1,84 +1,61 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - +from ...types import UNSET, Response, Unset -def _get_kwargs( - project_id: str, - *, - template_name: str, - version: Union[None, Unset, int] = UNSET, - -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, template_name: str, version: Union[None, Unset, int] = UNSET) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} 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} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/templates/versions".format(project_id=project_id,), + "path": f"/projects/{project_id}/templates/versions", "params": params, } - 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -88,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - template_name: str, - version: Union[None, Unset, int] = UNSET, - + project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version By Name + """Get Template Version By Name Get a prompt template from a project. @@ -127,31 +99,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, template_name=template_name, version=version) - kwargs = _get_kwargs( - project_id=project_id, -template_name=template_name, -version=version, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - template_name: str, - version: Union[None, Unset, int] = UNSET, +def sync( + project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version By Name + """Get Template Version By Name Get a prompt template from a project. @@ -183,26 +143,15 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, client=client, template_name=template_name, version=version).parsed - return sync_detailed( - project_id=project_id, -client=client, -template_name=template_name, -version=version, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - template_name: str, - version: Union[None, Unset, int] = UNSET, - + project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version By Name + """Get Template Version By Name Get a prompt template from a project. @@ -234,31 +183,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -template_name=template_name, -version=version, + kwargs = _get_kwargs(project_id=project_id, template_name=template_name, version=version) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - template_name: str, - version: Union[None, Unset, int] = UNSET, +async def asyncio( + project_id: str, *, client: ApiClient, template_name: str, version: Union[None, Unset, int] = UNSET ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version By Name + """Get Template Version By Name Get a prompt template from a project. @@ -290,13 +227,8 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -client=client, -template_name=template_name, -version=version, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, client=client, template_name=template_name, version=version) + ).parsed 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 1cf88fc49..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 @@ -1,66 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_version_response import BasePromptTemplateVersionResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - template_id: str, - version: int, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, template_id: str, version: int) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/templates/{template_id}/versions/{version}".format(project_id=project_id,template_id=template_id,version=version,), + "path": f"/projects/{project_id}/templates/{template_id}/versions/{version}", } - 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateVersionResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,14 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, - + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version + """Get Template Version Get a specific version of a prompt template. @@ -106,31 +85,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, -version=version, + """ - ) + kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, +def sync( + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version + """Get Template Version Get a specific version of a prompt template. @@ -159,26 +126,15 @@ def sync( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, template_id=template_id, version=version, client=client).parsed - return sync_detailed( - project_id=project_id, -template_id=template_id, -version=version, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, - + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version + """Get Template Version Get a specific version of a prompt template. @@ -207,31 +163,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateVersionResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, -version=version, + kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, +async def asyncio( + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateVersionResponse, HTTPValidationError]]: - """ Get Template Version + """Get Template Version Get a specific version of a prompt template. @@ -260,13 +204,8 @@ async def asyncio( Returns: Union[BasePromptTemplateVersionResponse, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -template_id=template_id, -version=version, -client=client, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, template_id=template_id, version=version, client=client) + ).parsed 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 365325487..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_group_collaborators_response import ListGroupCollaboratorsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - template_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + template_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/templates/{template_id}/groups".format(template_id=template_id,), + "path": f"/templates/{template_id}/groups", "params": params, } - 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, ListGroupCollaboratorsResponse]]: - if response.status_code == : - response_200 = ListGroupCollaboratorsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: + if response.status_code == 200: + return ListGroupCollaboratorsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListGroupCollaboratorsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Prompt Template Collaborators + """List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - template_id=template_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Prompt Template Collaborators + """List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -143,26 +109,15 @@ def sync( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ + """ + return sync_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - template_id=template_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Prompt Template Collaborators + """List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListGroupCollaboratorsResponse]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListGroupCollaboratorsResponse]]: - """ List Group Prompt Template Collaborators + """List Group Prompt Template Collaborators List the groups with which the prompt template has been shared. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListGroupCollaboratorsResponse] - """ - + """ - return (await asyncio_detailed( - template_id=template_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 4a654e629..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 @@ -1,33 +1,22 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_user_collaborators_response import ListUserCollaboratorsResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - template_id: str, - *, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + template_id: str, *, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -35,44 +24,38 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/templates/{template_id}/users".format(template_id=template_id,), + "path": f"/templates/{template_id}/users", "params": params, } - 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, ListUserCollaboratorsResponse]]: - if response.status_code == : - response_200 = ListUserCollaboratorsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: + if response.status_code == 200: + return ListUserCollaboratorsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListUserCollaboratorsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -82,14 +65,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Prompt Template Collaborators + """List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -104,31 +82,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - template_id=template_id, -starting_token=starting_token, -limit=limit, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +def sync( + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Prompt Template Collaborators + """List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -143,26 +109,15 @@ def sync( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ + """ + return sync_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - template_id=template_id, -client=client, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Prompt Template Collaborators + """List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -177,31 +132,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListUserCollaboratorsResponse]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(template_id=template_id, starting_token=starting_token, limit=limit) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, +async def asyncio( + template_id: str, *, client: ApiClient, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> Optional[Union[HTTPValidationError, ListUserCollaboratorsResponse]]: - """ List User Prompt Template Collaborators + """List User Prompt Template Collaborators List the users with which the prompt template has been shared. @@ -216,13 +159,8 @@ async def asyncio( Returns: Union[HTTPValidationError, ListUserCollaboratorsResponse] - """ - + """ - return (await asyncio_detailed( - template_id=template_id, -client=client, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed(template_id=template_id, client=client, starting_token=starting_token, limit=limit) + ).parsed 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 84178103e..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 @@ -1,21 +1,17 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_prompt_template_version_params import ListPromptTemplateVersionParams from ...models.list_prompt_template_version_response import ListPromptTemplateVersionResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( @@ -24,12 +20,8 @@ def _get_kwargs( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -37,20 +29,17 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/templates/{template_id}/versions/query".format(template_id=template_id,), + "path": f"/templates/{template_id}/versions/query", "params": params, } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -59,26 +48,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - if response.status_code == : - response_200 = ListPromptTemplateVersionResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: + if response.status_code == 200: + return ListPromptTemplateVersionResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListPromptTemplateVersionResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -94,9 +80,8 @@ def sync_detailed( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """ Query Template Versions + """Query Template Versions Query versions of a specific prompt template. @@ -128,23 +113,15 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]] - """ - - - kwargs = _get_kwargs( - template_id=template_id, -body=body, -starting_token=starting_token, -limit=limit, + """ - ) + kwargs = _get_kwargs(template_id=template_id, body=body, starting_token=starting_token, limit=limit) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( template_id: str, *, @@ -152,9 +129,8 @@ def sync( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """ Query Template Versions + """Query Template Versions Query versions of a specific prompt template. @@ -186,18 +162,13 @@ def sync( Returns: Union[HTTPValidationError, ListPromptTemplateVersionResponse] - """ - + """ return sync_detailed( - template_id=template_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - + template_id=template_id, client=client, body=body, starting_token=starting_token, limit=limit ).parsed + async def asyncio_detailed( template_id: str, *, @@ -205,9 +176,8 @@ async def asyncio_detailed( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """ Query Template Versions + """Query Template Versions Query versions of a specific prompt template. @@ -239,23 +209,15 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateVersionResponse]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - template_id=template_id, -body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( template_id: str, *, @@ -263,9 +225,8 @@ async def asyncio( body: ListPromptTemplateVersionParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListPromptTemplateVersionResponse]]: - """ Query Template Versions + """Query Template Versions Query versions of a specific prompt template. @@ -297,14 +258,10 @@ async def asyncio( Returns: Union[HTTPValidationError, ListPromptTemplateVersionResponse] - """ - + """ - return (await asyncio_detailed( - template_id=template_id, -client=client, -body=body, -starting_token=starting_token, -limit=limit, - - )).parsed + return ( + await asyncio_detailed( + template_id=template_id, client=client, body=body, starting_token=starting_token, limit=limit + ) + ).parsed 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 dfd7b71b7..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 @@ -1,34 +1,23 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.list_prompt_template_params import ListPromptTemplateParams from ...models.list_prompt_template_response import ListPromptTemplateResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - *, - body: ListPromptTemplateParams, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + *, body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -36,10 +25,8 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -49,7 +36,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -58,26 +44,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: - if response.status_code == : - response_200 = ListPromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: + if response.status_code == 200: + return ListPromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, ListPromptTemplateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -92,9 +75,8 @@ def sync_detailed( body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """ Query Templates + """Query Templates Query prompt templates the user has access to. @@ -123,31 +105,23 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """ Query Templates + """Query Templates Query prompt templates the user has access to. @@ -176,16 +150,10 @@ def sync( Returns: Union[HTTPValidationError, ListPromptTemplateResponse] - """ + """ + return sync_detailed(client=client, body=body, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - client=client, -body=body, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( *, @@ -193,9 +161,8 @@ async def asyncio_detailed( body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """ Query Templates + """Query Templates Query prompt templates the user has access to. @@ -224,31 +191,23 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, ListPromptTemplateResponse]] - """ + """ + kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, body: ListPromptTemplateParams, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, ListPromptTemplateResponse]]: - """ Query Templates + """Query Templates Query prompt templates the user has access to. @@ -277,13 +236,6 @@ async def asyncio( Returns: Union[HTTPValidationError, ListPromptTemplateResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, -starting_token=starting_token, -limit=limit, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body, starting_token=starting_token, limit=limit)).parsed 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 1bd6625e7..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 @@ -1,34 +1,23 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.render_template_request import RenderTemplateRequest from ...models.render_template_response import RenderTemplateResponse -from ...types import UNSET, Unset -from typing import cast -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - *, - body: RenderTemplateRequest, - starting_token: Union[Unset, int] = 0, - limit: Union[Unset, int] = 100, - + *, body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100 ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} @@ -36,10 +25,8 @@ def _get_kwargs( params["limit"] = limit - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, @@ -49,7 +36,6 @@ def _get_kwargs( _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -58,26 +44,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: - if response.status_code == : - response_200 = RenderTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: + if response.status_code == 200: + return RenderTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, RenderTemplateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -92,9 +75,8 @@ def sync_detailed( body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: - """ Render Template + """Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -107,31 +89,23 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RenderTemplateResponse]] - """ - + """ - kwargs = _get_kwargs( - body=body, -starting_token=starting_token, -limit=limit, + kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( *, client: ApiClient, body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: - """ Render Template + """Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -144,16 +118,10 @@ def sync( Returns: Union[HTTPValidationError, RenderTemplateResponse] - """ + """ + return sync_detailed(client=client, body=body, starting_token=starting_token, limit=limit).parsed - return sync_detailed( - client=client, -body=body, -starting_token=starting_token, -limit=limit, - - ).parsed async def asyncio_detailed( *, @@ -161,9 +129,8 @@ async def asyncio_detailed( body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Response[Union[HTTPValidationError, RenderTemplateResponse]]: - """ Render Template + """Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -176,31 +143,23 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RenderTemplateResponse]] - """ + """ + kwargs = _get_kwargs(body=body, starting_token=starting_token, limit=limit) - kwargs = _get_kwargs( - body=body, -starting_token=starting_token, -limit=limit, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( *, client: ApiClient, body: RenderTemplateRequest, starting_token: Union[Unset, int] = 0, limit: Union[Unset, int] = 100, - ) -> Optional[Union[HTTPValidationError, RenderTemplateResponse]]: - """ Render Template + """Render Template Args: starting_token (Union[Unset, int]): Default: 0. @@ -213,13 +172,6 @@ async def asyncio( Returns: Union[HTTPValidationError, RenderTemplateResponse] - """ - - - return (await asyncio_detailed( - client=client, -body=body, -starting_token=starting_token, -limit=limit, + """ - )).parsed + return (await asyncio_detailed(client=client, body=body, starting_token=starting_token, limit=limit)).parsed 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 072924a09..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - version: int, -) -> dict[str, Any]: +def _get_kwargs(template_id: str, version: int) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/templates/{template_id}/versions/{version}".format(template_id=template_id,version=version,), + "path": f"/templates/{template_id}/versions/{version}", } - 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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - version: int, - *, - client: ApiClient, - + template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Global Template Version + """Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -103,29 +84,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -version=version, + kwargs = _get_kwargs(template_id=template_id, version=version) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - version: int, - *, - client: ApiClient, +def sync( + template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Global Template Version + """Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -153,24 +124,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - template_id=template_id, -version=version, -client=client, + return sync_detailed(template_id=template_id, version=version, client=client).parsed - ).parsed async def asyncio_detailed( - template_id: str, - version: int, - *, - client: ApiClient, - + template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Global Template Version + """Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -198,29 +160,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, version=version) - kwargs = _get_kwargs( - template_id=template_id, -version=version, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - version: int, - *, - client: ApiClient, +async def asyncio( + template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Global Template Version + """Set Selected Global Template Version Set a global prompt template version as the selected version. @@ -248,12 +200,6 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -version=version, -client=client, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, version=version, client=client)).parsed 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 b54f89ac8..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 @@ -1,66 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - template_id: str, - version: int, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, template_id: str, version: int) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}/templates/{template_id}/versions/{version}".format(project_id=project_id,template_id=template_id,version=version,), + "path": f"/projects/{project_id}/templates/{template_id}/versions/{version}", } - 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[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -70,14 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, - + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Template Version + """Set Selected Template Version Args: project_id (str): @@ -90,31 +69,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, -version=version, + """ - ) + kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, +def sync( + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Template Version + """Set Selected Template Version Args: project_id (str): @@ -127,26 +94,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, template_id=template_id, version=version, client=client).parsed - return sync_detailed( - project_id=project_id, -template_id=template_id, -version=version, -client=client, - - ).parsed async def asyncio_detailed( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, - + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Template Version + """Set Selected Template Version Args: project_id (str): @@ -159,31 +115,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -template_id=template_id, -version=version, + kwargs = _get_kwargs(project_id=project_id, template_id=template_id, version=version) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - template_id: str, - version: int, - *, - client: ApiClient, +async def asyncio( + project_id: str, template_id: str, version: int, *, client: ApiClient ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Set Selected Template Version + """Set Selected Template Version Args: project_id (str): @@ -196,13 +140,8 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -template_id=template_id, -version=version, -client=client, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, template_id=template_id, version=version, client=client) + ).parsed 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 23e9dea12..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.base_prompt_template_response import BasePromptTemplateResponse from ...models.http_validation_error import HTTPValidationError from ...models.update_prompt_template_request import UpdatePromptTemplateRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - *, - body: UpdatePromptTemplateRequest, -) -> dict[str, Any]: +def _get_kwargs(template_id: str, *, body: UpdatePromptTemplateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/templates/{template_id}".format(template_id=template_id,), + "path": f"/templates/{template_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = BasePromptTemplateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: + if response.status_code == 200: + return BasePromptTemplateResponse.from_dict(response.json()) + if response.status_code == 422: + 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[BasePromptTemplateResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - *, - client: ApiClient, - body: UpdatePromptTemplateRequest, - + template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Update Global Template + """Update Global Template Update a global prompt template. @@ -111,29 +91,19 @@ def sync_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -body=body, + kwargs = _get_kwargs(template_id=template_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - *, - client: ApiClient, - body: UpdatePromptTemplateRequest, +def sync( + template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Update Global Template + """Update Global Template Update a global prompt template. @@ -163,24 +133,15 @@ def sync( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - template_id=template_id, -client=client, -body=body, + return sync_detailed(template_id=template_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - template_id: str, - *, - client: ApiClient, - body: UpdatePromptTemplateRequest, - + template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest ) -> Response[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Update Global Template + """Update Global Template Update a global prompt template. @@ -210,29 +171,19 @@ async def asyncio_detailed( Returns: Response[Union[BasePromptTemplateResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, body=body) - kwargs = _get_kwargs( - template_id=template_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - *, - client: ApiClient, - body: UpdatePromptTemplateRequest, +async def asyncio( + template_id: str, *, client: ApiClient, body: UpdatePromptTemplateRequest ) -> Optional[Union[BasePromptTemplateResponse, HTTPValidationError]]: - """ Update Global Template + """Update Global Template Update a global prompt template. @@ -262,12 +213,6 @@ async def asyncio( Returns: Union[BasePromptTemplateResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, client=client, body=body)).parsed 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 a0e378608..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.collaborator_update import CollaboratorUpdate from ...models.group_collaborator import GroupCollaborator from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - group_id: str, - *, - body: CollaboratorUpdate, -) -> dict[str, Any]: +def _get_kwargs(template_id: str, group_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/templates/{template_id}/groups/{group_id}".format(template_id=template_id,group_id=group_id,), + "path": f"/templates/{template_id}/groups/{group_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - if response.status_code == : - response_200 = GroupCollaborator.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: + if response.status_code == 200: + return GroupCollaborator.from_dict(response.json()) + if response.status_code == 422: + 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[GroupCollaborator, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[GroupCollaborator, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Prompt Template Collaborator + """Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -group_id=group_id, -body=body, + kwargs = _get_kwargs(template_id=template_id, group_id=group_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +def sync( + template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Prompt Template Collaborator + """Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -137,26 +103,15 @@ def sync( Returns: Union[GroupCollaborator, HTTPValidationError] - """ - + """ - return sync_detailed( - template_id=template_id, -group_id=group_id, -client=client, -body=body, + return sync_detailed(template_id=template_id, group_id=group_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - template_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Prompt Template Collaborator + """Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[GroupCollaborator, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, group_id=group_id, body=body) - kwargs = _get_kwargs( - template_id=template_id, -group_id=group_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - group_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +async def asyncio( + template_id: str, group_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[GroupCollaborator, HTTPValidationError]]: - """ Update Group Prompt Template Collaborator + """Update Group Prompt Template Collaborator Update the sharing permissions of a group on a prompt template. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[GroupCollaborator, HTTPValidationError] - """ - - - return (await asyncio_detailed( - template_id=template_id, -group_id=group_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, group_id=group_id, client=client, body=body)).parsed 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 b9109a08e..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.collaborator_update import CollaboratorUpdate from ...models.http_validation_error import HTTPValidationError from ...models.user_collaborator import UserCollaborator -from typing import cast - - +from ...types import Response -def _get_kwargs( - template_id: str, - user_id: str, - *, - body: CollaboratorUpdate, -) -> dict[str, Any]: +def _get_kwargs(template_id: str, user_id: str, *, body: CollaboratorUpdate) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/templates/{template_id}/users/{user_id}".format(template_id=template_id,user_id=user_id,), + "path": f"/templates/{template_id}/users/{user_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - if response.status_code == : - response_200 = UserCollaborator.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, UserCollaborator]]: + if response.status_code == 200: + return UserCollaborator.from_dict(response.json()) + if response.status_code == 422: + 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, UserCollaborator]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, UserCollaborator]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - template_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Prompt Template Collaborator + """Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ - + """ - kwargs = _get_kwargs( - template_id=template_id, -user_id=user_id, -body=body, + kwargs = _get_kwargs(template_id=template_id, user_id=user_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - template_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +def sync( + template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Prompt Template Collaborator + """Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -137,26 +103,15 @@ def sync( Returns: Union[HTTPValidationError, UserCollaborator] - """ - + """ - return sync_detailed( - template_id=template_id, -user_id=user_id, -client=client, -body=body, + return sync_detailed(template_id=template_id, user_id=user_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - template_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, - + template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Response[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Prompt Template Collaborator + """Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, UserCollaborator]] - """ + """ + kwargs = _get_kwargs(template_id=template_id, user_id=user_id, body=body) - kwargs = _get_kwargs( - template_id=template_id, -user_id=user_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - template_id: str, - user_id: str, - *, - client: ApiClient, - body: CollaboratorUpdate, +async def asyncio( + template_id: str, user_id: str, *, client: ApiClient, body: CollaboratorUpdate ) -> Optional[Union[HTTPValidationError, UserCollaborator]]: - """ Update User Prompt Template Collaborator + """Update User Prompt Template Collaborator Update the sharing permissions of a user on a prompt template. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[HTTPValidationError, UserCollaborator] - """ - - - return (await asyncio_detailed( - template_id=template_id, -user_id=user_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(template_id=template_id, user_id=user_id, client=client, body=body)).parsed 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 c67203b60..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.stage_db import StageDB from ...models.stage_with_rulesets import StageWithRulesets -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: StageWithRulesets, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: StageWithRulesets) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/stages".format(project_id=project_id,), + "path": f"/projects/{project_id}/stages", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,22 +34,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: - if response.status_code == : - response_200 = StageDB.from_dict(response.json()) + if response.status_code == 200: + return StageDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -75,13 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: StageWithRulesets, - + project_id: str, *, client: ApiClient, body: StageWithRulesets ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Create Stage + """Create Stage Args: project_id (str): @@ -93,29 +69,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -body=body, + """ - ) + kwargs = _get_kwargs(project_id=project_id, body=body) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: StageWithRulesets, +def sync( + project_id: str, *, client: ApiClient, body: StageWithRulesets ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Create Stage + """Create Stage Args: project_id (str): @@ -127,24 +93,15 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ + return sync_detailed(project_id=project_id, client=client, body=body).parsed - return sync_detailed( - project_id=project_id, -client=client, -body=body, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: StageWithRulesets, - + project_id: str, *, client: ApiClient, body: StageWithRulesets ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Create Stage + """Create Stage Args: project_id (str): @@ -156,29 +113,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: StageWithRulesets, +async def asyncio( + project_id: str, *, client: ApiClient, body: StageWithRulesets ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Create Stage + """Create Stage Args: project_id (str): @@ -190,12 +137,6 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 df56595c3..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 @@ -1,63 +1,42 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.stage_db import StageDB -from ...types import UNSET, Unset -from typing import cast -from typing import cast, Union -from typing import Union - +from ...types import UNSET, Response, Unset def _get_kwargs( - project_id: str, - *, - stage_name: Union[None, Unset, str] = UNSET, - stage_id: Union[None, Unset, str] = UNSET, - + project_id: str, *, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET ) -> dict[str, Any]: headers: dict[str, Any] = {} - - - 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} - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/stages".format(project_id=project_id,), + "path": f"/projects/{project_id}/stages", "params": params, } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -65,22 +44,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: - if response.status_code == : - response_200 = StageDB.from_dict(response.json()) - - + if response.status_code == 200: + return StageDB.from_dict(response.json()) - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 422: + 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]]: @@ -98,9 +70,8 @@ def sync_detailed( client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, - ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Get Stage + """Get Stage Args: project_id (str): @@ -113,31 +84,23 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, stage_name=stage_name, stage_id=stage_id) - kwargs = _get_kwargs( - project_id=project_id, -stage_name=stage_name, -stage_id=stage_id, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) + def sync( project_id: str, *, client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, - ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Get Stage + """Get Stage Args: project_id (str): @@ -150,16 +113,10 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ + return sync_detailed(project_id=project_id, client=client, stage_name=stage_name, stage_id=stage_id).parsed - return sync_detailed( - project_id=project_id, -client=client, -stage_name=stage_name, -stage_id=stage_id, - - ).parsed async def asyncio_detailed( project_id: str, @@ -167,9 +124,8 @@ async def asyncio_detailed( client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, - ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Get Stage + """Get Stage Args: project_id (str): @@ -182,31 +138,23 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, stage_name=stage_name, stage_id=stage_id) - kwargs = _get_kwargs( - project_id=project_id, -stage_name=stage_name, -stage_id=stage_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) + async def asyncio( project_id: str, *, client: ApiClient, stage_name: Union[None, Unset, str] = UNSET, stage_id: Union[None, Unset, str] = UNSET, - ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Get Stage + """Get Stage Args: project_id (str): @@ -219,13 +167,8 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ - + """ - return (await asyncio_detailed( - project_id=project_id, -client=client, -stage_name=stage_name, -stage_id=stage_id, - - )).parsed + return ( + await asyncio_detailed(project_id=project_id, client=client, stage_name=stage_name, stage_id=stage_id) + ).parsed 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 06b9b5108..caadb4d67 100644 --- a/src/galileo/resources/api/protect/invoke_protect_invoke_post.py +++ b/src/galileo/resources/api/protect/invoke_protect_invoke_post.py @@ -71,7 +71,7 @@ def _parse_response_200(data: object) -> Union['InvokeResponse', 'ProtectRespons response_200 = _parse_response_200(response.json()) return response_200 - if response.status_code == : + if response.status_code == 422: response_422 = HTTPValidationError.from_dict(response.json()) 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 2bb1fe9a7..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 @@ -1,50 +1,34 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.stage_db import StageDB -from ...types import UNSET, Unset -from typing import cast -from typing import Union - - +from ...types import UNSET, Response, Unset -def _get_kwargs( - project_id: str, - stage_id: str, - *, - pause: Union[Unset, bool] = False, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, stage_id: str, *, pause: Union[Unset, bool] = False) -> dict[str, Any]: headers: dict[str, Any] = {} - - - params: dict[str, Any] = {} params["pause"] = pause - params = {k: v for k, v in params.items() if v is not UNSET and v is not None} - _kwargs: dict[str, Any] = { "method": RequestMethod.PUT, "return_raw_response": True, - "path": "/projects/{project_id}/stages/{stage_id}".format(project_id=project_id,stage_id=stage_id,), + "path": f"/projects/{project_id}/stages/{stage_id}", "params": params, } - headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" _kwargs["content_headers"] = headers @@ -52,22 +36,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: - if response.status_code == : - response_200 = StageDB.from_dict(response.json()) + if response.status_code == 200: + return StageDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -80,14 +57,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - stage_id: str, - *, - client: ApiClient, - pause: Union[Unset, bool] = False, - + project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Pause Stage + """Pause Stage Args: project_id (str): @@ -100,31 +72,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -stage_id=stage_id, -pause=pause, + kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, pause=pause) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - stage_id: str, - *, - client: ApiClient, - pause: Union[Unset, bool] = False, +def sync( + project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Pause Stage + """Pause Stage Args: project_id (str): @@ -137,26 +97,15 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ + return sync_detailed(project_id=project_id, stage_id=stage_id, client=client, pause=pause).parsed - return sync_detailed( - project_id=project_id, -stage_id=stage_id, -client=client, -pause=pause, - - ).parsed async def asyncio_detailed( - project_id: str, - stage_id: str, - *, - client: ApiClient, - pause: Union[Unset, bool] = False, - + project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Pause Stage + """Pause Stage Args: project_id (str): @@ -169,31 +118,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -stage_id=stage_id, -pause=pause, + kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, pause=pause) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - stage_id: str, - *, - client: ApiClient, - pause: Union[Unset, bool] = False, +async def asyncio( + project_id: str, stage_id: str, *, client: ApiClient, pause: Union[Unset, bool] = False ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Pause Stage + """Pause Stage Args: project_id (str): @@ -206,13 +143,6 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ - - - return (await asyncio_detailed( - project_id=project_id, -stage_id=stage_id, -client=client, -pause=pause, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, stage_id=stage_id, client=client, pause=pause)).parsed 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 9933208b2..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.rulesets_mixin import RulesetsMixin from ...models.stage_db import StageDB -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - stage_id: str, - *, - body: RulesetsMixin, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, stage_id: str, *, body: RulesetsMixin) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/stages/{stage_id}".format(project_id=project_id,stage_id=stage_id,), + "path": f"/projects/{project_id}/stages/{stage_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -48,22 +34,15 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, StageDB]]: - if response.status_code == : - response_200 = StageDB.from_dict(response.json()) + if response.status_code == 200: + return StageDB.from_dict(response.json()) + if response.status_code == 422: + return HTTPValidationError.from_dict(response.json()) - - return response_200 - if response.status_code == : - response_422 = 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]]: @@ -76,14 +55,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - stage_id: str, - *, - client: ApiClient, - body: RulesetsMixin, - + project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Update Stage + """Update Stage Args: project_id (str): @@ -96,31 +70,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ - - - kwargs = _get_kwargs( - project_id=project_id, -stage_id=stage_id, -body=body, + """ - ) + kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, body=body) - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - stage_id: str, - *, - client: ApiClient, - body: RulesetsMixin, +def sync( + project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Update Stage + """Update Stage Args: project_id (str): @@ -133,26 +95,15 @@ def sync( Returns: Union[HTTPValidationError, StageDB] - """ + """ + return sync_detailed(project_id=project_id, stage_id=stage_id, client=client, body=body).parsed - return sync_detailed( - project_id=project_id, -stage_id=stage_id, -client=client, -body=body, - - ).parsed async def asyncio_detailed( - project_id: str, - stage_id: str, - *, - client: ApiClient, - body: RulesetsMixin, - + project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin ) -> Response[Union[HTTPValidationError, StageDB]]: - """ Update Stage + """Update Stage Args: project_id (str): @@ -165,31 +116,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, StageDB]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -stage_id=stage_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, stage_id=stage_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - stage_id: str, - *, - client: ApiClient, - body: RulesetsMixin, +async def asyncio( + project_id: str, stage_id: str, *, client: ApiClient, body: RulesetsMixin ) -> Optional[Union[HTTPValidationError, StageDB]]: - """ Update Stage + """Update Stage Args: project_id (str): @@ -202,13 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, StageDB] - """ - - - return (await asyncio_detailed( - project_id=project_id, -stage_id=stage_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, stage_id=stage_id, client=client, body=body)).parsed 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 e0f1474a2..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_scorer_settings_response import RunScorerSettingsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - run_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, run_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/runs/{run_id}/scorer-settings".format(project_id=project_id,run_id=run_id,), + "path": f"/projects/{project_id}/runs/{run_id}/scorer-settings", } - 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, RunScorerSettingsResponse]]: - if response.status_code == : - response_200 = RunScorerSettingsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: + if response.status_code == 200: + return RunScorerSettingsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, RunScorerSettingsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - + project_id: str, run_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Get Settings + """Get Settings Args: project_id (str): @@ -87,29 +68,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, + kwargs = _get_kwargs(project_id=project_id, run_id=run_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - run_id: str, - *, - client: ApiClient, +def sync( + project_id: str, run_id: str, *, client: ApiClient ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Get Settings + """Get Settings Args: project_id (str): @@ -121,24 +92,15 @@ def sync( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -run_id=run_id, -client=client, + return sync_detailed(project_id=project_id, run_id=run_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - + project_id: str, run_id: str, *, client: ApiClient ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Get Settings + """Get Settings Args: project_id (str): @@ -150,29 +112,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, run_id=run_id) - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - run_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, run_id: str, *, client: ApiClient ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Get Settings + """Get Settings Args: project_id (str): @@ -184,12 +136,6 @@ async def asyncio( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -run_id=run_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client)).parsed 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 8da1ba9ca..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_scorer_settings_patch_request import RunScorerSettingsPatchRequest from ...models.run_scorer_settings_response import RunScorerSettingsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - run_id: str, - *, - body: RunScorerSettingsPatchRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, run_id: str, *, body: RunScorerSettingsPatchRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/projects/{project_id}/runs/{run_id}/scorer-settings".format(project_id=project_id,run_id=run_id,), + "path": f"/projects/{project_id}/runs/{run_id}/scorer-settings", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - if response.status_code == : - response_200 = RunScorerSettingsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: + if response.status_code == 200: + return RunScorerSettingsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, RunScorerSettingsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, - + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -96,31 +74,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, +def sync( + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -133,26 +99,15 @@ def sync( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -run_id=run_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, run_id=run_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, - + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -165,31 +120,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, +async def asyncio( + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -202,13 +145,6 @@ async def asyncio( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -run_id=run_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client, body=body)).parsed 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 38f3d24d9..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.run_scorer_settings_patch_request import RunScorerSettingsPatchRequest from ...models.run_scorer_settings_response import RunScorerSettingsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - run_id: str, - *, - body: RunScorerSettingsPatchRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, run_id: str, *, body: RunScorerSettingsPatchRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/runs/{run_id}/scorer-settings".format(project_id=project_id,run_id=run_id,), + "path": f"/projects/{project_id}/runs/{run_id}/scorer-settings", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - if response.status_code == : - response_200 = RunScorerSettingsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: + if response.status_code == 200: + return RunScorerSettingsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, RunScorerSettingsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, - + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -96,31 +74,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, +def sync( + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -133,26 +99,15 @@ def sync( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -run_id=run_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, run_id=run_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, - + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Response[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -165,31 +120,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, RunScorerSettingsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, run_id=run_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -run_id=run_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - run_id: str, - *, - client: ApiClient, - body: RunScorerSettingsPatchRequest, +async def asyncio( + project_id: str, run_id: str, *, client: ApiClient, body: RunScorerSettingsPatchRequest ) -> Optional[Union[HTTPValidationError, RunScorerSettingsResponse]]: - """ Upsert Scorers Config + """Upsert Scorers Config Args: project_id (str): @@ -202,13 +145,6 @@ async def asyncio( Returns: Union[HTTPValidationError, RunScorerSettingsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -run_id=run_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, run_id=run_id, client=client, body=body)).parsed 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 9b50e5aac..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.session_create_request import SessionCreateRequest from ...models.session_create_response import SessionCreateResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: SessionCreateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: SessionCreateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/sessions".format(project_id=project_id,), + "path": f"/projects/{project_id}/sessions", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: - if response.status_code == : - response_200 = SessionCreateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: + if response.status_code == 200: + return SessionCreateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, SessionCreateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: SessionCreateRequest, - + project_id: str, *, client: ApiClient, body: SessionCreateRequest ) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: - """ Create Session + """Create Session Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, SessionCreateResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: SessionCreateRequest, +def sync( + project_id: str, *, client: ApiClient, body: SessionCreateRequest ) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: - """ Create Session + """Create Session Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, SessionCreateResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: SessionCreateRequest, - + project_id: str, *, client: ApiClient, body: SessionCreateRequest ) -> Response[Union[HTTPValidationError, SessionCreateResponse]]: - """ Create Session + """Create Session Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, SessionCreateResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: SessionCreateRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: SessionCreateRequest ) -> Optional[Union[HTTPValidationError, SessionCreateResponse]]: - """ Create Session + """Create Session Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, SessionCreateResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 3019db0b2..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_delete_request import LogRecordsDeleteRequest from ...models.log_records_delete_response import LogRecordsDeleteResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsDeleteRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/sessions/delete".format(project_id=project_id,), + "path": f"/projects/{project_id}/sessions/delete", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - if response.status_code == : - response_200 = LogRecordsDeleteResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: + if response.status_code == 200: + return LogRecordsDeleteResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsDeleteResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Sessions + """Delete Sessions Delete all session records that match the provided filters. @@ -97,29 +77,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Sessions + """Delete Sessions Delete all session records that match the provided filters. @@ -135,24 +105,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Sessions + """Delete Sessions Delete all session records that match the provided filters. @@ -168,29 +129,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Sessions + """Delete Sessions Delete all session records that match the provided filters. @@ -206,12 +157,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 9c2f9dbd4..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_delete_request import LogRecordsDeleteRequest from ...models.log_records_delete_response import LogRecordsDeleteResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsDeleteRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/spans/delete".format(project_id=project_id,), + "path": f"/projects/{project_id}/spans/delete", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - if response.status_code == : - response_200 = LogRecordsDeleteResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: + if response.status_code == 200: + return LogRecordsDeleteResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsDeleteResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Spans + """Delete Spans Delete all span records that match the provided filters. @@ -97,29 +77,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Spans + """Delete Spans Delete all span records that match the provided filters. @@ -135,24 +105,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Spans + """Delete Spans Delete all span records that match the provided filters. @@ -168,29 +129,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Spans + """Delete Spans Delete all span records that match the provided filters. @@ -206,12 +157,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 2635e7c0f..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_delete_request import LogRecordsDeleteRequest from ...models.log_records_delete_response import LogRecordsDeleteResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsDeleteRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsDeleteRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/traces/delete".format(project_id=project_id,), + "path": f"/projects/{project_id}/traces/delete", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - if response.status_code == : - response_200 = LogRecordsDeleteResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: + if response.status_code == 200: + return LogRecordsDeleteResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsDeleteResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Traces + """Delete Traces Delete all trace records that match the provided filters. @@ -97,29 +77,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Traces + """Delete Traces Delete all trace records that match the provided filters. @@ -135,24 +105,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Response[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Traces + """Delete Traces Delete all trace records that match the provided filters. @@ -168,29 +129,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsDeleteResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsDeleteRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsDeleteRequest ) -> Optional[Union[HTTPValidationError, LogRecordsDeleteResponse]]: - """ Delete Traces + """Delete Traces Delete all trace records that match the provided filters. @@ -206,12 +157,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsDeleteResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 8d31867c3..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 @@ -1,42 +1,29 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_export_request import LogRecordsExportRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsExportRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsExportRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/export_records".format(project_id=project_id,), + "path": f"/projects/{project_id}/export_records", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,19 +33,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -69,33 +51,17 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ parsed=_parse_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, - ) +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 client.request(**kwargs) - return response def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsExportRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsExportRequest ) -> Response[Union[Any, HTTPValidationError]]: - """ Export Records + """Export Records Args: project_id (str): @@ -108,29 +74,19 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsExportRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsExportRequest ) -> Optional[Union[Any, HTTPValidationError]]: - """ Export Records + """Export Records Args: project_id (str): @@ -143,24 +99,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ + """ + return sync_detailed(project_id=project_id, client=client, body=body).parsed - return sync_detailed( - project_id=project_id, -client=client, -body=body, - - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsExportRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsExportRequest ) -> Response[Union[Any, HTTPValidationError]]: - """ Export Records + """Export Records Args: project_id (str): @@ -173,29 +120,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsExportRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsExportRequest ) -> Optional[Union[Any, HTTPValidationError]]: - """ Export Records + """Export Records Args: project_id (str): @@ -208,12 +145,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 0bcdf8382..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.aggregated_trace_view_request import AggregatedTraceViewRequest from ...models.aggregated_trace_view_response import AggregatedTraceViewResponse from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: AggregatedTraceViewRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: AggregatedTraceViewRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/traces/aggregated".format(project_id=project_id,), + "path": f"/projects/{project_id}/traces/aggregated", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - if response.status_code == : - response_200 = AggregatedTraceViewResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: + if response.status_code == 200: + return AggregatedTraceViewResponse.from_dict(response.json()) + if response.status_code == 422: + 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[AggregatedTraceViewResponse, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: AggregatedTraceViewRequest, - + project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest ) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """ Get Aggregated Trace View + """Get Aggregated Trace View Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[AggregatedTraceViewResponse, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: AggregatedTraceViewRequest, +def sync( + project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest ) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """ Get Aggregated Trace View + """Get Aggregated Trace View Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[AggregatedTraceViewResponse, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: AggregatedTraceViewRequest, - + project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest ) -> Response[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """ Get Aggregated Trace View + """Get Aggregated Trace View Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[AggregatedTraceViewResponse, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: AggregatedTraceViewRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: AggregatedTraceViewRequest ) -> Optional[Union[AggregatedTraceViewResponse, HTTPValidationError]]: - """ Get Aggregated Trace View + """Get Aggregated Trace View Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[AggregatedTraceViewResponse, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 ad0ca8939..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.extended_session_record_with_children import ExtendedSessionRecordWithChildren from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - session_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, session_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/sessions/{session_id}".format(project_id=project_id,session_id=session_id,), + "path": f"/projects/{project_id}/sessions/{session_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[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - if response.status_code == : - response_200 = ExtendedSessionRecordWithChildren.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: + if response.status_code == 200: + return ExtendedSessionRecordWithChildren.from_dict(response.json()) + if response.status_code == 422: + 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[ExtendedSessionRecordWithChildren, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - session_id: str, - *, - client: ApiClient, - + project_id: str, session_id: str, *, client: ApiClient ) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """ Get Session + """Get Session Args: project_id (str): @@ -87,29 +68,19 @@ def sync_detailed( Returns: Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -session_id=session_id, + kwargs = _get_kwargs(project_id=project_id, session_id=session_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - session_id: str, - *, - client: ApiClient, +def sync( + project_id: str, session_id: str, *, client: ApiClient ) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """ Get Session + """Get Session Args: project_id (str): @@ -121,24 +92,15 @@ def sync( Returns: Union[ExtendedSessionRecordWithChildren, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -session_id=session_id, -client=client, + return sync_detailed(project_id=project_id, session_id=session_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - session_id: str, - *, - client: ApiClient, - + project_id: str, session_id: str, *, client: ApiClient ) -> Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """ Get Session + """Get Session Args: project_id (str): @@ -150,29 +112,19 @@ async def asyncio_detailed( Returns: Response[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, session_id=session_id) - kwargs = _get_kwargs( - project_id=project_id, -session_id=session_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - session_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, session_id: str, *, client: ApiClient ) -> Optional[Union[ExtendedSessionRecordWithChildren, HTTPValidationError]]: - """ Get Session + """Get Session Args: project_id (str): @@ -184,12 +136,6 @@ async def asyncio( Returns: Union[ExtendedSessionRecordWithChildren, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -session_id=session_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, session_id=session_id, client=client)).parsed 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 85fb7b6c2..4919a36fc 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 @@ -99,7 +99,7 @@ def _parse_response_200(data: object) -> Union['ExtendedAgentSpanRecordWithChild response_200 = _parse_response_200(response.json()) return response_200 - if response.status_code == : + if response.status_code == 422: response_422 = HTTPValidationError.from_dict(response.json()) 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 3b7afd3dd..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 @@ -1,65 +1,50 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.extended_trace_record_with_children import ExtendedTraceRecordWithChildren from ...models.http_validation_error import HTTPValidationError -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - trace_id: str, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, trace_id: str) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.GET, "return_raw_response": True, - "path": "/projects/{project_id}/traces/{trace_id}".format(project_id=project_id,trace_id=trace_id,), + "path": f"/projects/{project_id}/traces/{trace_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[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - if response.status_code == : - response_200 = ExtendedTraceRecordWithChildren.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: + if response.status_code == 200: + return ExtendedTraceRecordWithChildren.from_dict(response.json()) + if response.status_code == 422: + 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[ExtendedTraceRecordWithChildren, HTTPValidationError]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -69,13 +54,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - trace_id: str, - *, - client: ApiClient, - + project_id: str, trace_id: str, *, client: ApiClient ) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """ Get Trace + """Get Trace Args: project_id (str): @@ -87,29 +68,19 @@ def sync_detailed( Returns: Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -trace_id=trace_id, + kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - trace_id: str, - *, - client: ApiClient, +def sync( + project_id: str, trace_id: str, *, client: ApiClient ) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """ Get Trace + """Get Trace Args: project_id (str): @@ -121,24 +92,15 @@ def sync( Returns: Union[ExtendedTraceRecordWithChildren, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -trace_id=trace_id, -client=client, + return sync_detailed(project_id=project_id, trace_id=trace_id, client=client).parsed - ).parsed async def asyncio_detailed( - project_id: str, - trace_id: str, - *, - client: ApiClient, - + project_id: str, trace_id: str, *, client: ApiClient ) -> Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """ Get Trace + """Get Trace Args: project_id (str): @@ -150,29 +112,19 @@ async def asyncio_detailed( Returns: Response[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id) - kwargs = _get_kwargs( - project_id=project_id, -trace_id=trace_id, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - trace_id: str, - *, - client: ApiClient, +async def asyncio( + project_id: str, trace_id: str, *, client: ApiClient ) -> Optional[Union[ExtendedTraceRecordWithChildren, HTTPValidationError]]: - """ Get Trace + """Get Trace Args: project_id (str): @@ -184,12 +136,6 @@ async def asyncio( Returns: Union[ExtendedTraceRecordWithChildren, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -trace_id=trace_id, -client=client, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, trace_id=trace_id, client=client)).parsed 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 4e3235e10..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_spans_ingest_request import LogSpansIngestRequest from ...models.log_spans_ingest_response import LogSpansIngestResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogSpansIngestRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogSpansIngestRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/spans".format(project_id=project_id,), + "path": f"/projects/{project_id}/spans", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: - if response.status_code == : - response_200 = LogSpansIngestResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: + if response.status_code == 200: + return LogSpansIngestResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogSpansIngestResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogSpansIngestRequest, - + project_id: str, *, client: ApiClient, body: LogSpansIngestRequest ) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: - """ Log Spans + """Log Spans Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogSpansIngestResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogSpansIngestRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogSpansIngestRequest ) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: - """ Log Spans + """Log Spans Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogSpansIngestResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogSpansIngestRequest, - + project_id: str, *, client: ApiClient, body: LogSpansIngestRequest ) -> Response[Union[HTTPValidationError, LogSpansIngestResponse]]: - """ Log Spans + """Log Spans Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogSpansIngestResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogSpansIngestRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogSpansIngestRequest ) -> Optional[Union[HTTPValidationError, LogSpansIngestResponse]]: - """ Log Spans + """Log Spans Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogSpansIngestResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 bbc44989e..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_traces_ingest_request import LogTracesIngestRequest from ...models.log_traces_ingest_response import LogTracesIngestResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogTracesIngestRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogTracesIngestRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/traces".format(project_id=project_id,), + "path": f"/projects/{project_id}/traces", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: - if response.status_code == : - response_200 = LogTracesIngestResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: + if response.status_code == 200: + return LogTracesIngestResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogTracesIngestResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogTracesIngestRequest, - + project_id: str, *, client: ApiClient, body: LogTracesIngestRequest ) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: - """ Log Traces + """Log Traces Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogTracesIngestResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogTracesIngestRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogTracesIngestRequest ) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: - """ Log Traces + """Log Traces Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogTracesIngestResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogTracesIngestRequest, - + project_id: str, *, client: ApiClient, body: LogTracesIngestRequest ) -> Response[Union[HTTPValidationError, LogTracesIngestResponse]]: - """ Log Traces + """Log Traces Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogTracesIngestResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogTracesIngestRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogTracesIngestRequest ) -> Optional[Union[HTTPValidationError, LogTracesIngestResponse]]: - """ Log Traces + """Log Traces Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogTracesIngestResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 e7d601d14..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse from ...models.metrics_testing_available_columns_request import MetricsTestingAvailableColumnsRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: MetricsTestingAvailableColumnsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: MetricsTestingAvailableColumnsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/metrics-testing/available_columns".format(project_id=project_id,), + "path": f"/projects/{project_id}/metrics-testing/available_columns", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == : - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == 200: + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsAvailableColumnsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: MetricsTestingAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Metrics Testing Available Columns + """Metrics Testing Available Columns Args: project_id (str): @@ -94,29 +74,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: MetricsTestingAvailableColumnsRequest, +def sync( + project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Metrics Testing Available Columns + """Metrics Testing Available Columns Args: project_id (str): @@ -129,24 +99,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: MetricsTestingAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Metrics Testing Available Columns + """Metrics Testing Available Columns Args: project_id (str): @@ -159,29 +120,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: MetricsTestingAvailableColumnsRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: MetricsTestingAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Metrics Testing Available Columns + """Metrics Testing Available Columns Args: project_id (str): @@ -194,12 +145,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 255c0d49b..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_metrics_query_request import LogRecordsMetricsQueryRequest from ...models.log_records_metrics_response import LogRecordsMetricsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsMetricsQueryRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsMetricsQueryRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/metrics/search".format(project_id=project_id,), + "path": f"/projects/{project_id}/metrics/search", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - if response.status_code == : - response_200 = LogRecordsMetricsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: + if response.status_code == 200: + return LogRecordsMetricsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsMetricsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsMetricsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """ Query Metrics + """Query Metrics Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsMetricsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsMetricsQueryRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """ Query Metrics + """Query Metrics Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsMetricsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsMetricsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """ Query Metrics + """Query Metrics Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsMetricsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsMetricsQueryRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsMetricsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsMetricsResponse]]: - """ Query Metrics + """Query Metrics Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsMetricsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 841852aaa..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_query_request import LogRecordsQueryRequest from ...models.log_records_query_response import LogRecordsQueryResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsQueryRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/sessions/search".format(project_id=project_id,), + "path": f"/projects/{project_id}/sessions/search", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - if response.status_code == : - response_200 = LogRecordsQueryResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: + if response.status_code == 200: + return LogRecordsQueryResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsQueryResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Sessions + """Query Sessions Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Sessions + """Query Sessions Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Sessions + """Query Sessions Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Sessions + """Query Sessions Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 5c1b382f5..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_query_request import LogRecordsQueryRequest from ...models.log_records_query_response import LogRecordsQueryResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsQueryRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/spans/search".format(project_id=project_id,), + "path": f"/projects/{project_id}/spans/search", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - if response.status_code == : - response_200 = LogRecordsQueryResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: + if response.status_code == 200: + return LogRecordsQueryResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsQueryResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Spans + """Query Spans Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Spans + """Query Spans Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Spans + """Query Spans Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Spans + """Query Spans Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 c7cf74ce8..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_query_request import LogRecordsQueryRequest from ...models.log_records_query_response import LogRecordsQueryResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsQueryRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsQueryRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/traces/search".format(project_id=project_id,), + "path": f"/projects/{project_id}/traces/search", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - if response.status_code == : - response_200 = LogRecordsQueryResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: + if response.status_code == 200: + return LogRecordsQueryResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsQueryResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Traces + """Query Traces Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Traces + """Query Traces Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Response[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Traces + """Query Traces Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsQueryResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsQueryRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsQueryRequest ) -> Optional[Union[HTTPValidationError, LogRecordsQueryResponse]]: - """ Query Traces + """Query Traces Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsQueryResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 d167180c2..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 @@ -1,42 +1,29 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.recompute_log_records_metrics_request import RecomputeLogRecordsMetricsRequest -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: RecomputeLogRecordsMetricsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: RecomputeLogRecordsMetricsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/recompute-metrics".format(project_id=project_id,), + "path": f"/projects/{project_id}/recompute-metrics", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,19 +33,14 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]: - if response.status_code == : - response_200 = response.json() - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) + if response.status_code == 200: + return response.json() + if response.status_code == 422: + 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]]: @@ -71,13 +53,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: RecomputeLogRecordsMetricsRequest, - + project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest ) -> Response[Union[Any, HTTPValidationError]]: - """ Recompute Metrics + """Recompute Metrics Args: project_id (str): @@ -92,29 +70,19 @@ def sync_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: RecomputeLogRecordsMetricsRequest, +def sync( + project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest ) -> Optional[Union[Any, HTTPValidationError]]: - """ Recompute Metrics + """Recompute Metrics Args: project_id (str): @@ -129,24 +97,15 @@ def sync( Returns: Union[Any, HTTPValidationError] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: RecomputeLogRecordsMetricsRequest, - + project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest ) -> Response[Union[Any, HTTPValidationError]]: - """ Recompute Metrics + """Recompute Metrics Args: project_id (str): @@ -161,29 +120,19 @@ async def asyncio_detailed( Returns: Response[Union[Any, HTTPValidationError]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: RecomputeLogRecordsMetricsRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: RecomputeLogRecordsMetricsRequest ) -> Optional[Union[Any, HTTPValidationError]]: - """ Recompute Metrics + """Recompute Metrics Args: project_id (str): @@ -198,12 +147,6 @@ async def asyncio( Returns: Union[Any, HTTPValidationError] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 a3084e251..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_available_columns_request import LogRecordsAvailableColumnsRequest from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsAvailableColumnsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/sessions/available_columns".format(project_id=project_id,), + "path": f"/projects/{project_id}/sessions/available_columns", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == : - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == 200: + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsAvailableColumnsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Sessions Available Columns + """Sessions Available Columns Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Sessions Available Columns + """Sessions Available Columns Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Sessions Available Columns + """Sessions Available Columns Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Sessions Available Columns + """Sessions Available Columns Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 2ab4be41c..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_available_columns_request import LogRecordsAvailableColumnsRequest from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsAvailableColumnsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/spans/available_columns".format(project_id=project_id,), + "path": f"/projects/{project_id}/spans/available_columns", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == : - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == 200: + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsAvailableColumnsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Spans Available Columns + """Spans Available Columns Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Spans Available Columns + """Spans Available Columns Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Spans Available Columns + """Spans Available Columns Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Spans Available Columns + """Spans Available Columns Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 8eb9a6e9a..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 @@ -1,43 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_records_available_columns_request import LogRecordsAvailableColumnsRequest from ...models.log_records_available_columns_response import LogRecordsAvailableColumnsResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - *, - body: LogRecordsAvailableColumnsRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, *, body: LogRecordsAvailableColumnsRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.POST, "return_raw_response": True, - "path": "/projects/{project_id}/traces/available_columns".format(project_id=project_id,), + "path": f"/projects/{project_id}/traces/available_columns", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -46,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - if response.status_code == : - response_200 = LogRecordsAvailableColumnsResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: + if response.status_code == 200: + return LogRecordsAvailableColumnsResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogRecordsAvailableColumnsResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -75,13 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Traces Available Columns + """Traces Available Columns Args: project_id (str): @@ -93,29 +73,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, +def sync( + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Traces Available Columns + """Traces Available Columns Args: project_id (str): @@ -127,24 +97,15 @@ def sync( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, - + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Traces Available Columns + """Traces Available Columns Args: project_id (str): @@ -156,29 +117,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - *, - client: ApiClient, - body: LogRecordsAvailableColumnsRequest, +async def asyncio( + project_id: str, *, client: ApiClient, body: LogRecordsAvailableColumnsRequest ) -> Optional[Union[HTTPValidationError, LogRecordsAvailableColumnsResponse]]: - """ Traces Available Columns + """Traces Available Columns Args: project_id (str): @@ -190,12 +141,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogRecordsAvailableColumnsResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, client=client, body=body)).parsed 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 b7f9130d4..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_span_update_request import LogSpanUpdateRequest from ...models.log_span_update_response import LogSpanUpdateResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - span_id: str, - *, - body: LogSpanUpdateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, span_id: str, *, body: LogSpanUpdateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/projects/{project_id}/spans/{span_id}".format(project_id=project_id,span_id=span_id,), + "path": f"/projects/{project_id}/spans/{span_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: - if response.status_code == : - response_200 = LogSpanUpdateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: + if response.status_code == 200: + return LogSpanUpdateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogSpanUpdateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - span_id: str, - *, - client: ApiClient, - body: LogSpanUpdateRequest, - + project_id: str, span_id: str, *, client: ApiClient, body: LogSpanUpdateRequest ) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """ Update Span + """Update Span Update a span with the given ID. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogSpanUpdateResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -span_id=span_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, span_id=span_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - span_id: str, - *, - client: ApiClient, - body: LogSpanUpdateRequest, +def sync( + project_id: str, span_id: str, *, client: ApiClient, body: LogSpanUpdateRequest ) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """ Update Span + """Update Span Update a span with the given ID. @@ -137,26 +103,15 @@ def sync( Returns: Union[HTTPValidationError, LogSpanUpdateResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -span_id=span_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, span_id=span_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - span_id: str, - *, - client: ApiClient, - body: LogSpanUpdateRequest, - + project_id: str, span_id: str, *, client: ApiClient, body: LogSpanUpdateRequest ) -> Response[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """ Update Span + """Update Span Update a span with the given ID. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogSpanUpdateResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, span_id=span_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -span_id=span_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - span_id: str, - *, - client: ApiClient, - body: LogSpanUpdateRequest, +async def asyncio( + project_id: str, span_id: str, *, client: ApiClient, body: LogSpanUpdateRequest ) -> Optional[Union[HTTPValidationError, LogSpanUpdateResponse]]: - """ Update Span + """Update Span Update a span with the given ID. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogSpanUpdateResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -span_id=span_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, span_id=span_id, client=client, body=body)).parsed 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 ad60c958e..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 @@ -1,44 +1,30 @@ from http import HTTPStatus -from typing import Any, Optional, Union, cast +from typing import Any, Optional, Union 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 ... import errors from ...models.http_validation_error import HTTPValidationError from ...models.log_trace_update_request import LogTraceUpdateRequest from ...models.log_trace_update_response import LogTraceUpdateResponse -from typing import cast - - +from ...types import Response -def _get_kwargs( - project_id: str, - trace_id: str, - *, - body: LogTraceUpdateRequest, -) -> dict[str, Any]: +def _get_kwargs(project_id: str, trace_id: str, *, body: LogTraceUpdateRequest) -> dict[str, Any]: headers: dict[str, Any] = {} - - - - - _kwargs: dict[str, Any] = { "method": RequestMethod.PATCH, "return_raw_response": True, - "path": "/projects/{project_id}/traces/{trace_id}".format(project_id=project_id,trace_id=trace_id,), + "path": f"/projects/{project_id}/traces/{trace_id}", } _kwargs["json"] = body.to_dict() - headers["Content-Type"] = "application/json" headers["X-Galileo-SDK"] = f"galileo-python/{get_package_version()}" @@ -47,26 +33,23 @@ def _get_kwargs( return _kwargs -def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: - if response.status_code == : - response_200 = LogTraceUpdateResponse.from_dict(response.json()) - - - - return response_200 - if response.status_code == : - response_422 = HTTPValidationError.from_dict(response.json()) - +def _parse_response( + *, client: ApiClient, response: httpx.Response +) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: + if response.status_code == 200: + return LogTraceUpdateResponse.from_dict(response.json()) + if response.status_code == 422: + 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, LogTraceUpdateResponse]]: +def _build_response( + *, client: ApiClient, response: httpx.Response +) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: return Response( status_code=HTTPStatus(response.status_code), content=response.content, @@ -76,14 +59,9 @@ def _build_response(*, client: ApiClient, response: httpx.Response) -> Response[ def sync_detailed( - project_id: str, - trace_id: str, - *, - client: ApiClient, - body: LogTraceUpdateRequest, - + project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest ) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """ Update Trace + """Update Trace Update a trace with the given ID. @@ -98,31 +76,19 @@ def sync_detailed( Returns: Response[Union[HTTPValidationError, LogTraceUpdateResponse]] - """ - + """ - kwargs = _get_kwargs( - project_id=project_id, -trace_id=trace_id, -body=body, + kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id, body=body) - ) - - response = client.request( - **kwargs, - ) + response = client.request(**kwargs) return _build_response(client=client, response=response) -def sync( - project_id: str, - trace_id: str, - *, - client: ApiClient, - body: LogTraceUpdateRequest, +def sync( + project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest ) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """ Update Trace + """Update Trace Update a trace with the given ID. @@ -137,26 +103,15 @@ def sync( Returns: Union[HTTPValidationError, LogTraceUpdateResponse] - """ - + """ - return sync_detailed( - project_id=project_id, -trace_id=trace_id, -client=client, -body=body, + return sync_detailed(project_id=project_id, trace_id=trace_id, client=client, body=body).parsed - ).parsed async def asyncio_detailed( - project_id: str, - trace_id: str, - *, - client: ApiClient, - body: LogTraceUpdateRequest, - + project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest ) -> Response[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """ Update Trace + """Update Trace Update a trace with the given ID. @@ -171,31 +126,19 @@ async def asyncio_detailed( Returns: Response[Union[HTTPValidationError, LogTraceUpdateResponse]] - """ + """ + kwargs = _get_kwargs(project_id=project_id, trace_id=trace_id, body=body) - kwargs = _get_kwargs( - project_id=project_id, -trace_id=trace_id, -body=body, - - ) - - response = await client.arequest( - **kwargs - ) + response = await client.arequest(**kwargs) return _build_response(client=client, response=response) -async def asyncio( - project_id: str, - trace_id: str, - *, - client: ApiClient, - body: LogTraceUpdateRequest, +async def asyncio( + project_id: str, trace_id: str, *, client: ApiClient, body: LogTraceUpdateRequest ) -> Optional[Union[HTTPValidationError, LogTraceUpdateResponse]]: - """ Update Trace + """Update Trace Update a trace with the given ID. @@ -210,13 +153,6 @@ async def asyncio( Returns: Union[HTTPValidationError, LogTraceUpdateResponse] - """ - - - return (await asyncio_detailed( - project_id=project_id, -trace_id=trace_id, -client=client, -body=body, + """ - )).parsed + return (await asyncio_detailed(project_id=project_id, trace_id=trace_id, client=client, body=body)).parsed From 9fe1beeb6d54f3269ca881936ff6d3f43271559f Mon Sep 17 00:00:00 2001 From: Nachiket Paranjape Date: Wed, 24 Sep 2025 17:48:14 -0700 Subject: [PATCH 3/3] patch fixer --- scripts/patch_status_codes.py | 51 ++++++++++++------- ..._project_id_runs_run_id_jobs_latest_get.py | 2 +- .../api/protect/invoke_protect_invoke_post.py | 2 +- ...n_projects_project_id_spans_span_id_get.py | 2 +- 4 files changed, 35 insertions(+), 22 deletions(-) diff --git a/scripts/patch_status_codes.py b/scripts/patch_status_codes.py index fff2a371e..db5d21027 100755 --- a/scripts/patch_status_codes.py +++ b/scripts/patch_status_codes.py @@ -28,24 +28,37 @@ def fix_status_codes_in_file(file_path: Path) -> bool: try: content = file_path.read_text(encoding="utf-8") original_content = content - - # Pattern to match: 'if response.status_code == :' followed by 'response_XXX' - # This captures the status code from the response variable name - pattern = r'if response\.status_code == :\s*\n(\s*)response_(\d+)' - - def replace_status_code(match): - indent = match.group(1) - status_code = match.group(2) - return f'if response.status_code == {status_code}:\n{indent}response_{status_code}' - - content = re.sub(pattern, replace_status_code, content, flags=re.MULTILINE) - + + # 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 @@ -56,15 +69,15 @@ def fix_status_codes_in_directory(directory: Path) -> int: 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 @@ -72,10 +85,10 @@ 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 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 a867c9824..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 @@ -42,7 +42,7 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['JobDB', None]]]: - if response.status_code == : + if response.status_code == 200: def _parse_response_200(data: object) -> Union['JobDB', None]: if data is None: return data 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 caadb4d67..8a0f354f4 100644 --- a/src/galileo/resources/api/protect/invoke_protect_invoke_post.py +++ b/src/galileo/resources/api/protect/invoke_protect_invoke_post.py @@ -48,7 +48,7 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['InvokeResponse', 'ProtectResponse']]]: - if response.status_code == : + if response.status_code == 200: def _parse_response_200(data: object) -> Union['InvokeResponse', 'ProtectResponse']: try: if not isinstance(data, dict): 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 4919a36fc..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 @@ -46,7 +46,7 @@ def _get_kwargs( def _parse_response(*, client: ApiClient, response: httpx.Response) -> Optional[Union[HTTPValidationError, Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']]]: - if response.status_code == : + if response.status_code == 200: def _parse_response_200(data: object) -> Union['ExtendedAgentSpanRecordWithChildren', 'ExtendedLlmSpanRecord', 'ExtendedRetrieverSpanRecordWithChildren', 'ExtendedToolSpanRecordWithChildren', 'ExtendedWorkflowSpanRecordWithChildren']: try: if not isinstance(data, dict):