nvidia-nat-llama-index 1.3.0rc3__py3-none-any.whl → 1.3.0rc4__py3-none-any.whl

nat/plugins/llama_index/llm.py

@@ -44,6 +44,12 @@ def _patch_llm_based_on_config(client: ModelType, llm_config: LLMBaseConfig) ->
 
         @override
         def inject(self, messages: Sequence[ChatMessage], *args, **kwargs) -> FunctionArgumentWrapper:
+            # Attempt to inject the system prompt into the first system message
+            for message in messages:
+                if message.role == "system":
+                    message.content = f"{self.system_prompt}\n\n{message.content}"
+                    return FunctionArgumentWrapper(messages, *args, **kwargs)
+            # If no system message found, prepend a new one
             new_messages = [ChatMessage(role="system", content=self.system_prompt)] + list(messages)
             return FunctionArgumentWrapper(new_messages, *args, **kwargs)
 

{nvidia_nat_llama_index-1.3.0rc3.dist-info → nvidia_nat_llama_index-1.3.0rc4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-llama-index
-Version: 1.3.0rc3
+Version: 1.3.0rc4
 Summary: Subpackage for Llama-Index integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,7 +16,7 @@ Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
-Requires-Dist: nvidia-nat==v1.3.0-rc3
+Requires-Dist: nvidia-nat==v1.3.0-rc4
 Requires-Dist: llama-index-core~=0.12.21
 Requires-Dist: llama-index-embeddings-azure-openai~=0.3.9
 Requires-Dist: llama-index-embeddings-nvidia~=0.3.1

{nvidia_nat_llama_index-1.3.0rc3.dist-info → nvidia_nat_llama_index-1.3.0rc4.dist-info}/RECORD

@@ -1,13 +1,13 @@
 nat/meta/pypi.md,sha256=s9C3pgWB0HLIXTx5QPryNOWN0O2fIRIap0p9_zCHlTs,1112
 nat/plugins/llama_index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/plugins/llama_index/embedder.py,sha256=4c4uAa19UArqA_Npe5mwEnUhCaguTbyWcAntBzcPEeQ,3493
-nat/plugins/llama_index/llm.py,sha256=ms6qwcBISWhiiOhIUXFNuhpDenTIS5s7U47wB3fEgg0,4944
+nat/plugins/llama_index/llm.py,sha256=zddtT5byVK_yC-UdFxc124Uzv7Irt_rcf53f1DcqOx0,5327
 nat/plugins/llama_index/register.py,sha256=1x_b8u6cuQwh4Iz_7TcIFWXvLIL9IIKUPE-zR9d6ug8,859
 nat/plugins/llama_index/tool_wrapper.py,sha256=VFKMIIeLdWqHwW2Ax11E2w-_9w3ow6Iuhra1Hk78RYM,1387
-nvidia_nat_llama_index-1.3.0rc3.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_llama_index-1.3.0rc3.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_llama_index-1.3.0rc3.dist-info/METADATA,sha256=KQwnzgwx3LJts-o5V0FXaR-6EHKyBJBAIsxPhmoA9bU,2411
-nvidia_nat_llama_index-1.3.0rc3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_llama_index-1.3.0rc3.dist-info/entry_points.txt,sha256=2LqRRju5448P2v8B3y6TSPnk-nOd5T3AmV5JibCnoQc,68
-nvidia_nat_llama_index-1.3.0rc3.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_llama_index-1.3.0rc3.dist-info/RECORD,,
+nvidia_nat_llama_index-1.3.0rc4.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_llama_index-1.3.0rc4.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_llama_index-1.3.0rc4.dist-info/METADATA,sha256=y7nD1i7rUO9KnuDc35a7o7oXUu0UJhkmDBe6m3FLbcs,2411
+nvidia_nat_llama_index-1.3.0rc4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_llama_index-1.3.0rc4.dist-info/entry_points.txt,sha256=2LqRRju5448P2v8B3y6TSPnk-nOd5T3AmV5JibCnoQc,68
+nvidia_nat_llama_index-1.3.0rc4.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_llama_index-1.3.0rc4.dist-info/RECORD,,