aiqtoolkit 1.2.0.dev0__py3-none-any.whl → 1.2.0rc2__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 (220) hide show
  1. aiq/agent/base.py +170 -8
  2. aiq/agent/dual_node.py +1 -1
  3. aiq/agent/react_agent/agent.py +146 -112
  4. aiq/agent/react_agent/prompt.py +1 -6
  5. aiq/agent/react_agent/register.py +36 -35
  6. aiq/agent/rewoo_agent/agent.py +36 -35
  7. aiq/agent/rewoo_agent/register.py +2 -2
  8. aiq/agent/tool_calling_agent/agent.py +3 -7
  9. aiq/agent/tool_calling_agent/register.py +1 -1
  10. aiq/authentication/__init__.py +14 -0
  11. aiq/authentication/api_key/__init__.py +14 -0
  12. aiq/authentication/api_key/api_key_auth_provider.py +92 -0
  13. aiq/authentication/api_key/api_key_auth_provider_config.py +124 -0
  14. aiq/authentication/api_key/register.py +26 -0
  15. aiq/authentication/exceptions/__init__.py +14 -0
  16. aiq/authentication/exceptions/api_key_exceptions.py +38 -0
  17. aiq/authentication/exceptions/auth_code_grant_exceptions.py +86 -0
  18. aiq/authentication/exceptions/call_back_exceptions.py +38 -0
  19. aiq/authentication/exceptions/request_exceptions.py +54 -0
  20. aiq/authentication/http_basic_auth/__init__.py +0 -0
  21. aiq/authentication/http_basic_auth/http_basic_auth_provider.py +81 -0
  22. aiq/authentication/http_basic_auth/register.py +30 -0
  23. aiq/authentication/interfaces.py +93 -0
  24. aiq/authentication/oauth2/__init__.py +14 -0
  25. aiq/authentication/oauth2/oauth2_auth_code_flow_provider.py +107 -0
  26. aiq/authentication/oauth2/oauth2_auth_code_flow_provider_config.py +39 -0
  27. aiq/authentication/oauth2/register.py +25 -0
  28. aiq/authentication/register.py +21 -0
  29. aiq/builder/builder.py +64 -2
  30. aiq/builder/component_utils.py +16 -3
  31. aiq/builder/context.py +37 -0
  32. aiq/builder/eval_builder.py +43 -2
  33. aiq/builder/function.py +44 -12
  34. aiq/builder/function_base.py +1 -1
  35. aiq/builder/intermediate_step_manager.py +6 -8
  36. aiq/builder/user_interaction_manager.py +3 -0
  37. aiq/builder/workflow.py +23 -18
  38. aiq/builder/workflow_builder.py +421 -61
  39. aiq/cli/commands/info/list_mcp.py +103 -16
  40. aiq/cli/commands/sizing/__init__.py +14 -0
  41. aiq/cli/commands/sizing/calc.py +294 -0
  42. aiq/cli/commands/sizing/sizing.py +27 -0
  43. aiq/cli/commands/start.py +2 -1
  44. aiq/cli/entrypoint.py +2 -0
  45. aiq/cli/register_workflow.py +80 -0
  46. aiq/cli/type_registry.py +151 -30
  47. aiq/data_models/api_server.py +124 -12
  48. aiq/data_models/authentication.py +231 -0
  49. aiq/data_models/common.py +35 -7
  50. aiq/data_models/component.py +17 -9
  51. aiq/data_models/component_ref.py +33 -0
  52. aiq/data_models/config.py +60 -3
  53. aiq/data_models/dataset_handler.py +2 -1
  54. aiq/data_models/embedder.py +1 -0
  55. aiq/data_models/evaluate.py +23 -0
  56. aiq/data_models/function_dependencies.py +8 -0
  57. aiq/data_models/interactive.py +10 -1
  58. aiq/data_models/intermediate_step.py +38 -5
  59. aiq/data_models/its_strategy.py +30 -0
  60. aiq/data_models/llm.py +1 -0
  61. aiq/data_models/memory.py +1 -0
  62. aiq/data_models/object_store.py +44 -0
  63. aiq/data_models/profiler.py +1 -0
  64. aiq/data_models/retry_mixin.py +35 -0
  65. aiq/data_models/span.py +187 -0
  66. aiq/data_models/telemetry_exporter.py +2 -2
  67. aiq/embedder/nim_embedder.py +2 -1
  68. aiq/embedder/openai_embedder.py +2 -1
  69. aiq/eval/config.py +19 -1
  70. aiq/eval/dataset_handler/dataset_handler.py +87 -2
  71. aiq/eval/evaluate.py +208 -27
  72. aiq/eval/evaluator/base_evaluator.py +73 -0
  73. aiq/eval/evaluator/evaluator_model.py +1 -0
  74. aiq/eval/intermediate_step_adapter.py +11 -5
  75. aiq/eval/rag_evaluator/evaluate.py +55 -15
  76. aiq/eval/rag_evaluator/register.py +6 -1
  77. aiq/eval/remote_workflow.py +7 -2
  78. aiq/eval/runners/__init__.py +14 -0
  79. aiq/eval/runners/config.py +39 -0
  80. aiq/eval/runners/multi_eval_runner.py +54 -0
  81. aiq/eval/trajectory_evaluator/evaluate.py +22 -65
  82. aiq/eval/tunable_rag_evaluator/evaluate.py +150 -168
  83. aiq/eval/tunable_rag_evaluator/register.py +2 -0
  84. aiq/eval/usage_stats.py +41 -0
  85. aiq/eval/utils/output_uploader.py +10 -1
  86. aiq/eval/utils/weave_eval.py +184 -0
  87. aiq/experimental/__init__.py +0 -0
  88. aiq/experimental/decorators/__init__.py +0 -0
  89. aiq/experimental/decorators/experimental_warning_decorator.py +130 -0
  90. aiq/experimental/inference_time_scaling/__init__.py +0 -0
  91. aiq/experimental/inference_time_scaling/editing/__init__.py +0 -0
  92. aiq/experimental/inference_time_scaling/editing/iterative_plan_refinement_editor.py +147 -0
  93. aiq/experimental/inference_time_scaling/editing/llm_as_a_judge_editor.py +204 -0
  94. aiq/experimental/inference_time_scaling/editing/motivation_aware_summarization.py +107 -0
  95. aiq/experimental/inference_time_scaling/functions/__init__.py +0 -0
  96. aiq/experimental/inference_time_scaling/functions/execute_score_select_function.py +105 -0
  97. aiq/experimental/inference_time_scaling/functions/its_tool_orchestration_function.py +205 -0
  98. aiq/experimental/inference_time_scaling/functions/its_tool_wrapper_function.py +146 -0
  99. aiq/experimental/inference_time_scaling/functions/plan_select_execute_function.py +224 -0
  100. aiq/experimental/inference_time_scaling/models/__init__.py +0 -0
  101. aiq/experimental/inference_time_scaling/models/editor_config.py +132 -0
  102. aiq/experimental/inference_time_scaling/models/its_item.py +48 -0
  103. aiq/experimental/inference_time_scaling/models/scoring_config.py +112 -0
  104. aiq/experimental/inference_time_scaling/models/search_config.py +120 -0
  105. aiq/experimental/inference_time_scaling/models/selection_config.py +154 -0
  106. aiq/experimental/inference_time_scaling/models/stage_enums.py +43 -0
  107. aiq/experimental/inference_time_scaling/models/strategy_base.py +66 -0
  108. aiq/experimental/inference_time_scaling/models/tool_use_config.py +41 -0
  109. aiq/experimental/inference_time_scaling/register.py +36 -0
  110. aiq/experimental/inference_time_scaling/scoring/__init__.py +0 -0
  111. aiq/experimental/inference_time_scaling/scoring/llm_based_agent_scorer.py +168 -0
  112. aiq/experimental/inference_time_scaling/scoring/llm_based_plan_scorer.py +168 -0
  113. aiq/experimental/inference_time_scaling/scoring/motivation_aware_scorer.py +111 -0
  114. aiq/experimental/inference_time_scaling/search/__init__.py +0 -0
  115. aiq/experimental/inference_time_scaling/search/multi_llm_planner.py +128 -0
  116. aiq/experimental/inference_time_scaling/search/multi_query_retrieval_search.py +122 -0
  117. aiq/experimental/inference_time_scaling/search/single_shot_multi_plan_planner.py +128 -0
  118. aiq/experimental/inference_time_scaling/selection/__init__.py +0 -0
  119. aiq/experimental/inference_time_scaling/selection/best_of_n_selector.py +63 -0
  120. aiq/experimental/inference_time_scaling/selection/llm_based_agent_output_selector.py +131 -0
  121. aiq/experimental/inference_time_scaling/selection/llm_based_output_merging_selector.py +159 -0
  122. aiq/experimental/inference_time_scaling/selection/llm_based_plan_selector.py +128 -0
  123. aiq/experimental/inference_time_scaling/selection/threshold_selector.py +58 -0
  124. aiq/front_ends/console/authentication_flow_handler.py +233 -0
  125. aiq/front_ends/console/console_front_end_plugin.py +11 -2
  126. aiq/front_ends/fastapi/auth_flow_handlers/__init__.py +0 -0
  127. aiq/front_ends/fastapi/auth_flow_handlers/http_flow_handler.py +27 -0
  128. aiq/front_ends/fastapi/auth_flow_handlers/websocket_flow_handler.py +107 -0
  129. aiq/front_ends/fastapi/fastapi_front_end_config.py +93 -9
  130. aiq/front_ends/fastapi/fastapi_front_end_controller.py +68 -0
  131. aiq/front_ends/fastapi/fastapi_front_end_plugin.py +14 -1
  132. aiq/front_ends/fastapi/fastapi_front_end_plugin_worker.py +537 -52
  133. aiq/front_ends/fastapi/html_snippets/__init__.py +14 -0
  134. aiq/front_ends/fastapi/html_snippets/auth_code_grant_success.py +35 -0
  135. aiq/front_ends/fastapi/job_store.py +47 -25
  136. aiq/front_ends/fastapi/main.py +2 -0
  137. aiq/front_ends/fastapi/message_handler.py +108 -89
  138. aiq/front_ends/fastapi/step_adaptor.py +2 -1
  139. aiq/llm/aws_bedrock_llm.py +57 -0
  140. aiq/llm/nim_llm.py +2 -1
  141. aiq/llm/openai_llm.py +3 -2
  142. aiq/llm/register.py +1 -0
  143. aiq/meta/pypi.md +12 -12
  144. aiq/object_store/__init__.py +20 -0
  145. aiq/object_store/in_memory_object_store.py +74 -0
  146. aiq/object_store/interfaces.py +84 -0
  147. aiq/object_store/models.py +36 -0
  148. aiq/object_store/register.py +20 -0
  149. aiq/observability/__init__.py +14 -0
  150. aiq/observability/exporter/__init__.py +14 -0
  151. aiq/observability/exporter/base_exporter.py +449 -0
  152. aiq/observability/exporter/exporter.py +78 -0
  153. aiq/observability/exporter/file_exporter.py +33 -0
  154. aiq/observability/exporter/processing_exporter.py +269 -0
  155. aiq/observability/exporter/raw_exporter.py +52 -0
  156. aiq/observability/exporter/span_exporter.py +264 -0
  157. aiq/observability/exporter_manager.py +335 -0
  158. aiq/observability/mixin/__init__.py +14 -0
  159. aiq/observability/mixin/batch_config_mixin.py +26 -0
  160. aiq/observability/mixin/collector_config_mixin.py +23 -0
  161. aiq/observability/mixin/file_mixin.py +288 -0
  162. aiq/observability/mixin/file_mode.py +23 -0
  163. aiq/observability/mixin/resource_conflict_mixin.py +134 -0
  164. aiq/observability/mixin/serialize_mixin.py +61 -0
  165. aiq/observability/mixin/type_introspection_mixin.py +183 -0
  166. aiq/observability/processor/__init__.py +14 -0
  167. aiq/observability/processor/batching_processor.py +316 -0
  168. aiq/observability/processor/intermediate_step_serializer.py +28 -0
  169. aiq/observability/processor/processor.py +68 -0
  170. aiq/observability/register.py +36 -39
  171. aiq/observability/utils/__init__.py +14 -0
  172. aiq/observability/utils/dict_utils.py +236 -0
  173. aiq/observability/utils/time_utils.py +31 -0
  174. aiq/profiler/calc/__init__.py +14 -0
  175. aiq/profiler/calc/calc_runner.py +623 -0
  176. aiq/profiler/calc/calculations.py +288 -0
  177. aiq/profiler/calc/data_models.py +176 -0
  178. aiq/profiler/calc/plot.py +345 -0
  179. aiq/profiler/callbacks/langchain_callback_handler.py +22 -10
  180. aiq/profiler/data_models.py +24 -0
  181. aiq/profiler/inference_metrics_model.py +3 -0
  182. aiq/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py +8 -0
  183. aiq/profiler/inference_optimization/data_models.py +2 -2
  184. aiq/profiler/inference_optimization/llm_metrics.py +2 -2
  185. aiq/profiler/profile_runner.py +61 -21
  186. aiq/runtime/loader.py +9 -3
  187. aiq/runtime/runner.py +23 -9
  188. aiq/runtime/session.py +25 -7
  189. aiq/runtime/user_metadata.py +2 -3
  190. aiq/tool/chat_completion.py +74 -0
  191. aiq/tool/code_execution/README.md +152 -0
  192. aiq/tool/code_execution/code_sandbox.py +151 -72
  193. aiq/tool/code_execution/local_sandbox/.gitignore +1 -0
  194. aiq/tool/code_execution/local_sandbox/local_sandbox_server.py +139 -24
  195. aiq/tool/code_execution/local_sandbox/sandbox.requirements.txt +3 -1
  196. aiq/tool/code_execution/local_sandbox/start_local_sandbox.sh +27 -2
  197. aiq/tool/code_execution/register.py +7 -3
  198. aiq/tool/code_execution/test_code_execution_sandbox.py +414 -0
  199. aiq/tool/mcp/exceptions.py +142 -0
  200. aiq/tool/mcp/mcp_client.py +41 -6
  201. aiq/tool/mcp/mcp_tool.py +3 -2
  202. aiq/tool/register.py +1 -0
  203. aiq/tool/server_tools.py +6 -3
  204. aiq/utils/exception_handlers/automatic_retries.py +289 -0
  205. aiq/utils/exception_handlers/mcp.py +211 -0
  206. aiq/utils/io/model_processing.py +28 -0
  207. aiq/utils/log_utils.py +37 -0
  208. aiq/utils/string_utils.py +38 -0
  209. aiq/utils/type_converter.py +18 -2
  210. aiq/utils/type_utils.py +87 -0
  211. {aiqtoolkit-1.2.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/METADATA +53 -21
  212. aiqtoolkit-1.2.0rc2.dist-info/RECORD +436 -0
  213. {aiqtoolkit-1.2.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/WHEEL +1 -1
  214. {aiqtoolkit-1.2.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/entry_points.txt +3 -0
  215. aiq/front_ends/fastapi/websocket.py +0 -148
  216. aiq/observability/async_otel_listener.py +0 -429
  217. aiqtoolkit-1.2.0.dev0.dist-info/RECORD +0 -316
  218. {aiqtoolkit-1.2.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/licenses/LICENSE-3rd-party.txt +0 -0
  219. {aiqtoolkit-1.2.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/licenses/LICENSE.md +0 -0
  220. {aiqtoolkit-1.2.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,436 @@
1
+ aiq/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
+ aiq/agent/base.py,sha256=ZUOfDRBgZD0eUGnlvAbmg-qDwqnUInQ6_ZXXgIWxCAw,9098
3
+ aiq/agent/dual_node.py,sha256=EOYpYzhaY-m1t2W3eiQrBjSfNjYMDttAwtzEEEcYP4s,2353
4
+ aiq/agent/register.py,sha256=EATlFFl7ov5HNGySLcPv1T7jzV-Jy-jPVkUzSXDT-7s,1005
5
+ aiq/agent/react_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
+ aiq/agent/react_agent/agent.py,sha256=Ljm93VqDRsWSIk4n7d11DOuemcNtlp12m8nglEUeyGc,19465
7
+ aiq/agent/react_agent/output_parser.py,sha256=m7K6wRwtckBBpAHqOf3BZ9mqZLwrP13Kxz5fvNxbyZE,4219
8
+ aiq/agent/react_agent/prompt.py,sha256=iGPBU6kh1xbp4QsU1p3o4A0JDov23J1EVM3HSAX6S0A,1713
9
+ aiq/agent/react_agent/register.py,sha256=O0IYtUQ_TVtmaBleYKdpthkspX2fAhB0l8DGYLoE6F0,8128
10
+ aiq/agent/reasoning_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
+ aiq/agent/reasoning_agent/reasoning_agent.py,sha256=1grbjv8c3neFgcTAl8qYeyPJ3B6gcEqkR6I68LOFPT8,9453
12
+ aiq/agent/rewoo_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
13
+ aiq/agent/rewoo_agent/agent.py,sha256=C8zUpbEFJ0De1tzMNvpxY66Qll2ekjAqSKCIcL0ftLA,19009
14
+ aiq/agent/rewoo_agent/prompt.py,sha256=2XsuI-db_qmH02ypx_IDvi6jTak15cqt_4pZkUv9TFk,3929
15
+ aiq/agent/rewoo_agent/register.py,sha256=krm0dUqM5RZpojiLZRpTgAsE0KGqa2NS2QCtlG70EJE,8128
16
+ aiq/agent/tool_calling_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
17
+ aiq/agent/tool_calling_agent/agent.py,sha256=e6VOeBnX_cHXUbnGW8N-ByRtHg4GrbsJecFhA9w0Ksk,5718
18
+ aiq/agent/tool_calling_agent/register.py,sha256=kqcN2uovVBQxrIx5MszBS65opbhBrCRlAw00TlG2i30,5408
19
+ aiq/authentication/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
20
+ aiq/authentication/interfaces.py,sha256=jfsbVx0MTrxZonyTEH0YFcSJoyFHVkfVHFhm9v3jMrY,3317
21
+ aiq/authentication/register.py,sha256=dYChd2HRv-uv4m8Ebo2sLfbVnksT8D3jEWA-QT-MzX0,947
22
+ aiq/authentication/api_key/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
23
+ aiq/authentication/api_key/api_key_auth_provider.py,sha256=08YhKWSS0t5ov1sagLC9DN8xM2Asep8eu5m5yeAzzj8,3947
24
+ aiq/authentication/api_key/api_key_auth_provider_config.py,sha256=U9Rx93XVfEFXzT_fikkE0b3bX4eP8VRB2b52_YjAFjY,5472
25
+ aiq/authentication/api_key/register.py,sha256=2W7GMWNNa_cZvB1uU4an5kNBrYVBY-kX0tgZ9daAF6Y,1147
26
+ aiq/authentication/exceptions/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
27
+ aiq/authentication/exceptions/api_key_exceptions.py,sha256=6wnz951BI77rFYuHxoHOthz-y5oE08uxsuM6G5EvOyM,1545
28
+ aiq/authentication/exceptions/auth_code_grant_exceptions.py,sha256=ycXQRJcKzNTezFGoJ2FuV5suTNteK6obeRfd1pEEpdU,3515
29
+ aiq/authentication/exceptions/call_back_exceptions.py,sha256=2Ik3x6IJdVQ4op9ES_KE2ooCxPoTeM3gCJpqxpxOwto,1481
30
+ aiq/authentication/exceptions/request_exceptions.py,sha256=2fjwZX7QLJk9J10YGHa7T-Q9j5j44iYYKHsFc25lh3w,2049
31
+ aiq/authentication/http_basic_auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
32
+ aiq/authentication/http_basic_auth/http_basic_auth_provider.py,sha256=ulGk37FfmZN3didNWTo826zn-E1hd4t2tVMNxSevRlc,3455
33
+ aiq/authentication/http_basic_auth/register.py,sha256=JFkxcJIPWffiq05r6xj4ShMdjK8iVb9GGC8zOSGYNgs,1235
34
+ aiq/authentication/oauth2/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
35
+ aiq/authentication/oauth2/oauth2_auth_code_flow_provider.py,sha256=6xztmBSsL1Pwaddbo2qX7CaFEguerTJoq4IuKDXb78s,4566
36
+ aiq/authentication/oauth2/oauth2_auth_code_flow_provider_config.py,sha256=SXDGsZAW5LRG1co7Dd0LrAX2wFLrjmf6sx7hSwo4mcM,2204
37
+ aiq/authentication/oauth2/register.py,sha256=zUQlp9L0KcqrUAzx9wLWleBj2snghMR5q10cHuTtPPI,1228
38
+ aiq/builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
39
+ aiq/builder/builder.py,sha256=vePvLcie6wnbqkMOdnMwtJRIbN_E8HUpcuOYNEWrwjU,10084
40
+ aiq/builder/component_utils.py,sha256=KD8c43j095cy9-Zry2Qv89toj5O4ruQXkRKe-Rsk45w,13640
41
+ aiq/builder/context.py,sha256=-MayxRHpSqD9nWDdoedN3rwTkn7Pl_6E80VrNlcBwyQ,10977
42
+ aiq/builder/embedder.py,sha256=M-n2oXSlwE7u-nmpzRbPq6UVWtFtMvrEdKqqsbKQZLk,965
43
+ aiq/builder/eval_builder.py,sha256=P2yqhPkjvkUi_ZwEsBNJi_qTSmHjraH0_9LXPGnLR7s,6614
44
+ aiq/builder/evaluator.py,sha256=O6Gu0cUwQkrPxPX29Vf_-RopgijxPnhy7mhg_j-9A84,1162
45
+ aiq/builder/framework_enum.py,sha256=eYwHQifZ86dx-OTubVA3qhCLRqhB4ElMBYBGA0gYtic,885
46
+ aiq/builder/front_end.py,sha256=Xhvfi4VcDh5EoCtLr6AlLQfbRm8_TyugUc_IRfirN6Y,2225
47
+ aiq/builder/function.py,sha256=lDcMcAhtwYmFihpG3iXb6rkb0goKcJx5H7_WqNJhoRg,12775
48
+ aiq/builder/function_base.py,sha256=lzAFQtpreDx-XVPzInIzKwizpx8rVyI_hDz1m49ueSY,13113
49
+ aiq/builder/function_info.py,sha256=pGPIAL0tjVqLOJymIRB0boI9pzJGdXiPK3KiZvXQsqM,25266
50
+ aiq/builder/intermediate_step_manager.py,sha256=sENuXYQKOwnlEZ7f4lKZ63TTVa6FraT8s_ZM9-vwqa4,7614
51
+ aiq/builder/llm.py,sha256=DcoYCyschsRjkW_yGsa_Ci7ELSpk5KRbi9778Dm_B9c,951
52
+ aiq/builder/retriever.py,sha256=GM7L1T4NdNZKerFZiCfLcQOwsGoX0NRlF8my7SMq3l4,970
53
+ aiq/builder/user_interaction_manager.py,sha256=h84Xj9I_huAcCExENQWlAHlNA8w0TU9UM2NnXni-zUc,2966
54
+ aiq/builder/workflow.py,sha256=nlX1FS2jzEW0VA4FbO5ADqwfllUQaGwqPvtLkFx_OZo,6399
55
+ aiq/builder/workflow_builder.py,sha256=ObMeJHD6YUQSm-pCLIpOpjmu2jHxXWO3m5uL09ulrLQ,46369
56
+ aiq/cli/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
57
+ aiq/cli/entrypoint.py,sha256=vaju3I27FCIJCcsJOFE_d1_rvUMt34AqtJ2Zb3Bpwvc,4839
58
+ aiq/cli/main.py,sha256=yVTX5-5-21OOfG8qAdcK3M1fCQUxdr3G37Mb5OldPQc,1772
59
+ aiq/cli/register_workflow.py,sha256=Hmh1Wp6-XF4suTjnuBibB0mHykyf2FE9jJCvHfRJjJ4,21867
60
+ aiq/cli/type_registry.py,sha256=K_G0czwVWL20J2oyD7fkEoS914YsI8_ip5NbpfJ4AN8,46028
61
+ aiq/cli/cli_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
62
+ aiq/cli/cli_utils/config_override.py,sha256=WuX9ki1W0Z6jTqjm553U_owWFxbVzjbKRWGJINFPCvI,8919
63
+ aiq/cli/cli_utils/validation.py,sha256=GlKpoi3HfE5HELjmz5wk8ezGbb5iZeY0zmA3uxmCrBU,1302
64
+ aiq/cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
65
+ aiq/cli/commands/evaluate.py,sha256=_pqAuvrNKBf0DvGpZFO28vAKBWp6izMpaLbAVnP57_4,4783
66
+ aiq/cli/commands/start.py,sha256=8mnJYSToQ1XuYuRBRSonuOsjSL1wqSm8nwj3tH9cnAM,10097
67
+ aiq/cli/commands/uninstall.py,sha256=tTb5WsyRPPXo511yAGSvSG7U19swbQs8Cf_B4rh7mxQ,3248
68
+ aiq/cli/commands/validate.py,sha256=YfYNRK7m5te_jkKp1klhGp4PdUVCuDyVG4LRW1YXZk0,1669
69
+ aiq/cli/commands/configure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
70
+ aiq/cli/commands/configure/configure.py,sha256=b_rnfThV161rDmuRO0Jbke-10oSn4RJj2q4IdTvL3ng,1107
71
+ aiq/cli/commands/configure/channel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
72
+ aiq/cli/commands/configure/channel/add.py,sha256=BvDI-vPPo_mAj2i7FV_Cf0Q6hF3EqGi9USCAf07Q6S8,1077
73
+ aiq/cli/commands/configure/channel/channel.py,sha256=3y4lCk1PqqKPLlPtBo-B7jkUZ-KuTF2T0QTTIsJgZqo,1221
74
+ aiq/cli/commands/configure/channel/remove.py,sha256=zvLw1gRwF58erC1EWbPiWbPMRYXfNNcf2SKusBqsm24,1104
75
+ aiq/cli/commands/configure/channel/update.py,sha256=vsQzt4vFAg5PsmbRimrKoChtkGUSsVywusOrUxgLWd0,1100
76
+ aiq/cli/commands/info/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
77
+ aiq/cli/commands/info/info.py,sha256=ePgsfHfmfAkjg-MoS9AIfY0Zqm0GSEG3Blid821-woc,1359
78
+ aiq/cli/commands/info/list_channels.py,sha256=OVmb_BtCQiH5j2KVd0srwMUW90JUHyNIU7tqo2GdiGc,1292
79
+ aiq/cli/commands/info/list_components.py,sha256=1qkJazV_dsoCVd4T8Vjv3F7Kg7P7xmzMjLs2o0fkpws,4483
80
+ aiq/cli/commands/info/list_mcp.py,sha256=PKkWMW0xL9LTQghhG_wuhlvRqLQYLd6O3_wRkNJGA9c,8964
81
+ aiq/cli/commands/registry/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
82
+ aiq/cli/commands/registry/publish.py,sha256=uPKJ_pvHkDohvIYlGtkGSoo6KCftBtC-vRDUfmUyceE,3234
83
+ aiq/cli/commands/registry/pull.py,sha256=jMR9NP7rV4g1WuUxGXUXjMgv0tV3UAJTKrTZvstx-Tc,4119
84
+ aiq/cli/commands/registry/registry.py,sha256=oo-dPYoKqmGc9lqw3FN9ADg6mGMHwfM14hXxnzwQG1s,1337
85
+ aiq/cli/commands/registry/remove.py,sha256=ysyfA38NKFZpktjKqDkVK54e9AakpAmBGym2OF6xSuo,3915
86
+ aiq/cli/commands/registry/search.py,sha256=1AhW5Q5GL8SsR6fKgG0rbsd2E0nodm-ehYZwqX2SGgA,5125
87
+ aiq/cli/commands/sizing/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
88
+ aiq/cli/commands/sizing/calc.py,sha256=jJq4BHrdtaiPK9P7f-AblvEouq1fbFZxnGn3GGtWsHY,11007
89
+ aiq/cli/commands/sizing/sizing.py,sha256=-Hr9mz_ScEMtBbn6ijvmmWVk0WybLeX0Ryi4qhDiYQU,902
90
+ aiq/cli/commands/workflow/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
91
+ aiq/cli/commands/workflow/workflow.py,sha256=gzd_UXIMGq_NBRZiS_WHh3Dimuw9aTdncREVh8KItJI,1342
92
+ aiq/cli/commands/workflow/workflow_commands.py,sha256=0Iy0nOrMhJcxT_P8zL7pYDmNdUBHEHf0g3ZdYXOb-hw,11856
93
+ aiq/cli/commands/workflow/templates/__init__.py.j2,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
94
+ aiq/cli/commands/workflow/templates/config.yml.j2,sha256=KkZl1fOMVQVFBW-BD_d0Lu8kQgNBtjNpfojhSCPu4uA,222
95
+ aiq/cli/commands/workflow/templates/pyproject.toml.j2,sha256=tDV7-vbt8Of82OEdSOijtWS5YJdW1xw2S-r3a8lCbdo,733
96
+ aiq/cli/commands/workflow/templates/register.py.j2,sha256=SlOFmIZakPDu_E6DbIhUZ3yP8KhTrAQCFGBuhy9Fyg4,170
97
+ aiq/cli/commands/workflow/templates/workflow.py.j2,sha256=NRp0MP8GtZByk7lrHp2Y5_6iEopRK2Wyrt0v0_2qQeo,1226
98
+ aiq/data_models/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
99
+ aiq/data_models/api_server.py,sha256=Jx-IZb_lnddjeQxkIBPYSosoEdZYvRWmkVURAot-FbM,25404
100
+ aiq/data_models/authentication.py,sha256=TShP47-z9vIrkV3S4sadm0QF7YUkEWtsToKSBiHS3NI,7349
101
+ aiq/data_models/common.py,sha256=y_8AiWmTEaMjCMayVaFYddhv2AAou8Pr84isHgGxeUg,5874
102
+ aiq/data_models/component.py,sha256=_HeHlDmz2fgJlVfmz0tG_yCyes86hQNaQwSWz1FDQvk,1737
103
+ aiq/data_models/component_ref.py,sha256=_6DeNWJSHN3GSoAFe0RCQkCJ31_WA9GR3nP6vsxMj7Q,4594
104
+ aiq/data_models/config.py,sha256=I7-9kJqpr6IvqmTU4nnALjDmq8YRlrKhYzq6g7DrJeo,17124
105
+ aiq/data_models/dataset_handler.py,sha256=liMB3xRohkr4VTMmNWPvWi9qhbhlJQfQK36g5Rknweo,4027
106
+ aiq/data_models/discovery_metadata.py,sha256=OcITQc5VeML4bTHurrsMNiK_oB3z7wudMxcyN7LI8pY,12785
107
+ aiq/data_models/embedder.py,sha256=nPhthEQDtzAMGd8gFRB1ZfJpN5M9DJvv0h28ohHnTmI,1002
108
+ aiq/data_models/evaluate.py,sha256=WBeABZsIa6W04MPj24SRu4s-ty2PkJ7_4SLojXmj5Pk,4704
109
+ aiq/data_models/evaluator.py,sha256=bd2njsyQB2t6ClJ66gJiCjYHsQpWZwPD7rsU0J109TI,939
110
+ aiq/data_models/front_end.py,sha256=z8k6lSWjt1vMOYFbjWQxodpwAqPeuGS0hRBjsriDW2s,932
111
+ aiq/data_models/function.py,sha256=M_duXVXL5MvYe0WVLvqEgEzXs0UAYNSMfy9ZTpxuKPA,1013
112
+ aiq/data_models/function_dependencies.py,sha256=NOPUF7W-OxiJ76nI8MwgVs6kky-rQjgRkLeTJh-7pvg,2732
113
+ aiq/data_models/interactive.py,sha256=qOkxyYPQYEBIBMDAA1rjfYcdvf6-iCM4qPV8Awc4VGw,8794
114
+ aiq/data_models/intermediate_step.py,sha256=g__7FC2DcecSZ6L_VBsH6PYx0ViGdCgeU0iHdsvOEd8,11734
115
+ aiq/data_models/invocation_node.py,sha256=nDRylgzBfJduGA-lme9xN4P6BdOYj0L6ytLHnTuetMI,1618
116
+ aiq/data_models/its_strategy.py,sha256=C3fLz_IPDQQWlOKZhszz6mvqbcLzv9FZd_RWgoFlmiA,1119
117
+ aiq/data_models/llm.py,sha256=PCTeCPg2YnYk5tvSu7XOEVr2Cg_wSkjAdVsoiTg8Joo,968
118
+ aiq/data_models/logging.py,sha256=1QtVjIQ99PgMYUuzw4h1FAoPRteZY7uf3oFTqV3ONgA,940
119
+ aiq/data_models/memory.py,sha256=IKwe7CflCto30j4yI5yQtq8DXfMilAJ17S5NcsSDrOQ,1052
120
+ aiq/data_models/object_store.py,sha256=S8YY6i8ALgRPuggUI1FCG-xbvwPWuaCg1lJnZOx5scM,1515
121
+ aiq/data_models/profiler.py,sha256=z3IlEhj-veB4Yz85271bTkScSUkVwK50tR3dwlDRgcE,1781
122
+ aiq/data_models/registry_handler.py,sha256=g1rFaz4uSydMJn7qpdX-DNHJd_rNf8tXYN49dLDYHPo,968
123
+ aiq/data_models/retriever.py,sha256=UOfss4sru5ku5E8YZYN5qz4MVbFi2VwvpNUPVp9hsnQ,1202
124
+ aiq/data_models/retry_mixin.py,sha256=s7UAhAHhlwTJ3uz6POVaSD8Y5DwKnU8mmo7OkRzeaW8,1863
125
+ aiq/data_models/span.py,sha256=t93UrZA4IcyAob61hmCjHKKhdsKVdnFcNjykPffVFmk,6501
126
+ aiq/data_models/step_adaptor.py,sha256=h7nVAwdgbuHd1e1-SR5jY9nkDMBDGqzTzrl-4lBQX7o,2615
127
+ aiq/data_models/streaming.py,sha256=sSqJqLqb70qyw69_4R9QC2RMbRw7UjTLPdo3FYBUGxE,1159
128
+ aiq/data_models/swe_bench_model.py,sha256=dmgU1M-lL7bUO3gD7wkeDkffPf65cxF1ijgELkR3i64,1754
129
+ aiq/data_models/telemetry_exporter.py,sha256=KZMcxfiRT5iwPLVJHNBWUzdlmYwWEP54CqokYooyD40,998
130
+ aiq/embedder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
131
+ aiq/embedder/langchain_client.py,sha256=AWQ8lzo5xkV0ZfyOlrCUGF1EemZqcbeuqUrLi7YLs2I,1751
132
+ aiq/embedder/nim_embedder.py,sha256=9M_nDQ5adtQDG-fWnU8HdYjiHDV6QdjK-ZWmh_qlCJ0,2517
133
+ aiq/embedder/openai_embedder.py,sha256=F4n_ZlR_vmiG05Pr_14EsJtMNkhjWu1BxWfPc55Xqlg,2008
134
+ aiq/embedder/register.py,sha256=3MTZrfNQKp6AZTbfaA-PpTnyXiMyu-8HH9JnDCC0v9o,978
135
+ aiq/eval/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
136
+ aiq/eval/config.py,sha256=RI2XrEk_rlE8A9i-j1tjju7h29AI0rNPDVSwX6z-Mkk,2271
137
+ aiq/eval/evaluate.py,sha256=abDB1-7I3pCXsA7ZYqsVyy57YwF969SZqC_5_leLsVE,23436
138
+ aiq/eval/intermediate_step_adapter.py,sha256=4cSsGgFBvNjXnclk5FvZnQaFEdeulp7VEdRWKLcREAQ,4498
139
+ aiq/eval/register.py,sha256=QOHJqA2CQixeWMC9InyKbzXo1jByvrntD_m9-2Mvg9k,1076
140
+ aiq/eval/remote_workflow.py,sha256=t0uZcS56tu0Vw6jyao5aYP4muXlTB8geYvK6HWDfWgM,6018
141
+ aiq/eval/runtime_event_subscriber.py,sha256=2VM8MqmPc_EWPxxrDDR9naiioZirkJUfGwzbXQqbdZA,1906
142
+ aiq/eval/usage_stats.py,sha256=_d_ErIvSKDN8wuvOyPn01kqM7zlFpwVxqOonaCV5XtY,1319
143
+ aiq/eval/dataset_handler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
144
+ aiq/eval/dataset_handler/dataset_downloader.py,sha256=Zvfbd-fPOhB9n8ZiCBaBKW0y-5v97mQAy3dkBL0OFZ0,4553
145
+ aiq/eval/dataset_handler/dataset_filter.py,sha256=mop6wa4P_QtQ5QkfXv-hVBm3EMerfNECSTJGGDB1YWE,2115
146
+ aiq/eval/dataset_handler/dataset_handler.py,sha256=DJa25OX3iZkh3CTOWB5cH2qRFRSd1D-39p72_OsMrWQ,11161
147
+ aiq/eval/evaluator/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
148
+ aiq/eval/evaluator/base_evaluator.py,sha256=5kqOcTYNecnh9us_XvV58pj5tZI82NGkVN4tg9-R_ZE,3040
149
+ aiq/eval/evaluator/evaluator_model.py,sha256=5cxe3mqznlNGzv29v_VseYU7OzoT1eTf7hgSPQxytsM,1440
150
+ aiq/eval/rag_evaluator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
151
+ aiq/eval/rag_evaluator/evaluate.py,sha256=SNVSndTwaACs8guWnB1SPKXpnZB6trOhhtkK01MSlA8,8399
152
+ aiq/eval/rag_evaluator/register.py,sha256=vmUxgMJsI42scapLFLvFI6oqXgu9Rl_XhiNedy5-Cqw,5889
153
+ aiq/eval/runners/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
154
+ aiq/eval/runners/config.py,sha256=1iIOAQEgQs8y-JtNmFOO_1j3AKcnLMIZdHQJBsT2pD4,1403
155
+ aiq/eval/runners/multi_eval_runner.py,sha256=EXf3s5W8BxyLVhxWyZrejzoLfR9IPcsqyhPyuFNWefc,1986
156
+ aiq/eval/swe_bench_evaluator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
157
+ aiq/eval/swe_bench_evaluator/evaluate.py,sha256=kNukRruq1EM1RsGLvpVuC22xcP0gpn9acF3edGak9vY,9858
158
+ aiq/eval/swe_bench_evaluator/register.py,sha256=sTb74F7w4iuI0ROsEJ4bV13Nt1GEWQn7UvO2O0HXwXk,1537
159
+ aiq/eval/trajectory_evaluator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
160
+ aiq/eval/trajectory_evaluator/evaluate.py,sha256=Y51KMhJ9t8AoYWrQlrwipc2CtgIXA9IUGZTbKegtsnw,3257
161
+ aiq/eval/trajectory_evaluator/register.py,sha256=kktT4fu5_1Cou-iohD3YhQevsWiR3TA5NpFSweVz0eQ,1709
162
+ aiq/eval/tunable_rag_evaluator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
163
+ aiq/eval/tunable_rag_evaluator/evaluate.py,sha256=f4jfn9VVLmkOg631TQr2wy7hPwGMJMsQa4kmXsu0-Uc,13069
164
+ aiq/eval/tunable_rag_evaluator/register.py,sha256=q4p2rFyMzWmaINJc961ZV4jzIlAN4GfWsoImHo0ovsY,2558
165
+ aiq/eval/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
166
+ aiq/eval/utils/output_uploader.py,sha256=lkV63Jr97YuG1vr04uOZDvs9e1pGP4FbJykRxS2d7a4,5579
167
+ aiq/eval/utils/tqdm_position_registry.py,sha256=9CtpCk1wtYCSyieHPaSp8nlZu6EcNUOaUz2RTqfekrA,1286
168
+ aiq/eval/utils/weave_eval.py,sha256=zmRLPSYiOhWidINWTFiej2qg5_3AVrpN2HjEJCADmZw,7287
169
+ aiq/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
170
+ aiq/experimental/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
171
+ aiq/experimental/decorators/experimental_warning_decorator.py,sha256=Wz0byTMg9tLmPI40aB1I_Mc9I4F3dl78rr9S0-dreEQ,4947
172
+ aiq/experimental/inference_time_scaling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
173
+ aiq/experimental/inference_time_scaling/register.py,sha256=uFatD1RrY16-xcFyysA_bPRCUQDHRvSb53-l6I_qdo0,1600
174
+ aiq/experimental/inference_time_scaling/editing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
175
+ aiq/experimental/inference_time_scaling/editing/iterative_plan_refinement_editor.py,sha256=AjpySH2utebD2Pi6jlxUs4h_8Y_AdPhYluiPMD-gnFg,6162
176
+ aiq/experimental/inference_time_scaling/editing/llm_as_a_judge_editor.py,sha256=7PLhitsUAbATWtLVtTfwqzzqakyp4uxolVuwFqGbq3Y,8564
177
+ aiq/experimental/inference_time_scaling/editing/motivation_aware_summarization.py,sha256=Ai1iiQ3OdQQezgPcmBfUcPXw8LELe88NeO7oxKB_aXw,4528
178
+ aiq/experimental/inference_time_scaling/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
179
+ aiq/experimental/inference_time_scaling/functions/execute_score_select_function.py,sha256=2zQ388pBael0TWura_dMSON0RK42csieNP_v0cH0rRw,4565
180
+ aiq/experimental/inference_time_scaling/functions/its_tool_orchestration_function.py,sha256=TKgGlDVmqiDUZw9_cchDjnQEqbtFuzmfccH3MhSQkG8,8446
181
+ aiq/experimental/inference_time_scaling/functions/its_tool_wrapper_function.py,sha256=alz0qWK8ZtZalJzrGojv8Pz6uNdUP2cvLPaBjdBUgCU,6907
182
+ aiq/experimental/inference_time_scaling/functions/plan_select_execute_function.py,sha256=REkVEuu66TwVWh49u5XDw80-rpQdHSWow3VoRCJKncI,9870
183
+ aiq/experimental/inference_time_scaling/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
184
+ aiq/experimental/inference_time_scaling/models/editor_config.py,sha256=LjLBuW9HPUPj7wrEZ9WxGasiQ_oblZ3_RxHsaS0ZgXQ,7153
185
+ aiq/experimental/inference_time_scaling/models/its_item.py,sha256=N76rcN29ltUL7-lP3K7yIUJ7w0snZ5fBTLszOwQQZGc,2412
186
+ aiq/experimental/inference_time_scaling/models/scoring_config.py,sha256=X1mFSds9pnih3XhhLkcLxyN3kCJfk7Ygb5GunOKJbII,5500
187
+ aiq/experimental/inference_time_scaling/models/search_config.py,sha256=HSjBziUXtZFeglY23sJfeC5UT0w-1mOcSRRv6K5e1i0,6487
188
+ aiq/experimental/inference_time_scaling/models/selection_config.py,sha256=UsT2NCzJhT_U7HU32fpDbfJNgdNLxVHL9UzL-r8UyOI,7867
189
+ aiq/experimental/inference_time_scaling/models/stage_enums.py,sha256=UNQwUqRFoRe5pgasUbBRkRil7zs_NAsJlgFbWhcAU_Y,1284
190
+ aiq/experimental/inference_time_scaling/models/strategy_base.py,sha256=fw2lth61-MY_VMxHZWELcFvRwYv__PK3SWD-sYy96b0,2558
191
+ aiq/experimental/inference_time_scaling/models/tool_use_config.py,sha256=WX6Z2ODGElDA7Io8wBxtDkk3jUQGtHLukS6a-ivSZnE,1617
192
+ aiq/experimental/inference_time_scaling/scoring/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
193
+ aiq/experimental/inference_time_scaling/scoring/llm_based_agent_scorer.py,sha256=Ncom3-10E-S_-eLCPpg5rkDqV9hTLpryFPF5wAJ4vJM,6633
194
+ aiq/experimental/inference_time_scaling/scoring/llm_based_plan_scorer.py,sha256=OzvvlX3dIFcBrPj7__qgJBXsJ-1pVSrCCOdO7fgQAPI,6526
195
+ aiq/experimental/inference_time_scaling/scoring/motivation_aware_scorer.py,sha256=DOodNRmLqpDs628D3_nW9R_jKq4zE21T-qauzXkYo58,4613
196
+ aiq/experimental/inference_time_scaling/search/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
197
+ aiq/experimental/inference_time_scaling/search/multi_llm_planner.py,sha256=r0NuneI9fZjWlRlGvk4zeY31phepPRrAG9uKUU2xvvA,5796
198
+ aiq/experimental/inference_time_scaling/search/multi_query_retrieval_search.py,sha256=n2dO3P1R5al9NQziGgSSPPRXrUDZrcGSxWqKyW_tC5Q,5306
199
+ aiq/experimental/inference_time_scaling/search/single_shot_multi_plan_planner.py,sha256=zNIx6mMA75YsQRN3F90TBJzxgWHBnzm5YOboWE0R-bI,5716
200
+ aiq/experimental/inference_time_scaling/selection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
201
+ aiq/experimental/inference_time_scaling/selection/best_of_n_selector.py,sha256=mOLYHxcUugfGQx4XIWD22q8nm1eQK5zjTx_xvNaDCzA,2514
202
+ aiq/experimental/inference_time_scaling/selection/llm_based_agent_output_selector.py,sha256=t1lkvgBHw3M7Fx8pmUwkDIPvtDP3r0L3ZCtPedgay7g,5823
203
+ aiq/experimental/inference_time_scaling/selection/llm_based_output_merging_selector.py,sha256=biJr1dDVlKdiLQx44BXcgGs829waMgAdUafBp9yxNgY,6781
204
+ aiq/experimental/inference_time_scaling/selection/llm_based_plan_selector.py,sha256=2l7ZDUWpc_mjoIjJ0pU_sGmjQQoE6vdE8EV5pF0jLcA,5636
205
+ aiq/experimental/inference_time_scaling/selection/threshold_selector.py,sha256=ncqeUOCyRyIJDyGyP0xqIncCBKBxHoq7Jjko28LF18o,2440
206
+ aiq/front_ends/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
207
+ aiq/front_ends/register.py,sha256=OKv1xi-g8WHtUMuIPhwjG6wOYqaGDD-Q9vDtKtT9d1Y,889
208
+ aiq/front_ends/console/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
209
+ aiq/front_ends/console/authentication_flow_handler.py,sha256=pKgaAnBfr3Zc5V3yPiyU-oVZyCC-dNwjvhLn1gSxnUE,9566
210
+ aiq/front_ends/console/console_front_end_config.py,sha256=vI81IK9GJll0t9P-3yb1JcUda5PLfdvyKc-636ZSqYU,1327
211
+ aiq/front_ends/console/console_front_end_plugin.py,sha256=3qOLXKI6JCE33oh2MylrxMm_EonZlUCGjxZrPvTJ_4k,4535
212
+ aiq/front_ends/console/register.py,sha256=a84M0jWUFTgOQVyrUiS7UJcxx84i1zhCb1yRkjhapiQ,1159
213
+ aiq/front_ends/cron/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
214
+ aiq/front_ends/fastapi/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
215
+ aiq/front_ends/fastapi/fastapi_front_end_config.py,sha256=jhyc4_29bWXAxQsuPmAFGzVoC490-8NUP68IlCpGz0c,10619
216
+ aiq/front_ends/fastapi/fastapi_front_end_controller.py,sha256=vs-VIWHd-YW1w2t-ifYqJ0jy5_SoD-11VyeojW_up5Y,2596
217
+ aiq/front_ends/fastapi/fastapi_front_end_plugin.py,sha256=EKhkDS4_Gmx8pEQqK3Dyt1oPLp6gly4lLZZmJvdVyio,4428
218
+ aiq/front_ends/fastapi/fastapi_front_end_plugin_worker.py,sha256=sXy2ghwiWd8UAAooECpes-TtEtYR4fSXSsHgo3oZbjI,52498
219
+ aiq/front_ends/fastapi/intermediate_steps_subscriber.py,sha256=2Y3ZXfiu8d85qJRWbT3-ex6iFDuXO6TnNtjQ6Cjl-tc,3131
220
+ aiq/front_ends/fastapi/job_store.py,sha256=AZC0a8miKqSI9YbW19-AA_GPITVY9bYHXn9H8saNrfQ,6389
221
+ aiq/front_ends/fastapi/main.py,sha256=HdH_KwqRYDGbUJjx894ex3UOJCsiu77Qpx0XSHK4XN8,2917
222
+ aiq/front_ends/fastapi/message_handler.py,sha256=_NrOQ02vNJ4Ad6vDkcuIkmkhRXQ2DHe2BwUR8o_oaB0,13975
223
+ aiq/front_ends/fastapi/message_validator.py,sha256=NqjeIG0InGAS6yooEnTaYwjfy3qtQHNgmdJ4zZlxgSQ,17407
224
+ aiq/front_ends/fastapi/register.py,sha256=-Vr6HEDy7L1IIBQZy6VFKZWWKYNtSI-cxk3l4ZPPLko,1159
225
+ aiq/front_ends/fastapi/response_helpers.py,sha256=JnOOvurlkhh6akpzkKDNh1xCi5ClQTePyW_ScEY1pLo,9111
226
+ aiq/front_ends/fastapi/step_adaptor.py,sha256=YcCrvT91-94XOINr0uxi73OkAePibQHn_Bd-alU4oCY,12577
227
+ aiq/front_ends/fastapi/auth_flow_handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
228
+ aiq/front_ends/fastapi/auth_flow_handlers/http_flow_handler.py,sha256=FI_TdA4vKy0lpT4KoIX5RE_RzzwZIyvapqOdf9aG2qM,1280
229
+ aiq/front_ends/fastapi/auth_flow_handlers/websocket_flow_handler.py,sha256=EUh3b6xjXVvCHTnjaf9gT8ilXWcYguLTcTI-c80pUls,4862
230
+ aiq/front_ends/fastapi/html_snippets/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
231
+ aiq/front_ends/fastapi/html_snippets/auth_code_grant_success.py,sha256=BNpWwzmA58UM0GK4kZXG4PHJy_5K9ihaVHu8SgCs5JA,1131
232
+ aiq/front_ends/mcp/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
233
+ aiq/front_ends/mcp/mcp_front_end_config.py,sha256=Mp2mgCh1pJkegih5DEJ3t0OKEiCiCg8-rM5vnLpLS6U,1470
234
+ aiq/front_ends/mcp/mcp_front_end_plugin.py,sha256=XXyu5A5zWLH-eZgLDGcdCELXkwC1YPb94z6wddadH8I,3677
235
+ aiq/front_ends/mcp/register.py,sha256=9u5AJVH5UXiniP9bmsOjpfWVzQLHFlUPfL5HZ10Qwnw,1179
236
+ aiq/front_ends/mcp/tool_converter.py,sha256=Pgb06Gtb8MVWeV_dAaI4Tl0Hono_mSuJAHRxTvSONHQ,9840
237
+ aiq/front_ends/simple_base/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
238
+ aiq/front_ends/simple_base/simple_front_end_plugin_base.py,sha256=HzcJFHZUZFnZJdw4YcRTG6nQbTDoBQSgKO6vwifSomk,1684
239
+ aiq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
240
+ aiq/llm/aws_bedrock_llm.py,sha256=FFlwS7xZYSTRuPWU_vxvTbmPTbuZpN0ByVxYukEThM4,2836
241
+ aiq/llm/nim_llm.py,sha256=_9WwR4Pt9RCg8RG9oBeYxSt94KcZsMEVoKOFjig_2Zo,2179
242
+ aiq/llm/openai_llm.py,sha256=T4GOcQQUIdUcVsdFr3ocWuMSBxZ2kcSO8OL3YdXRK9o,2230
243
+ aiq/llm/register.py,sha256=GMsFzhupP_rwSVkIoJUT8j0CRhWyC4X58ihnO_l9OkM,899
244
+ aiq/llm/utils/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
245
+ aiq/llm/utils/env_config_value.py,sha256=SBpppIRszDXNcEEG2L6kVojY7LH_EpHfK7bu92TGrE4,3568
246
+ aiq/llm/utils/error.py,sha256=gFFDG_v_3hBZVWpcD7HWkno-NBHDjXae7qDGnfiCNwA,820
247
+ aiq/memory/__init__.py,sha256=sNqiLatqyTNSBUKKgmInOvCmebkuDHRTErZaeOaE8C8,844
248
+ aiq/memory/interfaces.py,sha256=lyj1TGr3Fhibul8Y64Emj-BUEqDotmmFoVCEMqTujUA,5531
249
+ aiq/memory/models.py,sha256=c5dA7nKHQ4AS1_ptQZcfC_oXO495-ehocnf_qXTE6c8,4319
250
+ aiq/meta/module_to_distro.json,sha256=1XV7edobFrdDKvsSoynfodXg_hczUWpDrQzGkW9qqEs,28
251
+ aiq/meta/pypi.md,sha256=N1fvWaio3KhnAw9yigeM-oWaLuT5i_C7U_2UVzyPbks,4386
252
+ aiq/object_store/__init__.py,sha256=7JInpxFZUdqigaBaXDzUj0KTlv_2tiM-SuTOlSvadkg,847
253
+ aiq/object_store/in_memory_object_store.py,sha256=_k_5V3sOZh-S3UtsVaKWPnQaeQc7eKY0-Ui6Tx2uXW4,2453
254
+ aiq/object_store/interfaces.py,sha256=5NbsE9TccihOf5ScG04hE1eNOaiajOZIUOeK_Kvukk8,2519
255
+ aiq/object_store/models.py,sha256=yAEa7oZgax7OwObynv0MFCMo43xTGps5xz3SIY6-0VM,1425
256
+ aiq/object_store/register.py,sha256=M93V17RxBXzGZaAmWboDw-S2XP7lrMEyzdlxXqC0f30,788
257
+ aiq/observability/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
258
+ aiq/observability/exporter_manager.py,sha256=50Vzsagf84s87yAiGXQNWXr7pooTEkA5SAUikUko02o,13866
259
+ aiq/observability/register.py,sha256=hR3uB2aoB6YQatbOvgBy9UMV9Nc28KJYSsbx2eUKUPg,4267
260
+ aiq/observability/exporter/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
261
+ aiq/observability/exporter/base_exporter.py,sha256=uVr2qssNwJPVJP0P9fD7tNVSdUmj4SWx4GVHqkA1_3s,16639
262
+ aiq/observability/exporter/exporter.py,sha256=QjEtbaTC7LTQpuzdcvRK7gUs6OIlHzQnhzD2pGmXyws,2391
263
+ aiq/observability/exporter/file_exporter.py,sha256=Z1DhUSpTu1SmN282pPRTU20QJZox49jZ75ARsUkgZAk,1496
264
+ aiq/observability/exporter/processing_exporter.py,sha256=4X-LIiJFWryo8ZRrgmz2A5X-JyU3EbVirT6y8rWL9Ck,12310
265
+ aiq/observability/exporter/raw_exporter.py,sha256=9MFukCkJE7qHBWf2V2rnsCdzL1HRYRarcudveTsvS8w,1862
266
+ aiq/observability/exporter/span_exporter.py,sha256=9KHedqEAZnPyoPNbVLb4yGsT_2LCYPSCbh3XLCoN9NA,12062
267
+ aiq/observability/mixin/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
268
+ aiq/observability/mixin/batch_config_mixin.py,sha256=DixQq-jRhBFJvpOX-gq7GvPmZCPOXQdacylyEuhZ6y0,1399
269
+ aiq/observability/mixin/collector_config_mixin.py,sha256=3iptkRH9N6JgcsPq7GyjjJVAoxjd-l42UKE7iSF4Hq8,1087
270
+ aiq/observability/mixin/file_mixin.py,sha256=GLD0YMxTrvXLnA4_l8ZnYk5QShO2EmStNhsNh853yBI,12293
271
+ aiq/observability/mixin/file_mode.py,sha256=Rq7l8UegECub5QCyCAYwhyL_Jul386gW-ANmtMmv2G4,837
272
+ aiq/observability/mixin/resource_conflict_mixin.py,sha256=mcUp3Qinmhiepq3DyRvp9IaKGYtJfDgQVB-MuyVkWvk,5243
273
+ aiq/observability/mixin/serialize_mixin.py,sha256=DgRHJpXCz9qHFYzhlTTx8Dkj297EylCKK3ydGrH5zOw,2478
274
+ aiq/observability/mixin/type_introspection_mixin.py,sha256=VCb68SY_hitWrWLaK2UHQLkjn2jsgxSn9593T2N3zC0,6637
275
+ aiq/observability/processor/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
276
+ aiq/observability/processor/batching_processor.py,sha256=unPc4bZcBgLzMTPVp2x9ezXOPsrZj9LZvrMDsld26KU,13441
277
+ aiq/observability/processor/intermediate_step_serializer.py,sha256=UQgcHauq568TqVmF_FrInsE5Q_Piryrf_fDnuJbRtAc,1249
278
+ aiq/observability/processor/processor.py,sha256=WY-olVcVe2VNB9iopxVIj4xwVODlJ8UcHKqPBh7UqsE,2526
279
+ aiq/observability/utils/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
280
+ aiq/observability/utils/dict_utils.py,sha256=DcNhZ0mgcJ-QQfsCl9QSGL-m_jTuHhr1N-v43ZCAMik,7371
281
+ aiq/observability/utils/time_utils.py,sha256=V8m-e3ldUgwv031B17y29yLXIowdlTH4QW8xDw9WKvk,1071
282
+ aiq/plugins/.namespace,sha256=Gace0pOC3ETEJf-TBVuNw0TQV6J_KtOPpEiSzMH-odo,215
283
+ aiq/profiler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
284
+ aiq/profiler/data_frame_row.py,sha256=vudqk1ZzZtlZln2Ir43mPl3nwNc0pQlhwbtdY9oSKtI,1755
285
+ aiq/profiler/data_models.py,sha256=xt6pRyfqfm5u4WerD7eTUDxxu9Twsx8qQplAul4wXwU,1026
286
+ aiq/profiler/inference_metrics_model.py,sha256=Thz3OHBDzGrpPYaOm8m8_pNeEA_q0yDlUUDHFkQ3U90,1481
287
+ aiq/profiler/intermediate_property_adapter.py,sha256=XZ_A8f2S5M-EJSkErY6I750Y8HAZPdXsr6Cpb1wXlNM,3537
288
+ aiq/profiler/profile_runner.py,sha256=aNwzcgw9udNh_rgTjtPCScAfvn7ug63LVzW13AvhWRY,22363
289
+ aiq/profiler/utils.py,sha256=hNh_JfxXDrACIp4usXtlriTfVuYUkk3Pv-x74K34MQg,8180
290
+ aiq/profiler/calc/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
291
+ aiq/profiler/calc/calc_runner.py,sha256=r65olaY-AjuE0lCwlC7Ob9MJ3opopYzGuTqDwx5UOd4,30379
292
+ aiq/profiler/calc/calculations.py,sha256=GUDceQOs52QMmHg0u6BVgq_cmFKjXLbuQdNyd7UVZ0A,12327
293
+ aiq/profiler/calc/data_models.py,sha256=Wq3REnYWiJR5EWMH-ZzouJRo2NUi78DfyKevNmXKqX8,5662
294
+ aiq/profiler/calc/plot.py,sha256=GW1cK8U2w6hIzKp6ItciCmxHR0KeUCw7n5EvpcPsoio,12222
295
+ aiq/profiler/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
296
+ aiq/profiler/callbacks/agno_callback_handler.py,sha256=aDAUY6GDIUtly6KowXXKUqLc7NbE6khg1aXT1AritaA,14930
297
+ aiq/profiler/callbacks/base_callback_class.py,sha256=BFZMkb-dPoHAnM_4jVXX08hD8Vi2n8thyftVlUxfE24,745
298
+ aiq/profiler/callbacks/langchain_callback_handler.py,sha256=tG9OX2C3Rsqpew2Jaj2LLkZkwEztbeQ-oLVIBUDPL48,12511
299
+ aiq/profiler/callbacks/llama_index_callback_handler.py,sha256=AXUtfsUwif-rUFoVaRnMeSe29TcpT9Z8nxWcJWD2PiQ,9334
300
+ aiq/profiler/callbacks/semantic_kernel_callback_handler.py,sha256=uYlq0Lku6U1AMg5GoPpJoZm6lvgIjLQN5_uIbswL4bA,11157
301
+ aiq/profiler/callbacks/token_usage_base_model.py,sha256=X0b_AbBgVQAAbgbDMim-3S3XdQ7PaPs9qMUACvMAe5o,1104
302
+ aiq/profiler/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
303
+ aiq/profiler/decorators/framework_wrapper.py,sha256=wMkZJPBkBe1uqf-JIqvkg80hkxno4cBMnQERFziFmi8,5438
304
+ aiq/profiler/decorators/function_tracking.py,sha256=jwBebxAUxbSu8aWPae-_hOel3x6pZ50NCvaEriHXpVw,11110
305
+ aiq/profiler/forecasting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
306
+ aiq/profiler/forecasting/config.py,sha256=5BhMa8csuPCjEnTaNQjo_2IoO7esh1ch02MoSWkvwPw,791
307
+ aiq/profiler/forecasting/model_trainer.py,sha256=fnWqUvgrgv3fFMguItmvcSWDQwlra4hJXnCTr2u1MbQ,2456
308
+ aiq/profiler/forecasting/models/__init__.py,sha256=pLpWi7u1UrguKIYD-BYu8IExvJLX_U2cuW3Ifp3zCOY,937
309
+ aiq/profiler/forecasting/models/forecasting_base_model.py,sha256=6-oe3jn-X9_m3QvkWAEJ9m7sMLlZ7VDt6yfNyWK-_18,1219
310
+ aiq/profiler/forecasting/models/linear_model.py,sha256=quIKTY0NxaKB4-VIffP6YHmnWXuqp1FV0a2Nt9nYWPI,6993
311
+ aiq/profiler/forecasting/models/random_forest_regressor.py,sha256=139cOJTnJKkYOOU9ZDxPq9BvJ5nj1sV0JZIcjpy1JfU,9533
312
+ aiq/profiler/inference_optimization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
313
+ aiq/profiler/inference_optimization/data_models.py,sha256=lK6EPAgEPtJMMpe3SFoDqx4BntJVo8fMQzLfqOMKpB0,11884
314
+ aiq/profiler/inference_optimization/llm_metrics.py,sha256=uvd2KMPPiEjZbuLqK7iJFK8UUrOUb5F3n53T76vKzd8,9482
315
+ aiq/profiler/inference_optimization/prompt_caching.py,sha256=LGfxJG4R2y4vMFoiFztJkdeBivhBOO4sk7cKY-llTXk,6505
316
+ aiq/profiler/inference_optimization/token_uniqueness.py,sha256=OCNlVmemMLS2kt0OZIXOGt8MbrTy5mbdhSMPYHs31a4,4571
317
+ aiq/profiler/inference_optimization/workflow_runtimes.py,sha256=lnGa0eTpHiDEbx9rX-tcx100qSd6amePLlgb4Gx7JBc,2664
318
+ aiq/profiler/inference_optimization/bottleneck_analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
319
+ aiq/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py,sha256=yr81PsQ4TcrEnuPDlRwhL9Hcox3gO855DsS-BDo00u0,16732
320
+ aiq/profiler/inference_optimization/bottleneck_analysis/simple_stack_analysis.py,sha256=VZLBgsIUGOkY0ZUCLHQM4LpBQpJBM5JKRTUBGyoOFWU,11100
321
+ aiq/profiler/inference_optimization/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
322
+ aiq/profiler/inference_optimization/experimental/concurrency_spike_analysis.py,sha256=J-oMRCEnd6I1XFXiyLUu8VPR745ptnzgzvn0Opsi208,16953
323
+ aiq/profiler/inference_optimization/experimental/prefix_span_analysis.py,sha256=nD46SCVi7zwxdPXRN5c61O58_OgMA2WCfaZdRJRqgP4,16600
324
+ aiq/registry_handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
325
+ aiq/registry_handlers/metadata_factory.py,sha256=Urr_7mLLpoU0VPjl2Nknl9aZhzaAAwb66ydTGBBrIwc,2778
326
+ aiq/registry_handlers/package_utils.py,sha256=WCENHOO55Vhwv9DbLYjR9SMu_ERzxMKX6oO9ywe9cpM,7468
327
+ aiq/registry_handlers/register.py,sha256=k2gvV0htVpfMdzsRvZGnTZp17JA2Na8sO9ocMaxDSmo,902
328
+ aiq/registry_handlers/registry_handler_base.py,sha256=BYLH6R9y3pySBDH9HKUR16rM0atFRt594IDyyhgCQVQ,5842
329
+ aiq/registry_handlers/local/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
330
+ aiq/registry_handlers/local/local_handler.py,sha256=oHfXdxHJgZFnkQh9ApGqArEWQGwGBKTd6LAQIEPTlWM,7647
331
+ aiq/registry_handlers/local/register_local.py,sha256=WYFGadNPmdtF9YdevaAbv3OqX1nNzZHa2g__2orlzZk,1357
332
+ aiq/registry_handlers/pypi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
333
+ aiq/registry_handlers/pypi/pypi_handler.py,sha256=95Wu9FnXrgTqHLevD2Tf_2iBh4XYqX3HG44HPPlKmGc,10187
334
+ aiq/registry_handlers/pypi/register_pypi.py,sha256=m5STJNk_RVxidqCgT7l6UUnGbcVevyyJ65XWkMhahtg,1864
335
+ aiq/registry_handlers/rest/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
336
+ aiq/registry_handlers/rest/register_rest.py,sha256=kpqIfo7Pwrz1VUd4--vv3DFNlVimDWQKQybHdFWEbyE,2561
337
+ aiq/registry_handlers/rest/rest_handler.py,sha256=R2RcoHzUsg9JElImdeu7hM8-weNe3wIxhmDAVghsaU4,9482
338
+ aiq/registry_handlers/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
339
+ aiq/registry_handlers/schemas/headers.py,sha256=LnVfCMNc3vRr-lRdFB8Cuv9Db5Ct-ACTapjWLaRg2os,1549
340
+ aiq/registry_handlers/schemas/package.py,sha256=aAK-yRkRLvNXRi77Au2_90OdcBFcP3IQ3DxJXTfXUN0,2429
341
+ aiq/registry_handlers/schemas/publish.py,sha256=UIz5S3ez1UV5Rckf7k7c27duTDg2f2MVvlERA14Z2Q0,2200
342
+ aiq/registry_handlers/schemas/pull.py,sha256=eV1MrkVOW0ojk_8THuYOJ4gbGnsA7MhO-jxB9KfoKqE,2661
343
+ aiq/registry_handlers/schemas/remove.py,sha256=FKZjWKuIAPh8-NkvFfjLwD58jxR4vlXCoM7CJyq-ikM,1390
344
+ aiq/registry_handlers/schemas/search.py,sha256=_b9FBhCmSXrihvSlwyYVSDxn6F2sKSIsbQ6_VUQlNbI,3228
345
+ aiq/registry_handlers/schemas/status.py,sha256=U4c5vWsu1SqewSRSgR9ch5xji8xPSUdzaXh1IA4tByA,1473
346
+ aiq/retriever/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
347
+ aiq/retriever/interface.py,sha256=1LttaQyC7ICEoFPBT60rqpaNXl4bNSwoadlEt1A87pA,1342
348
+ aiq/retriever/models.py,sha256=Qob546CP3ADj6J-hdJUnBOTMOU63HjI6RzEoPXETt1g,2387
349
+ aiq/retriever/register.py,sha256=5cqDqwK8CIIzhnLfkeOSedHOSZy-A6H7FlmrdS2TbBw,904
350
+ aiq/retriever/milvus/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
351
+ aiq/retriever/milvus/register.py,sha256=Lavli6FKWve2Roj-AsRgyGlPXbLb-vLXp9rqnSH6660,4027
352
+ aiq/retriever/milvus/retriever.py,sha256=E8aG5usZbWqXc5hTqOTWYG6YNxhTf8oi4hloRjUkZFk,9508
353
+ aiq/retriever/nemo_retriever/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
354
+ aiq/retriever/nemo_retriever/register.py,sha256=ODV-TZfXzDs1VJHHLdj2kC05odirtlQZSeh9c1zw8AQ,2893
355
+ aiq/retriever/nemo_retriever/retriever.py,sha256=IvScUr9XuDLiMR__I3QsboLaM52N5D5Qu94qtTOGQw8,6958
356
+ aiq/runtime/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
357
+ aiq/runtime/loader.py,sha256=CqZzeIpN_aWUEASr4JBtQjmVtSv2DNpu6dFha_4iRI8,7119
358
+ aiq/runtime/runner.py,sha256=CqmlVAYfrBh3ml3t2n3V693RaNyxtK9ScWT4S-Isbr8,6365
359
+ aiq/runtime/session.py,sha256=i1pIqopZCBgGJqVUskKLiBnZYH-lTdMhvFu56dXAU5A,6206
360
+ aiq/runtime/user_metadata.py,sha256=9EiBc-EEJzOdpf3Q1obHqAdY_kRlJ1T0TVvY0Jonk6o,3692
361
+ aiq/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
362
+ aiq/settings/global_settings.py,sha256=Utm7GEiZTYTBZsxJt2TLp4bNWWMscU4b47MAEJehIrU,12036
363
+ aiq/test/.namespace,sha256=Gace0pOC3ETEJf-TBVuNw0TQV6J_KtOPpEiSzMH-odo,215
364
+ aiq/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
365
+ aiq/tool/chat_completion.py,sha256=kq60g4yFJPHrY13P-TgfWbFTc9yJ_ByIcMOLEEGKKAw,3157
366
+ aiq/tool/datetime_tools.py,sha256=6yvTO4cCaxko7-wK6fdCXUwNJFwxuEllfR5S57uo02k,1664
367
+ aiq/tool/document_search.py,sha256=w3D3r5ZBS2jYmVAZZ7lC7xCoi25bA1RePoFjjlV1Zog,6793
368
+ aiq/tool/nvidia_rag.py,sha256=9mS3igONo1RywxXNj_ITh2-qD91x1R0f7uhOWMZQX3o,4178
369
+ aiq/tool/register.py,sha256=Lwl6l_eEzS8LAELxClmniNhhLluRVZFYXhsk2ocQhNg,1491
370
+ aiq/tool/retriever.py,sha256=DnuU4khpJkd4epDBGQsowDOqDBKFiLQrnyKXgU6IRW8,3724
371
+ aiq/tool/server_tools.py,sha256=7wnB-bfFb2lcGSJuG4Wu9PL_N4cDYRu29zFXxIIa9-8,3201
372
+ aiq/tool/code_execution/README.md,sha256=GQy3pPOVspFHQdclQCCweIAY2r8rVZ6bXyCY2FcEc6M,4090
373
+ aiq/tool/code_execution/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
374
+ aiq/tool/code_execution/code_sandbox.py,sha256=ibe6BoYdWzmQWHdVNweVP-QW1WRGirfhvFr0iqm4fI8,10159
375
+ aiq/tool/code_execution/register.py,sha256=i3MNxCb3lBeeSKMwQeg6X2oUpgUKdEoOw2bqUEIqy1E,3322
376
+ aiq/tool/code_execution/test_code_execution_sandbox.py,sha256=mK8xzaz7tKYKN9CrbcLtAATvCsI_50BZqTf1dY-x9QI,14776
377
+ aiq/tool/code_execution/utils.py,sha256=__W-T1kaphFKYSc2AydQW8lCdvD7zAccarvs7XVFTtI,4194
378
+ aiq/tool/code_execution/local_sandbox/.gitignore,sha256=BrV-H5osDtwwIx0eieoexolpnaJvc2DQLV15j95Qtyg,19
379
+ aiq/tool/code_execution/local_sandbox/Dockerfile.sandbox,sha256=CYqZ5jbBwk3ge8pg87aLjhzWERauScwya5naYq0vb5o,2316
380
+ aiq/tool/code_execution/local_sandbox/__init__.py,sha256=k25Kqr_PrH4s0YCfzVzC9vaBAiu8yI428C4HX-qUcqA,615
381
+ aiq/tool/code_execution/local_sandbox/local_sandbox_server.py,sha256=Zl20CYKNbCClV7Q8uEY65m-I7WiYyLKdP7DKF8_8BiE,6949
382
+ aiq/tool/code_execution/local_sandbox/sandbox.requirements.txt,sha256=R86yJ6mcUhfD9_ZU-rSMdaojR6MU41hcH4pE3vAGmcE,43
383
+ aiq/tool/code_execution/local_sandbox/start_local_sandbox.sh,sha256=gnPuzbneKZ61YvzaGIYSUdcyKMLuchYPti3zGLaNCZY,2055
384
+ aiq/tool/github_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
385
+ aiq/tool/github_tools/create_github_commit.py,sha256=5aHFfI-YSvtZaL4RcyZHVCLJX0qULL0PmEAM0QVezVQ,6604
386
+ aiq/tool/github_tools/create_github_issue.py,sha256=SXSjCC8P4uOmGmKbuZIJdwRmO3VwQM80pGZxEgevIMk,3428
387
+ aiq/tool/github_tools/create_github_pr.py,sha256=MZn2XLcXfa2-vLBQ-pJM1n9-ghsR85YMD85G-QX6XaQ,4945
388
+ aiq/tool/github_tools/get_github_file.py,sha256=q0vbDfq5nG6kYDUhkzMUJN3C81lp88Odl3ZOyiqTqNw,4486
389
+ aiq/tool/github_tools/get_github_issue.py,sha256=vwLNkNOszLlymkQju0cR8BNvfdH4EnmAfLCKFCcyR1o,6532
390
+ aiq/tool/github_tools/get_github_pr.py,sha256=b7eCOqrVoejGjRwmUVdU45uF07ihbY8lRacMYOSgMrY,9716
391
+ aiq/tool/github_tools/update_github_issue.py,sha256=TUElxUuzjZr_QldL_48RcqSx0A9b23NB_lA82QwFjkM,4103
392
+ aiq/tool/mcp/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
393
+ aiq/tool/mcp/exceptions.py,sha256=D3pGUO3hPfC4QSWZP24BceAnt1IGeIoZAWlZbIw28uA,5980
394
+ aiq/tool/mcp/mcp_client.py,sha256=ir4XcPPiSBtI3tjZcMIidq6eh-g_JARovFdmQYHtiSo,8963
395
+ aiq/tool/mcp/mcp_tool.py,sha256=HGNauVwgW87VTezY1YiEjoNyQTYvI_QxnRftwI0-Z7E,4105
396
+ aiq/tool/memory_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
397
+ aiq/tool/memory_tools/add_memory_tool.py,sha256=9EjB3DpYhxwasz7o3O8Rq__Ys5986fciv44ahC6mVCo,3349
398
+ aiq/tool/memory_tools/delete_memory_tool.py,sha256=wdB_I8y-1D1OpNtBi6ZOg36vvNkbaxp-yvdqFMc2Suk,2532
399
+ aiq/tool/memory_tools/get_memory_tool.py,sha256=-i0Bt5xYeapbbd2wtAgPc0pOv0Dx4jK1-yyHG7YCeQ0,2749
400
+ aiq/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
401
+ aiq/utils/debugging_utils.py,sha256=6M4JhbHDNDnfmSRGmHvT5IgEeWSHBore3VngdE_PMqc,1332
402
+ aiq/utils/log_utils.py,sha256=dZLHt7qFqLlpPqMMFO9UVtSkOpMjFwz9tkmbAfOiNlg,1355
403
+ aiq/utils/metadata_utils.py,sha256=lGYvc8Gk0az4qZDGeRbVz4L7B_b-Gnjss8JT4goqL5I,2897
404
+ aiq/utils/optional_imports.py,sha256=jQSVBc2fBSRw-2d6r8cEwvh5-di2EUUPakuuo9QbbwA,4039
405
+ aiq/utils/producer_consumer_queue.py,sha256=AcSYkAMBxLx06A5Xdy960PP3AJ7YaSPGJ7rbN_hJsjI,6599
406
+ aiq/utils/string_utils.py,sha256=71HuIzGx7rF8ocTmeoUBpnCi1Qf1yynYlNLLIKP4BVs,1415
407
+ aiq/utils/type_converter.py,sha256=w711BBOwzhtJo2si722C0wPtACFwZvOaKZQDwScAunY,9461
408
+ aiq/utils/type_utils.py,sha256=G-SYlk4BkJv4L225myUhwBY6-Z7wOgR9K49dVUtV8SA,14896
409
+ aiq/utils/url_utils.py,sha256=UzDP_xaS6brWTu7vAws0B4jZyrITIK9Si3U6pZBZqDE,1028
410
+ aiq/utils/data_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
411
+ aiq/utils/data_models/schema_validator.py,sha256=IT74_H4qWSUYae8rgDK-gyBjHJGjzUmf_tVKv0ovbN0,1599
412
+ aiq/utils/exception_handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
413
+ aiq/utils/exception_handlers/automatic_retries.py,sha256=Sgu6zSth3no4xsV2x-dRxWjbTnyLvDkCWP3y_PsmPFE,11963
414
+ aiq/utils/exception_handlers/mcp.py,sha256=uqOjgPG1vqbzBcRdwB9MG1ggljXKRhVo-CZFqAbsIek,7625
415
+ aiq/utils/exception_handlers/schemas.py,sha256=VynNMWHYqpoTfDEpU8Nyw80k04k4Q-0Xr0gFr8wAxpQ,3835
416
+ aiq/utils/io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
417
+ aiq/utils/io/model_processing.py,sha256=bEbH_tIgZQvPlEJKVV4kye_Y9UU96W4k2mKuckGErHA,936
418
+ aiq/utils/io/yaml_tools.py,sha256=5UfXkJsCWZ4hk8RcAL8FWso6EmmxE_sWwSfj9PLMY44,3317
419
+ aiq/utils/reactive/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
420
+ aiq/utils/reactive/observable.py,sha256=Sag3IsgGnw2ax-2fX7WKW4HG4BRleyr0AIzpXAQpa8Q,2285
421
+ aiq/utils/reactive/observer.py,sha256=nHbvC84NgqtuHTWzKKNy7p4kngLPi6-9pisHdjfjCUE,2602
422
+ aiq/utils/reactive/subject.py,sha256=1LgHQXI7hf66KNveR_1XCFSBumgEu-vrhykV6gudkjA,4879
423
+ aiq/utils/reactive/subscription.py,sha256=FyspYkhxP1vAjdoh8I29y1objHXvwmQERdbhJM0w4jk,1701
424
+ aiq/utils/reactive/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
425
+ aiq/utils/reactive/base/observable_base.py,sha256=wHQ0dUdrj7d6HJjbAUNx3pzLf10gEsmlzmD3kMxS2xI,2375
426
+ aiq/utils/reactive/base/observer_base.py,sha256=UAlyAY_ky4q2t0P81RVFo2Bs_R7z5Nde55vRWqXUgz8,1875
427
+ aiq/utils/reactive/base/subject_base.py,sha256=Ed-AC6P7cT3qkW1EXjzbd5M9WpVoeN_9KCe3OM3FLU4,2521
428
+ aiq/utils/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
429
+ aiq/utils/settings/global_settings.py,sha256=U9TCLdoZsKq5qOVGjREipGVv9e-FlStzqy5zv82_VYk,7454
430
+ aiqtoolkit-1.2.0rc2.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
431
+ aiqtoolkit-1.2.0rc2.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
432
+ aiqtoolkit-1.2.0rc2.dist-info/METADATA,sha256=SZwCOh4Pv7hbJMosDBuaGY4svIioO4aeivAQR0ifrN0,21558
433
+ aiqtoolkit-1.2.0rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
434
+ aiqtoolkit-1.2.0rc2.dist-info/entry_points.txt,sha256=iZR3yrf1liXfbcLqn5_pUkLhZyr1bUw_Qh1d2i7gsv4,625
435
+ aiqtoolkit-1.2.0rc2.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
436
+ aiqtoolkit-1.2.0rc2.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.7.1)
2
+ Generator: setuptools (80.9.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -1,8 +1,11 @@
1
1
  [aiq.components]
2
2
  aiq_agents = aiq.agent.register
3
+ aiq_authentication = aiq.authentication.register
3
4
  aiq_embedders = aiq.embedder.register
4
5
  aiq_evaluators = aiq.eval.register
6
+ aiq_inference_time_scaling = aiq.experimental.inference_time_scaling.register
5
7
  aiq_llms = aiq.llm.register
8
+ aiq_object_stores = aiq.object_store.register
6
9
  aiq_observability = aiq.observability.register
7
10
  aiq_retrievers = aiq.retriever.register
8
11
  aiq_tools = aiq.tool.register
@@ -1,148 +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
- import asyncio
17
- import logging
18
- import typing
19
- from collections.abc import Awaitable
20
- from collections.abc import Callable
21
- from typing import Any
22
-
23
- from fastapi import WebSocket
24
- from fastapi import WebSocketException
25
- from starlette.endpoints import WebSocketEndpoint
26
- from starlette.websockets import WebSocketDisconnect
27
-
28
- from aiq.data_models.api_server import AIQChatRequest
29
- from aiq.data_models.api_server import AIQChatResponse
30
- from aiq.data_models.api_server import AIQChatResponseChunk
31
- from aiq.data_models.api_server import AIQResponsePayloadOutput
32
- from aiq.data_models.api_server import AIQResponseSerializable
33
- from aiq.data_models.api_server import WebSocketMessageStatus
34
- from aiq.data_models.api_server import WorkflowSchemaType
35
- from aiq.front_ends.fastapi.message_handler import MessageHandler
36
- from aiq.front_ends.fastapi.response_helpers import generate_streaming_response
37
- from aiq.front_ends.fastapi.step_adaptor import StepAdaptor
38
- from aiq.runtime.session import AIQSessionManager
39
-
40
- logger = logging.getLogger(__name__)
41
-
42
-
43
- class AIQWebSocket(WebSocketEndpoint):
44
- encoding = "json"
45
-
46
- def __init__(self, session_manager: AIQSessionManager, step_adaptor: StepAdaptor, *args, **kwargs):
47
- self._session_manager: AIQSessionManager = session_manager
48
- self._message_handler: MessageHandler = MessageHandler(self)
49
- self._process_response_event: asyncio.Event = asyncio.Event()
50
- self._workflow_schema_type: dict[str, Callable[..., Awaitable[Any]]] = {
51
- WorkflowSchemaType.GENERATE_STREAM: self.process_generate_stream,
52
- WorkflowSchemaType.CHAT_STREAM: self.process_chat_stream,
53
- WorkflowSchemaType.GENERATE: self.process_generate,
54
- WorkflowSchemaType.CHAT: self.process_chat
55
- }
56
- self._step_adaptor = step_adaptor
57
- super().__init__(*args, **kwargs)
58
-
59
- @property
60
- def workflow_schema_type(self) -> dict[str, Callable[..., Awaitable[Any]]]:
61
- return self._workflow_schema_type
62
-
63
- @property
64
- def process_response_event(self) -> asyncio.Event:
65
- return self._process_response_event
66
-
67
- async def on_connect(self, websocket: WebSocket):
68
- try:
69
- # Accept the websocket connection
70
- await websocket.accept()
71
- try:
72
- # Start background message processors
73
- self._message_handler.process_messages_task = asyncio.create_task(
74
- self._message_handler.process_messages())
75
-
76
- self._message_handler.process_out_going_messages_task = asyncio.create_task(
77
- self._message_handler.process_out_going_messages(websocket))
78
-
79
- except asyncio.CancelledError:
80
- pass
81
-
82
- except (WebSocketDisconnect, WebSocketException):
83
- logger.error("A WebSocket error occured during `on_connect`. Ignoring the connection.", exc_info=True)
84
-
85
- async def on_send(self, websocket: WebSocket, data: dict[str, str]):
86
- try:
87
- await websocket.send_json(data)
88
- except (WebSocketDisconnect, WebSocketException, Exception):
89
- logger.error("A WebSocket error occurred during `on_send`. Ignoring the connection.", exc_info=True)
90
-
91
- async def on_receive(self, websocket: WebSocket, data: dict[str, Any]):
92
- try:
93
- await self._message_handler.messages_queue.put(data)
94
- except (Exception):
95
- logger.error("An unxpected error occurred during `on_receive`. Ignoring the exception", exc_info=True)
96
-
97
- async def on_disconnect(self, websocket: WebSocket, close_code: Any):
98
- try:
99
- if self._message_handler.process_messages_task:
100
- self._message_handler.process_messages_task.cancel()
101
-
102
- if self._message_handler.process_out_going_messages_task:
103
- self._message_handler.process_out_going_messages_task.cancel()
104
-
105
- if self._message_handler.background_task:
106
- self._message_handler.background_task.cancel()
107
-
108
- except (WebSocketDisconnect, asyncio.CancelledError):
109
- pass
110
-
111
- async def _process_message(self,
112
- payload: typing.Any,
113
- result_type: type | None = None,
114
- output_type: type | None = None) -> None:
115
-
116
- async with self._session_manager.session(
117
- user_input_callback=self._message_handler.human_interaction) as session:
118
-
119
- async for value in generate_streaming_response(payload,
120
- session_manager=session,
121
- streaming=True,
122
- step_adaptor=self._step_adaptor,
123
- result_type=result_type,
124
- output_type=output_type):
125
-
126
- await self._process_response_event.wait()
127
-
128
- if not isinstance(value, AIQResponseSerializable):
129
- value = AIQResponsePayloadOutput(payload=value)
130
-
131
- await self._message_handler.create_websocket_message(data_model=value,
132
- status=WebSocketMessageStatus.IN_PROGRESS)
133
-
134
- async def process_generate_stream(self, payload: str):
135
-
136
- return await self._process_message(payload, result_type=None, output_type=None)
137
-
138
- async def process_chat_stream(self, payload: AIQChatRequest):
139
-
140
- return await self._process_message(payload, result_type=AIQChatResponse, output_type=AIQChatResponseChunk)
141
-
142
- async def process_generate(self, payload: typing.Any):
143
-
144
- return await self._process_message(payload)
145
-
146
- async def process_chat(self, payload: AIQChatRequest):
147
-
148
- return await self._process_message(payload, result_type=AIQChatResponse)