nvidia-nat-llama-index 1.3.0a20250906__py3-none-any.whl → 1.3.0a20250910__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 (6) hide show
  1. nat/plugins/llama_index/llm.py +4 -4
  2. {nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.dist-info}/METADATA +2 -2
  3. {nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.dist-info}/RECORD +6 -6
  4. {nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.dist-info}/WHEEL +0 -0
  5. {nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.dist-info}/entry_points.txt +0 -0
  6. {nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.dist-info}/top_level.txt +0 -0
nat/plugins/llama_index/llm.py CHANGED
@@ -74,7 +74,7 @@ async def aws_bedrock_llama_index(llm_config: AWSBedrockModelConfig, _builder: B
74
74
  from llama_index.llms.bedrock import Bedrock
75
75
 
76
76
  # LlamaIndex uses context_size instead of max_tokens
77
- llm = Bedrock(**llm_config.model_dump(exclude={"type", "top_p"}, by_alias=True), )
77
+ llm = Bedrock(**llm_config.model_dump(exclude={"type", "top_p", "thinking"}, by_alias=True))
78
78
 
79
79
  yield _patch_llm_based_on_config(llm, llm_config)
80
80
 
@@ -84,7 +84,7 @@ async def azure_openai_llama_index(llm_config: AzureOpenAIModelConfig, _builder:
84
84
 
85
85
  from llama_index.llms.azure_openai import AzureOpenAI
86
86
 
87
- llm = AzureOpenAI(**llm_config.model_dump(exclude={"type"}, by_alias=True))
87
+ llm = AzureOpenAI(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True))
88
88
 
89
89
  yield _patch_llm_based_on_config(llm, llm_config)
90
90
 
@@ -94,7 +94,7 @@ async def nim_llama_index(llm_config: NIMModelConfig, _builder: Builder):
94
94
 
95
95
  from llama_index.llms.nvidia import NVIDIA
96
96
 
97
- llm = NVIDIA(**llm_config.model_dump(exclude={"type"}, by_alias=True, exclude_none=True))
97
+ llm = NVIDIA(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True, exclude_none=True))
98
98
 
99
99
  yield _patch_llm_based_on_config(llm, llm_config)
100
100
 
@@ -104,6 +104,6 @@ async def openai_llama_index(llm_config: OpenAIModelConfig, _builder: Builder):
104
104
 
105
105
  from llama_index.llms.openai import OpenAI
106
106
 
107
- llm = OpenAI(**llm_config.model_dump(exclude={"type"}, by_alias=True, exclude_none=True))
107
+ llm = OpenAI(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True, exclude_none=True))
108
108
 
109
109
  yield _patch_llm_based_on_config(llm, llm_config)
{nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.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.0a20250906
3
+ Version: 1.3.0a20250910
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,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==v1.3.0a20250906
12
+ Requires-Dist: nvidia-nat==v1.3.0a20250910
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
{nvidia_nat_llama_index-1.3.0a20250906.dist-info → nvidia_nat_llama_index-1.3.0a20250910.dist-info}/RECORD RENAMED
@@ -1,11 +1,11 @@
1
1
  nat/meta/pypi.md,sha256=s9C3pgWB0HLIXTx5QPryNOWN0O2fIRIap0p9_zCHlTs,1112
2
2
  nat/plugins/llama_index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  nat/plugins/llama_index/embedder.py,sha256=6US_DR4dWwHuhRnf2KlN8TVIV7wk7NHTDFyBwgMyEQc,3093
4
- nat/plugins/llama_index/llm.py,sha256=XJZ7U5dXMX-k4K_dIufqS4QUwavhiQZVOj4r7NvADWo,4454
4
+ nat/plugins/llama_index/llm.py,sha256=0M1m1sjGwVMNibiAzkH2_hnAfdC-Ptc6PM9IKDmfSL8,4500
5
5
  nat/plugins/llama_index/register.py,sha256=1x_b8u6cuQwh4Iz_7TcIFWXvLIL9IIKUPE-zR9d6ug8,859
6
6
  nat/plugins/llama_index/tool_wrapper.py,sha256=VFKMIIeLdWqHwW2Ax11E2w-_9w3ow6Iuhra1Hk78RYM,1387
7
- nvidia_nat_llama_index-1.3.0a20250906.dist-info/METADATA,sha256=ZgZNWh4Z9vzhBrokDnJlLoRrq5L5OAIQ3LQ-dKpe_tQ,2069
8
- nvidia_nat_llama_index-1.3.0a20250906.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
- nvidia_nat_llama_index-1.3.0a20250906.dist-info/entry_points.txt,sha256=2LqRRju5448P2v8B3y6TSPnk-nOd5T3AmV5JibCnoQc,68
10
- nvidia_nat_llama_index-1.3.0a20250906.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
11
- nvidia_nat_llama_index-1.3.0a20250906.dist-info/RECORD,,
7
+ nvidia_nat_llama_index-1.3.0a20250910.dist-info/METADATA,sha256=fUzCPdJwoPg4biQ5NqGT2Fzg7nD8Edni_23vgPguc14,2069
8
+ nvidia_nat_llama_index-1.3.0a20250910.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
+ nvidia_nat_llama_index-1.3.0a20250910.dist-info/entry_points.txt,sha256=2LqRRju5448P2v8B3y6TSPnk-nOd5T3AmV5JibCnoQc,68
10
+ nvidia_nat_llama_index-1.3.0a20250910.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
11
+ nvidia_nat_llama_index-1.3.0a20250910.dist-info/RECORD,,