aiqtoolkit-semantic-kernel 1.2.0a20250707__py3-none-any.whl → 1.2.0rc1__py3-none-any.whl

aiq/plugins/semantic_kernel/llm.py

@@ -16,7 +16,9 @@
 from aiq.builder.builder import Builder
 from aiq.builder.framework_enum import LLMFrameworkEnum
 from aiq.cli.register_workflow import register_llm_client
+from aiq.data_models.retry_mixin import RetryMixin
 from aiq.llm.openai_llm import OpenAIModelConfig
+from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
 @register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.SEMANTIC_KERNEL)
@@ -30,4 +32,10 @@ async def openai_semantic_kernel(llm_config: OpenAIModelConfig, builder: Builder
 
     llm = OpenAIChatCompletion(ai_model_id=config_obj.get("model"))
 
+    if isinstance(llm_config, RetryMixin):
+        llm = patch_with_retry(llm,
+                               retries=llm_config.num_retries,
+                               retry_codes=llm_config.retry_on_status_codes,
+                               retry_on_messages=llm_config.retry_on_errors)
+
     yield llm

{aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info → aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-semantic-kernel
-Version: 1.2.0a20250707
+Version: 1.2.0rc1
 Summary: Subpackage for Semantic-Kernel integration in AIQtoolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: aiqtoolkit==v1.2.0a20250707
+Requires-Dist: aiqtoolkit~=1.2
 Requires-Dist: semantic-kernel~=1.24.0
 
 <!--

aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info/RECORD

@@ -0,0 +1,10 @@
+aiq/meta/pypi.md,sha256=Fz0ye_uf6risEpiRBtYo301nUqmAuVCSquYhJMZQlzQ,1092
+aiq/plugins/semantic_kernel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiq/plugins/semantic_kernel/llm.py,sha256=l13R5Wv50iQ8OQ2juAFCoPoll6ZtIbojTnWjyGjepGI,1744
+aiq/plugins/semantic_kernel/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
+aiq/plugins/semantic_kernel/tool_wrapper.py,sha256=miogE5cp3o5H-KxaKp1dMqqJZnV-UMsYb1PQ4ox-xKY,7175
+aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info/METADATA,sha256=8k48IusIVpGzXIcJEWgbwX1x1AamYDG_8eFKN9OLi64,1439
+aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info/entry_points.txt,sha256=qGCOsf9tkWvc5EUonXrUI96cDrTJ3qoueyy3DgClK3M,76
+aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit_semantic_kernel-1.2.0rc1.dist-info/RECORD,,

aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info/RECORD

@@ -1,10 +0,0 @@
-aiq/meta/pypi.md,sha256=Fz0ye_uf6risEpiRBtYo301nUqmAuVCSquYhJMZQlzQ,1092
-aiq/plugins/semantic_kernel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/semantic_kernel/llm.py,sha256=WzzdOZXgWEUFzmfgfgLSTFjbvv0GDlaKIiyDChjANnQ,1320
-aiq/plugins/semantic_kernel/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
-aiq/plugins/semantic_kernel/tool_wrapper.py,sha256=miogE5cp3o5H-KxaKp1dMqqJZnV-UMsYb1PQ4ox-xKY,7175
-aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info/METADATA,sha256=JXU23UNGMKVZX5Y97ae7626LNRCM9CJ_adFn4ejp5cI,1457
-aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info/entry_points.txt,sha256=qGCOsf9tkWvc5EUonXrUI96cDrTJ3qoueyy3DgClK3M,76
-aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_semantic_kernel-1.2.0a20250707.dist-info/RECORD,,