aiqtoolkit 1.2.0a20250706__py3-none-any.whl → 1.2.0a20250730__py3-none-any.whl

aiq/utils/type_utils.py

@@ -395,3 +395,90 @@ class DecomposedType:
         converters.append(_convert_to_schema)
 
         return schema
+
+    @staticmethod
+    def extract_generic_parameters_from_class(target_class: type,
+                                              expected_param_count: int | None = None) -> tuple[type, ...]:
+        """
+        Extract generic type parameters from a class's inheritance chain.
+
+        This method searches through __orig_bases__ to find generic parameters,
+        which is useful for classes that inherit from generic base classes.
+
+        Parameters
+        ----------
+        target_class : type
+            The class to extract parameters from
+        expected_param_count : int | None, optional
+            Expected number of parameters. If specified, only matches with this count are considered.
+
+        Returns
+        -------
+        tuple[type, ...]
+            Tuple of generic type parameters found
+
+        Raises
+        ------
+        ValueError
+            If no generic parameters matching the expected count are found
+
+        Examples
+        --------
+        >>> class MyClass(SomeGeneric[int, str, bool]):
+        ...     pass
+        >>> DecomposedType.extract_generic_parameters_from_class(MyClass, 3)
+        (int, str, bool)
+        """
+        for base_cls in getattr(target_class, '__orig_bases__', []):
+            base_cls_args = typing.get_args(base_cls)
+
+            if expected_param_count is None or len(base_cls_args) == expected_param_count:
+                if base_cls_args:  # Only return if we actually found parameters
+                    return base_cls_args
+
+        if expected_param_count is not None:
+            raise ValueError(
+                f"Could not find generic parameters with count {expected_param_count} for class {target_class}")
+        raise ValueError(f"Could not find any generic parameters for class {target_class}")
+
+    @staticmethod
+    def is_type_compatible(source_type: type, target_type: type) -> bool:
+        """
+        Check if a source type is compatible with a target type.
+
+        This handles direct compatibility and special cases like batch compatibility
+        where list[T] can be compatible with targets that expect T.
+
+        Parameters
+        ----------
+        source_type : type
+            The source type to check
+        target_type : type
+            The target type to check compatibility with
+
+        Returns
+        -------
+        bool
+            True if types are compatible, False otherwise
+        """
+        # Direct compatibility check
+        try:
+            if issubclass(source_type, target_type):
+                return True
+        except TypeError:
+            # Handle generic types that can't use issubclass
+            pass
+
+        # Check if source outputs list[T] and target expects T
+        source_decomposed = DecomposedType(source_type)
+        if source_decomposed.origin is list and source_decomposed.args:
+            inner_type = source_decomposed.args[0]
+            try:
+                if issubclass(inner_type, target_type):
+                    return True
+            except TypeError:
+                # If we can't use issubclass, check type equality
+                if inner_type == target_type:
+                    return True
+
+        return False

{aiqtoolkit-1.2.0a20250706.dist-info → aiqtoolkit-1.2.0a20250730.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit
-Version: 1.2.0a20250706
+Version: 1.2.0a20250730
 Summary: NVIDIA Agent Intelligence toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -212,18 +212,21 @@ Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
 Requires-Dist: aioboto3>=11.0.0
+Requires-Dist: authlib~=1.3.1
 Requires-Dist: click~=8.1
 Requires-Dist: colorama~=0.4.6
+Requires-Dist: datasets~=4.0
 Requires-Dist: expandvars~=1.0
 Requires-Dist: fastapi~=0.115.5
 Requires-Dist: httpx~=0.27
 Requires-Dist: jinja2~=3.1
 Requires-Dist: jsonpath-ng~=1.7
-Requires-Dist: mcp>=1.0.0
+Requires-Dist: mcp~=1.10
 Requires-Dist: networkx~=3.4
 Requires-Dist: numpy~=1.26
 Requires-Dist: openinference-semantic-conventions~=0.1.14
 Requires-Dist: openpyxl~=3.1
+Requires-Dist: pkce==1.0.3
 Requires-Dist: pkginfo~=1.12
 Requires-Dist: platformdirs~=4.3
 Requires-Dist: pydantic==2.10.*
@@ -231,6 +234,7 @@ Requires-Dist: pymilvus~=2.4
 Requires-Dist: PyYAML~=6.0
 Requires-Dist: ragas~=0.2.14
 Requires-Dist: rich~=13.9
+Requires-Dist: tabulate~=0.9
 Requires-Dist: uvicorn[standard]~=0.32.0
 Requires-Dist: wikipedia~=1.4
 Provides-Extra: agno
@@ -245,28 +249,52 @@ Provides-Extra: llama-index
 Requires-Dist: aiqtoolkit-llama-index; extra == "llama-index"
 Provides-Extra: mem0ai
 Requires-Dist: aiqtoolkit-mem0ai; extra == "mem0ai"
+Provides-Extra: opentelemetry
+Requires-Dist: aiqtoolkit-opentelemetry; extra == "opentelemetry"
+Provides-Extra: phoenix
+Requires-Dist: aiqtoolkit-phoenix; extra == "phoenix"
+Provides-Extra: ragaai
+Requires-Dist: aiqtoolkit-ragaai; extra == "ragaai"
+Provides-Extra: mysql
+Requires-Dist: aiqtoolkit-mysql; extra == "mysql"
+Provides-Extra: redis
+Requires-Dist: aiqtoolkit-redis; extra == "redis"
+Provides-Extra: s3
+Requires-Dist: aiqtoolkit-s3; extra == "s3"
 Provides-Extra: semantic-kernel
 Requires-Dist: aiqtoolkit-semantic-kernel; extra == "semantic-kernel"
 Provides-Extra: telemetry
-Requires-Dist: arize-phoenix~=6.1; extra == "telemetry"
-Requires-Dist: opentelemetry-api~=1.2; extra == "telemetry"
-Requires-Dist: opentelemetry-sdk~=1.3; extra == "telemetry"
+Requires-Dist: aiqtoolkit-opentelemetry; extra == "telemetry"
+Requires-Dist: aiqtoolkit-phoenix; extra == "telemetry"
+Requires-Dist: aiqtoolkit-weave; extra == "telemetry"
+Requires-Dist: aiqtoolkit-ragaai; extra == "telemetry"
 Provides-Extra: weave
 Requires-Dist: aiqtoolkit-weave; extra == "weave"
 Provides-Extra: zep-cloud
 Requires-Dist: aiqtoolkit-zep-cloud; extra == "zep-cloud"
 Provides-Extra: examples
+Requires-Dist: aiq_agno_personal_finance; extra == "examples"
 Requires-Dist: aiq_alert_triage_agent; extra == "examples"
+Requires-Dist: aiq_automated_description_generation; extra == "examples"
 Requires-Dist: aiq_email_phishing_analyzer; extra == "examples"
 Requires-Dist: aiq_multi_frameworks; extra == "examples"
 Requires-Dist: aiq_plot_charts; extra == "examples"
+Requires-Dist: aiq_por_to_jiratickets; extra == "examples"
+Requires-Dist: aiq_profiler_agent; extra == "examples"
+Requires-Dist: aiq_redact_pii; extra == "examples"
 Requires-Dist: aiq_semantic_kernel_demo; extra == "examples"
+Requires-Dist: aiq_simple_auth; extra == "examples"
+Requires-Dist: aiq_simple_web_query; extra == "examples"
+Requires-Dist: aiq_simple_web_query_eval; extra == "examples"
 Requires-Dist: aiq_simple_calculator; extra == "examples"
-Requires-Dist: aiq_simple; extra == "examples"
+Requires-Dist: aiq_simple_calculator_custom_routes; extra == "examples"
+Requires-Dist: aiq_simple_calculator_eval; extra == "examples"
+Requires-Dist: aiq_simple_calculator_mcp; extra == "examples"
+Requires-Dist: aiq_simple_calculator_observability; extra == "examples"
+Requires-Dist: aiq_simple_calculator_hitl; extra == "examples"
+Requires-Dist: aiq_simple_rag; 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"
+Requires-Dist: aiq_user_report; extra == "examples"
 Provides-Extra: profiling
 Requires-Dist: matplotlib~=3.9; extra == "profiling"
 Requires-Dist: prefixspan~=0.5.2; extra == "profiling"