nvidia-nat-langchain 1.4.0a20251029__py3-none-any.whl → 1.4.0a20251031__py3-none-any.whl

nat/plugins/langchain/tools/tavily_internet_search.py

@@ -13,9 +13,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from pydantic import Field
+
 from nat.builder.builder import Builder
 from nat.builder.function_info import FunctionInfo
 from nat.cli.register_workflow import register_function
+from nat.data_models.common import SerializableSecretStr
+from nat.data_models.common import get_secret_value
 from nat.data_models.function import FunctionBaseConfig
 
 
@@ -26,7 +30,7 @@ class TavilyInternetSearchToolConfig(FunctionBaseConfig, name="tavily_internet_s
     Requires a TAVILY_API_KEY.
     """
     max_results: int = 3
-    api_key: str = ""
+    api_key: SerializableSecretStr = Field(default="", description="The API key for the Tavily service.")
 
 
 @register_function(config_type=TavilyInternetSearchToolConfig)
@@ -36,7 +40,8 @@ async def tavily_internet_search(tool_config: TavilyInternetSearchToolConfig, bu
     from langchain_tavily import TavilySearch
 
     if not os.environ.get("TAVILY_API_KEY"):
-        os.environ["TAVILY_API_KEY"] = tool_config.api_key
+        if tool_config.api_key:
+            os.environ["TAVILY_API_KEY"] = get_secret_value(tool_config.api_key)
     # This tavily tool requires an API Key and it must be set as an environment variable (TAVILY_API_KEY)
     # Refer to create_customize_workflow.md for instructions of getting the API key
 

{nvidia_nat_langchain-1.4.0a20251029.dist-info → nvidia_nat_langchain-1.4.0a20251031.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-langchain
-Version: 1.4.0a20251029
+Version: 1.4.0a20251031
 Summary: Subpackage for LangChain/LangGraph integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,7 +16,7 @@ Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
-Requires-Dist: nvidia-nat==v1.4.0a20251029
+Requires-Dist: nvidia-nat==v1.4.0a20251031
 Requires-Dist: langchain~=0.3.27
 Requires-Dist: langchain-aws~=0.2.31
 Requires-Dist: langchain-core~=0.3.75

{nvidia_nat_langchain-1.4.0a20251029.dist-info → nvidia_nat_langchain-1.4.0a20251031.dist-info}/RECORD

@@ -8,12 +8,12 @@ nat/plugins/langchain/tool_wrapper.py,sha256=Zgb2_XB4bEhjPPeqS-ZH_OJT_pcQmteX7u0
 nat/plugins/langchain/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/plugins/langchain/tools/code_generation_tool.py,sha256=f5pna0WMOx3QOS4WnaMFKD7tBZ1-tS0PfI0IMYobtTQ,2723
 nat/plugins/langchain/tools/register.py,sha256=uemxqLxcNk1bGX4crV52oMphLTZWonStzkXwTZeG2Rw,889
-nat/plugins/langchain/tools/tavily_internet_search.py,sha256=UFMP1xh_kC3fydMQBeV-oDZ-M7jnLcs5OkMSzgm7mng,2653
+nat/plugins/langchain/tools/tavily_internet_search.py,sha256=W5sdZ9hobPc3xbnWPSbtFBClIn14EM8xT0XUVF2HpWo,2928
 nat/plugins/langchain/tools/wikipedia_search.py,sha256=431YwLsjoC_mdvMZ_gY0Q37Uqaue2ASnAHpwr4jWCaU,2197
-nvidia_nat_langchain-1.4.0a20251029.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_langchain-1.4.0a20251029.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_langchain-1.4.0a20251029.dist-info/METADATA,sha256=oF7Cd7VE2x8BGddcrxW6PxqzSqLBt7hXZpJgdqKw4ok,2263
-nvidia_nat_langchain-1.4.0a20251029.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_langchain-1.4.0a20251029.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
-nvidia_nat_langchain-1.4.0a20251029.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_langchain-1.4.0a20251029.dist-info/RECORD,,
+nvidia_nat_langchain-1.4.0a20251031.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_langchain-1.4.0a20251031.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_langchain-1.4.0a20251031.dist-info/METADATA,sha256=IMGSc_w655owDdGmTDmqkUsI_uhAVG9YBthllbyh45g,2263
+nvidia_nat_langchain-1.4.0a20251031.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_langchain-1.4.0a20251031.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
+nvidia_nat_langchain-1.4.0a20251031.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_langchain-1.4.0a20251031.dist-info/RECORD,,