aiqtoolkit 1.2.0.dev0__py3-none-any.whl → 1.2.0rc2__py3-none-any.whl

aiq/utils/type_converter.py

@@ -70,7 +70,7 @@ class TypeConverter:
         self._converters.setdefault(to_type, OrderedDict())[from_type] = converter
         # to do(MDD): If needed, sort by specificity here.
 
-    def try_convert(self, data, to_type: type[_T]) -> _T | None:
+    def _convert(self, data, to_type: type[_T]) -> _T | None:
         """
         Attempts to convert `data` into `to_type`. Returns None if no path is found.
         """
@@ -100,7 +100,7 @@ class TypeConverter:
         Converts or raises ValueError if no path is found.
         We also give the parent a chance if self fails.
         """
-        result = self.try_convert(data, to_type)
+        result = self._convert(data, to_type)
         if result is None and self._parent:
             # fallback on parent entirely
             return self._parent.convert(data, to_type)
@@ -109,6 +109,18 @@ class TypeConverter:
             return result
         raise ValueError(f"Cannot convert type {type(data)} to {to_type}. No match found.")
 
+    def try_convert(self, data, to_type: type[_T]) -> _T:
+        """
+        Converts with graceful error handling. If conversion fails, returns the original data
+        and continues processing.
+        """
+        try:
+            return self.convert(data, to_type)
+        except ValueError:
+            logger.warning("Type conversion failed, using original value. From %s to %s", type(data), to_type)
+            # Return original data, let downstream code handle it
+            return data
+
     # -------------------------------------------------
     # INTERNAL DIRECT CONVERSION (with parent fallback)
     # -------------------------------------------------
@@ -221,6 +233,10 @@ class GlobalTypeConverter:
     def convert(data, to_type: type[_T]) -> _T:
         return GlobalTypeConverter._global_converter.convert(data, to_type)
 
+    @staticmethod
+    def try_convert(data, to_type: type[_T]) -> _T:
+        return GlobalTypeConverter._global_converter.try_convert(data, to_type)
+
 
 TypeConverter._global_initialized = True
 

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.0.dev0.dist-info → aiqtoolkit-1.2.0rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit
-Version: 1.2.0.dev0
+Version: 1.2.0rc2
 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,44 +234,73 @@ 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
 Requires-Dist: aiqtoolkit-agno; extra == "agno"
 Provides-Extra: crewai
 Requires-Dist: aiqtoolkit-crewai; extra == "crewai"
+Provides-Extra: ingestion
+Requires-Dist: lxml~=5.4; extra == "ingestion"
 Provides-Extra: langchain
 Requires-Dist: aiqtoolkit-langchain; extra == "langchain"
 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"
 Requires-Dist: scikit-learn~=1.6; extra == "profiling"
+Provides-Extra: gunicorn
+Requires-Dist: gunicorn~=23.0; extra == "gunicorn"
 Dynamic: license-file
 
 <!--
@@ -288,7 +320,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/AIQToolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
+![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
 
 # NVIDIA Agent Intelligence Toolkit
 
@@ -296,26 +328,26 @@ AIQ toolkit is a flexible library designed to seamlessly integrate your enterpri
 
 ## Key Features
 
-- [**Framework Agnostic:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/extend/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
-- [**Reusability:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/extend/sharing-components.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
-- [**Rapid Development:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/tutorials/index.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
-- [**Profiling:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/workflows/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
-- [**Observability:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/workflows/observe/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool, with examples using [Phoenix](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/workflows/observe/observe-workflow-with-phoenix.html) and [W&B Weave](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/workflows/observe/observe-workflow-with-weave.html).
-- [**Evaluation System:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/workflows/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
-- [**User Interface:**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/quick-start/launching-ui.html) Use the AIQ toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
-- [**MCP Compatibility**](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as AIQ toolkit functions.
+- [**Framework Agnostic:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/extend/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
+- [**Reusability:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/extend/sharing-components.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
+- [**Rapid Development:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/tutorials/index.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
+- [**Profiling:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
+- [**Observability:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool, with examples using [Phoenix](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) and [W&B Weave](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/observe/observe-workflow-with-weave.html).
+- [**Evaluation System:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
+- [**User Interface:**](https://docs.nvidia.com/aiqtoolkit/1.2.0/quick-start/launching-ui.html) Use the AIQ toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
+- [**MCP Compatibility**](https://docs.nvidia.com/aiqtoolkit/1.2.0/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as AIQ toolkit functions.
 
 With AIQ toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
 
 ## Links
- * [Documentation](https://docs.nvidia.com/aiqtoolkit/v1.2.0-dev/index.html): Explore the full documentation for AIQ toolkit.
+ * [Documentation](https://docs.nvidia.com/aiqtoolkit/1.2.0/index.html): Explore the full documentation for AIQ toolkit.
 
 ## First time user?
- If this is your first time using AIQ toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/AIQToolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with AIQ toolkit applications and need to add AIQ toolkit as a dependency to their project.
+ If this is your first time using AIQ toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/NeMo-Agent-Toolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with AIQ toolkit applications and need to add AIQ toolkit as a dependency to their project.
 
 ## Feedback
 
-We would love to hear from you! Please file an issue on [GitHub](https://github.com/NVIDIA/AIQToolkit/issues) if you have any feedback or feature requests.
+We would love to hear from you! Please file an issue on [GitHub](https://github.com/NVIDIA/NeMo-Agent-Toolkit/issues) if you have any feedback or feature requests.
 
 ## Acknowledgements