aiqtoolkit 1.2.0rc4__py3-none-any.whl → 1.2rc9__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of aiqtoolkit might be problematic. Click here for more details.
- aiqtoolkit-1.2rc9.dist-info/METADATA +29 -0
- aiqtoolkit-1.2rc9.dist-info/RECORD +4 -0
- aiqtoolkit-1.2rc9.dist-info/top_level.txt +1 -0
- aiq/agent/__init__.py +0 -0
- aiq/agent/base.py +0 -239
- aiq/agent/dual_node.py +0 -67
- aiq/agent/react_agent/__init__.py +0 -0
- aiq/agent/react_agent/agent.py +0 -355
- aiq/agent/react_agent/output_parser.py +0 -104
- aiq/agent/react_agent/prompt.py +0 -41
- aiq/agent/react_agent/register.py +0 -149
- aiq/agent/reasoning_agent/__init__.py +0 -0
- aiq/agent/reasoning_agent/reasoning_agent.py +0 -225
- aiq/agent/register.py +0 -23
- aiq/agent/rewoo_agent/__init__.py +0 -0
- aiq/agent/rewoo_agent/agent.py +0 -411
- aiq/agent/rewoo_agent/prompt.py +0 -108
- aiq/agent/rewoo_agent/register.py +0 -158
- aiq/agent/tool_calling_agent/__init__.py +0 -0
- aiq/agent/tool_calling_agent/agent.py +0 -119
- aiq/agent/tool_calling_agent/register.py +0 -106
- aiq/authentication/__init__.py +0 -14
- aiq/authentication/api_key/__init__.py +0 -14
- aiq/authentication/api_key/api_key_auth_provider.py +0 -96
- aiq/authentication/api_key/api_key_auth_provider_config.py +0 -124
- aiq/authentication/api_key/register.py +0 -26
- aiq/authentication/exceptions/__init__.py +0 -14
- aiq/authentication/exceptions/api_key_exceptions.py +0 -38
- aiq/authentication/exceptions/auth_code_grant_exceptions.py +0 -86
- aiq/authentication/exceptions/call_back_exceptions.py +0 -38
- aiq/authentication/exceptions/request_exceptions.py +0 -54
- aiq/authentication/http_basic_auth/__init__.py +0 -0
- aiq/authentication/http_basic_auth/http_basic_auth_provider.py +0 -81
- aiq/authentication/http_basic_auth/register.py +0 -30
- aiq/authentication/interfaces.py +0 -93
- aiq/authentication/oauth2/__init__.py +0 -14
- aiq/authentication/oauth2/oauth2_auth_code_flow_provider.py +0 -107
- aiq/authentication/oauth2/oauth2_auth_code_flow_provider_config.py +0 -39
- aiq/authentication/oauth2/register.py +0 -25
- aiq/authentication/register.py +0 -21
- aiq/builder/__init__.py +0 -0
- aiq/builder/builder.py +0 -285
- aiq/builder/component_utils.py +0 -316
- aiq/builder/context.py +0 -264
- aiq/builder/embedder.py +0 -24
- aiq/builder/eval_builder.py +0 -161
- aiq/builder/evaluator.py +0 -29
- aiq/builder/framework_enum.py +0 -24
- aiq/builder/front_end.py +0 -73
- aiq/builder/function.py +0 -344
- aiq/builder/function_base.py +0 -380
- aiq/builder/function_info.py +0 -627
- aiq/builder/intermediate_step_manager.py +0 -174
- aiq/builder/llm.py +0 -25
- aiq/builder/retriever.py +0 -25
- aiq/builder/user_interaction_manager.py +0 -74
- aiq/builder/workflow.py +0 -148
- aiq/builder/workflow_builder.py +0 -1117
- aiq/cli/__init__.py +0 -14
- aiq/cli/cli_utils/__init__.py +0 -0
- aiq/cli/cli_utils/config_override.py +0 -231
- aiq/cli/cli_utils/validation.py +0 -37
- aiq/cli/commands/__init__.py +0 -0
- aiq/cli/commands/configure/__init__.py +0 -0
- aiq/cli/commands/configure/channel/__init__.py +0 -0
- aiq/cli/commands/configure/channel/add.py +0 -28
- aiq/cli/commands/configure/channel/channel.py +0 -36
- aiq/cli/commands/configure/channel/remove.py +0 -30
- aiq/cli/commands/configure/channel/update.py +0 -30
- aiq/cli/commands/configure/configure.py +0 -33
- aiq/cli/commands/evaluate.py +0 -139
- aiq/cli/commands/info/__init__.py +0 -14
- aiq/cli/commands/info/info.py +0 -39
- aiq/cli/commands/info/list_channels.py +0 -32
- aiq/cli/commands/info/list_components.py +0 -129
- aiq/cli/commands/info/list_mcp.py +0 -213
- aiq/cli/commands/registry/__init__.py +0 -14
- aiq/cli/commands/registry/publish.py +0 -88
- aiq/cli/commands/registry/pull.py +0 -118
- aiq/cli/commands/registry/registry.py +0 -38
- aiq/cli/commands/registry/remove.py +0 -108
- aiq/cli/commands/registry/search.py +0 -155
- aiq/cli/commands/sizing/__init__.py +0 -14
- aiq/cli/commands/sizing/calc.py +0 -297
- aiq/cli/commands/sizing/sizing.py +0 -27
- aiq/cli/commands/start.py +0 -246
- aiq/cli/commands/uninstall.py +0 -81
- aiq/cli/commands/validate.py +0 -47
- aiq/cli/commands/workflow/__init__.py +0 -14
- aiq/cli/commands/workflow/templates/__init__.py.j2 +0 -0
- aiq/cli/commands/workflow/templates/config.yml.j2 +0 -16
- aiq/cli/commands/workflow/templates/pyproject.toml.j2 +0 -22
- aiq/cli/commands/workflow/templates/register.py.j2 +0 -5
- aiq/cli/commands/workflow/templates/workflow.py.j2 +0 -36
- aiq/cli/commands/workflow/workflow.py +0 -37
- aiq/cli/commands/workflow/workflow_commands.py +0 -313
- aiq/cli/entrypoint.py +0 -135
- aiq/cli/main.py +0 -44
- aiq/cli/register_workflow.py +0 -488
- aiq/cli/type_registry.py +0 -1000
- aiq/data_models/__init__.py +0 -14
- aiq/data_models/api_server.py +0 -694
- aiq/data_models/authentication.py +0 -231
- aiq/data_models/common.py +0 -171
- aiq/data_models/component.py +0 -54
- aiq/data_models/component_ref.py +0 -168
- aiq/data_models/config.py +0 -406
- aiq/data_models/dataset_handler.py +0 -123
- aiq/data_models/discovery_metadata.py +0 -335
- aiq/data_models/embedder.py +0 -27
- aiq/data_models/evaluate.py +0 -127
- aiq/data_models/evaluator.py +0 -26
- aiq/data_models/front_end.py +0 -26
- aiq/data_models/function.py +0 -30
- aiq/data_models/function_dependencies.py +0 -72
- aiq/data_models/interactive.py +0 -246
- aiq/data_models/intermediate_step.py +0 -302
- aiq/data_models/invocation_node.py +0 -38
- aiq/data_models/its_strategy.py +0 -30
- aiq/data_models/llm.py +0 -27
- aiq/data_models/logging.py +0 -26
- aiq/data_models/memory.py +0 -27
- aiq/data_models/object_store.py +0 -44
- aiq/data_models/profiler.py +0 -54
- aiq/data_models/registry_handler.py +0 -26
- aiq/data_models/retriever.py +0 -30
- aiq/data_models/retry_mixin.py +0 -35
- aiq/data_models/span.py +0 -187
- aiq/data_models/step_adaptor.py +0 -64
- aiq/data_models/streaming.py +0 -33
- aiq/data_models/swe_bench_model.py +0 -54
- aiq/data_models/telemetry_exporter.py +0 -26
- aiq/embedder/__init__.py +0 -0
- aiq/embedder/langchain_client.py +0 -41
- aiq/embedder/nim_embedder.py +0 -59
- aiq/embedder/openai_embedder.py +0 -43
- aiq/embedder/register.py +0 -24
- aiq/eval/__init__.py +0 -14
- aiq/eval/config.py +0 -60
- aiq/eval/dataset_handler/__init__.py +0 -0
- aiq/eval/dataset_handler/dataset_downloader.py +0 -106
- aiq/eval/dataset_handler/dataset_filter.py +0 -52
- aiq/eval/dataset_handler/dataset_handler.py +0 -254
- aiq/eval/evaluate.py +0 -506
- aiq/eval/evaluator/__init__.py +0 -14
- aiq/eval/evaluator/base_evaluator.py +0 -73
- aiq/eval/evaluator/evaluator_model.py +0 -45
- aiq/eval/intermediate_step_adapter.py +0 -99
- aiq/eval/rag_evaluator/__init__.py +0 -0
- aiq/eval/rag_evaluator/evaluate.py +0 -178
- aiq/eval/rag_evaluator/register.py +0 -143
- aiq/eval/register.py +0 -23
- aiq/eval/remote_workflow.py +0 -133
- aiq/eval/runners/__init__.py +0 -14
- aiq/eval/runners/config.py +0 -39
- aiq/eval/runners/multi_eval_runner.py +0 -54
- aiq/eval/runtime_event_subscriber.py +0 -52
- aiq/eval/swe_bench_evaluator/__init__.py +0 -0
- aiq/eval/swe_bench_evaluator/evaluate.py +0 -215
- aiq/eval/swe_bench_evaluator/register.py +0 -36
- aiq/eval/trajectory_evaluator/__init__.py +0 -0
- aiq/eval/trajectory_evaluator/evaluate.py +0 -75
- aiq/eval/trajectory_evaluator/register.py +0 -40
- aiq/eval/tunable_rag_evaluator/__init__.py +0 -0
- aiq/eval/tunable_rag_evaluator/evaluate.py +0 -245
- aiq/eval/tunable_rag_evaluator/register.py +0 -52
- aiq/eval/usage_stats.py +0 -41
- aiq/eval/utils/__init__.py +0 -0
- aiq/eval/utils/output_uploader.py +0 -140
- aiq/eval/utils/tqdm_position_registry.py +0 -40
- aiq/eval/utils/weave_eval.py +0 -184
- aiq/experimental/__init__.py +0 -0
- aiq/experimental/decorators/__init__.py +0 -0
- aiq/experimental/decorators/experimental_warning_decorator.py +0 -130
- aiq/experimental/inference_time_scaling/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/editing/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/editing/iterative_plan_refinement_editor.py +0 -147
- aiq/experimental/inference_time_scaling/editing/llm_as_a_judge_editor.py +0 -204
- aiq/experimental/inference_time_scaling/editing/motivation_aware_summarization.py +0 -107
- aiq/experimental/inference_time_scaling/functions/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/functions/execute_score_select_function.py +0 -105
- aiq/experimental/inference_time_scaling/functions/its_tool_orchestration_function.py +0 -205
- aiq/experimental/inference_time_scaling/functions/its_tool_wrapper_function.py +0 -146
- aiq/experimental/inference_time_scaling/functions/plan_select_execute_function.py +0 -224
- aiq/experimental/inference_time_scaling/models/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/models/editor_config.py +0 -132
- aiq/experimental/inference_time_scaling/models/its_item.py +0 -48
- aiq/experimental/inference_time_scaling/models/scoring_config.py +0 -112
- aiq/experimental/inference_time_scaling/models/search_config.py +0 -120
- aiq/experimental/inference_time_scaling/models/selection_config.py +0 -154
- aiq/experimental/inference_time_scaling/models/stage_enums.py +0 -43
- aiq/experimental/inference_time_scaling/models/strategy_base.py +0 -66
- aiq/experimental/inference_time_scaling/models/tool_use_config.py +0 -41
- aiq/experimental/inference_time_scaling/register.py +0 -36
- aiq/experimental/inference_time_scaling/scoring/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/scoring/llm_based_agent_scorer.py +0 -168
- aiq/experimental/inference_time_scaling/scoring/llm_based_plan_scorer.py +0 -168
- aiq/experimental/inference_time_scaling/scoring/motivation_aware_scorer.py +0 -111
- aiq/experimental/inference_time_scaling/search/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/search/multi_llm_planner.py +0 -128
- aiq/experimental/inference_time_scaling/search/multi_query_retrieval_search.py +0 -122
- aiq/experimental/inference_time_scaling/search/single_shot_multi_plan_planner.py +0 -128
- aiq/experimental/inference_time_scaling/selection/__init__.py +0 -0
- aiq/experimental/inference_time_scaling/selection/best_of_n_selector.py +0 -63
- aiq/experimental/inference_time_scaling/selection/llm_based_agent_output_selector.py +0 -131
- aiq/experimental/inference_time_scaling/selection/llm_based_output_merging_selector.py +0 -159
- aiq/experimental/inference_time_scaling/selection/llm_based_plan_selector.py +0 -128
- aiq/experimental/inference_time_scaling/selection/threshold_selector.py +0 -58
- aiq/front_ends/__init__.py +0 -14
- aiq/front_ends/console/__init__.py +0 -14
- aiq/front_ends/console/authentication_flow_handler.py +0 -233
- aiq/front_ends/console/console_front_end_config.py +0 -32
- aiq/front_ends/console/console_front_end_plugin.py +0 -96
- aiq/front_ends/console/register.py +0 -25
- aiq/front_ends/cron/__init__.py +0 -14
- aiq/front_ends/fastapi/__init__.py +0 -14
- aiq/front_ends/fastapi/auth_flow_handlers/__init__.py +0 -0
- aiq/front_ends/fastapi/auth_flow_handlers/http_flow_handler.py +0 -27
- aiq/front_ends/fastapi/auth_flow_handlers/websocket_flow_handler.py +0 -107
- aiq/front_ends/fastapi/fastapi_front_end_config.py +0 -234
- aiq/front_ends/fastapi/fastapi_front_end_controller.py +0 -68
- aiq/front_ends/fastapi/fastapi_front_end_plugin.py +0 -116
- aiq/front_ends/fastapi/fastapi_front_end_plugin_worker.py +0 -1092
- aiq/front_ends/fastapi/html_snippets/__init__.py +0 -14
- aiq/front_ends/fastapi/html_snippets/auth_code_grant_success.py +0 -35
- aiq/front_ends/fastapi/intermediate_steps_subscriber.py +0 -80
- aiq/front_ends/fastapi/job_store.py +0 -183
- aiq/front_ends/fastapi/main.py +0 -72
- aiq/front_ends/fastapi/message_handler.py +0 -298
- aiq/front_ends/fastapi/message_validator.py +0 -345
- aiq/front_ends/fastapi/register.py +0 -25
- aiq/front_ends/fastapi/response_helpers.py +0 -195
- aiq/front_ends/fastapi/step_adaptor.py +0 -321
- aiq/front_ends/mcp/__init__.py +0 -14
- aiq/front_ends/mcp/mcp_front_end_config.py +0 -32
- aiq/front_ends/mcp/mcp_front_end_plugin.py +0 -93
- aiq/front_ends/mcp/register.py +0 -27
- aiq/front_ends/mcp/tool_converter.py +0 -242
- aiq/front_ends/register.py +0 -22
- aiq/front_ends/simple_base/__init__.py +0 -14
- aiq/front_ends/simple_base/simple_front_end_plugin_base.py +0 -54
- aiq/llm/__init__.py +0 -0
- aiq/llm/aws_bedrock_llm.py +0 -57
- aiq/llm/nim_llm.py +0 -46
- aiq/llm/openai_llm.py +0 -46
- aiq/llm/register.py +0 -23
- aiq/llm/utils/__init__.py +0 -14
- aiq/llm/utils/env_config_value.py +0 -94
- aiq/llm/utils/error.py +0 -17
- aiq/memory/__init__.py +0 -20
- aiq/memory/interfaces.py +0 -183
- aiq/memory/models.py +0 -112
- aiq/meta/module_to_distro.json +0 -3
- aiq/meta/pypi.md +0 -58
- aiq/object_store/__init__.py +0 -20
- aiq/object_store/in_memory_object_store.py +0 -76
- aiq/object_store/interfaces.py +0 -84
- aiq/object_store/models.py +0 -36
- aiq/object_store/register.py +0 -20
- aiq/observability/__init__.py +0 -14
- aiq/observability/exporter/__init__.py +0 -14
- aiq/observability/exporter/base_exporter.py +0 -449
- aiq/observability/exporter/exporter.py +0 -78
- aiq/observability/exporter/file_exporter.py +0 -33
- aiq/observability/exporter/processing_exporter.py +0 -322
- aiq/observability/exporter/raw_exporter.py +0 -52
- aiq/observability/exporter/span_exporter.py +0 -265
- aiq/observability/exporter_manager.py +0 -335
- aiq/observability/mixin/__init__.py +0 -14
- aiq/observability/mixin/batch_config_mixin.py +0 -26
- aiq/observability/mixin/collector_config_mixin.py +0 -23
- aiq/observability/mixin/file_mixin.py +0 -288
- aiq/observability/mixin/file_mode.py +0 -23
- aiq/observability/mixin/resource_conflict_mixin.py +0 -134
- aiq/observability/mixin/serialize_mixin.py +0 -61
- aiq/observability/mixin/type_introspection_mixin.py +0 -183
- aiq/observability/processor/__init__.py +0 -14
- aiq/observability/processor/batching_processor.py +0 -309
- aiq/observability/processor/callback_processor.py +0 -42
- aiq/observability/processor/intermediate_step_serializer.py +0 -28
- aiq/observability/processor/processor.py +0 -71
- aiq/observability/register.py +0 -96
- aiq/observability/utils/__init__.py +0 -14
- aiq/observability/utils/dict_utils.py +0 -236
- aiq/observability/utils/time_utils.py +0 -31
- aiq/plugins/.namespace +0 -1
- aiq/profiler/__init__.py +0 -0
- aiq/profiler/calc/__init__.py +0 -14
- aiq/profiler/calc/calc_runner.py +0 -627
- aiq/profiler/calc/calculations.py +0 -288
- aiq/profiler/calc/data_models.py +0 -188
- aiq/profiler/calc/plot.py +0 -345
- aiq/profiler/callbacks/__init__.py +0 -0
- aiq/profiler/callbacks/agno_callback_handler.py +0 -295
- aiq/profiler/callbacks/base_callback_class.py +0 -20
- aiq/profiler/callbacks/langchain_callback_handler.py +0 -290
- aiq/profiler/callbacks/llama_index_callback_handler.py +0 -205
- aiq/profiler/callbacks/semantic_kernel_callback_handler.py +0 -238
- aiq/profiler/callbacks/token_usage_base_model.py +0 -27
- aiq/profiler/data_frame_row.py +0 -51
- aiq/profiler/data_models.py +0 -24
- aiq/profiler/decorators/__init__.py +0 -0
- aiq/profiler/decorators/framework_wrapper.py +0 -131
- aiq/profiler/decorators/function_tracking.py +0 -254
- aiq/profiler/forecasting/__init__.py +0 -0
- aiq/profiler/forecasting/config.py +0 -18
- aiq/profiler/forecasting/model_trainer.py +0 -75
- aiq/profiler/forecasting/models/__init__.py +0 -22
- aiq/profiler/forecasting/models/forecasting_base_model.py +0 -40
- aiq/profiler/forecasting/models/linear_model.py +0 -196
- aiq/profiler/forecasting/models/random_forest_regressor.py +0 -268
- aiq/profiler/inference_metrics_model.py +0 -28
- aiq/profiler/inference_optimization/__init__.py +0 -0
- aiq/profiler/inference_optimization/bottleneck_analysis/__init__.py +0 -0
- aiq/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py +0 -460
- aiq/profiler/inference_optimization/bottleneck_analysis/simple_stack_analysis.py +0 -258
- aiq/profiler/inference_optimization/data_models.py +0 -386
- aiq/profiler/inference_optimization/experimental/__init__.py +0 -0
- aiq/profiler/inference_optimization/experimental/concurrency_spike_analysis.py +0 -468
- aiq/profiler/inference_optimization/experimental/prefix_span_analysis.py +0 -405
- aiq/profiler/inference_optimization/llm_metrics.py +0 -212
- aiq/profiler/inference_optimization/prompt_caching.py +0 -163
- aiq/profiler/inference_optimization/token_uniqueness.py +0 -107
- aiq/profiler/inference_optimization/workflow_runtimes.py +0 -72
- aiq/profiler/intermediate_property_adapter.py +0 -102
- aiq/profiler/profile_runner.py +0 -473
- aiq/profiler/utils.py +0 -184
- aiq/registry_handlers/__init__.py +0 -0
- aiq/registry_handlers/local/__init__.py +0 -0
- aiq/registry_handlers/local/local_handler.py +0 -176
- aiq/registry_handlers/local/register_local.py +0 -37
- aiq/registry_handlers/metadata_factory.py +0 -60
- aiq/registry_handlers/package_utils.py +0 -567
- aiq/registry_handlers/pypi/__init__.py +0 -0
- aiq/registry_handlers/pypi/pypi_handler.py +0 -251
- aiq/registry_handlers/pypi/register_pypi.py +0 -40
- aiq/registry_handlers/register.py +0 -21
- aiq/registry_handlers/registry_handler_base.py +0 -157
- aiq/registry_handlers/rest/__init__.py +0 -0
- aiq/registry_handlers/rest/register_rest.py +0 -56
- aiq/registry_handlers/rest/rest_handler.py +0 -237
- aiq/registry_handlers/schemas/__init__.py +0 -0
- aiq/registry_handlers/schemas/headers.py +0 -42
- aiq/registry_handlers/schemas/package.py +0 -68
- aiq/registry_handlers/schemas/publish.py +0 -63
- aiq/registry_handlers/schemas/pull.py +0 -82
- aiq/registry_handlers/schemas/remove.py +0 -36
- aiq/registry_handlers/schemas/search.py +0 -91
- aiq/registry_handlers/schemas/status.py +0 -47
- aiq/retriever/__init__.py +0 -0
- aiq/retriever/interface.py +0 -37
- aiq/retriever/milvus/__init__.py +0 -14
- aiq/retriever/milvus/register.py +0 -81
- aiq/retriever/milvus/retriever.py +0 -228
- aiq/retriever/models.py +0 -74
- aiq/retriever/nemo_retriever/__init__.py +0 -14
- aiq/retriever/nemo_retriever/register.py +0 -60
- aiq/retriever/nemo_retriever/retriever.py +0 -190
- aiq/retriever/register.py +0 -22
- aiq/runtime/__init__.py +0 -14
- aiq/runtime/loader.py +0 -215
- aiq/runtime/runner.py +0 -190
- aiq/runtime/session.py +0 -158
- aiq/runtime/user_metadata.py +0 -130
- aiq/settings/__init__.py +0 -0
- aiq/settings/global_settings.py +0 -318
- aiq/test/.namespace +0 -1
- aiq/tool/__init__.py +0 -0
- aiq/tool/chat_completion.py +0 -74
- aiq/tool/code_execution/README.md +0 -151
- aiq/tool/code_execution/__init__.py +0 -0
- aiq/tool/code_execution/code_sandbox.py +0 -267
- aiq/tool/code_execution/local_sandbox/.gitignore +0 -1
- aiq/tool/code_execution/local_sandbox/Dockerfile.sandbox +0 -60
- aiq/tool/code_execution/local_sandbox/__init__.py +0 -13
- aiq/tool/code_execution/local_sandbox/local_sandbox_server.py +0 -198
- aiq/tool/code_execution/local_sandbox/sandbox.requirements.txt +0 -6
- aiq/tool/code_execution/local_sandbox/start_local_sandbox.sh +0 -50
- aiq/tool/code_execution/register.py +0 -74
- aiq/tool/code_execution/test_code_execution_sandbox.py +0 -414
- aiq/tool/code_execution/utils.py +0 -100
- aiq/tool/datetime_tools.py +0 -42
- aiq/tool/document_search.py +0 -141
- aiq/tool/github_tools/__init__.py +0 -0
- aiq/tool/github_tools/create_github_commit.py +0 -133
- aiq/tool/github_tools/create_github_issue.py +0 -87
- aiq/tool/github_tools/create_github_pr.py +0 -106
- aiq/tool/github_tools/get_github_file.py +0 -106
- aiq/tool/github_tools/get_github_issue.py +0 -166
- aiq/tool/github_tools/get_github_pr.py +0 -256
- aiq/tool/github_tools/update_github_issue.py +0 -100
- aiq/tool/mcp/__init__.py +0 -14
- aiq/tool/mcp/exceptions.py +0 -142
- aiq/tool/mcp/mcp_client.py +0 -255
- aiq/tool/mcp/mcp_tool.py +0 -96
- aiq/tool/memory_tools/__init__.py +0 -0
- aiq/tool/memory_tools/add_memory_tool.py +0 -79
- aiq/tool/memory_tools/delete_memory_tool.py +0 -67
- aiq/tool/memory_tools/get_memory_tool.py +0 -72
- aiq/tool/nvidia_rag.py +0 -95
- aiq/tool/register.py +0 -38
- aiq/tool/retriever.py +0 -89
- aiq/tool/server_tools.py +0 -66
- aiq/utils/__init__.py +0 -0
- aiq/utils/data_models/__init__.py +0 -0
- aiq/utils/data_models/schema_validator.py +0 -58
- aiq/utils/debugging_utils.py +0 -43
- aiq/utils/dump_distro_mapping.py +0 -32
- aiq/utils/exception_handlers/__init__.py +0 -0
- aiq/utils/exception_handlers/automatic_retries.py +0 -289
- aiq/utils/exception_handlers/mcp.py +0 -211
- aiq/utils/exception_handlers/schemas.py +0 -114
- aiq/utils/io/__init__.py +0 -0
- aiq/utils/io/model_processing.py +0 -28
- aiq/utils/io/yaml_tools.py +0 -119
- aiq/utils/log_utils.py +0 -37
- aiq/utils/metadata_utils.py +0 -74
- aiq/utils/optional_imports.py +0 -142
- aiq/utils/producer_consumer_queue.py +0 -178
- aiq/utils/reactive/__init__.py +0 -0
- aiq/utils/reactive/base/__init__.py +0 -0
- aiq/utils/reactive/base/observable_base.py +0 -65
- aiq/utils/reactive/base/observer_base.py +0 -55
- aiq/utils/reactive/base/subject_base.py +0 -79
- aiq/utils/reactive/observable.py +0 -59
- aiq/utils/reactive/observer.py +0 -76
- aiq/utils/reactive/subject.py +0 -131
- aiq/utils/reactive/subscription.py +0 -49
- aiq/utils/settings/__init__.py +0 -0
- aiq/utils/settings/global_settings.py +0 -197
- aiq/utils/string_utils.py +0 -38
- aiq/utils/type_converter.py +0 -290
- aiq/utils/type_utils.py +0 -484
- aiq/utils/url_utils.py +0 -27
- aiqtoolkit-1.2.0rc4.dist-info/METADATA +0 -363
- aiqtoolkit-1.2.0rc4.dist-info/RECORD +0 -438
- aiqtoolkit-1.2.0rc4.dist-info/entry_points.txt +0 -20
- aiqtoolkit-1.2.0rc4.dist-info/licenses/LICENSE-3rd-party.txt +0 -3686
- aiqtoolkit-1.2.0rc4.dist-info/licenses/LICENSE.md +0 -201
- aiqtoolkit-1.2.0rc4.dist-info/top_level.txt +0 -1
- {aiqtoolkit-1.2.0rc4.dist-info → aiqtoolkit-1.2rc9.dist-info}/WHEEL +0 -0
|
@@ -1,124 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
import logging
|
|
17
|
-
import re
|
|
18
|
-
import string
|
|
19
|
-
|
|
20
|
-
from pydantic import Field
|
|
21
|
-
from pydantic import field_validator
|
|
22
|
-
|
|
23
|
-
from aiq.authentication.exceptions.api_key_exceptions import APIKeyFieldError
|
|
24
|
-
from aiq.authentication.exceptions.api_key_exceptions import HeaderNameFieldError
|
|
25
|
-
from aiq.authentication.exceptions.api_key_exceptions import HeaderPrefixFieldError
|
|
26
|
-
from aiq.data_models.authentication import AuthProviderBaseConfig
|
|
27
|
-
from aiq.data_models.authentication import HeaderAuthScheme
|
|
28
|
-
|
|
29
|
-
logger = logging.getLogger(__name__)
|
|
30
|
-
|
|
31
|
-
# Strict RFC 7230 compliant header name regex
|
|
32
|
-
HEADER_NAME_REGEX = re.compile(r"^[!#$%&'*+\-.^_`|~0-9a-zA-Z]+$")
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
class APIKeyAuthProviderConfig(AuthProviderBaseConfig, name="api_key"):
|
|
36
|
-
"""
|
|
37
|
-
API Key authentication configuration model.
|
|
38
|
-
"""
|
|
39
|
-
|
|
40
|
-
raw_key: str = Field(description=("Raw API token or credential to be injected into the request parameter. "
|
|
41
|
-
"Used for 'bearer','x-api-key','custom', and other schemes. "))
|
|
42
|
-
|
|
43
|
-
auth_scheme: HeaderAuthScheme = Field(default=HeaderAuthScheme.BEARER,
|
|
44
|
-
description=("The HTTP authentication scheme to use. "
|
|
45
|
-
"Supported schemes: BEARER, X_API_KEY, BASIC, CUSTOM."))
|
|
46
|
-
|
|
47
|
-
custom_header_name: str | None = Field(description="The HTTP header name that MUST be used in conjunction "
|
|
48
|
-
"with the custom_header_prefix when HeaderAuthScheme is CUSTOM.",
|
|
49
|
-
default=None)
|
|
50
|
-
custom_header_prefix: str | None = Field(description="The HTTP header prefix that MUST be used in conjunction "
|
|
51
|
-
"with the custom_header_name when HeaderAuthScheme is CUSTOM.",
|
|
52
|
-
default=None)
|
|
53
|
-
|
|
54
|
-
@field_validator('raw_key')
|
|
55
|
-
@classmethod
|
|
56
|
-
def validate_raw_key(cls, value: str) -> str:
|
|
57
|
-
if not value:
|
|
58
|
-
raise APIKeyFieldError('value_missing', 'raw_key field value is required.')
|
|
59
|
-
|
|
60
|
-
if len(value) < 8:
|
|
61
|
-
raise APIKeyFieldError(
|
|
62
|
-
'value_too_short',
|
|
63
|
-
'raw_key field value must be at least 8 characters long for security. '
|
|
64
|
-
f'Got: {len(value)} characters.')
|
|
65
|
-
|
|
66
|
-
if len(value.strip()) != len(value):
|
|
67
|
-
raise APIKeyFieldError('whitespace_found',
|
|
68
|
-
'raw_key field value cannot have leading or trailing whitespace.')
|
|
69
|
-
|
|
70
|
-
if any(c in string.whitespace for c in value):
|
|
71
|
-
raise APIKeyFieldError('contains_whitespace', 'raw_key must not contain any '
|
|
72
|
-
'whitespace characters.')
|
|
73
|
-
|
|
74
|
-
return value
|
|
75
|
-
|
|
76
|
-
@field_validator('custom_header_name')
|
|
77
|
-
@classmethod
|
|
78
|
-
def validate_custom_header_name(cls, value: str) -> str:
|
|
79
|
-
if not value:
|
|
80
|
-
raise HeaderNameFieldError('value_missing', 'custom_header_name is required.')
|
|
81
|
-
|
|
82
|
-
if value != value.strip():
|
|
83
|
-
raise HeaderNameFieldError('whitespace_found',
|
|
84
|
-
'custom_header_name field value cannot have leading or trailing whitespace.')
|
|
85
|
-
|
|
86
|
-
if any(c in string.whitespace for c in value):
|
|
87
|
-
raise HeaderNameFieldError('contains_whitespace',
|
|
88
|
-
'custom_header_name must not contain any whitespace characters.')
|
|
89
|
-
|
|
90
|
-
if not HEADER_NAME_REGEX.fullmatch(value):
|
|
91
|
-
raise HeaderNameFieldError(
|
|
92
|
-
'invalid_format',
|
|
93
|
-
'custom_header_name must match the HTTP token syntax: ASCII letters, digits, or allowed symbols.')
|
|
94
|
-
|
|
95
|
-
return value
|
|
96
|
-
|
|
97
|
-
@field_validator('custom_header_prefix')
|
|
98
|
-
@classmethod
|
|
99
|
-
def validate_custom_header_prefix(cls, value: str) -> str:
|
|
100
|
-
if not value:
|
|
101
|
-
raise HeaderPrefixFieldError('value_missing', 'custom_header_prefix is required.')
|
|
102
|
-
|
|
103
|
-
if value != value.strip():
|
|
104
|
-
raise HeaderPrefixFieldError(
|
|
105
|
-
'whitespace_found', 'custom_header_prefix field value cannot have '
|
|
106
|
-
'leading or trailing whitespace.')
|
|
107
|
-
|
|
108
|
-
if any(c in string.whitespace for c in value):
|
|
109
|
-
raise HeaderPrefixFieldError('contains_whitespace',
|
|
110
|
-
'custom_header_prefix must not contain any whitespace characters.')
|
|
111
|
-
|
|
112
|
-
if not value.isascii():
|
|
113
|
-
raise HeaderPrefixFieldError('invalid_format', 'custom_header_prefix must be ASCII.')
|
|
114
|
-
|
|
115
|
-
return value
|
|
116
|
-
|
|
117
|
-
@field_validator('raw_key', mode='after')
|
|
118
|
-
@classmethod
|
|
119
|
-
def validate_raw_key_after(cls, value: str) -> str:
|
|
120
|
-
if not value:
|
|
121
|
-
raise APIKeyFieldError('value_missing', 'raw_key field value is '
|
|
122
|
-
'required after construction.')
|
|
123
|
-
|
|
124
|
-
return value
|
|
@@ -1,26 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
from aiq.authentication.api_key.api_key_auth_provider_config import APIKeyAuthProviderConfig
|
|
17
|
-
from aiq.builder.builder import Builder
|
|
18
|
-
from aiq.cli.register_workflow import register_auth_provider
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
@register_auth_provider(config_type=APIKeyAuthProviderConfig)
|
|
22
|
-
async def api_key_client(config: APIKeyAuthProviderConfig, builder: Builder):
|
|
23
|
-
|
|
24
|
-
from aiq.authentication.api_key.api_key_auth_provider import APIKeyAuthProvider
|
|
25
|
-
|
|
26
|
-
yield APIKeyAuthProvider(config=config)
|
|
@@ -1,14 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
@@ -1,38 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
class APIKeyFieldError(Exception):
|
|
18
|
-
"""Raised when API Key Config api_key field validation fails unexpectedly."""
|
|
19
|
-
|
|
20
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
21
|
-
self.error_code = error_code
|
|
22
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
class HeaderNameFieldError(Exception):
|
|
26
|
-
"""Raised when API Key Config header_name field validation fails unexpectedly."""
|
|
27
|
-
|
|
28
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
29
|
-
self.error_code = error_code
|
|
30
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
class HeaderPrefixFieldError(Exception):
|
|
34
|
-
"""Raised when API Key Config header_prefix field validation fails unexpectedly."""
|
|
35
|
-
|
|
36
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
37
|
-
self.error_code = error_code
|
|
38
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
@@ -1,86 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
class AuthCodeGrantFlowError(Exception):
|
|
18
|
-
"""Raised when Auth Code Grant Flow fails unexpectedly."""
|
|
19
|
-
|
|
20
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
21
|
-
self.error_code = error_code
|
|
22
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
class AuthCodeGrantFlowRefreshTokenError(Exception):
|
|
26
|
-
"""Raised when Auth Code Grant Flow requesting access token using refresh flow fails unexpectedly. """
|
|
27
|
-
|
|
28
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
29
|
-
self.error_code = error_code
|
|
30
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
class AuthCodeGrantConfigClientServerUrlFieldError(Exception):
|
|
34
|
-
"""Raised when Auth Code Grant Config client server URL field validation fails unexpectedly."""
|
|
35
|
-
|
|
36
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
37
|
-
self.error_code = error_code
|
|
38
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
class AuthCodeGrantConfigAuthorizationUrlFieldError(Exception):
|
|
42
|
-
"""Raised when Auth Code Grant Config authorization_url and authorization_token_url field validation fails."""
|
|
43
|
-
|
|
44
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
45
|
-
self.error_code = error_code
|
|
46
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
class AuthCodeGrantConfigConsentPromptKeyFieldError(Exception):
|
|
50
|
-
"""Raised when Auth Code Grant Config consent_prompt_key field validation fails unexpectedly."""
|
|
51
|
-
|
|
52
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
53
|
-
self.error_code = error_code
|
|
54
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
class AuthCodeGrantConfigClientSecretFieldError(Exception):
|
|
58
|
-
"""Raised when Auth Code Grant Config client_secret field validation fails unexpectedly."""
|
|
59
|
-
|
|
60
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
61
|
-
self.error_code = error_code
|
|
62
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
class AuthCodeGrantConfigClientIDFieldError(Exception):
|
|
66
|
-
"""Raised when Auth Code Grant Config client_id field validation fails unexpectedly."""
|
|
67
|
-
|
|
68
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
69
|
-
self.error_code = error_code
|
|
70
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
class AuthCodeGrantConfigScopeFieldError(Exception):
|
|
74
|
-
"""Raised when Auth Code Grant Config scope field validation fails unexpectedly."""
|
|
75
|
-
|
|
76
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
77
|
-
self.error_code = error_code
|
|
78
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
class AuthCodeGrantConfigAudienceFieldError(Exception):
|
|
82
|
-
"""Raised when Auth Code Grant Config audience field validation fails unexpectedly."""
|
|
83
|
-
|
|
84
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
85
|
-
self.error_code = error_code
|
|
86
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
@@ -1,38 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
class AuthenticationError(Exception):
|
|
18
|
-
"""Raised when user authentication fails unexpectedly."""
|
|
19
|
-
|
|
20
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
21
|
-
self.error_code = error_code
|
|
22
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
class OAuthClientConsoleError(Exception):
|
|
26
|
-
"""Raised when user authentication fails unexpectedly."""
|
|
27
|
-
|
|
28
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
29
|
-
self.error_code = error_code
|
|
30
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
class OAuthClientServerError(Exception):
|
|
34
|
-
"""Raised when user authentication fails unexpectedly."""
|
|
35
|
-
|
|
36
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
37
|
-
self.error_code = error_code
|
|
38
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
@@ -1,54 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
class BaseUrlValidationError(Exception):
|
|
18
|
-
"""Raised when HTTP Base URL validation fails unexpectedly."""
|
|
19
|
-
|
|
20
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
21
|
-
self.error_code = error_code
|
|
22
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
class HTTPMethodValidationError(Exception):
|
|
26
|
-
"""Raised when HTTP Method validation fails unexpectedly."""
|
|
27
|
-
|
|
28
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
29
|
-
self.error_code = error_code
|
|
30
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
class QueryParameterValidationError(Exception):
|
|
34
|
-
"""Raised when HTTP Query Parameter validation fails unexpectedly."""
|
|
35
|
-
|
|
36
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
37
|
-
self.error_code = error_code
|
|
38
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
class HTTPHeaderValidationError(Exception):
|
|
42
|
-
"""Raised when HTTP Header validation fails unexpectedly."""
|
|
43
|
-
|
|
44
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
45
|
-
self.error_code = error_code
|
|
46
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
class BodyValidationError(Exception):
|
|
50
|
-
"""Raised when HTTP Body validation fails unexpectedly."""
|
|
51
|
-
|
|
52
|
-
def __init__(self, error_code: str, message: str, *args):
|
|
53
|
-
self.error_code = error_code
|
|
54
|
-
super().__init__(f"[{error_code}] {message}", *args)
|
|
File without changes
|
|
@@ -1,81 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
from pydantic import SecretStr
|
|
17
|
-
|
|
18
|
-
from aiq.authentication.interfaces import AuthProviderBase
|
|
19
|
-
from aiq.builder.context import AIQContext
|
|
20
|
-
from aiq.data_models.authentication import AuthenticatedContext
|
|
21
|
-
from aiq.data_models.authentication import AuthFlowType
|
|
22
|
-
from aiq.data_models.authentication import AuthProviderBaseConfig
|
|
23
|
-
from aiq.data_models.authentication import AuthResult
|
|
24
|
-
from aiq.data_models.authentication import BasicAuthCred
|
|
25
|
-
from aiq.data_models.authentication import BearerTokenCred
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
class HTTPBasicAuthProvider(AuthProviderBase):
|
|
29
|
-
"""
|
|
30
|
-
Abstract base class for HTTP Basic Authentication exchangers.
|
|
31
|
-
"""
|
|
32
|
-
|
|
33
|
-
def __init__(self, config: AuthProviderBaseConfig):
|
|
34
|
-
"""
|
|
35
|
-
Initialize the HTTP Basic Auth Exchanger with the given configuration.
|
|
36
|
-
"""
|
|
37
|
-
super().__init__(config)
|
|
38
|
-
|
|
39
|
-
self._authenticated_tokens: dict[str, AuthResult] = {}
|
|
40
|
-
|
|
41
|
-
async def authenticate(self, user_id: str | None = None) -> AuthResult:
|
|
42
|
-
"""
|
|
43
|
-
Performs simple HTTP Authentication using the provided user ID.
|
|
44
|
-
"""
|
|
45
|
-
|
|
46
|
-
context = AIQContext.get()
|
|
47
|
-
|
|
48
|
-
if user_id is None and hasattr(context, "metadata") and hasattr(
|
|
49
|
-
context.metadata, "cookies") and context.metadata.cookies is not None:
|
|
50
|
-
session_id = context.metadata.cookies.get("aiqtoolkit-session", None)
|
|
51
|
-
if not session_id:
|
|
52
|
-
raise RuntimeError("Authentication failed. No session ID found. Cannot identify user.")
|
|
53
|
-
|
|
54
|
-
user_id = session_id
|
|
55
|
-
|
|
56
|
-
if user_id and user_id in self._authenticated_tokens:
|
|
57
|
-
return self._authenticated_tokens[user_id]
|
|
58
|
-
|
|
59
|
-
auth_callback = context.user_auth_callback
|
|
60
|
-
|
|
61
|
-
try:
|
|
62
|
-
auth_context: AuthenticatedContext = await auth_callback(self.config, AuthFlowType.HTTP_BASIC)
|
|
63
|
-
except RuntimeError as e:
|
|
64
|
-
raise RuntimeError(f"Authentication callback failed: {str(e)}. Did you forget to set a "
|
|
65
|
-
f"callback handler for your frontend?") from e
|
|
66
|
-
|
|
67
|
-
basic_auth_credentials = BasicAuthCred(username=SecretStr(auth_context.metadata.get("username", "")),
|
|
68
|
-
password=SecretStr(auth_context.metadata.get("password", "")))
|
|
69
|
-
|
|
70
|
-
# Get the auth token from the headers of auth context
|
|
71
|
-
bearer_token = auth_context.headers.get("Authorization", "").split(" ")[-1]
|
|
72
|
-
if not bearer_token:
|
|
73
|
-
raise RuntimeError("Authentication failed: No Authorization header found in the response.")
|
|
74
|
-
|
|
75
|
-
bearer_token_cred = BearerTokenCred(token=SecretStr(bearer_token), scheme="Basic")
|
|
76
|
-
|
|
77
|
-
auth_result = AuthResult(credentials=[basic_auth_credentials, bearer_token_cred])
|
|
78
|
-
|
|
79
|
-
self._authenticated_tokens[user_id] = auth_result
|
|
80
|
-
|
|
81
|
-
return auth_result
|
|
@@ -1,30 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
from aiq.builder.builder import Builder
|
|
17
|
-
from aiq.cli.register_workflow import register_auth_provider
|
|
18
|
-
from aiq.data_models.authentication import AuthProviderBaseConfig
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
class HTTPBasicAuthProviderConfig(AuthProviderBaseConfig, name="http_basic"):
|
|
22
|
-
pass
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
@register_auth_provider(config_type=HTTPBasicAuthProviderConfig)
|
|
26
|
-
async def http_basic_auth_provider(config: HTTPBasicAuthProviderConfig, builder: Builder):
|
|
27
|
-
|
|
28
|
-
from aiq.authentication.http_basic_auth.http_basic_auth_provider import HTTPBasicAuthProvider
|
|
29
|
-
|
|
30
|
-
yield HTTPBasicAuthProvider(config)
|
aiq/authentication/interfaces.py
DELETED
|
@@ -1,93 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|
|
15
|
-
|
|
16
|
-
import typing
|
|
17
|
-
from abc import ABC
|
|
18
|
-
from abc import abstractmethod
|
|
19
|
-
|
|
20
|
-
from aiq.data_models.authentication import AuthenticatedContext
|
|
21
|
-
from aiq.data_models.authentication import AuthFlowType
|
|
22
|
-
from aiq.data_models.authentication import AuthProviderBaseConfig
|
|
23
|
-
from aiq.data_models.authentication import AuthProviderBaseConfigT
|
|
24
|
-
from aiq.data_models.authentication import AuthResult
|
|
25
|
-
|
|
26
|
-
AUTHORIZATION_HEADER = "Authorization"
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
class AuthProviderBase(typing.Generic[AuthProviderBaseConfigT], ABC):
|
|
30
|
-
"""
|
|
31
|
-
Base class for authenticating to API services.
|
|
32
|
-
This class provides an interface for authenticating to API services.
|
|
33
|
-
"""
|
|
34
|
-
|
|
35
|
-
def __init__(self, config: AuthProviderBaseConfigT):
|
|
36
|
-
"""
|
|
37
|
-
Initialize the AuthProviderBase with the given configuration.
|
|
38
|
-
|
|
39
|
-
Args:
|
|
40
|
-
config (AuthProviderBaseConfig): Configuration items for authentication.
|
|
41
|
-
"""
|
|
42
|
-
self._config = config
|
|
43
|
-
|
|
44
|
-
@property
|
|
45
|
-
def config(self) -> AuthProviderBaseConfigT:
|
|
46
|
-
"""
|
|
47
|
-
Returns the auth provider configuration object.
|
|
48
|
-
|
|
49
|
-
Returns
|
|
50
|
-
-------
|
|
51
|
-
AuthProviderBaseConfigT
|
|
52
|
-
The auth provider configuration object.
|
|
53
|
-
"""
|
|
54
|
-
return self._config
|
|
55
|
-
|
|
56
|
-
@abstractmethod
|
|
57
|
-
async def authenticate(self, user_id: str | None = None) -> AuthResult:
|
|
58
|
-
"""
|
|
59
|
-
Perform the authentication process for the client.
|
|
60
|
-
|
|
61
|
-
This method handles the necessary steps to authenticate the client with the
|
|
62
|
-
target API service, which may include obtaining tokens, refreshing credentials,
|
|
63
|
-
or completing multi-step authentication flows.
|
|
64
|
-
|
|
65
|
-
Raises:
|
|
66
|
-
NotImplementedError: Must be implemented by subclasses.
|
|
67
|
-
"""
|
|
68
|
-
# This method will call the frontend FlowHandlerBase `authenticate` method
|
|
69
|
-
pass
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
class FlowHandlerBase(ABC):
|
|
73
|
-
"""
|
|
74
|
-
Handles front-end specifc flows for authentication clients.
|
|
75
|
-
|
|
76
|
-
Each front end will define a FlowHandler that will implement the authenticate method.
|
|
77
|
-
|
|
78
|
-
The `authenticate` method will be stored as the callback in the AIQContextState.user_auth_callback
|
|
79
|
-
"""
|
|
80
|
-
|
|
81
|
-
@abstractmethod
|
|
82
|
-
async def authenticate(self, config: AuthProviderBaseConfig, method: AuthFlowType) -> AuthenticatedContext:
|
|
83
|
-
"""
|
|
84
|
-
Perform the authentication process for the client.
|
|
85
|
-
|
|
86
|
-
This method handles the necessary steps to authenticate the client with the
|
|
87
|
-
target API service, which may include obtaining tokens, refreshing credentials,
|
|
88
|
-
or completing multistep authentication flows.
|
|
89
|
-
|
|
90
|
-
Raises:
|
|
91
|
-
NotImplementedError: Must be implemented by subclasses.
|
|
92
|
-
"""
|
|
93
|
-
pass
|
|
@@ -1,14 +0,0 @@
|
|
|
1
|
-
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
2
|
-
# SPDX-License-Identifier: Apache-2.0
|
|
3
|
-
#
|
|
4
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
# you may not use this file except in compliance with the License.
|
|
6
|
-
# You may obtain a copy of the License at
|
|
7
|
-
#
|
|
8
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
#
|
|
10
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
# See the License for the specific language governing permissions and
|
|
14
|
-
# limitations under the License.
|