agentiq-llama-index 1.0.0rc8__py3-none-any.whl

agentiq_llama_index-1.0.0rc8.dist-info/METADATA

@@ -0,0 +1,13 @@
+Metadata-Version: 2.2
+Name: agentiq-llama-index
+Version: 1.0.0rc8
+Summary: Subpackage for Llama-Index integration in AgentIQ
+Keywords: ai,rag,agents
+Classifier: Programming Language :: Python
+Requires-Python: >=3.12
+Requires-Dist: agentiq
+Requires-Dist: llama-index-core==0.12.15
+Requires-Dist: llama-index-embeddings-nvidia==0.3.1
+Requires-Dist: llama-index-llms-nvidia==0.3.1
+Requires-Dist: llama-index-readers-file==0.4.4
+Requires-Dist: llama-index==0.12.15

agentiq_llama_index-1.0.0rc8.dist-info/RECORD

@@ -0,0 +1,9 @@
+aiq/plugins/llama_index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiq/plugins/llama_index/llm.py,sha256=3F-hNlqgkrTz_yd1WtarRmN1GhcLOTQCizuXGiKbBYw,1848
+aiq/plugins/llama_index/register.py,sha256=da1IYMrYe4MUtlwzBGIiukPazf7WU40j5X3n2KCjNts,868
+aiq/plugins/llama_index/tool_wrapper.py,sha256=n7E6uzVtcyFXSWtkYG8yrtqRCsXVHdEU38eTrWLSlrU,1387
+agentiq_llama_index-1.0.0rc8.dist-info/METADATA,sha256=mQEsBs7coneoi2IVnmYXNWOyx8_7e7m0m-fOZ9ILjZE,461
+agentiq_llama_index-1.0.0rc8.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+agentiq_llama_index-1.0.0rc8.dist-info/entry_points.txt,sha256=2_FES--T1uOL8QdHJWWhuFipXoSK8uo0DR8olysK2Ck,68
+agentiq_llama_index-1.0.0rc8.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+agentiq_llama_index-1.0.0rc8.dist-info/RECORD,,

agentiq_llama_index-1.0.0rc8.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (76.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

agentiq_llama_index-1.0.0rc8.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[aiq.components]
+aiq_llama_index = aiq.plugins.llama_index.register

agentiq_llama_index-1.0.0rc8.dist-info/top_level.txt

@@ -0,0 +1 @@
+aiq

aiq/plugins/llama_index/llm.py

@@ -0,0 +1,53 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from aiq.builder.builder import Builder
+from aiq.builder.framework_enum import LLMFrameworkEnum
+from aiq.cli.register_workflow import register_llm_client
+from aiq.llm.nim_llm import NIMModelConfig
+from aiq.llm.openai_llm import OpenAIModelConfig
+
+
+@register_llm_client(config_type=NIMModelConfig, wrapper_type=LLMFrameworkEnum.LLAMA_INDEX)
+async def nim_llama_index(llm_config: NIMModelConfig, builder: Builder):
+
+    from llama_index.llms.nvidia import NVIDIA
+
+    kwargs = llm_config.model_dump(exclude={"type"}, by_alias=True)
+
+    if ("base_url" in kwargs and kwargs["base_url"] is None):
+        del kwargs["base_url"]
+
+    llm = NVIDIA(**kwargs)
+
+    yield llm
+
+
+@register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.LLAMA_INDEX)
+async def openai_llama_index(llm_config: OpenAIModelConfig, builder: Builder):
+
+    from llama_index.llms.openai import OpenAI
+
+    kwargs = llm_config.model_dump(exclude={"type"}, by_alias=True)
+
+    if ("base_url" in kwargs and kwargs["base_url"] is None):
+        del kwargs["base_url"]
+
+    llm = OpenAI(**kwargs)
+
+    # Disable content blocks
+    llm.supports_content_blocks = False
+
+    yield llm

aiq/plugins/llama_index/register.py

@@ -0,0 +1,23 @@
+# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# pylint: disable=unused-import
+# flake8: noqa
+# isort:skip_file
+
+# Import any providers which need to be automatically registered here
+
+from . import llm
+from . import tool_wrapper

aiq/plugins/llama_index/tool_wrapper.py

@@ -0,0 +1,32 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from aiq.builder.builder import Builder
+from aiq.builder.framework_enum import LLMFrameworkEnum
+from aiq.builder.function import Function
+from aiq.cli.register_workflow import register_tool_wrapper
+
+
+@register_tool_wrapper(wrapper_type=LLMFrameworkEnum.LLAMA_INDEX)
+def langchain_tool_wrapper(name: str, fn: Function, builder: Builder):
+
+    from llama_index.core.tools import FunctionTool
+
+    assert fn.input_schema is not None, "Tool must have input schema"
+
+    return FunctionTool.from_defaults(async_fn=fn.acall_invoke,
+                                      name=name,
+                                      description=fn.description,
+                                      fn_schema=fn.input_schema)