aiqtoolkit 1.2.0rc3__py3-none-any.whl → 1.2.0rc5__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.

Files changed (440) hide show
  1. aiqtoolkit-1.2.0rc5.dist-info/METADATA +29 -0
  2. aiqtoolkit-1.2.0rc5.dist-info/RECORD +4 -0
  3. aiqtoolkit-1.2.0rc5.dist-info/top_level.txt +1 -0
  4. aiq/agent/__init__.py +0 -0
  5. aiq/agent/base.py +0 -239
  6. aiq/agent/dual_node.py +0 -67
  7. aiq/agent/react_agent/__init__.py +0 -0
  8. aiq/agent/react_agent/agent.py +0 -355
  9. aiq/agent/react_agent/output_parser.py +0 -104
  10. aiq/agent/react_agent/prompt.py +0 -41
  11. aiq/agent/react_agent/register.py +0 -149
  12. aiq/agent/reasoning_agent/__init__.py +0 -0
  13. aiq/agent/reasoning_agent/reasoning_agent.py +0 -225
  14. aiq/agent/register.py +0 -23
  15. aiq/agent/rewoo_agent/__init__.py +0 -0
  16. aiq/agent/rewoo_agent/agent.py +0 -411
  17. aiq/agent/rewoo_agent/prompt.py +0 -108
  18. aiq/agent/rewoo_agent/register.py +0 -158
  19. aiq/agent/tool_calling_agent/__init__.py +0 -0
  20. aiq/agent/tool_calling_agent/agent.py +0 -119
  21. aiq/agent/tool_calling_agent/register.py +0 -106
  22. aiq/authentication/__init__.py +0 -14
  23. aiq/authentication/api_key/__init__.py +0 -14
  24. aiq/authentication/api_key/api_key_auth_provider.py +0 -92
  25. aiq/authentication/api_key/api_key_auth_provider_config.py +0 -124
  26. aiq/authentication/api_key/register.py +0 -26
  27. aiq/authentication/exceptions/__init__.py +0 -14
  28. aiq/authentication/exceptions/api_key_exceptions.py +0 -38
  29. aiq/authentication/exceptions/auth_code_grant_exceptions.py +0 -86
  30. aiq/authentication/exceptions/call_back_exceptions.py +0 -38
  31. aiq/authentication/exceptions/request_exceptions.py +0 -54
  32. aiq/authentication/http_basic_auth/__init__.py +0 -0
  33. aiq/authentication/http_basic_auth/http_basic_auth_provider.py +0 -81
  34. aiq/authentication/http_basic_auth/register.py +0 -30
  35. aiq/authentication/interfaces.py +0 -93
  36. aiq/authentication/oauth2/__init__.py +0 -14
  37. aiq/authentication/oauth2/oauth2_auth_code_flow_provider.py +0 -107
  38. aiq/authentication/oauth2/oauth2_auth_code_flow_provider_config.py +0 -39
  39. aiq/authentication/oauth2/register.py +0 -25
  40. aiq/authentication/register.py +0 -21
  41. aiq/builder/__init__.py +0 -0
  42. aiq/builder/builder.py +0 -285
  43. aiq/builder/component_utils.py +0 -316
  44. aiq/builder/context.py +0 -264
  45. aiq/builder/embedder.py +0 -24
  46. aiq/builder/eval_builder.py +0 -161
  47. aiq/builder/evaluator.py +0 -29
  48. aiq/builder/framework_enum.py +0 -24
  49. aiq/builder/front_end.py +0 -73
  50. aiq/builder/function.py +0 -344
  51. aiq/builder/function_base.py +0 -380
  52. aiq/builder/function_info.py +0 -627
  53. aiq/builder/intermediate_step_manager.py +0 -174
  54. aiq/builder/llm.py +0 -25
  55. aiq/builder/retriever.py +0 -25
  56. aiq/builder/user_interaction_manager.py +0 -74
  57. aiq/builder/workflow.py +0 -148
  58. aiq/builder/workflow_builder.py +0 -1117
  59. aiq/cli/__init__.py +0 -14
  60. aiq/cli/cli_utils/__init__.py +0 -0
  61. aiq/cli/cli_utils/config_override.py +0 -231
  62. aiq/cli/cli_utils/validation.py +0 -37
  63. aiq/cli/commands/__init__.py +0 -0
  64. aiq/cli/commands/configure/__init__.py +0 -0
  65. aiq/cli/commands/configure/channel/__init__.py +0 -0
  66. aiq/cli/commands/configure/channel/add.py +0 -28
  67. aiq/cli/commands/configure/channel/channel.py +0 -36
  68. aiq/cli/commands/configure/channel/remove.py +0 -30
  69. aiq/cli/commands/configure/channel/update.py +0 -30
  70. aiq/cli/commands/configure/configure.py +0 -33
  71. aiq/cli/commands/evaluate.py +0 -139
  72. aiq/cli/commands/info/__init__.py +0 -14
  73. aiq/cli/commands/info/info.py +0 -39
  74. aiq/cli/commands/info/list_channels.py +0 -32
  75. aiq/cli/commands/info/list_components.py +0 -129
  76. aiq/cli/commands/info/list_mcp.py +0 -213
  77. aiq/cli/commands/registry/__init__.py +0 -14
  78. aiq/cli/commands/registry/publish.py +0 -88
  79. aiq/cli/commands/registry/pull.py +0 -118
  80. aiq/cli/commands/registry/registry.py +0 -38
  81. aiq/cli/commands/registry/remove.py +0 -108
  82. aiq/cli/commands/registry/search.py +0 -155
  83. aiq/cli/commands/sizing/__init__.py +0 -14
  84. aiq/cli/commands/sizing/calc.py +0 -297
  85. aiq/cli/commands/sizing/sizing.py +0 -27
  86. aiq/cli/commands/start.py +0 -246
  87. aiq/cli/commands/uninstall.py +0 -81
  88. aiq/cli/commands/validate.py +0 -47
  89. aiq/cli/commands/workflow/__init__.py +0 -14
  90. aiq/cli/commands/workflow/templates/__init__.py.j2 +0 -0
  91. aiq/cli/commands/workflow/templates/config.yml.j2 +0 -16
  92. aiq/cli/commands/workflow/templates/pyproject.toml.j2 +0 -22
  93. aiq/cli/commands/workflow/templates/register.py.j2 +0 -5
  94. aiq/cli/commands/workflow/templates/workflow.py.j2 +0 -36
  95. aiq/cli/commands/workflow/workflow.py +0 -37
  96. aiq/cli/commands/workflow/workflow_commands.py +0 -313
  97. aiq/cli/entrypoint.py +0 -135
  98. aiq/cli/main.py +0 -44
  99. aiq/cli/register_workflow.py +0 -488
  100. aiq/cli/type_registry.py +0 -1000
  101. aiq/data_models/__init__.py +0 -14
  102. aiq/data_models/api_server.py +0 -694
  103. aiq/data_models/authentication.py +0 -231
  104. aiq/data_models/common.py +0 -171
  105. aiq/data_models/component.py +0 -54
  106. aiq/data_models/component_ref.py +0 -168
  107. aiq/data_models/config.py +0 -406
  108. aiq/data_models/dataset_handler.py +0 -123
  109. aiq/data_models/discovery_metadata.py +0 -286
  110. aiq/data_models/embedder.py +0 -27
  111. aiq/data_models/evaluate.py +0 -127
  112. aiq/data_models/evaluator.py +0 -26
  113. aiq/data_models/front_end.py +0 -26
  114. aiq/data_models/function.py +0 -30
  115. aiq/data_models/function_dependencies.py +0 -72
  116. aiq/data_models/interactive.py +0 -246
  117. aiq/data_models/intermediate_step.py +0 -302
  118. aiq/data_models/invocation_node.py +0 -38
  119. aiq/data_models/its_strategy.py +0 -30
  120. aiq/data_models/llm.py +0 -27
  121. aiq/data_models/logging.py +0 -26
  122. aiq/data_models/memory.py +0 -27
  123. aiq/data_models/object_store.py +0 -44
  124. aiq/data_models/profiler.py +0 -54
  125. aiq/data_models/registry_handler.py +0 -26
  126. aiq/data_models/retriever.py +0 -30
  127. aiq/data_models/retry_mixin.py +0 -35
  128. aiq/data_models/span.py +0 -187
  129. aiq/data_models/step_adaptor.py +0 -64
  130. aiq/data_models/streaming.py +0 -33
  131. aiq/data_models/swe_bench_model.py +0 -54
  132. aiq/data_models/telemetry_exporter.py +0 -26
  133. aiq/embedder/__init__.py +0 -0
  134. aiq/embedder/langchain_client.py +0 -41
  135. aiq/embedder/nim_embedder.py +0 -59
  136. aiq/embedder/openai_embedder.py +0 -43
  137. aiq/embedder/register.py +0 -24
  138. aiq/eval/__init__.py +0 -14
  139. aiq/eval/config.py +0 -60
  140. aiq/eval/dataset_handler/__init__.py +0 -0
  141. aiq/eval/dataset_handler/dataset_downloader.py +0 -106
  142. aiq/eval/dataset_handler/dataset_filter.py +0 -52
  143. aiq/eval/dataset_handler/dataset_handler.py +0 -254
  144. aiq/eval/evaluate.py +0 -506
  145. aiq/eval/evaluator/__init__.py +0 -14
  146. aiq/eval/evaluator/base_evaluator.py +0 -73
  147. aiq/eval/evaluator/evaluator_model.py +0 -45
  148. aiq/eval/intermediate_step_adapter.py +0 -99
  149. aiq/eval/rag_evaluator/__init__.py +0 -0
  150. aiq/eval/rag_evaluator/evaluate.py +0 -178
  151. aiq/eval/rag_evaluator/register.py +0 -143
  152. aiq/eval/register.py +0 -23
  153. aiq/eval/remote_workflow.py +0 -133
  154. aiq/eval/runners/__init__.py +0 -14
  155. aiq/eval/runners/config.py +0 -39
  156. aiq/eval/runners/multi_eval_runner.py +0 -54
  157. aiq/eval/runtime_event_subscriber.py +0 -52
  158. aiq/eval/swe_bench_evaluator/__init__.py +0 -0
  159. aiq/eval/swe_bench_evaluator/evaluate.py +0 -215
  160. aiq/eval/swe_bench_evaluator/register.py +0 -36
  161. aiq/eval/trajectory_evaluator/__init__.py +0 -0
  162. aiq/eval/trajectory_evaluator/evaluate.py +0 -75
  163. aiq/eval/trajectory_evaluator/register.py +0 -40
  164. aiq/eval/tunable_rag_evaluator/__init__.py +0 -0
  165. aiq/eval/tunable_rag_evaluator/evaluate.py +0 -245
  166. aiq/eval/tunable_rag_evaluator/register.py +0 -52
  167. aiq/eval/usage_stats.py +0 -41
  168. aiq/eval/utils/__init__.py +0 -0
  169. aiq/eval/utils/output_uploader.py +0 -140
  170. aiq/eval/utils/tqdm_position_registry.py +0 -40
  171. aiq/eval/utils/weave_eval.py +0 -184
  172. aiq/experimental/__init__.py +0 -0
  173. aiq/experimental/decorators/__init__.py +0 -0
  174. aiq/experimental/decorators/experimental_warning_decorator.py +0 -130
  175. aiq/experimental/inference_time_scaling/__init__.py +0 -0
  176. aiq/experimental/inference_time_scaling/editing/__init__.py +0 -0
  177. aiq/experimental/inference_time_scaling/editing/iterative_plan_refinement_editor.py +0 -147
  178. aiq/experimental/inference_time_scaling/editing/llm_as_a_judge_editor.py +0 -204
  179. aiq/experimental/inference_time_scaling/editing/motivation_aware_summarization.py +0 -107
  180. aiq/experimental/inference_time_scaling/functions/__init__.py +0 -0
  181. aiq/experimental/inference_time_scaling/functions/execute_score_select_function.py +0 -105
  182. aiq/experimental/inference_time_scaling/functions/its_tool_orchestration_function.py +0 -205
  183. aiq/experimental/inference_time_scaling/functions/its_tool_wrapper_function.py +0 -146
  184. aiq/experimental/inference_time_scaling/functions/plan_select_execute_function.py +0 -224
  185. aiq/experimental/inference_time_scaling/models/__init__.py +0 -0
  186. aiq/experimental/inference_time_scaling/models/editor_config.py +0 -132
  187. aiq/experimental/inference_time_scaling/models/its_item.py +0 -48
  188. aiq/experimental/inference_time_scaling/models/scoring_config.py +0 -112
  189. aiq/experimental/inference_time_scaling/models/search_config.py +0 -120
  190. aiq/experimental/inference_time_scaling/models/selection_config.py +0 -154
  191. aiq/experimental/inference_time_scaling/models/stage_enums.py +0 -43
  192. aiq/experimental/inference_time_scaling/models/strategy_base.py +0 -66
  193. aiq/experimental/inference_time_scaling/models/tool_use_config.py +0 -41
  194. aiq/experimental/inference_time_scaling/register.py +0 -36
  195. aiq/experimental/inference_time_scaling/scoring/__init__.py +0 -0
  196. aiq/experimental/inference_time_scaling/scoring/llm_based_agent_scorer.py +0 -168
  197. aiq/experimental/inference_time_scaling/scoring/llm_based_plan_scorer.py +0 -168
  198. aiq/experimental/inference_time_scaling/scoring/motivation_aware_scorer.py +0 -111
  199. aiq/experimental/inference_time_scaling/search/__init__.py +0 -0
  200. aiq/experimental/inference_time_scaling/search/multi_llm_planner.py +0 -128
  201. aiq/experimental/inference_time_scaling/search/multi_query_retrieval_search.py +0 -122
  202. aiq/experimental/inference_time_scaling/search/single_shot_multi_plan_planner.py +0 -128
  203. aiq/experimental/inference_time_scaling/selection/__init__.py +0 -0
  204. aiq/experimental/inference_time_scaling/selection/best_of_n_selector.py +0 -63
  205. aiq/experimental/inference_time_scaling/selection/llm_based_agent_output_selector.py +0 -131
  206. aiq/experimental/inference_time_scaling/selection/llm_based_output_merging_selector.py +0 -159
  207. aiq/experimental/inference_time_scaling/selection/llm_based_plan_selector.py +0 -128
  208. aiq/experimental/inference_time_scaling/selection/threshold_selector.py +0 -58
  209. aiq/front_ends/__init__.py +0 -14
  210. aiq/front_ends/console/__init__.py +0 -14
  211. aiq/front_ends/console/authentication_flow_handler.py +0 -233
  212. aiq/front_ends/console/console_front_end_config.py +0 -32
  213. aiq/front_ends/console/console_front_end_plugin.py +0 -96
  214. aiq/front_ends/console/register.py +0 -25
  215. aiq/front_ends/cron/__init__.py +0 -14
  216. aiq/front_ends/fastapi/__init__.py +0 -14
  217. aiq/front_ends/fastapi/auth_flow_handlers/__init__.py +0 -0
  218. aiq/front_ends/fastapi/auth_flow_handlers/http_flow_handler.py +0 -27
  219. aiq/front_ends/fastapi/auth_flow_handlers/websocket_flow_handler.py +0 -107
  220. aiq/front_ends/fastapi/fastapi_front_end_config.py +0 -234
  221. aiq/front_ends/fastapi/fastapi_front_end_controller.py +0 -68
  222. aiq/front_ends/fastapi/fastapi_front_end_plugin.py +0 -116
  223. aiq/front_ends/fastapi/fastapi_front_end_plugin_worker.py +0 -1092
  224. aiq/front_ends/fastapi/html_snippets/__init__.py +0 -14
  225. aiq/front_ends/fastapi/html_snippets/auth_code_grant_success.py +0 -35
  226. aiq/front_ends/fastapi/intermediate_steps_subscriber.py +0 -80
  227. aiq/front_ends/fastapi/job_store.py +0 -183
  228. aiq/front_ends/fastapi/main.py +0 -72
  229. aiq/front_ends/fastapi/message_handler.py +0 -298
  230. aiq/front_ends/fastapi/message_validator.py +0 -345
  231. aiq/front_ends/fastapi/register.py +0 -25
  232. aiq/front_ends/fastapi/response_helpers.py +0 -195
  233. aiq/front_ends/fastapi/step_adaptor.py +0 -321
  234. aiq/front_ends/mcp/__init__.py +0 -14
  235. aiq/front_ends/mcp/mcp_front_end_config.py +0 -32
  236. aiq/front_ends/mcp/mcp_front_end_plugin.py +0 -93
  237. aiq/front_ends/mcp/register.py +0 -27
  238. aiq/front_ends/mcp/tool_converter.py +0 -242
  239. aiq/front_ends/register.py +0 -22
  240. aiq/front_ends/simple_base/__init__.py +0 -14
  241. aiq/front_ends/simple_base/simple_front_end_plugin_base.py +0 -54
  242. aiq/llm/__init__.py +0 -0
  243. aiq/llm/aws_bedrock_llm.py +0 -57
  244. aiq/llm/nim_llm.py +0 -46
  245. aiq/llm/openai_llm.py +0 -46
  246. aiq/llm/register.py +0 -23
  247. aiq/llm/utils/__init__.py +0 -14
  248. aiq/llm/utils/env_config_value.py +0 -94
  249. aiq/llm/utils/error.py +0 -17
  250. aiq/memory/__init__.py +0 -20
  251. aiq/memory/interfaces.py +0 -183
  252. aiq/memory/models.py +0 -112
  253. aiq/meta/module_to_distro.json +0 -3
  254. aiq/meta/pypi.md +0 -58
  255. aiq/object_store/__init__.py +0 -20
  256. aiq/object_store/in_memory_object_store.py +0 -74
  257. aiq/object_store/interfaces.py +0 -84
  258. aiq/object_store/models.py +0 -36
  259. aiq/object_store/register.py +0 -20
  260. aiq/observability/__init__.py +0 -14
  261. aiq/observability/exporter/__init__.py +0 -14
  262. aiq/observability/exporter/base_exporter.py +0 -449
  263. aiq/observability/exporter/exporter.py +0 -78
  264. aiq/observability/exporter/file_exporter.py +0 -33
  265. aiq/observability/exporter/processing_exporter.py +0 -322
  266. aiq/observability/exporter/raw_exporter.py +0 -52
  267. aiq/observability/exporter/span_exporter.py +0 -265
  268. aiq/observability/exporter_manager.py +0 -335
  269. aiq/observability/mixin/__init__.py +0 -14
  270. aiq/observability/mixin/batch_config_mixin.py +0 -26
  271. aiq/observability/mixin/collector_config_mixin.py +0 -23
  272. aiq/observability/mixin/file_mixin.py +0 -288
  273. aiq/observability/mixin/file_mode.py +0 -23
  274. aiq/observability/mixin/resource_conflict_mixin.py +0 -134
  275. aiq/observability/mixin/serialize_mixin.py +0 -61
  276. aiq/observability/mixin/type_introspection_mixin.py +0 -183
  277. aiq/observability/processor/__init__.py +0 -14
  278. aiq/observability/processor/batching_processor.py +0 -309
  279. aiq/observability/processor/callback_processor.py +0 -42
  280. aiq/observability/processor/intermediate_step_serializer.py +0 -28
  281. aiq/observability/processor/processor.py +0 -71
  282. aiq/observability/register.py +0 -96
  283. aiq/observability/utils/__init__.py +0 -14
  284. aiq/observability/utils/dict_utils.py +0 -236
  285. aiq/observability/utils/time_utils.py +0 -31
  286. aiq/plugins/.namespace +0 -1
  287. aiq/profiler/__init__.py +0 -0
  288. aiq/profiler/calc/__init__.py +0 -14
  289. aiq/profiler/calc/calc_runner.py +0 -627
  290. aiq/profiler/calc/calculations.py +0 -288
  291. aiq/profiler/calc/data_models.py +0 -188
  292. aiq/profiler/calc/plot.py +0 -345
  293. aiq/profiler/callbacks/__init__.py +0 -0
  294. aiq/profiler/callbacks/agno_callback_handler.py +0 -295
  295. aiq/profiler/callbacks/base_callback_class.py +0 -20
  296. aiq/profiler/callbacks/langchain_callback_handler.py +0 -290
  297. aiq/profiler/callbacks/llama_index_callback_handler.py +0 -205
  298. aiq/profiler/callbacks/semantic_kernel_callback_handler.py +0 -238
  299. aiq/profiler/callbacks/token_usage_base_model.py +0 -27
  300. aiq/profiler/data_frame_row.py +0 -51
  301. aiq/profiler/data_models.py +0 -24
  302. aiq/profiler/decorators/__init__.py +0 -0
  303. aiq/profiler/decorators/framework_wrapper.py +0 -131
  304. aiq/profiler/decorators/function_tracking.py +0 -254
  305. aiq/profiler/forecasting/__init__.py +0 -0
  306. aiq/profiler/forecasting/config.py +0 -18
  307. aiq/profiler/forecasting/model_trainer.py +0 -75
  308. aiq/profiler/forecasting/models/__init__.py +0 -22
  309. aiq/profiler/forecasting/models/forecasting_base_model.py +0 -40
  310. aiq/profiler/forecasting/models/linear_model.py +0 -196
  311. aiq/profiler/forecasting/models/random_forest_regressor.py +0 -268
  312. aiq/profiler/inference_metrics_model.py +0 -28
  313. aiq/profiler/inference_optimization/__init__.py +0 -0
  314. aiq/profiler/inference_optimization/bottleneck_analysis/__init__.py +0 -0
  315. aiq/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py +0 -460
  316. aiq/profiler/inference_optimization/bottleneck_analysis/simple_stack_analysis.py +0 -258
  317. aiq/profiler/inference_optimization/data_models.py +0 -386
  318. aiq/profiler/inference_optimization/experimental/__init__.py +0 -0
  319. aiq/profiler/inference_optimization/experimental/concurrency_spike_analysis.py +0 -468
  320. aiq/profiler/inference_optimization/experimental/prefix_span_analysis.py +0 -405
  321. aiq/profiler/inference_optimization/llm_metrics.py +0 -212
  322. aiq/profiler/inference_optimization/prompt_caching.py +0 -163
  323. aiq/profiler/inference_optimization/token_uniqueness.py +0 -107
  324. aiq/profiler/inference_optimization/workflow_runtimes.py +0 -72
  325. aiq/profiler/intermediate_property_adapter.py +0 -102
  326. aiq/profiler/profile_runner.py +0 -473
  327. aiq/profiler/utils.py +0 -184
  328. aiq/registry_handlers/__init__.py +0 -0
  329. aiq/registry_handlers/local/__init__.py +0 -0
  330. aiq/registry_handlers/local/local_handler.py +0 -176
  331. aiq/registry_handlers/local/register_local.py +0 -37
  332. aiq/registry_handlers/metadata_factory.py +0 -60
  333. aiq/registry_handlers/package_utils.py +0 -198
  334. aiq/registry_handlers/pypi/__init__.py +0 -0
  335. aiq/registry_handlers/pypi/pypi_handler.py +0 -251
  336. aiq/registry_handlers/pypi/register_pypi.py +0 -40
  337. aiq/registry_handlers/register.py +0 -21
  338. aiq/registry_handlers/registry_handler_base.py +0 -157
  339. aiq/registry_handlers/rest/__init__.py +0 -0
  340. aiq/registry_handlers/rest/register_rest.py +0 -56
  341. aiq/registry_handlers/rest/rest_handler.py +0 -237
  342. aiq/registry_handlers/schemas/__init__.py +0 -0
  343. aiq/registry_handlers/schemas/headers.py +0 -42
  344. aiq/registry_handlers/schemas/package.py +0 -68
  345. aiq/registry_handlers/schemas/publish.py +0 -63
  346. aiq/registry_handlers/schemas/pull.py +0 -82
  347. aiq/registry_handlers/schemas/remove.py +0 -36
  348. aiq/registry_handlers/schemas/search.py +0 -91
  349. aiq/registry_handlers/schemas/status.py +0 -47
  350. aiq/retriever/__init__.py +0 -0
  351. aiq/retriever/interface.py +0 -37
  352. aiq/retriever/milvus/__init__.py +0 -14
  353. aiq/retriever/milvus/register.py +0 -81
  354. aiq/retriever/milvus/retriever.py +0 -228
  355. aiq/retriever/models.py +0 -74
  356. aiq/retriever/nemo_retriever/__init__.py +0 -14
  357. aiq/retriever/nemo_retriever/register.py +0 -60
  358. aiq/retriever/nemo_retriever/retriever.py +0 -190
  359. aiq/retriever/register.py +0 -22
  360. aiq/runtime/__init__.py +0 -14
  361. aiq/runtime/loader.py +0 -194
  362. aiq/runtime/runner.py +0 -190
  363. aiq/runtime/session.py +0 -158
  364. aiq/runtime/user_metadata.py +0 -130
  365. aiq/settings/__init__.py +0 -0
  366. aiq/settings/global_settings.py +0 -318
  367. aiq/test/.namespace +0 -1
  368. aiq/tool/__init__.py +0 -0
  369. aiq/tool/chat_completion.py +0 -74
  370. aiq/tool/code_execution/README.md +0 -152
  371. aiq/tool/code_execution/__init__.py +0 -0
  372. aiq/tool/code_execution/code_sandbox.py +0 -267
  373. aiq/tool/code_execution/local_sandbox/.gitignore +0 -1
  374. aiq/tool/code_execution/local_sandbox/Dockerfile.sandbox +0 -60
  375. aiq/tool/code_execution/local_sandbox/__init__.py +0 -13
  376. aiq/tool/code_execution/local_sandbox/local_sandbox_server.py +0 -198
  377. aiq/tool/code_execution/local_sandbox/sandbox.requirements.txt +0 -6
  378. aiq/tool/code_execution/local_sandbox/start_local_sandbox.sh +0 -50
  379. aiq/tool/code_execution/register.py +0 -74
  380. aiq/tool/code_execution/test_code_execution_sandbox.py +0 -414
  381. aiq/tool/code_execution/utils.py +0 -100
  382. aiq/tool/datetime_tools.py +0 -42
  383. aiq/tool/document_search.py +0 -141
  384. aiq/tool/github_tools/__init__.py +0 -0
  385. aiq/tool/github_tools/create_github_commit.py +0 -133
  386. aiq/tool/github_tools/create_github_issue.py +0 -87
  387. aiq/tool/github_tools/create_github_pr.py +0 -106
  388. aiq/tool/github_tools/get_github_file.py +0 -106
  389. aiq/tool/github_tools/get_github_issue.py +0 -166
  390. aiq/tool/github_tools/get_github_pr.py +0 -256
  391. aiq/tool/github_tools/update_github_issue.py +0 -100
  392. aiq/tool/mcp/__init__.py +0 -14
  393. aiq/tool/mcp/exceptions.py +0 -142
  394. aiq/tool/mcp/mcp_client.py +0 -255
  395. aiq/tool/mcp/mcp_tool.py +0 -96
  396. aiq/tool/memory_tools/__init__.py +0 -0
  397. aiq/tool/memory_tools/add_memory_tool.py +0 -79
  398. aiq/tool/memory_tools/delete_memory_tool.py +0 -67
  399. aiq/tool/memory_tools/get_memory_tool.py +0 -72
  400. aiq/tool/nvidia_rag.py +0 -95
  401. aiq/tool/register.py +0 -38
  402. aiq/tool/retriever.py +0 -89
  403. aiq/tool/server_tools.py +0 -66
  404. aiq/utils/__init__.py +0 -0
  405. aiq/utils/data_models/__init__.py +0 -0
  406. aiq/utils/data_models/schema_validator.py +0 -58
  407. aiq/utils/debugging_utils.py +0 -43
  408. aiq/utils/exception_handlers/__init__.py +0 -0
  409. aiq/utils/exception_handlers/automatic_retries.py +0 -289
  410. aiq/utils/exception_handlers/mcp.py +0 -211
  411. aiq/utils/exception_handlers/schemas.py +0 -114
  412. aiq/utils/io/__init__.py +0 -0
  413. aiq/utils/io/model_processing.py +0 -28
  414. aiq/utils/io/yaml_tools.py +0 -119
  415. aiq/utils/log_utils.py +0 -37
  416. aiq/utils/metadata_utils.py +0 -74
  417. aiq/utils/optional_imports.py +0 -142
  418. aiq/utils/producer_consumer_queue.py +0 -178
  419. aiq/utils/reactive/__init__.py +0 -0
  420. aiq/utils/reactive/base/__init__.py +0 -0
  421. aiq/utils/reactive/base/observable_base.py +0 -65
  422. aiq/utils/reactive/base/observer_base.py +0 -55
  423. aiq/utils/reactive/base/subject_base.py +0 -79
  424. aiq/utils/reactive/observable.py +0 -59
  425. aiq/utils/reactive/observer.py +0 -76
  426. aiq/utils/reactive/subject.py +0 -131
  427. aiq/utils/reactive/subscription.py +0 -49
  428. aiq/utils/settings/__init__.py +0 -0
  429. aiq/utils/settings/global_settings.py +0 -197
  430. aiq/utils/string_utils.py +0 -38
  431. aiq/utils/type_converter.py +0 -290
  432. aiq/utils/type_utils.py +0 -484
  433. aiq/utils/url_utils.py +0 -27
  434. aiqtoolkit-1.2.0rc3.dist-info/METADATA +0 -363
  435. aiqtoolkit-1.2.0rc3.dist-info/RECORD +0 -437
  436. aiqtoolkit-1.2.0rc3.dist-info/entry_points.txt +0 -20
  437. aiqtoolkit-1.2.0rc3.dist-info/licenses/LICENSE-3rd-party.txt +0 -3686
  438. aiqtoolkit-1.2.0rc3.dist-info/licenses/LICENSE.md +0 -201
  439. aiqtoolkit-1.2.0rc3.dist-info/top_level.txt +0 -1
  440. {aiqtoolkit-1.2.0rc3.dist-info → aiqtoolkit-1.2.0rc5.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)
@@ -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.