nvidia-nat-langchain 1.3.0a20250902__py3-none-any.whl → 1.3.0a20250904__py3-none-any.whl

nat/plugins/langchain/llm.py

@@ -78,10 +78,13 @@ def _patch_llm_based_on_config(client: ModelType, llm_config: LLMBaseConfig) ->
                 if all(isinstance(m, BaseMessage) for m in messages):
                     new_messages = [system_message, *list(messages)]
                     return FunctionArgumentWrapper(new_messages, *args, **kwargs)
-                raise ValueError(
-                    "Unsupported sequence element types for LanguageModelInput; expected Sequence[BaseMessage].")
-            else:
-                return FunctionArgumentWrapper(messages, *args, **kwargs)
+            raise ValueError(f"Unsupported message type: {type(messages)}")
+
+    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)
 
     if isinstance(llm_config, ThinkingMixin) and llm_config.thinking_system_prompt is not None:
         client = patch_with_thinking(
@@ -96,12 +99,6 @@ def _patch_llm_based_on_config(client: ModelType, llm_config: LLMBaseConfig) ->
                 ],
             ))
 
-    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)
-
     return client
 
 

{nvidia_nat_langchain-1.3.0a20250902.dist-info → nvidia_nat_langchain-1.3.0a20250904.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-langchain
-Version: 1.3.0a20250902
+Version: 1.3.0a20250904
 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.0a20250902
+Requires-Dist: nvidia-nat==v1.3.0a20250904
 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.0a20250902.dist-info → nvidia_nat_langchain-1.3.0a20250904.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=7bHXRcLwCeqG6ZQZQ35UAb46k5bN3UUL02OtG7RdL4Y,3282
-nat/plugins/langchain/llm.py,sha256=VbBq5wuTz8VjQjiUVE-bBfsQr2Zkx5wwNiTbvi-Ngaw,6537
+nat/plugins/langchain/llm.py,sha256=YqsfDWhmv_liJ9K-ix6twsRMVE6gxzyE088DOCjDWcE,6373
 nat/plugins/langchain/register.py,sha256=jgq6wSJoGQIZFJhS8RbUs25cLgNJjCkFu4M6qaWJS_4,906
 nat/plugins/langchain/retriever.py,sha256=SWbXXOezEUuPACnmSSU497NAmEVEMj2SrFJGodkRg34,2644
 nat/plugins/langchain/tool_wrapper.py,sha256=Zgb2_XB4bEhjPPeqS-ZH_OJT_pcQmteX7u03N_qCLfc,2121
@@ -10,8 +10,8 @@ nat/plugins/langchain/tools/code_generation_tool.py,sha256=qL3HBiOQzVPLw4EiUOWes
 nat/plugins/langchain/tools/register.py,sha256=uemxqLxcNk1bGX4crV52oMphLTZWonStzkXwTZeG2Rw,889
 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.0a20250902.dist-info/METADATA,sha256=WcD7aNtzozuHqJaqFBRePOBA2_rHX3QAQP8y9-MfQCs,1735
-nvidia_nat_langchain-1.3.0a20250902.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_langchain-1.3.0a20250902.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
-nvidia_nat_langchain-1.3.0a20250902.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_langchain-1.3.0a20250902.dist-info/RECORD,,
+nvidia_nat_langchain-1.3.0a20250904.dist-info/METADATA,sha256=_A2l291pvwOrJdAgHwxJmvEh7Q6N1FNDAVIOx53TlBs,1735
+nvidia_nat_langchain-1.3.0a20250904.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_langchain-1.3.0a20250904.dist-info/entry_points.txt,sha256=4deXsMn97I012HhDw0UjoqcZ8eEoZ7BnqaRx5QmzebY,123
+nvidia_nat_langchain-1.3.0a20250904.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_langchain-1.3.0a20250904.dist-info/RECORD,,