aiqtoolkit 1.1.0a20250505__py3-none-any.whl → 1.1.0a20250507__py3-none-any.whl

aiq/data_models/discovery_metadata.py

@@ -150,7 +150,11 @@ class DiscoveryMetadata(BaseModel):
                 logger.error("Encountered issue getting distro_name for module %s", module.__name__)
                 return DiscoveryMetadata(status=DiscoveryStatusEnum.FAILURE)
 
-            version = importlib.metadata.version(distro_name) if distro_name != "" else ""
+            try:
+                version = importlib.metadata.version(distro_name) if distro_name != "" else ""
+            except importlib.metadata.PackageNotFoundError:
+                logger.warning("Package metadata not found for %s", distro_name)
+                version = ""
         except Exception as e:
             logger.exception("Encountered issue extracting module metadata for %s: %s", config_type, e, exc_info=True)
             return DiscoveryMetadata(status=DiscoveryStatusEnum.FAILURE)
@@ -185,7 +189,11 @@ class DiscoveryMetadata(BaseModel):
             module = inspect.getmodule(fn)
             root_package: str = module.__package__.split(".")[0]
             root_package = DiscoveryMetadata.get_distribution_name(root_package)
-            version = importlib.metadata.version(root_package) if root_package != "" else ""
+            try:
+                version = importlib.metadata.version(root_package) if root_package != "" else ""
+            except importlib.metadata.PackageNotFoundError:
+                logger.warning("Package metadata not found for %s", root_package)
+                version = ""
         except Exception as e:
             logger.exception("Encountered issue extracting module metadata for %s: %s", fn, e, exc_info=True)
             return DiscoveryMetadata(status=DiscoveryStatusEnum.FAILURE)
@@ -213,10 +221,15 @@ class DiscoveryMetadata(BaseModel):
 
         try:
             package_name = DiscoveryMetadata.get_distribution_name(package_name)
-            metadata = importlib.metadata.metadata(package_name)
-            description = metadata.get("Summary", "")
-            if (package_version is None):
-                package_version = importlib.metadata.version(package_name)
+            try:
+                metadata = importlib.metadata.metadata(package_name)
+                description = metadata.get("Summary", "")
+                if (package_version is None):
+                    package_version = importlib.metadata.version(package_name)
+            except importlib.metadata.PackageNotFoundError:
+                logger.warning("Package metadata not found for %s", package_name)
+                description = ""
+                package_version = package_version or ""
         except Exception as e:
             logger.exception("Encountered issue extracting module metadata for %s: %s", package_name, e, exc_info=True)
             return DiscoveryMetadata(status=DiscoveryStatusEnum.FAILURE)
@@ -252,7 +265,11 @@ class DiscoveryMetadata(BaseModel):
             module = inspect.getmodule(config_type)
             root_package: str = module.__package__.split(".")[0]
             root_package = DiscoveryMetadata.get_distribution_name(root_package)
-            version = importlib.metadata.version(root_package) if root_package != "" else ""
+            try:
+                version = importlib.metadata.version(root_package) if root_package != "" else ""
+            except importlib.metadata.PackageNotFoundError:
+                logger.warning("Package metadata not found for %s", root_package)
+                version = ""
         except Exception as e:
             logger.exception("Encountered issue extracting module metadata for %s: %s", config_type, e, exc_info=True)
             return DiscoveryMetadata(status=DiscoveryStatusEnum.FAILURE)

aiq/eval/evaluate.py

@@ -177,8 +177,10 @@ class EvaluationRun:  # pylint: disable=too-many-public-methods
         workflow_output_file.parent.mkdir(parents=True, exist_ok=True)
 
         # Write the workflow output to a file (this can be used for re-running the evaluation)
-        workflow_output = dataset_handler.publish_eval_input(
-            self.eval_input, self.eval_config.general.output.workflow_output_step_filter)
+
+        step_filter = self.eval_config.general.output.workflow_output_step_filter \
+            if self.eval_config.general.output else None
+        workflow_output = dataset_handler.publish_eval_input(self.eval_input, step_filter)
         with open(workflow_output_file, "w", encoding="utf-8") as f:
             # set indent to 2 for pretty printing
             f.write(workflow_output)

aiq/runtime/session.py

@@ -123,7 +123,11 @@ class AIQSessionManager:
     def set_request_attributes(self, request: Request) -> None:
         """
         Extracts and sets request attributes from an HTTP request.
+        If request is None, no attributes are set.
         """
+        if request is None:
+            return
+
         self._context.metadata._request.method = request.method
         self._context.metadata._request.url_path = request.url.path
         self._context.metadata._request.url_port = request.url.port

{aiqtoolkit-1.1.0a20250505.dist-info → aiqtoolkit-1.1.0a20250507.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit
-Version: 1.1.0a20250505
+Version: 1.1.0a20250507
 Summary: Agent Intelligence Toolkit (AIQ Toolkit)
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -254,6 +254,7 @@ Requires-Dist: aiqtoolkit-weave; extra == "weave"
 Provides-Extra: zep-cloud
 Requires-Dist: aiqtoolkit-zep-cloud; extra == "zep-cloud"
 Provides-Extra: examples
+Requires-Dist: aiq_alert_triage_agent; extra == "examples"
 Requires-Dist: aiq_email_phishing_analyzer; extra == "examples"
 Requires-Dist: aiq_multi_frameworks; extra == "examples"
 Requires-Dist: aiq_plot_charts; extra == "examples"
@@ -263,6 +264,7 @@ Requires-Dist: aiq_simple; extra == "examples"
 Requires-Dist: aiq_swe_bench; extra == "examples"
 Requires-Dist: aiq_automated_description_generation; extra == "examples"
 Requires-Dist: aiq_agno_personal_finance; extra == "examples"
+Requires-Dist: aiq_profiler_agent; extra == "examples"
 Provides-Extra: profiling
 Requires-Dist: matplotlib~=3.9; extra == "profiling"
 Requires-Dist: prefixspan~=0.5.2; extra == "profiling"

{aiqtoolkit-1.1.0a20250505.dist-info → aiqtoolkit-1.1.0a20250507.dist-info}/RECORD

@@ -79,7 +79,7 @@ aiq/data_models/component.py,sha256=x6jm1Fhn1k1hGu-5AjM0ywuyvs6ztaZfapD8bLUXSqc,
 aiq/data_models/component_ref.py,sha256=GyyIf4k80aUIn6LV9r84m5imbiVhpdaY7uKMMpYpbzU,3872
 aiq/data_models/config.py,sha256=ERLjZY0iqexZ-gSXsCSN1UqgNeiwkEjWdYJEdKqeYTY,14116
 aiq/data_models/dataset_handler.py,sha256=SifWhFHtxTMEjrXaXOYQgBOSKfWOzkc6OtOoPJ39pD4,3978
-aiq/data_models/discovery_metadata.py,sha256=nF3SfGO_XER2bIPaSlTfXOCBbrZvWPuf0GVRUcbUXZk,11946
+aiq/data_models/discovery_metadata.py,sha256=OcITQc5VeML4bTHurrsMNiK_oB3z7wudMxcyN7LI8pY,12785
 aiq/data_models/embedder.py,sha256=0v917IiohVA_7zdF7hoO_zQcmNe4hQEFhh4fxRiYBbk,940
 aiq/data_models/evaluate.py,sha256=Llu9_H840nbV_nAimcFQaTeK6oLRmGac9UKsoaLlL58,3786
 aiq/data_models/evaluator.py,sha256=bd2njsyQB2t6ClJ66gJiCjYHsQpWZwPD7rsU0J109TI,939
@@ -106,7 +106,7 @@ aiq/embedder/openai_embedder.py,sha256=5FO3xsyNvEmbLBsZb3xsCpbN1Soxio4yf4b5gTPVx
 aiq/embedder/register.py,sha256=3MTZrfNQKp6AZTbfaA-PpTnyXiMyu-8HH9JnDCC0v9o,978
 aiq/eval/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/eval/config.py,sha256=IlOr2o618kbkXP0G1F-AklZfsKYVos9UB4Dvlxf66xk,1431
-aiq/eval/evaluate.py,sha256=MI7ufASKJGrdsiIVUd4jmcRxwgyZ_8dOY39uyVZKy6Y,14543
+aiq/eval/evaluate.py,sha256=WPGLBeJ46mwIlnprbtia1cm2MwMqZ-GskXoTn6R4oV0,14624
 aiq/eval/intermediate_step_adapter.py,sha256=D645cfnncBEYc-LhYNwr0NwweCr1kTLENt1pLNvmVvU,4211
 aiq/eval/register.py,sha256=QOHJqA2CQixeWMC9InyKbzXo1jByvrntD_m9-2Mvg9k,1076
 aiq/eval/remote_workflow.py,sha256=Fb7Z6gdP2L_gqyWB--AEWfcXe9xPpQ_hPsf9lmqGXjI,5524
@@ -245,7 +245,7 @@ aiq/retriever/nemo_retriever/retriever.py,sha256=IvScUr9XuDLiMR__I3QsboLaM52N5D5
 aiq/runtime/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/runtime/loader.py,sha256=jw_ZPM1vBJkyDtuXeyx42ACpu3EyXeimbo3Fb6k9-gM,6811
 aiq/runtime/runner.py,sha256=WUAiliqI5Se9OgRmimpeFdrl38d9gRTJQ8uf59lvk7U,5836
-aiq/runtime/session.py,sha256=ITKEJlRHzYUEzmqnXjy0fGQJlUE49-kjiEUfTJHAD4Y,4940
+aiq/runtime/session.py,sha256=Hd92_MjYkPNdjbuoxQVV5EOJ0d_X2re449O6XHDlJZI,5039
 aiq/runtime/user_metadata.py,sha256=d7K5CFdOvaXpP2NCUIFBY2lHxB1FnjqV4yOlT4C7fuU,3697
 aiq/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/settings/global_settings.py,sha256=Utm7GEiZTYTBZsxJt2TLp4bNWWMscU4b47MAEJehIrU,12036
@@ -305,10 +305,10 @@ aiq/utils/reactive/base/observer_base.py,sha256=UAlyAY_ky4q2t0P81RVFo2Bs_R7z5Nde
 aiq/utils/reactive/base/subject_base.py,sha256=Ed-AC6P7cT3qkW1EXjzbd5M9WpVoeN_9KCe3OM3FLU4,2521
 aiq/utils/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/utils/settings/global_settings.py,sha256=U9TCLdoZsKq5qOVGjREipGVv9e-FlStzqy5zv82_VYk,7454
-aiqtoolkit-1.1.0a20250505.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
-aiqtoolkit-1.1.0a20250505.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-aiqtoolkit-1.1.0a20250505.dist-info/METADATA,sha256=uuCDEvh_Wh1fTQFUPuGzwMkEuJrdpyKrD8bWydh1D1k,20024
-aiqtoolkit-1.1.0a20250505.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-aiqtoolkit-1.1.0a20250505.dist-info/entry_points.txt,sha256=gRlPfR5g21t328WNEQ4CcEz80S1sJNS8A7rMDYnzl4A,452
-aiqtoolkit-1.1.0a20250505.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit-1.1.0a20250505.dist-info/RECORD,,
+aiqtoolkit-1.1.0a20250507.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
+aiqtoolkit-1.1.0a20250507.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+aiqtoolkit-1.1.0a20250507.dist-info/METADATA,sha256=h-CFJbZB-02tWaQMQWwo-4NrH8B598M1LwOfvQwT_Ys,20138
+aiqtoolkit-1.1.0a20250507.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+aiqtoolkit-1.1.0a20250507.dist-info/entry_points.txt,sha256=gRlPfR5g21t328WNEQ4CcEz80S1sJNS8A7rMDYnzl4A,452
+aiqtoolkit-1.1.0a20250507.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit-1.1.0a20250507.dist-info/RECORD,,