nvidia-nat-llama-index 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.
Files changed (7) hide show
  1. nat/plugins/llama_index/llm.py +11 -0
  2. {nvidia_nat_llama_index-1.3.0rc1.dist-info → nvidia_nat_llama_index-1.4.0a20251008.dist-info}/METADATA +3 -2
  3. nvidia_nat_llama_index-1.4.0a20251008.dist-info/RECORD +11 -0
  4. nvidia_nat_llama_index-1.3.0rc1.dist-info/RECORD +0 -11
  5. {nvidia_nat_llama_index-1.3.0rc1.dist-info → nvidia_nat_llama_index-1.4.0a20251008.dist-info}/WHEEL +0 -0
  6. {nvidia_nat_llama_index-1.3.0rc1.dist-info → nvidia_nat_llama_index-1.4.0a20251008.dist-info}/entry_points.txt +0 -0
  7. {nvidia_nat_llama_index-1.3.0rc1.dist-info → nvidia_nat_llama_index-1.4.0a20251008.dist-info}/top_level.txt +0 -0
nat/plugins/llama_index/llm.py CHANGED
@@ -24,6 +24,7 @@ from nat.data_models.retry_mixin import RetryMixin
24
24
  from nat.data_models.thinking_mixin import ThinkingMixin
25
25
  from nat.llm.aws_bedrock_llm import AWSBedrockModelConfig
26
26
  from nat.llm.azure_openai_llm import AzureOpenAIModelConfig
27
+ from nat.llm.litellm_llm import LiteLlmModelConfig
27
28
  from nat.llm.nim_llm import NIMModelConfig
28
29
  from nat.llm.openai_llm import OpenAIModelConfig
29
30
  from nat.llm.utils.thinking import BaseThinkingInjector
@@ -107,3 +108,13 @@ async def openai_llama_index(llm_config: OpenAIModelConfig, _builder: Builder):
107
108
  llm = OpenAI(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True, exclude_none=True))
108
109
 
109
110
  yield _patch_llm_based_on_config(llm, llm_config)
111
+
112
+
113
+ @register_llm_client(config_type=LiteLlmModelConfig, wrapper_type=LLMFrameworkEnum.LLAMA_INDEX)
114
+ async def litellm_llama_index(llm_config: LiteLlmModelConfig, _builder: Builder):
115
+
116
+ from llama_index.llms.litellm import LiteLLM
117
+
118
+ llm = LiteLLM(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True, exclude_none=True))
119
+
120
+ yield _patch_llm_based_on_config(llm, llm_config)
{nvidia_nat_llama_index-1.3.0rc1.dist-info → nvidia_nat_llama_index-1.4.0a20251008.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat-llama-index
3
- Version: 1.3.0rc1
3
+ Version: 1.4.0a20251008
4
4
  Summary: Subpackage for Llama-Index integration in NeMo Agent toolkit
5
5
  Keywords: ai,rag,agents
6
6
  Classifier: Programming Language :: Python
@@ -9,13 +9,14 @@ 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==v1.3.0-rc1
12
+ Requires-Dist: nvidia-nat==v1.4.0a20251008
13
13
  Requires-Dist: llama-index-core~=0.12.21
14
14
  Requires-Dist: llama-index-embeddings-azure-openai~=0.3.9
15
15
  Requires-Dist: llama-index-embeddings-nvidia~=0.3.1
16
16
  Requires-Dist: llama-index-embeddings-openai~=0.3.1
17
17
  Requires-Dist: llama-index-llms-azure-openai~=0.3.2
18
18
  Requires-Dist: llama-index-llms-bedrock~=0.3.8
19
+ Requires-Dist: llama-index-llms-litellm~=0.5.1
19
20
  Requires-Dist: llama-index-llms-nvidia~=0.3.1
20
21
  Requires-Dist: llama-index-llms-openai~=0.3.42
21
22
  Requires-Dist: llama-index-readers-file~=0.4.4
nvidia_nat_llama_index-1.4.0a20251008.dist-info/RECORD ADDED
@@ -0,0 +1,11 @@
1
+ nat/meta/pypi.md,sha256=s9C3pgWB0HLIXTx5QPryNOWN0O2fIRIap0p9_zCHlTs,1112
2
+ nat/plugins/llama_index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
+ nat/plugins/llama_index/embedder.py,sha256=4c4uAa19UArqA_Npe5mwEnUhCaguTbyWcAntBzcPEeQ,3493
4
+ nat/plugins/llama_index/llm.py,sha256=ms6qwcBISWhiiOhIUXFNuhpDenTIS5s7U47wB3fEgg0,4944
5
+ nat/plugins/llama_index/register.py,sha256=1x_b8u6cuQwh4Iz_7TcIFWXvLIL9IIKUPE-zR9d6ug8,859
6
+ nat/plugins/llama_index/tool_wrapper.py,sha256=VFKMIIeLdWqHwW2Ax11E2w-_9w3ow6Iuhra1Hk78RYM,1387
7
+ nvidia_nat_llama_index-1.4.0a20251008.dist-info/METADATA,sha256=oWOAlsqnXZrhvGlC_LtGm54_ae4kluMGbbcaEm68ihM,2116
8
+ nvidia_nat_llama_index-1.4.0a20251008.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
+ nvidia_nat_llama_index-1.4.0a20251008.dist-info/entry_points.txt,sha256=2LqRRju5448P2v8B3y6TSPnk-nOd5T3AmV5JibCnoQc,68
10
+ nvidia_nat_llama_index-1.4.0a20251008.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
11
+ nvidia_nat_llama_index-1.4.0a20251008.dist-info/RECORD,,
nvidia_nat_llama_index-1.3.0rc1.dist-info/RECORD DELETED
@@ -1,11 +0,0 @@
1
- nat/meta/pypi.md,sha256=s9C3pgWB0HLIXTx5QPryNOWN0O2fIRIap0p9_zCHlTs,1112
2
- nat/plugins/llama_index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- nat/plugins/llama_index/embedder.py,sha256=4c4uAa19UArqA_Npe5mwEnUhCaguTbyWcAntBzcPEeQ,3493
4
- nat/plugins/llama_index/llm.py,sha256=0M1m1sjGwVMNibiAzkH2_hnAfdC-Ptc6PM9IKDmfSL8,4500
5
- nat/plugins/llama_index/register.py,sha256=1x_b8u6cuQwh4Iz_7TcIFWXvLIL9IIKUPE-zR9d6ug8,859
6
- nat/plugins/llama_index/tool_wrapper.py,sha256=VFKMIIeLdWqHwW2Ax11E2w-_9w3ow6Iuhra1Hk78RYM,1387
7
- nvidia_nat_llama_index-1.3.0rc1.dist-info/METADATA,sha256=RwXDLjxwnnOjiWB-XC5y6olMd0ckHTzRIDlE3SAafyg,2058
8
- nvidia_nat_llama_index-1.3.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
- nvidia_nat_llama_index-1.3.0rc1.dist-info/entry_points.txt,sha256=2LqRRju5448P2v8B3y6TSPnk-nOd5T3AmV5JibCnoQc,68
10
- nvidia_nat_llama_index-1.3.0rc1.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
11
- nvidia_nat_llama_index-1.3.0rc1.dist-info/RECORD,,