nvidia-nat-langchain 1.2.0rc5__py3-none-any.whl → 1.2.0rc6__py3-none-any.whl

{aiq → nat}/meta/pypi.md

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![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")
+![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")
 
 # NVIDIA NeMo Agent Toolkit Subpackage
 This is a subpackage for LangChain and LangGraph integration in NeMo Agent toolkit.

{aiq → nat}/plugins/langchain/embedder.py

@@ -12,13 +12,15 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+# pylint: disable=unused-argument
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_embedder_client
-from aiq.data_models.retry_mixin import RetryMixin
-from aiq.embedder.openai_embedder import OpenAIEmbedderModelConfig
-from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_embedder_client
+from nat.data_models.retry_mixin import RetryMixin
+from nat.embedder.nim_embedder import NIMEmbedderModelConfig
+from nat.embedder.openai_embedder import OpenAIEmbedderModelConfig
+from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
 @register_embedder_client(config_type=OpenAIEmbedderModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
@@ -35,3 +37,19 @@ async def openai_langchain(embedder_config: OpenAIEmbedderModelConfig, builder:
                                   retry_on_messages=embedder_config.retry_on_errors)
 
     yield client
+
+
+@register_embedder_client(config_type=NIMEmbedderModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def nim_langchain(embedder_config: NIMEmbedderModelConfig, builder: Builder):
+
+    from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings
+
+    client = NVIDIAEmbeddings(**embedder_config.model_dump(exclude={"type"}, by_alias=True))
+
+    if isinstance(embedder_config, RetryMixin):
+        client = patch_with_retry(client,
+                                  retries=embedder_config.num_retries,
+                                  retry_codes=embedder_config.retry_on_status_codes,
+                                  retry_on_messages=embedder_config.retry_on_errors)
+
+    yield client

{aiq → nat}/plugins/langchain/llm.py

@@ -13,14 +13,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_llm_client
-from aiq.data_models.retry_mixin import RetryMixin
-from aiq.llm.aws_bedrock_llm import AWSBedrockModelConfig
-from aiq.llm.nim_llm import NIMModelConfig
-from aiq.llm.openai_llm import OpenAIModelConfig
-from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_llm_client
+from nat.data_models.retry_mixin import RetryMixin
+from nat.llm.aws_bedrock_llm import AWSBedrockModelConfig
+from nat.llm.nim_llm import NIMModelConfig
+from nat.llm.openai_llm import OpenAIModelConfig
+from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
 @register_llm_client(config_type=NIMModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)

{aiq → nat}/plugins/langchain/retriever.py

@@ -13,17 +13,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_retriever_client
-from aiq.retriever.milvus.register import MilvusRetrieverConfig
-from aiq.retriever.nemo_retriever.register import NemoRetrieverConfig
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_retriever_client
+from nat.retriever.milvus.register import MilvusRetrieverConfig
+from nat.retriever.nemo_retriever.register import NemoRetrieverConfig
 
 
 @register_retriever_client(config_type=NemoRetrieverConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
 async def nemo_langchain(retriever_config: NemoRetrieverConfig, builder: Builder):
-    from aiq.retriever.nemo_retriever.retriever import NemoLangchainRetriever
-    from aiq.retriever.nemo_retriever.retriever import NemoRetriever
+    from nat.retriever.nemo_retriever.retriever import NemoLangchainRetriever
+    from nat.retriever.nemo_retriever.retriever import NemoRetriever
 
     retriever = NemoRetriever(**retriever_config.model_dump(exclude={"type", "top_k", "collection_name"}))
     optional_fields = ["collection_name", "top_k", "output_fields"]

{aiq → nat}/plugins/langchain/tool_wrapper.py

@@ -13,20 +13,40 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.builder.function import Function
-from aiq.cli.register_workflow import register_tool_wrapper
+import logging
+
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.builder.function import Function
+from nat.cli.register_workflow import register_tool_wrapper
+
+logger = logging.getLogger(__name__)
 
 
 @register_tool_wrapper(wrapper_type=LLMFrameworkEnum.LANGCHAIN)
-def langchain_tool_wrapper(name: str, fn: Function, builder: Builder):
+def langchain_tool_wrapper(name: str, fn: Function, builder: Builder):  # pylint: disable=unused-argument
+
+    import asyncio
 
     from langchain_core.tools.structured import StructuredTool
 
     assert fn.input_schema is not None, "Tool must have input schema"
 
+    loop = asyncio.get_running_loop()
+
+    # Provide a sync wrapper for the tool to support synchronous tool calls
+    def _sync_fn(*args, **kwargs):
+        logger.warning("Invoking a synchronous tool call, performance may be degraded: `%s`", fn.instance_name)
+        return loop.run_until_complete(fn.acall_invoke(*args, **kwargs))
+
+    if fn.description is None:
+        logger.warning("No description set for `%s` falling back to instance name: `%s`",
+                       type(fn).__name__,
+                       fn.instance_name)
+        _sync_fn.__doc__ = fn.instance_name
+
     return StructuredTool.from_function(coroutine=fn.acall_invoke,
+                                        func=_sync_fn,
                                         name=name,
                                         description=fn.description,
                                         args_schema=fn.input_schema)

{aiq → nat}/plugins/langchain/tools/code_generation_tool.py

@@ -15,12 +15,12 @@
 
 import logging
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.builder.function_info import FunctionInfo
-from aiq.cli.register_workflow import register_function
-from aiq.data_models.component_ref import LLMRef
-from aiq.data_models.function import FunctionBaseConfig
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.builder.function_info import FunctionInfo
+from nat.cli.register_workflow import register_function
+from nat.data_models.component_ref import LLMRef
+from nat.data_models.function import FunctionBaseConfig
 
 log = logging.getLogger(__name__)
 

{aiq → nat}/plugins/langchain/tools/tavily_internet_search.py

@@ -13,10 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.builder.builder import Builder
-from aiq.builder.function_info import FunctionInfo
-from aiq.cli.register_workflow import register_function
-from aiq.data_models.function import FunctionBaseConfig
+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.function import FunctionBaseConfig
 
 
 # Internet Search tool
@@ -49,7 +49,7 @@ async def tavily_internet_search(tool_config: TavilyInternetSearchToolConfig, bu
             [f'<Document href="{doc["url"]}"/>\n{doc["content"]}\n</Document>' for doc in search_docs])
         return web_search_results
 
-    # Create a Generic AIQ Toolkit tool that can be used with any supported LLM framework
+    # 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.

{aiq → nat}/plugins/langchain/tools/wikipedia_search.py

@@ -13,10 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.builder.builder import Builder
-from aiq.builder.function_info import FunctionInfo
-from aiq.cli.register_workflow import register_function
-from aiq.data_models.function import FunctionBaseConfig
+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.function import FunctionBaseConfig
 
 
 # Wikipedia Search tool
@@ -41,7 +41,7 @@ async def wiki_search(tool_config: WikiSearchToolConfig, builder: Builder):
         ])
         return wiki_search_results
 
-    # Create an AIQ Toolkit wiki search tool that can be used with any supported LLM framework
+    # Create a NAT wiki search tool that can be used with any supported LLM framework
     yield FunctionInfo.from_fn(
         _wiki_search,
         description=("""This tool retrieves relevant contexts from wikipedia search for the given question.

{nvidia_nat_langchain-1.2.0rc5.dist-info → nvidia_nat_langchain-1.2.0rc6.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-langchain
-Version: 1.2.0rc5
-Summary: Subpackage for LangChain and LangGraph integration in AIQtoolkit
+Version: 1.2.0rc6
+Summary: Subpackage for LangChain and LangGraph integration in NeMo Agent toolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat~=1.2
+Requires-Dist: nvidia-nat==v1.2.0-rc6
 Requires-Dist: langchain-aws~=0.2.1
 Requires-Dist: langchain-core~=0.3.7
 Requires-Dist: langchain-nvidia-ai-endpoints~=0.3.5
@@ -32,7 +32,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![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")
+![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")
 
 # NVIDIA NeMo Agent Toolkit Subpackage
 This is a subpackage for LangChain and LangGraph integration in NeMo Agent toolkit.

nvidia_nat_langchain-1.2.0rc6.dist-info/RECORD

@@ -0,0 +1,17 @@
+nat/meta/pypi.md,sha256=-RewrXPwhrT6398iluvXb5lefn18PybmvRFhmZF7KVI,1124
+nat/plugins/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/langchain/embedder.py,sha256=UcDB4GYY0vW3pUiFqqKhNMgAmEZdRSUJUazjThkPHuQ,2519
+nat/plugins/langchain/llm.py,sha256=i_DffRmQ9PEELuXFYR2ouU0LRkPKAjwZi1z65xGrxws,3293
+nat/plugins/langchain/register.py,sha256=UwxFY-HsZ5n32XPpqDtLvBamiw1Pdavtf2oYSK_XGtY,938
+nat/plugins/langchain/retriever.py,sha256=SWbXXOezEUuPACnmSSU497NAmEVEMj2SrFJGodkRg34,2644
+nat/plugins/langchain/tool_wrapper.py,sha256=dLK09a8lrmytNkIuhTaSByuLdW23aVmIcr3REyuJTMA,2156
+nat/plugins/langchain/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/langchain/tools/code_generation_tool.py,sha256=qL3HBiOQzVPLw4EiUOWeswckuVX8ynG2UQXYBLxR_gI,2724
+nat/plugins/langchain/tools/register.py,sha256=3cf4RH2tQ_qOtZviwXQUqK5dKiVUJQYach4djxGpcOU,921
+nat/plugins/langchain/tools/tavily_internet_search.py,sha256=AnnLRY1xSU4DOzxbB8nFZRjHngXpqatPVOJ7yWV7jVw,2612
+nat/plugins/langchain/tools/wikipedia_search.py,sha256=431YwLsjoC_mdvMZ_gY0Q37Uqaue2ASnAHpwr4jWCaU,2197
+nvidia_nat_langchain-1.2.0rc6.dist-info/METADATA,sha256=pllMoYf-NSsiYN89RNHphv4a3FhIzxQcqj-RH0v2QLs,1724
+nvidia_nat_langchain-1.2.0rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_langchain-1.2.0rc6.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
+nvidia_nat_langchain-1.2.0rc6.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_langchain-1.2.0rc6.dist-info/RECORD,,

nvidia_nat_langchain-1.2.0rc6.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[nat.components]
+nat_langchain = nat.plugins.langchain.register
+nat_langchain_tools = nat.plugins.langchain.tools.register

nvidia_nat_langchain-1.2.0rc6.dist-info/top_level.txt

@@ -0,0 +1 @@
+nat

nvidia_nat_langchain-1.2.0rc5.dist-info/RECORD

@@ -1,17 +0,0 @@
-aiq/meta/pypi.md,sha256=jxJ0L4icshSNDFo-5xV1OQrSqK9R31Lx4K6Tg3oXP44,1135
-aiq/plugins/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/langchain/embedder.py,sha256=MU63gh_hWNojlB6AG8WpER-3GIUan5CUIpm6_YOf9xg,1727
-aiq/plugins/langchain/llm.py,sha256=5M-Q0DKycb_RWLj7SQWmEOpkpSHlXijC0Cus9deKhv8,3293
-aiq/plugins/langchain/register.py,sha256=UwxFY-HsZ5n32XPpqDtLvBamiw1Pdavtf2oYSK_XGtY,938
-aiq/plugins/langchain/retriever.py,sha256=P6PdppeQbROalO69w1Q4U-1f6hpdMor0fOEATb4SDw0,2644
-aiq/plugins/langchain/tool_wrapper.py,sha256=XDh9JBC5X9pTmTtfBsc4zLAjDMIjaYV5P8g8LIB83W0,1407
-aiq/plugins/langchain/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/langchain/tools/code_generation_tool.py,sha256=ArBhVXgk9KMvF4VHz4SKwGCSbmG-RAN9DmsCljRY4Uo,2724
-aiq/plugins/langchain/tools/register.py,sha256=3cf4RH2tQ_qOtZviwXQUqK5dKiVUJQYach4djxGpcOU,921
-aiq/plugins/langchain/tools/tavily_internet_search.py,sha256=zB-GpqGzJ1_73_DjOlhsEBajyFEnvYVMlD_ObOr7lSM,2620
-aiq/plugins/langchain/tools/wikipedia_search.py,sha256=qPPQg0nUwjrVTTmo3CH9s13QyiNBnsaue6vFEggqCLo,2206
-nvidia_nat_langchain-1.2.0rc5.dist-info/METADATA,sha256=QGFTWeNsT7losdEf_VhAtv7E9PZRHJJH5a3xkCZ-wAc,1720
-nvidia_nat_langchain-1.2.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_langchain-1.2.0rc5.dist-info/entry_points.txt,sha256=WoapTU8bMDFxwRqT6eATj3l6kr15fHwifvfsjpuxAnA,123
-nvidia_nat_langchain-1.2.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-nvidia_nat_langchain-1.2.0rc5.dist-info/RECORD,,

nvidia_nat_langchain-1.2.0rc5.dist-info/entry_points.txt

@@ -1,3 +0,0 @@
-[aiq.components]
-aiq_langchain = aiq.plugins.langchain.register
-aiq_langchain_tools = aiq.plugins.langchain.tools.register

nvidia_nat_langchain-1.2.0rc5.dist-info/top_level.txt

@@ -1 +0,0 @@
-aiq