nvidia-nat-agno 1.3.0rc1__py3-none-any.whl → 1.3.0rc2__py3-none-any.whl

nat/plugins/agno/llm.py

@@ -21,6 +21,7 @@ from nat.cli.register_workflow import register_llm_client
 from nat.data_models.llm import LLMBaseConfig
 from nat.data_models.retry_mixin import RetryMixin
 from nat.data_models.thinking_mixin import ThinkingMixin
+from nat.llm.litellm_llm import LiteLlmModelConfig
 from nat.llm.nim_llm import NIMModelConfig
 from nat.llm.openai_llm import OpenAIModelConfig
 from nat.llm.utils.thinking import BaseThinkingInjector
@@ -99,3 +100,20 @@ async def openai_agno(llm_config: OpenAIModelConfig, _builder: Builder):
     client = OpenAIChat(**config_obj, id=llm_config.model_name)
 
     yield _patch_llm_based_on_config(client, llm_config)
+
+
+@register_llm_client(config_type=LiteLlmModelConfig, wrapper_type=LLMFrameworkEnum.AGNO)
+async def litellm_agno(llm_config: LiteLlmModelConfig, _builder: Builder):
+
+    from agno.models.litellm.chat import LiteLLM
+
+    client = LiteLLM(
+        **llm_config.model_dump(
+            exclude={"type", "thinking", "model_name"},
+            by_alias=True,
+            exclude_none=True,
+        ),
+        id=llm_config.model_name,
+    )
+
+    yield _patch_llm_based_on_config(client, llm_config)

nat/plugins/agno/tool_wrapper.py

@@ -148,7 +148,7 @@ def execute_agno_tool(name: str,
         List of required fields for validation
     loop : asyncio.AbstractEventLoop
         The event loop to use for async execution
-    **kwargs : Any
+    kwargs : Any
         The arguments to pass to the function
 
     Returns

{nvidia_nat_agno-1.3.0rc1.dist-info → nvidia_nat_agno-1.3.0rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-agno
-Version: 1.3.0rc1
+Version: 1.3.0rc2
 Summary: Subpackage for Agno integration in NeMo Agent toolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
@@ -9,7 +9,7 @@ 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[openai]==v1.3.0-rc1
+Requires-Dist: nvidia-nat[openai]==v1.3.0-rc2
 Requires-Dist: agno~=1.2.3
 Requires-Dist: google-search-results~=2.4.2
 

{nvidia_nat_agno-1.3.0rc1.dist-info → nvidia_nat_agno-1.3.0rc2.dist-info}/RECORD

@@ -1,13 +1,13 @@
 nat/meta/pypi.md,sha256=tZD7hiOSYWgiAdddD1eIJ8T5ipZwEIjnd8ilgmasdmw,1198
 nat/plugins/agno/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/plugins/agno/llm.py,sha256=op02DUgf_DwYFxOwmUmd7NhFIf6TTXj313sFV6GOZZM,3828
+nat/plugins/agno/llm.py,sha256=thW9_jyaPgU0HvMJLVIOSqR3xCIg5F8mmZd2d9Hj9pQ,4374
 nat/plugins/agno/register.py,sha256=q-es1KVb_PTHOFszltym7e7Pj2jmyieih_Ve-cguHI8,788
-nat/plugins/agno/tool_wrapper.py,sha256=Iq8v0uO6HfSVQvcRwZOk4TMRpRyWUMM4s-XAtZx_Hj8,15729
+nat/plugins/agno/tool_wrapper.py,sha256=MZLPeh1FHt15TBNRvNqGXxWzMhd4ROxuUfVdtUzwv5E,15727
 nat/plugins/agno/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/plugins/agno/tools/register.py,sha256=mpNgD1r51EoYlQiEAqPX15wCtx2alDcMIQiR8F_ub-M,743
 nat/plugins/agno/tools/serp_api_tool.py,sha256=AJQH6-1iEUUrk_nzfZ3zZqutEKhJ_LMOUJi_iol65Sc,4442
-nvidia_nat_agno-1.3.0rc1.dist-info/METADATA,sha256=8EKugYRK44pr-1SWNsOJGiobrJwL86I_iqcVD02yBAY,1731
-nvidia_nat_agno-1.3.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_agno-1.3.0rc1.dist-info/entry_points.txt,sha256=qRhuHKj2WmdJkLpbVXpYkdtc2cZdG4LPlBsABG2ImVI,103
-nvidia_nat_agno-1.3.0rc1.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_agno-1.3.0rc1.dist-info/RECORD,,
+nvidia_nat_agno-1.3.0rc2.dist-info/METADATA,sha256=atR790gQHrO7Rg7rYocwh3iyydFpvCI4N8ZkiraGCs8,1731
+nvidia_nat_agno-1.3.0rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_agno-1.3.0rc2.dist-info/entry_points.txt,sha256=qRhuHKj2WmdJkLpbVXpYkdtc2cZdG4LPlBsABG2ImVI,103
+nvidia_nat_agno-1.3.0rc2.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_agno-1.3.0rc2.dist-info/RECORD,,