nvidia-nat 1.2.0a20250813__py3-none-any.whl → 1.2.0rc6__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 (116) hide show
  1. nat/agent/react_agent/register.py +1 -1
  2. nat/agent/reasoning_agent/reasoning_agent.py +3 -3
  3. nat/agent/rewoo_agent/prompt.py +11 -12
  4. nat/agent/rewoo_agent/register.py +30 -32
  5. nat/authentication/http_basic_auth/http_basic_auth_provider.py +1 -1
  6. nat/authentication/interfaces.py +1 -1
  7. nat/authentication/oauth2/oauth2_auth_code_flow_provider.py +2 -2
  8. nat/builder/component_utils.py +5 -5
  9. nat/builder/context.py +5 -5
  10. nat/builder/front_end.py +6 -6
  11. nat/builder/function_base.py +4 -4
  12. nat/builder/function_info.py +1 -1
  13. nat/builder/intermediate_step_manager.py +4 -4
  14. nat/builder/user_interaction_manager.py +3 -3
  15. nat/builder/workflow_builder.py +2 -2
  16. nat/cli/cli_utils/validation.py +1 -1
  17. nat/cli/commands/configure/channel/add.py +1 -1
  18. nat/cli/commands/configure/channel/channel.py +1 -3
  19. nat/cli/commands/configure/channel/remove.py +1 -1
  20. nat/cli/commands/configure/channel/update.py +1 -1
  21. nat/cli/commands/configure/configure.py +2 -2
  22. nat/cli/commands/evaluate.py +1 -1
  23. nat/cli/commands/info/info.py +2 -4
  24. nat/cli/commands/info/list_components.py +2 -2
  25. nat/cli/commands/info/list_mcp.py +9 -9
  26. nat/cli/commands/registry/publish.py +3 -3
  27. nat/cli/commands/registry/pull.py +3 -3
  28. nat/cli/commands/registry/registry.py +1 -3
  29. nat/cli/commands/registry/remove.py +3 -3
  30. nat/cli/commands/registry/search.py +3 -3
  31. nat/cli/commands/start.py +4 -4
  32. nat/cli/commands/workflow/templates/pyproject.toml.j2 +4 -4
  33. nat/cli/commands/workflow/workflow_commands.py +6 -6
  34. nat/data_models/api_server.py +38 -31
  35. nat/data_models/component_ref.py +9 -9
  36. nat/data_models/dataset_handler.py +56 -10
  37. nat/data_models/discovery_metadata.py +21 -50
  38. nat/data_models/evaluate.py +2 -2
  39. nat/data_models/intermediate_step.py +4 -4
  40. nat/embedder/register.py +0 -2
  41. nat/eval/dataset_handler/dataset_handler.py +118 -5
  42. nat/eval/evaluate.py +1 -1
  43. nat/eval/evaluator/evaluator_model.py +3 -3
  44. nat/eval/rag_evaluator/evaluate.py +1 -1
  45. nat/eval/swe_bench_evaluator/evaluate.py +2 -2
  46. nat/experimental/test_time_compute/editing/motivation_aware_summarization.py +1 -1
  47. nat/experimental/test_time_compute/functions/plan_select_execute_function.py +4 -4
  48. nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py +1 -1
  49. nat/experimental/test_time_compute/scoring/llm_based_agent_scorer.py +1 -1
  50. nat/experimental/test_time_compute/scoring/llm_based_plan_scorer.py +1 -1
  51. nat/experimental/test_time_compute/search/multi_llm_planner.py +1 -1
  52. nat/experimental/test_time_compute/search/multi_query_retrieval_search.py +1 -1
  53. nat/experimental/test_time_compute/search/single_shot_multi_plan_planner.py +1 -1
  54. nat/experimental/test_time_compute/selection/llm_based_agent_output_selector.py +1 -1
  55. nat/experimental/test_time_compute/selection/llm_based_output_merging_selector.py +1 -1
  56. nat/experimental/test_time_compute/selection/llm_based_plan_selector.py +1 -1
  57. nat/front_ends/console/console_front_end_config.py +1 -1
  58. nat/front_ends/fastapi/fastapi_front_end_config.py +6 -7
  59. nat/front_ends/fastapi/fastapi_front_end_plugin.py +4 -4
  60. nat/front_ends/fastapi/fastapi_front_end_plugin_worker.py +13 -13
  61. nat/front_ends/fastapi/intermediate_steps_subscriber.py +1 -1
  62. nat/front_ends/fastapi/main.py +6 -6
  63. nat/front_ends/fastapi/message_handler.py +14 -3
  64. nat/front_ends/fastapi/message_validator.py +6 -8
  65. nat/front_ends/fastapi/response_helpers.py +3 -3
  66. nat/front_ends/mcp/mcp_front_end_config.py +3 -2
  67. nat/front_ends/mcp/mcp_front_end_plugin_worker.py +4 -4
  68. nat/front_ends/mcp/tool_converter.py +15 -16
  69. nat/memory/__init__.py +2 -2
  70. nat/meta/pypi.md +8 -8
  71. nat/object_store/__init__.py +2 -2
  72. nat/observability/exporter/base_exporter.py +1 -1
  73. nat/observability/exporter/raw_exporter.py +1 -1
  74. nat/observability/exporter_manager.py +1 -1
  75. nat/profiler/callbacks/agno_callback_handler.py +4 -4
  76. nat/profiler/callbacks/llama_index_callback_handler.py +2 -2
  77. nat/profiler/callbacks/semantic_kernel_callback_handler.py +1 -1
  78. nat/profiler/decorators/function_tracking.py +1 -1
  79. nat/profiler/forecasting/models/linear_model.py +3 -2
  80. nat/profiler/forecasting/models/random_forest_regressor.py +3 -2
  81. nat/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py +1 -1
  82. nat/profiler/inference_optimization/experimental/prefix_span_analysis.py +1 -1
  83. nat/profiler/profile_runner.py +1 -1
  84. nat/registry_handlers/local/local_handler.py +5 -5
  85. nat/registry_handlers/local/register_local.py +1 -1
  86. nat/registry_handlers/package_utils.py +17 -17
  87. nat/registry_handlers/pypi/pypi_handler.py +5 -5
  88. nat/registry_handlers/pypi/register_pypi.py +3 -3
  89. nat/registry_handlers/registry_handler_base.py +8 -8
  90. nat/registry_handlers/rest/register_rest.py +4 -4
  91. nat/registry_handlers/rest/rest_handler.py +6 -6
  92. nat/registry_handlers/schemas/package.py +3 -3
  93. nat/registry_handlers/schemas/publish.py +4 -4
  94. nat/registry_handlers/schemas/pull.py +4 -4
  95. nat/registry_handlers/schemas/search.py +7 -7
  96. nat/runtime/loader.py +21 -20
  97. nat/runtime/runner.py +2 -2
  98. nat/runtime/session.py +1 -1
  99. nat/settings/global_settings.py +2 -2
  100. nat/tool/chat_completion.py +1 -1
  101. nat/tool/code_execution/README.md +2 -2
  102. nat/tool/code_execution/test_code_execution_sandbox.py +1 -1
  103. nat/tool/mcp/exceptions.py +1 -1
  104. nat/tool/mcp/mcp_tool.py +1 -1
  105. nat/tool/retriever.py +3 -3
  106. nat/tool/server_tools.py +11 -11
  107. nat/utils/settings/global_settings.py +1 -1
  108. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/METADATA +9 -9
  109. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/RECORD +114 -116
  110. nat/embedder/langchain_client.py +0 -41
  111. nat/meta/module_to_distro.json +0 -4
  112. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/WHEEL +0 -0
  113. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/entry_points.txt +0 -0
  114. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/licenses/LICENSE-3rd-party.txt +0 -0
  115. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/licenses/LICENSE.md +0 -0
  116. {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/top_level.txt +0 -0
@@ -50,7 +50,7 @@ class TestCodeExecutionSandbox:
50
50
  except (ConnectionError, Timeout, RequestException):
51
51
  pytest.skip(
52
52
  f"Sandbox server is not running at {sandbox_config['url']}. "
53
- "Please start it with: cd src/aiq/tool/code_execution/local_sandbox && ./start_local_sandbox.sh")
53
+ "Please start it with: cd src/nat/tool/code_execution/local_sandbox && ./start_local_sandbox.sh")
54
54
 
55
55
  def execute_code(self, sandbox_config: dict[str, Any], code: str, language: str = "python") -> dict[str, Any]:
56
56
  """
@@ -108,7 +108,7 @@ class MCPToolNotFoundError(MCPError):
108
108
  url=url,
109
109
  category=MCPErrorCategory.TOOL_NOT_FOUND,
110
110
  suggestions=[
111
- "Use 'aiq info mcp --detail' to see available tools",
111
+ "Use 'nat info mcp --detail' to see available tools",
112
112
  "Check that the tool name is spelled correctly"
113
113
  ],
114
114
  original_exception=original_exception)
nat/tool/mcp/mcp_tool.py CHANGED
@@ -50,7 +50,7 @@ class MCPToolConfig(FunctionBaseConfig, name="mcp_tool_wrapper"):
50
50
  @register_function(config_type=MCPToolConfig)
51
51
  async def mcp_tool(config: MCPToolConfig, builder: Builder): # pylint: disable=unused-argument
52
52
  """
53
- Generate an AIQ Toolkit Function that wraps a tool provided by the MCP server.
53
+ Generate a NAT Function that wraps a tool provided by the MCP server.
54
54
  """
55
55
 
56
56
  from nat.tool.mcp.mcp_client import MCPBuilder
nat/tool/retriever.py CHANGED
@@ -30,9 +30,9 @@ from nat.retriever.models import RetrieverOutput
30
30
  logger = logging.getLogger(__name__)
31
31
 
32
32
 
33
- class RetrieverConfig(FunctionBaseConfig, name="aiq_retriever"):
33
+ class RetrieverConfig(FunctionBaseConfig, name="nat_retriever"):
34
34
  """
35
- AIQRetriever tool which provides a common interface for different vectorstores. Its
35
+ Retriever tool which provides a common interface for different vectorstores. Its
36
36
  configuration uses clients, which are the vectorstore-specific implementaiton of the retriever interface.
37
37
  """
38
38
  retriever: RetrieverRef = Field(description="The retriever instance name from the workflow configuration object.")
@@ -58,7 +58,7 @@ def _get_description_from_config(config: RetrieverConfig) -> str:
58
58
  @register_function(config_type=RetrieverConfig)
59
59
  async def retriever_tool(config: RetrieverConfig, builder: Builder):
60
60
  """
61
- Configure an AIQ Toolkit Retriever Tool which supports different clients such as Milvus and Nemo Retriever.
61
+ Configure a NAT Retriever Tool which supports different clients such as Milvus and Nemo Retriever.
62
62
 
63
63
  Args:
64
64
  config: A config object with required parameters 'client' and 'client_config'
nat/tool/server_tools.py CHANGED
@@ -38,18 +38,18 @@ async def current_request_attributes(config: RequestAttributesTool, builder: Bui
38
38
  async def _get_request_attributes(unused: str) -> str:
39
39
 
40
40
  from nat.builder.context import Context
41
- aiq_context = Context.get()
41
+ nat_context = Context.get()
42
42
 
43
- method: str | None = aiq_context.metadata.method
44
- url_path: str | None = aiq_context.metadata.url_path
45
- url_scheme: str | None = aiq_context.metadata.url_scheme
46
- headers: Headers | None = aiq_context.metadata.headers
47
- query_params: QueryParams | None = aiq_context.metadata.query_params
48
- path_params: dict[str, str] | None = aiq_context.metadata.path_params
49
- client_host: str | None = aiq_context.metadata.client_host
50
- client_port: int | None = aiq_context.metadata.client_port
51
- cookies: dict[str, str] | None = aiq_context.metadata.cookies
52
- conversation_id: str | None = aiq_context.conversation_id
43
+ method: str | None = nat_context.metadata.method
44
+ url_path: str | None = nat_context.metadata.url_path
45
+ url_scheme: str | None = nat_context.metadata.url_scheme
46
+ headers: Headers | None = nat_context.metadata.headers
47
+ query_params: QueryParams | None = nat_context.metadata.query_params
48
+ path_params: dict[str, str] | None = nat_context.metadata.path_params
49
+ client_host: str | None = nat_context.metadata.client_host
50
+ client_port: int | None = nat_context.metadata.client_port
51
+ cookies: dict[str, str] | None = nat_context.metadata.cookies
52
+ conversation_id: str | None = nat_context.conversation_id
53
53
 
54
54
  return (f"Method: {method}, "
55
55
  f"URL Path: {url_path}, "
@@ -66,7 +66,7 @@ def configure_registry_channel(config_type: RegistryHandlerBaseConfig, channel_n
66
66
 
67
67
 
68
68
  def add_channel_interative(channel_type: str) -> None:
69
- """Add a remote registry channel to publish/search/pull AIQ Toolkit plugin packages.
69
+ """Add a remote registry channel to publish/search/pull NAT plugin packages.
70
70
 
71
71
  Args:
72
72
  channel_type (str): They type of channel to configure.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat
3
- Version: 1.2.0a20250813
3
+ Version: 1.2.0rc6
4
4
  Summary: NVIDIA NeMo Agent toolkit
5
5
  Author: NVIDIA Corporation
6
6
  Maintainer: NVIDIA Corporation
@@ -320,11 +320,11 @@ See the License for the specific language governing permissions and
320
320
  limitations under the License.
321
321
  -->
322
322
 
323
- ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
323
+ ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
324
324
 
325
325
  # NVIDIA NeMo Agent Toolkit
326
326
 
327
- AIQ toolkit is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, AIQ toolkit enables true composability: build once and reuse anywhere.
327
+ NeMo Agent toolkit is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, NeMo Agent toolkit enables true composability: build once and reuse anywhere.
328
328
 
329
329
  ## Key Features
330
330
 
@@ -334,16 +334,16 @@ AIQ toolkit is a flexible library designed to seamlessly integrate your enterpri
334
334
  - [**Profiling:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
335
335
  - [**Observability:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool, with examples using [Phoenix](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) and [W&B Weave](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-weave.html).
336
336
  - [**Evaluation System:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
337
- - [**User Interface:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/quick-start/launching-ui.html) Use the AIQ toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
338
- - [**MCP Compatibility**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as AIQ toolkit functions.
337
+ - [**User Interface:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/quick-start/launching-ui.html) Use the NeMo Agent toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
338
+ - [**MCP Compatibility**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as NeMo Agent toolkit functions.
339
339
 
340
- With AIQ toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
340
+ With NeMo Agent toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
341
341
 
342
342
  ## Links
343
- * [Documentation](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/index.html): Explore the full documentation for AIQ toolkit.
343
+ * [Documentation](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/index.html): Explore the full documentation for NeMo Agent toolkit.
344
344
 
345
345
  ## First time user?
346
- If this is your first time using AIQ toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/NeMo-Agent-Toolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with AIQ toolkit applications and need to add AIQ toolkit as a dependency to their project.
346
+ If this is your first time using NeMo Agent toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/NeMo-Agent-Toolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with NeMo Agent toolkit applications and need to add NeMo Agent toolkit as a dependency to their project.
347
347
 
348
348
  ## Feedback
349
349
 
@@ -351,7 +351,7 @@ We would love to hear from you! Please file an issue on [GitHub](https://github.
351
351
 
352
352
  ## Acknowledgements
353
353
 
354
- We would like to thank the following open source projects that made AIQ toolkit possible:
354
+ We would like to thank the following open source projects that made NeMo Agent toolkit possible:
355
355
 
356
356
  - [CrewAI](https://github.com/crewAIInc/crewAI)
357
357
  - [FastAPI](https://github.com/tiangolo/fastapi)