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.
- nat/agent/react_agent/register.py +1 -1
- nat/agent/reasoning_agent/reasoning_agent.py +3 -3
- nat/agent/rewoo_agent/prompt.py +11 -12
- nat/agent/rewoo_agent/register.py +30 -32
- nat/authentication/http_basic_auth/http_basic_auth_provider.py +1 -1
- nat/authentication/interfaces.py +1 -1
- nat/authentication/oauth2/oauth2_auth_code_flow_provider.py +2 -2
- nat/builder/component_utils.py +5 -5
- nat/builder/context.py +5 -5
- nat/builder/front_end.py +6 -6
- nat/builder/function_base.py +4 -4
- nat/builder/function_info.py +1 -1
- nat/builder/intermediate_step_manager.py +4 -4
- nat/builder/user_interaction_manager.py +3 -3
- nat/builder/workflow_builder.py +2 -2
- nat/cli/cli_utils/validation.py +1 -1
- nat/cli/commands/configure/channel/add.py +1 -1
- nat/cli/commands/configure/channel/channel.py +1 -3
- nat/cli/commands/configure/channel/remove.py +1 -1
- nat/cli/commands/configure/channel/update.py +1 -1
- nat/cli/commands/configure/configure.py +2 -2
- nat/cli/commands/evaluate.py +1 -1
- nat/cli/commands/info/info.py +2 -4
- nat/cli/commands/info/list_components.py +2 -2
- nat/cli/commands/info/list_mcp.py +9 -9
- nat/cli/commands/registry/publish.py +3 -3
- nat/cli/commands/registry/pull.py +3 -3
- nat/cli/commands/registry/registry.py +1 -3
- nat/cli/commands/registry/remove.py +3 -3
- nat/cli/commands/registry/search.py +3 -3
- nat/cli/commands/start.py +4 -4
- nat/cli/commands/workflow/templates/pyproject.toml.j2 +4 -4
- nat/cli/commands/workflow/workflow_commands.py +6 -6
- nat/data_models/api_server.py +38 -31
- nat/data_models/component_ref.py +9 -9
- nat/data_models/dataset_handler.py +56 -10
- nat/data_models/discovery_metadata.py +21 -50
- nat/data_models/evaluate.py +2 -2
- nat/data_models/intermediate_step.py +4 -4
- nat/embedder/register.py +0 -2
- nat/eval/dataset_handler/dataset_handler.py +118 -5
- nat/eval/evaluate.py +1 -1
- nat/eval/evaluator/evaluator_model.py +3 -3
- nat/eval/rag_evaluator/evaluate.py +1 -1
- nat/eval/swe_bench_evaluator/evaluate.py +2 -2
- nat/experimental/test_time_compute/editing/motivation_aware_summarization.py +1 -1
- nat/experimental/test_time_compute/functions/plan_select_execute_function.py +4 -4
- nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py +1 -1
- nat/experimental/test_time_compute/scoring/llm_based_agent_scorer.py +1 -1
- nat/experimental/test_time_compute/scoring/llm_based_plan_scorer.py +1 -1
- nat/experimental/test_time_compute/search/multi_llm_planner.py +1 -1
- nat/experimental/test_time_compute/search/multi_query_retrieval_search.py +1 -1
- nat/experimental/test_time_compute/search/single_shot_multi_plan_planner.py +1 -1
- nat/experimental/test_time_compute/selection/llm_based_agent_output_selector.py +1 -1
- nat/experimental/test_time_compute/selection/llm_based_output_merging_selector.py +1 -1
- nat/experimental/test_time_compute/selection/llm_based_plan_selector.py +1 -1
- nat/front_ends/console/console_front_end_config.py +1 -1
- nat/front_ends/fastapi/fastapi_front_end_config.py +6 -7
- nat/front_ends/fastapi/fastapi_front_end_plugin.py +4 -4
- nat/front_ends/fastapi/fastapi_front_end_plugin_worker.py +13 -13
- nat/front_ends/fastapi/intermediate_steps_subscriber.py +1 -1
- nat/front_ends/fastapi/main.py +6 -6
- nat/front_ends/fastapi/message_handler.py +14 -3
- nat/front_ends/fastapi/message_validator.py +6 -8
- nat/front_ends/fastapi/response_helpers.py +3 -3
- nat/front_ends/mcp/mcp_front_end_config.py +3 -2
- nat/front_ends/mcp/mcp_front_end_plugin_worker.py +4 -4
- nat/front_ends/mcp/tool_converter.py +15 -16
- nat/memory/__init__.py +2 -2
- nat/meta/pypi.md +8 -8
- nat/object_store/__init__.py +2 -2
- nat/observability/exporter/base_exporter.py +1 -1
- nat/observability/exporter/raw_exporter.py +1 -1
- nat/observability/exporter_manager.py +1 -1
- nat/profiler/callbacks/agno_callback_handler.py +4 -4
- nat/profiler/callbacks/llama_index_callback_handler.py +2 -2
- nat/profiler/callbacks/semantic_kernel_callback_handler.py +1 -1
- nat/profiler/decorators/function_tracking.py +1 -1
- nat/profiler/forecasting/models/linear_model.py +3 -2
- nat/profiler/forecasting/models/random_forest_regressor.py +3 -2
- nat/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py +1 -1
- nat/profiler/inference_optimization/experimental/prefix_span_analysis.py +1 -1
- nat/profiler/profile_runner.py +1 -1
- nat/registry_handlers/local/local_handler.py +5 -5
- nat/registry_handlers/local/register_local.py +1 -1
- nat/registry_handlers/package_utils.py +17 -17
- nat/registry_handlers/pypi/pypi_handler.py +5 -5
- nat/registry_handlers/pypi/register_pypi.py +3 -3
- nat/registry_handlers/registry_handler_base.py +8 -8
- nat/registry_handlers/rest/register_rest.py +4 -4
- nat/registry_handlers/rest/rest_handler.py +6 -6
- nat/registry_handlers/schemas/package.py +3 -3
- nat/registry_handlers/schemas/publish.py +4 -4
- nat/registry_handlers/schemas/pull.py +4 -4
- nat/registry_handlers/schemas/search.py +7 -7
- nat/runtime/loader.py +21 -20
- nat/runtime/runner.py +2 -2
- nat/runtime/session.py +1 -1
- nat/settings/global_settings.py +2 -2
- nat/tool/chat_completion.py +1 -1
- nat/tool/code_execution/README.md +2 -2
- nat/tool/code_execution/test_code_execution_sandbox.py +1 -1
- nat/tool/mcp/exceptions.py +1 -1
- nat/tool/mcp/mcp_tool.py +1 -1
- nat/tool/retriever.py +3 -3
- nat/tool/server_tools.py +11 -11
- nat/utils/settings/global_settings.py +1 -1
- {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/METADATA +9 -9
- {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/RECORD +114 -116
- nat/embedder/langchain_client.py +0 -41
- nat/meta/module_to_distro.json +0 -4
- {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/WHEEL +0 -0
- {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/entry_points.txt +0 -0
- {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/licenses/LICENSE-3rd-party.txt +0 -0
- {nvidia_nat-1.2.0a20250813.dist-info → nvidia_nat-1.2.0rc6.dist-info}/licenses/LICENSE.md +0 -0
- {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/
|
|
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
|
"""
|
nat/tool/mcp/exceptions.py
CHANGED
|
@@ -108,7 +108,7 @@ class MCPToolNotFoundError(MCPError):
|
|
|
108
108
|
url=url,
|
|
109
109
|
category=MCPErrorCategory.TOOL_NOT_FOUND,
|
|
110
110
|
suggestions=[
|
|
111
|
-
"Use '
|
|
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
|
|
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="
|
|
33
|
+
class RetrieverConfig(FunctionBaseConfig, name="nat_retriever"):
|
|
34
34
|
"""
|
|
35
|
-
|
|
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
|
|
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
|
-
|
|
41
|
+
nat_context = Context.get()
|
|
42
42
|
|
|
43
|
-
method: str | None =
|
|
44
|
-
url_path: str | None =
|
|
45
|
-
url_scheme: str | None =
|
|
46
|
-
headers: Headers | None =
|
|
47
|
-
query_params: QueryParams | None =
|
|
48
|
-
path_params: dict[str, str] | None =
|
|
49
|
-
client_host: str | None =
|
|
50
|
-
client_port: int | None =
|
|
51
|
-
cookies: dict[str, str] | None =
|
|
52
|
-
conversation_id: str | None =
|
|
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
|
|
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.
|
|
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
|
-

|
|
324
324
|
|
|
325
325
|
# NVIDIA NeMo Agent Toolkit
|
|
326
326
|
|
|
327
|
-
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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)
|