aiqtoolkit-llama-index 1.2.0a20250530__py3-none-any.whl → 1.2.0a20250531__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.

Potentially problematic release.

This version of aiqtoolkit-llama-index might be problematic. Click here for more details.

Files changed (6) hide show
  1. aiq/plugins/llama_index/llm.py +12 -2
  2. {aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/METADATA +3 -2
  3. {aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/RECORD +6 -6
  4. {aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/WHEEL +0 -0
  5. {aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/entry_points.txt +0 -0
  6. {aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/top_level.txt +0 -0
aiq/plugins/llama_index/llm.py CHANGED
@@ -16,6 +16,7 @@
16
16
  from aiq.builder.builder import Builder
17
17
  from aiq.builder.framework_enum import LLMFrameworkEnum
18
18
  from aiq.cli.register_workflow import register_llm_client
19
+ from aiq.llm.aws_bedrock_llm import AWSBedrockModelConfig
19
20
  from aiq.llm.nim_llm import NIMModelConfig
20
21
  from aiq.llm.openai_llm import OpenAIModelConfig
21
22
 
@@ -47,7 +48,16 @@ async def openai_llama_index(llm_config: OpenAIModelConfig, builder: Builder):
47
48
 
48
49
  llm = OpenAI(**kwargs)
49
50
 
50
- # Disable content blocks
51
- llm.supports_content_blocks = False
51
+ yield llm
52
+
53
+
54
+ @register_llm_client(config_type=AWSBedrockModelConfig, wrapper_type=LLMFrameworkEnum.LLAMA_INDEX)
55
+ async def aws_bedrock_llama_index(llm_config: AWSBedrockModelConfig, builder: Builder):
56
+
57
+ from llama_index.llms.bedrock import Bedrock
58
+
59
+ kwargs = llm_config.model_dump(exclude={"type", "max_tokens"}, by_alias=True)
60
+
61
+ llm = Bedrock(**kwargs)
52
62
 
53
63
  yield llm
{aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/METADATA RENAMED
@@ -1,14 +1,15 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aiqtoolkit-llama-index
3
- Version: 1.2.0a20250530
3
+ Version: 1.2.0a20250531
4
4
  Summary: Subpackage for Llama-Index integration in AIQtoolkit
5
5
  Keywords: ai,rag,agents
6
6
  Classifier: Programming Language :: Python
7
7
  Requires-Python: <3.13,>=3.11
8
8
  Description-Content-Type: text/markdown
9
- Requires-Dist: aiqtoolkit==v1.2.0a20250530
9
+ Requires-Dist: aiqtoolkit==v1.2.0a20250531
10
10
  Requires-Dist: llama-index-core==0.12.21
11
11
  Requires-Dist: llama-index-embeddings-nvidia==0.3.1
12
+ Requires-Dist: llama-index-llms-bedrock==0.3.8
12
13
  Requires-Dist: llama-index-llms-nvidia==0.3.1
13
14
  Requires-Dist: llama-index-readers-file==0.4.4
14
15
  Requires-Dist: llama-index==0.12.21
{aiqtoolkit_llama_index-1.2.0a20250530.dist-info → aiqtoolkit_llama_index-1.2.0a20250531.dist-info}/RECORD RENAMED
@@ -1,10 +1,10 @@
1
1
  aiq/meta/pypi.md,sha256=ekC3K0mCeJGZYxfubRfTs5qFLWDwQGf5_iCNqrD8sCw,1080
2
2
  aiq/plugins/llama_index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- aiq/plugins/llama_index/llm.py,sha256=3F-hNlqgkrTz_yd1WtarRmN1GhcLOTQCizuXGiKbBYw,1848
3
+ aiq/plugins/llama_index/llm.py,sha256=-zHaLyUHEJO5__500PzFu5Gt8HT_OlNSKmb26xThysc,2202
4
4
  aiq/plugins/llama_index/register.py,sha256=da1IYMrYe4MUtlwzBGIiukPazf7WU40j5X3n2KCjNts,868
5
5
  aiq/plugins/llama_index/tool_wrapper.py,sha256=n7E6uzVtcyFXSWtkYG8yrtqRCsXVHdEU38eTrWLSlrU,1387
6
- aiqtoolkit_llama_index-1.2.0a20250530.dist-info/METADATA,sha256=OetaCvmdUQkraz91xwM4erD4F1pF81Ix9FfGoZCg8Fw,1620
7
- aiqtoolkit_llama_index-1.2.0a20250530.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
8
- aiqtoolkit_llama_index-1.2.0a20250530.dist-info/entry_points.txt,sha256=2_FES--T1uOL8QdHJWWhuFipXoSK8uo0DR8olysK2Ck,68
9
- aiqtoolkit_llama_index-1.2.0a20250530.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
10
- aiqtoolkit_llama_index-1.2.0a20250530.dist-info/RECORD,,
6
+ aiqtoolkit_llama_index-1.2.0a20250531.dist-info/METADATA,sha256=phv3DvXaCG_d8yuk0bXDBWay6JEBfBfsUGwoY8VxOyY,1667
7
+ aiqtoolkit_llama_index-1.2.0a20250531.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
8
+ aiqtoolkit_llama_index-1.2.0a20250531.dist-info/entry_points.txt,sha256=2_FES--T1uOL8QdHJWWhuFipXoSK8uo0DR8olysK2Ck,68
9
+ aiqtoolkit_llama_index-1.2.0a20250531.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
10
+ aiqtoolkit_llama_index-1.2.0a20250531.dist-info/RECORD,,