nvidia-nat 1.3.dev0__py3-none-any.whl → 1.3.0rc1__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.
Files changed (246) hide show
  1. aiq/__init__.py +66 -0
  2. nat/agent/base.py +40 -14
  3. nat/agent/dual_node.py +9 -4
  4. nat/agent/prompt_optimizer/prompt.py +68 -0
  5. nat/agent/prompt_optimizer/register.py +149 -0
  6. nat/agent/react_agent/agent.py +96 -57
  7. nat/agent/react_agent/prompt.py +4 -1
  8. nat/agent/react_agent/register.py +41 -21
  9. nat/agent/reasoning_agent/reasoning_agent.py +11 -9
  10. nat/agent/register.py +1 -1
  11. nat/agent/rewoo_agent/agent.py +332 -150
  12. nat/agent/rewoo_agent/prompt.py +22 -22
  13. nat/agent/rewoo_agent/register.py +49 -28
  14. nat/agent/tool_calling_agent/agent.py +156 -29
  15. nat/agent/tool_calling_agent/register.py +57 -38
  16. nat/authentication/api_key/api_key_auth_provider.py +2 -2
  17. nat/authentication/credential_validator/bearer_token_validator.py +557 -0
  18. nat/authentication/http_basic_auth/http_basic_auth_provider.py +1 -1
  19. nat/authentication/interfaces.py +5 -2
  20. nat/authentication/oauth2/oauth2_auth_code_flow_provider.py +40 -20
  21. nat/authentication/oauth2/oauth2_resource_server_config.py +124 -0
  22. nat/authentication/register.py +0 -1
  23. nat/builder/builder.py +56 -24
  24. nat/builder/component_utils.py +9 -5
  25. nat/builder/context.py +46 -11
  26. nat/builder/eval_builder.py +16 -11
  27. nat/builder/framework_enum.py +1 -0
  28. nat/builder/front_end.py +1 -1
  29. nat/builder/function.py +378 -8
  30. nat/builder/function_base.py +3 -3
  31. nat/builder/function_info.py +6 -8
  32. nat/builder/user_interaction_manager.py +2 -2
  33. nat/builder/workflow.py +13 -1
  34. nat/builder/workflow_builder.py +281 -76
  35. nat/cli/cli_utils/config_override.py +2 -2
  36. nat/cli/commands/evaluate.py +1 -1
  37. nat/cli/commands/info/info.py +16 -6
  38. nat/cli/commands/info/list_channels.py +1 -1
  39. nat/cli/commands/info/list_components.py +7 -8
  40. nat/cli/commands/mcp/__init__.py +14 -0
  41. nat/cli/commands/mcp/mcp.py +986 -0
  42. nat/cli/commands/object_store/__init__.py +14 -0
  43. nat/cli/commands/object_store/object_store.py +227 -0
  44. nat/cli/commands/optimize.py +90 -0
  45. nat/cli/commands/registry/publish.py +2 -2
  46. nat/cli/commands/registry/pull.py +2 -2
  47. nat/cli/commands/registry/remove.py +2 -2
  48. nat/cli/commands/registry/search.py +15 -17
  49. nat/cli/commands/start.py +16 -5
  50. nat/cli/commands/uninstall.py +1 -1
  51. nat/cli/commands/workflow/templates/config.yml.j2 +0 -1
  52. nat/cli/commands/workflow/templates/pyproject.toml.j2 +4 -1
  53. nat/cli/commands/workflow/templates/register.py.j2 +0 -1
  54. nat/cli/commands/workflow/workflow_commands.py +9 -13
  55. nat/cli/entrypoint.py +8 -10
  56. nat/cli/register_workflow.py +38 -4
  57. nat/cli/type_registry.py +79 -10
  58. nat/control_flow/__init__.py +0 -0
  59. nat/control_flow/register.py +20 -0
  60. nat/control_flow/router_agent/__init__.py +0 -0
  61. nat/control_flow/router_agent/agent.py +329 -0
  62. nat/control_flow/router_agent/prompt.py +48 -0
  63. nat/control_flow/router_agent/register.py +91 -0
  64. nat/control_flow/sequential_executor.py +166 -0
  65. nat/data_models/agent.py +34 -0
  66. nat/data_models/api_server.py +10 -10
  67. nat/data_models/authentication.py +23 -9
  68. nat/data_models/common.py +1 -1
  69. nat/data_models/component.py +2 -0
  70. nat/data_models/component_ref.py +11 -0
  71. nat/data_models/config.py +41 -17
  72. nat/data_models/dataset_handler.py +1 -1
  73. nat/data_models/discovery_metadata.py +4 -4
  74. nat/data_models/evaluate.py +4 -1
  75. nat/data_models/function.py +34 -0
  76. nat/data_models/function_dependencies.py +14 -6
  77. nat/data_models/gated_field_mixin.py +242 -0
  78. nat/data_models/intermediate_step.py +3 -3
  79. nat/data_models/optimizable.py +119 -0
  80. nat/data_models/optimizer.py +149 -0
  81. nat/data_models/swe_bench_model.py +1 -1
  82. nat/data_models/temperature_mixin.py +44 -0
  83. nat/data_models/thinking_mixin.py +86 -0
  84. nat/data_models/top_p_mixin.py +44 -0
  85. nat/embedder/azure_openai_embedder.py +46 -0
  86. nat/embedder/nim_embedder.py +1 -1
  87. nat/embedder/openai_embedder.py +2 -3
  88. nat/embedder/register.py +1 -1
  89. nat/eval/config.py +3 -1
  90. nat/eval/dataset_handler/dataset_handler.py +71 -7
  91. nat/eval/evaluate.py +86 -31
  92. nat/eval/evaluator/base_evaluator.py +1 -1
  93. nat/eval/evaluator/evaluator_model.py +13 -0
  94. nat/eval/intermediate_step_adapter.py +1 -1
  95. nat/eval/rag_evaluator/evaluate.py +2 -2
  96. nat/eval/rag_evaluator/register.py +3 -3
  97. nat/eval/register.py +4 -1
  98. nat/eval/remote_workflow.py +3 -3
  99. nat/eval/runtime_evaluator/__init__.py +14 -0
  100. nat/eval/runtime_evaluator/evaluate.py +123 -0
  101. nat/eval/runtime_evaluator/register.py +100 -0
  102. nat/eval/swe_bench_evaluator/evaluate.py +6 -6
  103. nat/eval/trajectory_evaluator/evaluate.py +1 -1
  104. nat/eval/trajectory_evaluator/register.py +1 -1
  105. nat/eval/tunable_rag_evaluator/evaluate.py +4 -7
  106. nat/eval/utils/eval_trace_ctx.py +89 -0
  107. nat/eval/utils/weave_eval.py +18 -9
  108. nat/experimental/decorators/experimental_warning_decorator.py +27 -7
  109. nat/experimental/test_time_compute/functions/plan_select_execute_function.py +7 -3
  110. nat/experimental/test_time_compute/functions/ttc_tool_orchestration_function.py +3 -3
  111. nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py +1 -1
  112. nat/experimental/test_time_compute/models/strategy_base.py +5 -4
  113. nat/experimental/test_time_compute/register.py +0 -1
  114. nat/experimental/test_time_compute/selection/llm_based_output_merging_selector.py +1 -3
  115. nat/front_ends/console/authentication_flow_handler.py +82 -30
  116. nat/front_ends/console/console_front_end_plugin.py +8 -5
  117. nat/front_ends/fastapi/auth_flow_handlers/websocket_flow_handler.py +52 -17
  118. nat/front_ends/fastapi/dask_client_mixin.py +65 -0
  119. nat/front_ends/fastapi/fastapi_front_end_config.py +36 -5
  120. nat/front_ends/fastapi/fastapi_front_end_controller.py +4 -4
  121. nat/front_ends/fastapi/fastapi_front_end_plugin.py +135 -4
  122. nat/front_ends/fastapi/fastapi_front_end_plugin_worker.py +481 -281
  123. nat/front_ends/fastapi/job_store.py +518 -99
  124. nat/front_ends/fastapi/main.py +11 -19
  125. nat/front_ends/fastapi/message_handler.py +13 -14
  126. nat/front_ends/fastapi/message_validator.py +17 -19
  127. nat/front_ends/fastapi/response_helpers.py +4 -4
  128. nat/front_ends/fastapi/step_adaptor.py +2 -2
  129. nat/front_ends/fastapi/utils.py +57 -0
  130. nat/front_ends/mcp/introspection_token_verifier.py +73 -0
  131. nat/front_ends/mcp/mcp_front_end_config.py +10 -1
  132. nat/front_ends/mcp/mcp_front_end_plugin.py +45 -13
  133. nat/front_ends/mcp/mcp_front_end_plugin_worker.py +116 -8
  134. nat/front_ends/mcp/tool_converter.py +44 -14
  135. nat/front_ends/register.py +0 -1
  136. nat/front_ends/simple_base/simple_front_end_plugin_base.py +3 -1
  137. nat/llm/aws_bedrock_llm.py +24 -12
  138. nat/llm/azure_openai_llm.py +57 -0
  139. nat/llm/litellm_llm.py +69 -0
  140. nat/llm/nim_llm.py +20 -8
  141. nat/llm/openai_llm.py +14 -6
  142. nat/llm/register.py +5 -1
  143. nat/llm/utils/env_config_value.py +2 -3
  144. nat/llm/utils/thinking.py +215 -0
  145. nat/meta/pypi.md +9 -9
  146. nat/object_store/models.py +2 -0
  147. nat/object_store/register.py +0 -1
  148. nat/observability/exporter/base_exporter.py +3 -3
  149. nat/observability/exporter/file_exporter.py +1 -1
  150. nat/observability/exporter/processing_exporter.py +309 -81
  151. nat/observability/exporter/span_exporter.py +1 -1
  152. nat/observability/exporter_manager.py +7 -7
  153. nat/observability/mixin/file_mixin.py +7 -7
  154. nat/observability/mixin/redaction_config_mixin.py +42 -0
  155. nat/observability/mixin/tagging_config_mixin.py +62 -0
  156. nat/observability/mixin/type_introspection_mixin.py +420 -107
  157. nat/observability/processor/batching_processor.py +5 -7
  158. nat/observability/processor/falsy_batch_filter_processor.py +55 -0
  159. nat/observability/processor/processor.py +3 -0
  160. nat/observability/processor/processor_factory.py +70 -0
  161. nat/observability/processor/redaction/__init__.py +24 -0
  162. nat/observability/processor/redaction/contextual_redaction_processor.py +125 -0
  163. nat/observability/processor/redaction/contextual_span_redaction_processor.py +66 -0
  164. nat/observability/processor/redaction/redaction_processor.py +177 -0
  165. nat/observability/processor/redaction/span_header_redaction_processor.py +92 -0
  166. nat/observability/processor/span_tagging_processor.py +68 -0
  167. nat/observability/register.py +6 -4
  168. nat/profiler/calc/calc_runner.py +3 -4
  169. nat/profiler/callbacks/agno_callback_handler.py +1 -1
  170. nat/profiler/callbacks/langchain_callback_handler.py +14 -7
  171. nat/profiler/callbacks/llama_index_callback_handler.py +3 -3
  172. nat/profiler/callbacks/semantic_kernel_callback_handler.py +3 -3
  173. nat/profiler/data_frame_row.py +1 -1
  174. nat/profiler/decorators/framework_wrapper.py +62 -13
  175. nat/profiler/decorators/function_tracking.py +160 -3
  176. nat/profiler/forecasting/models/forecasting_base_model.py +3 -1
  177. nat/profiler/inference_optimization/bottleneck_analysis/simple_stack_analysis.py +1 -1
  178. nat/profiler/inference_optimization/data_models.py +3 -3
  179. nat/profiler/inference_optimization/experimental/prefix_span_analysis.py +7 -8
  180. nat/profiler/inference_optimization/token_uniqueness.py +1 -1
  181. nat/profiler/parameter_optimization/__init__.py +0 -0
  182. nat/profiler/parameter_optimization/optimizable_utils.py +93 -0
  183. nat/profiler/parameter_optimization/optimizer_runtime.py +67 -0
  184. nat/profiler/parameter_optimization/parameter_optimizer.py +153 -0
  185. nat/profiler/parameter_optimization/parameter_selection.py +107 -0
  186. nat/profiler/parameter_optimization/pareto_visualizer.py +380 -0
  187. nat/profiler/parameter_optimization/prompt_optimizer.py +384 -0
  188. nat/profiler/parameter_optimization/update_helpers.py +66 -0
  189. nat/profiler/profile_runner.py +14 -9
  190. nat/profiler/utils.py +4 -2
  191. nat/registry_handlers/local/local_handler.py +2 -2
  192. nat/registry_handlers/package_utils.py +1 -2
  193. nat/registry_handlers/pypi/pypi_handler.py +23 -26
  194. nat/registry_handlers/register.py +3 -4
  195. nat/registry_handlers/rest/rest_handler.py +12 -13
  196. nat/retriever/milvus/retriever.py +2 -2
  197. nat/retriever/nemo_retriever/retriever.py +1 -1
  198. nat/retriever/register.py +0 -1
  199. nat/runtime/loader.py +2 -2
  200. nat/runtime/runner.py +3 -2
  201. nat/runtime/session.py +43 -8
  202. nat/settings/global_settings.py +16 -5
  203. nat/tool/chat_completion.py +5 -2
  204. nat/tool/code_execution/local_sandbox/local_sandbox_server.py +3 -3
  205. nat/tool/datetime_tools.py +49 -9
  206. nat/tool/document_search.py +2 -2
  207. nat/tool/github_tools.py +450 -0
  208. nat/tool/nvidia_rag.py +1 -1
  209. nat/tool/register.py +2 -9
  210. nat/tool/retriever.py +3 -2
  211. nat/utils/callable_utils.py +70 -0
  212. nat/utils/data_models/schema_validator.py +3 -3
  213. nat/utils/exception_handlers/automatic_retries.py +104 -51
  214. nat/utils/exception_handlers/schemas.py +1 -1
  215. nat/utils/io/yaml_tools.py +2 -2
  216. nat/utils/log_levels.py +25 -0
  217. nat/utils/reactive/base/observable_base.py +2 -2
  218. nat/utils/reactive/base/observer_base.py +1 -1
  219. nat/utils/reactive/observable.py +2 -2
  220. nat/utils/reactive/observer.py +4 -4
  221. nat/utils/reactive/subscription.py +1 -1
  222. nat/utils/settings/global_settings.py +6 -8
  223. nat/utils/type_converter.py +4 -3
  224. nat/utils/type_utils.py +9 -5
  225. {nvidia_nat-1.3.dev0.dist-info → nvidia_nat-1.3.0rc1.dist-info}/METADATA +49 -21
  226. {nvidia_nat-1.3.dev0.dist-info → nvidia_nat-1.3.0rc1.dist-info}/RECORD +233 -189
  227. {nvidia_nat-1.3.dev0.dist-info → nvidia_nat-1.3.0rc1.dist-info}/entry_points.txt +1 -0
  228. nvidia_nat-1.3.0rc1.dist-info/licenses/LICENSE-3rd-party.txt +5478 -0
  229. {nvidia_nat-1.3.dev0.dist-info → nvidia_nat-1.3.0rc1.dist-info}/top_level.txt +1 -0
  230. nat/cli/commands/info/list_mcp.py +0 -304
  231. nat/tool/github_tools/create_github_commit.py +0 -133
  232. nat/tool/github_tools/create_github_issue.py +0 -87
  233. nat/tool/github_tools/create_github_pr.py +0 -106
  234. nat/tool/github_tools/get_github_file.py +0 -106
  235. nat/tool/github_tools/get_github_issue.py +0 -166
  236. nat/tool/github_tools/get_github_pr.py +0 -256
  237. nat/tool/github_tools/update_github_issue.py +0 -100
  238. nat/tool/mcp/exceptions.py +0 -142
  239. nat/tool/mcp/mcp_client.py +0 -255
  240. nat/tool/mcp/mcp_tool.py +0 -96
  241. nat/utils/exception_handlers/mcp.py +0 -211
  242. nvidia_nat-1.3.dev0.dist-info/licenses/LICENSE-3rd-party.txt +0 -3686
  243. /nat/{tool/github_tools → agent/prompt_optimizer}/__init__.py +0 -0
  244. /nat/{tool/mcp → authentication/credential_validator}/__init__.py +0 -0
  245. {nvidia_nat-1.3.dev0.dist-info → nvidia_nat-1.3.0rc1.dist-info}/WHEEL +0 -0
  246. {nvidia_nat-1.3.dev0.dist-info → nvidia_nat-1.3.0rc1.dist-info}/licenses/LICENSE.md +0 -0

There are too many changes on this page to be displayed.

The amount of changes on this page would crash your brower.

You can still verify the content by downloading the package file manually.