nvidia-nat-langchain 1.3.0a20250929__py3-none-any.whl → 1.3.0a20250930__py3-none-any.whl

nat/plugins/langchain/tools/tavily_internet_search.py

@@ -33,7 +33,7 @@ class TavilyInternetSearchToolConfig(FunctionBaseConfig, name="tavily_internet_s
 async def tavily_internet_search(tool_config: TavilyInternetSearchToolConfig, builder: Builder):
     import os
 
-    from langchain_community.tools import TavilySearchResults
+    from langchain_tavily import TavilySearch
 
     if not os.environ.get("TAVILY_API_KEY"):
         os.environ["TAVILY_API_KEY"] = tool_config.api_key
@@ -41,20 +41,24 @@ async def tavily_internet_search(tool_config: TavilyInternetSearchToolConfig, bu
     # Refer to create_customize_workflow.md for instructions of getting the API key
 
     async def _tavily_internet_search(question: str) -> str:
+        """This tool retrieves relevant contexts from web search (using Tavily) for the given question.
+
+        Args:
+            question (str): The question to be answered.
+
+        Returns:
+            str: The web search results.
+        """
         # Search the web and get the requested amount of results
-        tavily_search = TavilySearchResults(max_results=tool_config.max_results)
+        tavily_search = TavilySearch(max_results=tool_config.max_results)
         search_docs = await tavily_search.ainvoke({'query': question})
         # Format
         web_search_results = "\n\n---\n\n".join(
-            [f'<Document href="{doc["url"]}"/>\n{doc["content"]}\n</Document>' for doc in search_docs])
+            [f'<Document href="{doc["url"]}"/>\n{doc["content"]}\n</Document>' for doc in search_docs["results"]])
         return web_search_results
 
     # Create a Generic NAT tool that can be used with any supported LLM framework
     yield FunctionInfo.from_fn(
         _tavily_internet_search,
-        description=("""This tool retrieves relevant contexts from web search (using Tavily) for the given question.
-
-                        Args:
-                            question (str): The question to be answered.
-                    """),
+        description=_tavily_internet_search.__doc__,
     )

{nvidia_nat_langchain-1.3.0a20250929.dist-info → nvidia_nat_langchain-1.3.0a20250930.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-langchain
-Version: 1.3.0a20250929
+Version: 1.3.0a20250930
 Summary: Subpackage for LangChain/LangGraph integration in NeMo Agent toolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
@@ -9,12 +9,13 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat==v1.3.0a20250929
+Requires-Dist: nvidia-nat==v1.3.0a20250930
 Requires-Dist: langchain-aws~=0.2.31
 Requires-Dist: langchain-core~=0.3.75
 Requires-Dist: langchain-milvus~=0.2.1
 Requires-Dist: langchain-nvidia-ai-endpoints~=0.3.17
 Requires-Dist: langchain-openai~=0.3.32
+Requires-Dist: langchain-tavily~=0.2.11
 Requires-Dist: langgraph~=0.6.7
 
 <!--

{nvidia_nat_langchain-1.3.0a20250929.dist-info → nvidia_nat_langchain-1.3.0a20250930.dist-info}/RECORD

@@ -8,10 +8,10 @@ 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=AnnLRY1xSU4DOzxbB8nFZRjHngXpqatPVOJ7yWV7jVw,2612
+nat/plugins/langchain/tools/tavily_internet_search.py,sha256=UFMP1xh_kC3fydMQBeV-oDZ-M7jnLcs5OkMSzgm7mng,2653
 nat/plugins/langchain/tools/wikipedia_search.py,sha256=431YwLsjoC_mdvMZ_gY0Q37Uqaue2ASnAHpwr4jWCaU,2197
-nvidia_nat_langchain-1.3.0a20250929.dist-info/METADATA,sha256=4k_8mdoA80wmclbjstdC5mvc5qK80nwGkZ-RWHkfOIk,1844
-nvidia_nat_langchain-1.3.0a20250929.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_langchain-1.3.0a20250929.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
-nvidia_nat_langchain-1.3.0a20250929.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_langchain-1.3.0a20250929.dist-info/RECORD,,
+nvidia_nat_langchain-1.3.0a20250930.dist-info/METADATA,sha256=gbhmJ8qBc7J0hDIB8BznPmbT0xOdwIQ8F0Hm_mNXR7k,1884
+nvidia_nat_langchain-1.3.0a20250930.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_langchain-1.3.0a20250930.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
+nvidia_nat_langchain-1.3.0a20250930.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_langchain-1.3.0a20250930.dist-info/RECORD,,