nvidia-nat-agno 1.3.0rc1__py3-none-any.whl → 1.4.0a20251008__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
nat/plugins/agno/llm.py CHANGED
@@ -21,6 +21,7 @@ from nat.cli.register_workflow import register_llm_client
21
21
  from nat.data_models.llm import LLMBaseConfig
22
22
  from nat.data_models.retry_mixin import RetryMixin
23
23
  from nat.data_models.thinking_mixin import ThinkingMixin
24
+ from nat.llm.litellm_llm import LiteLlmModelConfig
24
25
  from nat.llm.nim_llm import NIMModelConfig
25
26
  from nat.llm.openai_llm import OpenAIModelConfig
26
27
  from nat.llm.utils.thinking import BaseThinkingInjector
@@ -99,3 +100,20 @@ async def openai_agno(llm_config: OpenAIModelConfig, _builder: Builder):
99
100
  client = OpenAIChat(**config_obj, id=llm_config.model_name)
100
101
 
101
102
  yield _patch_llm_based_on_config(client, llm_config)
103
+
104
+
105
+ @register_llm_client(config_type=LiteLlmModelConfig, wrapper_type=LLMFrameworkEnum.AGNO)
106
+ async def litellm_agno(llm_config: LiteLlmModelConfig, _builder: Builder):
107
+
108
+ from agno.models.litellm.chat import LiteLLM
109
+
110
+ client = LiteLLM(
111
+ **llm_config.model_dump(
112
+ exclude={"type", "thinking", "model_name"},
113
+ by_alias=True,
114
+ exclude_none=True,
115
+ ),
116
+ id=llm_config.model_name,
117
+ )
118
+
119
+ yield _patch_llm_based_on_config(client, llm_config)
@@ -148,7 +148,7 @@ def execute_agno_tool(name: str,
148
148
  List of required fields for validation
149
149
  loop : asyncio.AbstractEventLoop
150
150
  The event loop to use for async execution
151
- **kwargs : Any
151
+ kwargs : Any
152
152
  The arguments to pass to the function
153
153
 
154
154
  Returns
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat-agno
3
- Version: 1.3.0rc1
3
+ Version: 1.4.0a20251008
4
4
  Summary: Subpackage for Agno integration in NeMo Agent toolkit
5
5
  Keywords: ai,rag,agents
6
6
  Classifier: Programming Language :: Python
@@ -9,7 +9,7 @@ Classifier: Programming Language :: Python :: 3.12
9
9
  Classifier: Programming Language :: Python :: 3.13
10
10
  Requires-Python: <3.14,>=3.11
11
11
  Description-Content-Type: text/markdown
12
- Requires-Dist: nvidia-nat[openai]==v1.3.0-rc1
12
+ Requires-Dist: nvidia-nat[openai]==v1.4.0a20251008
13
13
  Requires-Dist: agno~=1.2.3
14
14
  Requires-Dist: google-search-results~=2.4.2
15
15
 
@@ -0,0 +1,13 @@
1
+ nat/meta/pypi.md,sha256=tZD7hiOSYWgiAdddD1eIJ8T5ipZwEIjnd8ilgmasdmw,1198
2
+ nat/plugins/agno/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
+ nat/plugins/agno/llm.py,sha256=thW9_jyaPgU0HvMJLVIOSqR3xCIg5F8mmZd2d9Hj9pQ,4374
4
+ nat/plugins/agno/register.py,sha256=q-es1KVb_PTHOFszltym7e7Pj2jmyieih_Ve-cguHI8,788
5
+ nat/plugins/agno/tool_wrapper.py,sha256=MZLPeh1FHt15TBNRvNqGXxWzMhd4ROxuUfVdtUzwv5E,15727
6
+ nat/plugins/agno/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
+ nat/plugins/agno/tools/register.py,sha256=mpNgD1r51EoYlQiEAqPX15wCtx2alDcMIQiR8F_ub-M,743
8
+ nat/plugins/agno/tools/serp_api_tool.py,sha256=AJQH6-1iEUUrk_nzfZ3zZqutEKhJ_LMOUJi_iol65Sc,4442
9
+ nvidia_nat_agno-1.4.0a20251008.dist-info/METADATA,sha256=KkKSK9PfVPmy8OuTefyaOK9tdZqXWJgsIfUDMb1aslE,1742
10
+ nvidia_nat_agno-1.4.0a20251008.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
11
+ nvidia_nat_agno-1.4.0a20251008.dist-info/entry_points.txt,sha256=qRhuHKj2WmdJkLpbVXpYkdtc2cZdG4LPlBsABG2ImVI,103
12
+ nvidia_nat_agno-1.4.0a20251008.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
13
+ nvidia_nat_agno-1.4.0a20251008.dist-info/RECORD,,
@@ -1,13 +0,0 @@
1
- nat/meta/pypi.md,sha256=tZD7hiOSYWgiAdddD1eIJ8T5ipZwEIjnd8ilgmasdmw,1198
2
- nat/plugins/agno/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- nat/plugins/agno/llm.py,sha256=op02DUgf_DwYFxOwmUmd7NhFIf6TTXj313sFV6GOZZM,3828
4
- nat/plugins/agno/register.py,sha256=q-es1KVb_PTHOFszltym7e7Pj2jmyieih_Ve-cguHI8,788
5
- nat/plugins/agno/tool_wrapper.py,sha256=Iq8v0uO6HfSVQvcRwZOk4TMRpRyWUMM4s-XAtZx_Hj8,15729
6
- nat/plugins/agno/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
- nat/plugins/agno/tools/register.py,sha256=mpNgD1r51EoYlQiEAqPX15wCtx2alDcMIQiR8F_ub-M,743
8
- nat/plugins/agno/tools/serp_api_tool.py,sha256=AJQH6-1iEUUrk_nzfZ3zZqutEKhJ_LMOUJi_iol65Sc,4442
9
- nvidia_nat_agno-1.3.0rc1.dist-info/METADATA,sha256=8EKugYRK44pr-1SWNsOJGiobrJwL86I_iqcVD02yBAY,1731
10
- nvidia_nat_agno-1.3.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
11
- nvidia_nat_agno-1.3.0rc1.dist-info/entry_points.txt,sha256=qRhuHKj2WmdJkLpbVXpYkdtc2cZdG4LPlBsABG2ImVI,103
12
- nvidia_nat_agno-1.3.0rc1.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
13
- nvidia_nat_agno-1.3.0rc1.dist-info/RECORD,,