nvidia-nat-langchain 1.3.dev0__py3-none-any.whl → 1.3.0.dev2__py3-none-any.whl

nat/plugins/langchain/embedder.py

@@ -12,23 +12,25 @@
 # 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 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.azure_openai_embedder import AzureOpenAIEmbedderModelConfig
 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)
-async def openai_langchain(embedder_config: OpenAIEmbedderModelConfig, builder: Builder):
+@register_embedder_client(config_type=AzureOpenAIEmbedderModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def azure_openai_langchain(embedder_config: AzureOpenAIEmbedderModelConfig, builder: Builder):
 
-    from langchain_openai import OpenAIEmbeddings
+    from langchain_openai import AzureOpenAIEmbeddings
 
-    client = OpenAIEmbeddings(**embedder_config.model_dump(exclude={"type"}, by_alias=True))
+    client = AzureOpenAIEmbeddings(**embedder_config.model_dump(exclude={"type"}, by_alias=True))
 
     if isinstance(embedder_config, RetryMixin):
         client = patch_with_retry(client,
@@ -53,3 +55,19 @@ async def nim_langchain(embedder_config: NIMEmbedderModelConfig, builder: Builde
                                   retry_on_messages=embedder_config.retry_on_errors)
 
     yield client
+
+
+@register_embedder_client(config_type=OpenAIEmbedderModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def openai_langchain(embedder_config: OpenAIEmbedderModelConfig, builder: Builder):
+
+    from langchain_openai import OpenAIEmbeddings
+
+    client = OpenAIEmbeddings(**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

nat/plugins/langchain/llm.py

@@ -18,17 +18,18 @@ 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.azure_openai_llm import AzureOpenAIModelConfig
 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)
-async def nim_langchain(llm_config: NIMModelConfig, builder: Builder):
+@register_llm_client(config_type=AWSBedrockModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def aws_bedrock_langchain(llm_config: AWSBedrockModelConfig, _builder: Builder):
 
-    from langchain_nvidia_ai_endpoints import ChatNVIDIA
+    from langchain_aws import ChatBedrockConverse
 
-    client = ChatNVIDIA(**llm_config.model_dump(exclude={"type"}, by_alias=True))
+    client = ChatBedrockConverse(**llm_config.model_dump(exclude={"type", "context_size"}, by_alias=True))
 
     if isinstance(llm_config, RetryMixin):
         client = patch_with_retry(client,
@@ -39,18 +40,28 @@ async def nim_langchain(llm_config: NIMModelConfig, builder: Builder):
     yield client
 
 
-@register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
-async def openai_langchain(llm_config: OpenAIModelConfig, builder: Builder):
+@register_llm_client(config_type=AzureOpenAIModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def azure_openai_langchain(llm_config: AzureOpenAIModelConfig, _builder: Builder):
 
-    from langchain_openai import ChatOpenAI
+    from langchain_openai import AzureChatOpenAI
 
-    # Default kwargs for OpenAI to include usage metadata in the response. If the user has set stream_usage to False, we
-    # will not include this.
-    default_kwargs = {"stream_usage": True}
+    client = AzureChatOpenAI(**llm_config.model_dump(exclude={"type"}, by_alias=True))
 
-    kwargs = {**default_kwargs, **llm_config.model_dump(exclude={"type"}, by_alias=True)}
+    if isinstance(llm_config, RetryMixin):
+        client = patch_with_retry(client,
+                                  retries=llm_config.num_retries,
+                                  retry_codes=llm_config.retry_on_status_codes,
+                                  retry_on_messages=llm_config.retry_on_errors)
 
-    client = ChatOpenAI(**kwargs)
+    yield client
+
+
+@register_llm_client(config_type=NIMModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def nim_langchain(llm_config: NIMModelConfig, _builder: Builder):
+
+    from langchain_nvidia_ai_endpoints import ChatNVIDIA
+
+    client = ChatNVIDIA(**llm_config.model_dump(exclude={"type"}, by_alias=True))
 
     if isinstance(llm_config, RetryMixin):
         client = patch_with_retry(client,
@@ -61,12 +72,18 @@ async def openai_langchain(llm_config: OpenAIModelConfig, builder: Builder):
     yield client
 
 
-@register_llm_client(config_type=AWSBedrockModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
-async def aws_bedrock_langchain(llm_config: AWSBedrockModelConfig, builder: Builder):
+@register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
+async def openai_langchain(llm_config: OpenAIModelConfig, _builder: Builder):
 
-    from langchain_aws import ChatBedrockConverse
+    from langchain_openai import ChatOpenAI
 
-    client = ChatBedrockConverse(**llm_config.model_dump(exclude={"type", "context_size"}, by_alias=True))
+    # Default kwargs for OpenAI to include usage metadata in the response. If the user has set stream_usage to False, we
+    # will not include this.
+    default_kwargs = {"stream_usage": True}
+
+    kwargs = {**default_kwargs, **llm_config.model_dump(exclude={"type"}, by_alias=True)}
+
+    client = ChatOpenAI(**kwargs)
 
     if isinstance(llm_config, RetryMixin):
         client = patch_with_retry(client,

{nvidia_nat_langchain-1.3.dev0.dist-info → nvidia_nat_langchain-1.3.0.dev2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-langchain
-Version: 1.3.dev0
+Version: 1.3.0.dev2
 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==v1.3-dev
+Requires-Dist: nvidia-nat==v1.3.0-dev2
 Requires-Dist: langchain-aws~=0.2.1
 Requires-Dist: langchain-core~=0.3.7
 Requires-Dist: langchain-nvidia-ai-endpoints~=0.3.5

{nvidia_nat_langchain-1.3.dev0.dist-info → nvidia_nat_langchain-1.3.0.dev2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 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/embedder.py,sha256=uFC1M9nF4nJ1GpgUO2vz-iy3T8zGDVJXpnSqQbuKJkk,3317
+nat/plugins/langchain/llm.py,sha256=HJZsk0i72ZN0QTb1hZSUV4IsLmE0Ab0JOYYnvtqJfno,4013
 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
@@ -10,8 +10,8 @@ nat/plugins/langchain/tools/code_generation_tool.py,sha256=qL3HBiOQzVPLw4EiUOWes
 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.3.dev0.dist-info/METADATA,sha256=pZXhSbwaTCLrIWXDHlT2VigunZoMe3s1qak16B5W6Wc,1722
-nvidia_nat_langchain-1.3.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_langchain-1.3.dev0.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
-nvidia_nat_langchain-1.3.dev0.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_langchain-1.3.dev0.dist-info/RECORD,,
+nvidia_nat_langchain-1.3.0.dev2.dist-info/METADATA,sha256=bKpfzosrGdCkCXCQu00t2VhKdTf5nJFahypMBHNn3I0,1727
+nvidia_nat_langchain-1.3.0.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_langchain-1.3.0.dev2.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
+nvidia_nat_langchain-1.3.0.dev2.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_langchain-1.3.0.dev2.dist-info/RECORD,,