aiqtoolkit-crewai 1.2.0rc4__py3-none-any.whl → 1.2rc9__py3-none-any.whl

{aiqtoolkit_crewai-1.2.0rc4.dist-info → aiqtoolkit_crewai-1.2rc9.dist-info}/METADATA

@@ -1,13 +1,10 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-crewai
-Version: 1.2.0rc4
-Summary: Subpackage for CrewAI integration in AIQtoolkit
-Keywords: ai,rag,agents
+Version: 1.2rc9
+Summary: Transitional package for nvidia-nat-crewai, this package is deprecated and will be removed in the future.
 Classifier: Programming Language :: Python
-Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: aiqtoolkit==v1.2.0-rc4
-Requires-Dist: crewai~=0.95.0
+Requires-Dist: nvidia-nat[crewai]==v1.2-rc9
 
 <!--
 SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
@@ -26,9 +23,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/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
+# Transitional Package for `nvidia-nat-crewai`
+This is a transitional package for `nvidia-nat-crewai` to help ease the migration to `nvidia-nat-crewai`, and will be removed in a future release. It is recommended to use `nvidia-nat-crewai` directly for new projects.
 
-# NVIDIA Agent Intelligence Toolkit Subpackage
-This is a subpackage for CrewAI integration in AIQ toolkit.
-
-For more information about AIQ toolkit, please visit the [AIQ toolkit package](https://pypi.org/project/aiqtoolkit/).
+For more information about the NVIDIA NeMo Agent toolkit, please visit the [NeMo Agent toolkit package](https://pypi.org/project/nvidia-nat-crewai/).

aiqtoolkit_crewai-1.2rc9.dist-info/RECORD

@@ -0,0 +1,4 @@
+aiqtoolkit_crewai-1.2rc9.dist-info/METADATA,sha256=kZngIgFy4B8deHOEUn6zwxC9Mi4XHcYHwgl8HhushqA,1387
+aiqtoolkit_crewai-1.2rc9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_crewai-1.2rc9.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+aiqtoolkit_crewai-1.2rc9.dist-info/RECORD,,

aiqtoolkit_crewai-1.2rc9.dist-info/top_level.txt

@@ -0,0 +1 @@
+

aiq/meta/pypi.md

@@ -1,23 +0,0 @@
-<!--
-SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-SPDX-License-Identifier: Apache-2.0
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
--->
-
-![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 Subpackage
-This is a subpackage for CrewAI integration in AIQ toolkit.
-
-For more information about AIQ toolkit, please visit the [AIQ toolkit package](https://pypi.org/project/aiqtoolkit/).

aiq/plugins/crewai/crewai_callback_handler.py

@@ -1,205 +0,0 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import copy
-import logging
-import threading
-import time
-from collections.abc import Callable
-from typing import Any
-
-import litellm
-from crewai.tools import tool_usage
-
-from aiq.builder.context import AIQContext
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.data_models.intermediate_step import IntermediateStepPayload
-from aiq.data_models.intermediate_step import IntermediateStepType
-from aiq.data_models.intermediate_step import StreamEventData
-from aiq.data_models.intermediate_step import TraceMetadata
-from aiq.data_models.intermediate_step import UsageInfo
-from aiq.profiler.callbacks.base_callback_class import BaseProfilerCallback
-from aiq.profiler.callbacks.token_usage_base_model import TokenUsageBaseModel
-
-logger = logging.getLogger(__name__)
-
-
-class CrewAIProfilerHandler(BaseProfilerCallback):
-    """
-    A callback manager/handler for CrewAI that intercepts calls to:
-      - ToolUsage._use
-      - LLM Calls
-    to collect usage statistics (tokens, inputs, outputs, time intervals, etc.)
-    and store them in AIQ Toolkit's usage_stats queue for subsequent analysis.
-    """
-
-    def __init__(self) -> None:
-        super().__init__()
-        self._lock = threading.Lock()
-        self.last_call_ts = time.time()
-        self.step_manager = AIQContext.get().intermediate_step_manager
-
-        # Original references to CrewAI methods (for uninstrumenting if needed)
-        self._original_tool_use = None
-        self._original_llm_call = None
-
-    def instrument(self) -> None:
-        """
-        Monkey-patch the relevant CrewAI methods with usage-stat collection logic.
-        Assumes the 'crewai' library is installed.
-        """
-
-        # Save the originals
-        self._original_tool_use = getattr(tool_usage.ToolUsage, "_use", None)
-        self._original_llm_call = getattr(litellm, "completion", None)
-
-        # Patch if available
-        if self._original_tool_use:
-            tool_usage.ToolUsage._use = self._tool_use_monkey_patch()
-
-        if self._original_llm_call:
-            litellm.completion = self._llm_call_monkey_patch()
-
-        logger.debug("CrewAIProfilerHandler instrumentation applied successfully.")
-
-    def _tool_use_monkey_patch(self) -> Callable[..., Any]:
-        """
-        Returns a function that wraps calls to ToolUsage._use(...) with usage-logging.
-        """
-        original_func = self._original_tool_use
-
-        def wrapped_tool_use(tool_usage_instance, *args, **kwargs) -> Any:
-            """
-            Replicates _tool_use_wrapper logic without wrapt: collects usage stats,
-            calls the original, and captures output stats.
-            """
-            now = time.time()
-            tool_name = ""
-
-            try:
-                tool_info = kwargs.get("tool", "")
-
-                if tool_info:
-                    tool_name = tool_info.name
-            except Exception as e:
-                logger.exception("Error getting tool name: %s", e, exc_info=True)
-
-            try:
-                # Pre-call usage event
-                stats = IntermediateStepPayload(event_type=IntermediateStepType.TOOL_START,
-                                                framework=LLMFrameworkEnum.CREWAI,
-                                                name=tool_name,
-                                                data=StreamEventData(),
-                                                metadata=TraceMetadata(tool_inputs={
-                                                    "args": args, "kwargs": dict(kwargs)
-                                                }),
-                                                usage_info=UsageInfo(token_usage=TokenUsageBaseModel()))
-
-                self.step_manager.push_intermediate_step(stats)
-
-                self.last_call_ts = now
-
-                # Call the original _use(...)
-                result = original_func(tool_usage_instance, *args, **kwargs)
-                now = time.time()
-                # Post-call usage stats
-                usage_stat = IntermediateStepPayload(
-                    event_type=IntermediateStepType.TOOL_END,
-                    span_event_timestamp=now,
-                    framework=LLMFrameworkEnum.CREWAI,
-                    name=tool_name,
-                    data=StreamEventData(input={
-                        "args": args, "kwargs": dict(kwargs)
-                    }, output=str(result)),
-                    metadata=TraceMetadata(tool_outputs={"result": str(result)}),
-                    usage_info=UsageInfo(token_usage=TokenUsageBaseModel()),
-                )
-
-                self.step_manager.push_intermediate_step(usage_stat)
-
-                return result
-
-            except Exception as e:
-                logger.exception("ToolUsage._use error: %s", e)
-                raise
-
-        return wrapped_tool_use
-
-    def _llm_call_monkey_patch(self) -> Callable[..., Any]:
-        """
-        Returns a function that wraps calls to litellm.completion(...) with usage-logging.
-        """
-        original_func = self._original_llm_call
-
-        def wrapped_llm_call(*args, **kwargs) -> Any:
-            """
-            Replicates _llm_call_wrapper logic without wrapt: collects usage stats,
-            calls the original, and captures output stats.
-            """
-
-            now = time.time()
-            seconds_between_calls = int(now - self.last_call_ts)
-            model_name = kwargs.get('model', "")
-
-            model_input = ""
-            try:
-                for message in kwargs.get('messages', []):
-                    model_input += message.get('content', "")
-            except Exception as e:
-                logger.exception("Error getting model input: %s", e, exc_info=True)
-
-            # Record the start event
-            input_stats = IntermediateStepPayload(
-                event_type=IntermediateStepType.LLM_START,
-                framework=LLMFrameworkEnum.CREWAI,
-                name=model_name,
-                data=StreamEventData(input=model_input),
-                metadata=TraceMetadata(chat_inputs=copy.deepcopy(kwargs.get('messages', []))),
-                usage_info=UsageInfo(token_usage=TokenUsageBaseModel(),
-                                     num_llm_calls=1,
-                                     seconds_between_calls=seconds_between_calls))
-
-            self.step_manager.push_intermediate_step(input_stats)
-
-            # Call the original litellm.completion(...)
-            output = original_func(*args, **kwargs)
-
-            model_output = ""
-            try:
-                for choice in output.choices:
-                    msg = choice.model_extra["message"]
-                    model_output += msg.get('content', "")
-            except Exception as e:
-                logger.exception("Error getting model output: %s", e, exc_info=True)
-
-            now = time.time()
-            # Record the end event
-            output_stats = IntermediateStepPayload(
-                event_type=IntermediateStepType.LLM_END,
-                span_event_timestamp=now,
-                framework=LLMFrameworkEnum.CREWAI,
-                name=model_name,
-                data=StreamEventData(input=model_input, output=model_output),
-                metadata=TraceMetadata(chat_responses=output.choices[0].model_dump()),
-                usage_info=UsageInfo(token_usage=TokenUsageBaseModel(**output.model_extra['usage'].model_dump()),
-                                     num_llm_calls=1,
-                                     seconds_between_calls=seconds_between_calls))
-
-            self.step_manager.push_intermediate_step(output_stats)
-
-            # (Note: the original code did NOT update self.last_call_ts here)
-            return output
-
-        return wrapped_llm_call

aiq/plugins/crewai/llm.py

@@ -1,80 +0,0 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import os
-
-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.nim_llm import NIMModelConfig
-from aiq.llm.openai_llm import OpenAIModelConfig
-from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
-
-
-@register_llm_client(config_type=NIMModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
-async def nim_crewai(llm_config: NIMModelConfig, builder: Builder):
-
-    from crewai import LLM
-
-    config_obj = {
-        **llm_config.model_dump(exclude={"type"}, by_alias=True),
-        "model": f"nvidia_nim/{llm_config.model_name}",
-    }
-
-    # Because CrewAI uses a different environment variable for the API key, we need to set it here manually
-    if ("api_key" not in config_obj or config_obj["api_key"] is None):
-
-        if ("NVIDIA_NIM_API_KEY" in os.environ):
-            # Dont need to do anything. User has already set the correct key
-            pass
-        else:
-            nvidai_api_key = os.getenv("NVIDIA_API_KEY")
-
-            if (nvidai_api_key is not None):
-                # Transfer the key to the correct environment variable for LiteLLM
-                os.environ["NVIDIA_NIM_API_KEY"] = nvidai_api_key
-
-    client = LLM(**config_obj)
-
-    if isinstance(llm_config, RetryMixin):
-
-        client = patch_with_retry(client,
-                                  retries=llm_config.num_retries,
-                                  retry_codes=llm_config.retry_on_status_codes,
-                                  retry_on_messages=llm_config.retry_on_errors)
-
-    yield client
-
-
-@register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
-async def openai_crewai(llm_config: OpenAIModelConfig, builder: Builder):
-
-    from crewai import LLM
-
-    config_obj = {
-        **llm_config.model_dump(exclude={"type"}, by_alias=True),
-    }
-
-    client = LLM(**config_obj)
-
-    if isinstance(llm_config, RetryMixin):
-
-        client = patch_with_retry(client,
-                                  retries=llm_config.num_retries,
-                                  retry_codes=llm_config.retry_on_status_codes,
-                                  retry_on_messages=llm_config.retry_on_errors)
-
-    yield client

aiq/plugins/crewai/register.py

@@ -1,23 +0,0 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-# pylint: disable=unused-import
-# flake8: noqa
-# isort:skip_file
-
-# Import any providers which need to be automatically registered here
-
-from . import llm
-from . import tool_wrapper

aiq/plugins/crewai/tool_wrapper.py

@@ -1,40 +0,0 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import asyncio
-
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.builder.function import Function
-from aiq.cli.register_workflow import register_tool_wrapper
-
-
-@register_tool_wrapper(wrapper_type=LLMFrameworkEnum.CREWAI)
-def crewai_tool_wrapper(name: str, fn: Function, builder: Builder):
-
-    from crewai.tools.base_tool import Tool
-
-    # Capture the loop at the time this is called
-    loop = asyncio.get_event_loop()
-
-    # Capture the coroutine at the time this is called
-    runnable = fn.acall_invoke
-
-    # Because CrewAI tools are not async, we need to wrap the coroutine in a normal function
-    def wrapper(*args, **kwargs):
-
-        return asyncio.run_coroutine_threadsafe(runnable(*args, **kwargs), loop).result()
-
-    return Tool(name=name, description=fn.description or "", args_schema=fn.input_schema, func=wrapper)

aiqtoolkit_crewai-1.2.0rc4.dist-info/RECORD

@@ -1,11 +0,0 @@
-aiq/meta/pypi.md,sha256=mAJKaBmH4UyD5wCe3_BfPGv3ihPTg8BNfS3zu2Z4-70,1083
-aiq/plugins/crewai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/crewai/crewai_callback_handler.py,sha256=mFNWmtDsOUVsa6LKw7BBjBl7lsytAEaYKZfPgl9yY1o,8396
-aiq/plugins/crewai/llm.py,sha256=FsNt5TFYNpPU_FxB0dW0Cq_ZkviwJOmvivSYcVPtPvg,3000
-aiq/plugins/crewai/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
-aiq/plugins/crewai/tool_wrapper.py,sha256=LuL5VHHxOXf-CBd5E6kA6OsUz2N_aGin_r5KF59srL0,1569
-aiqtoolkit_crewai-1.2.0rc4.dist-info/METADATA,sha256=WBYEzReKrF9Q86t1negNJW0Fd6EZNLva9jOd0Co-uJo,1410
-aiqtoolkit_crewai-1.2.0rc4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit_crewai-1.2.0rc4.dist-info/entry_points.txt,sha256=gY-k52LFDNWYEcSYiFs78LRKWeGijnGCkwcOroG4T3E,58
-aiqtoolkit_crewai-1.2.0rc4.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_crewai-1.2.0rc4.dist-info/RECORD,,

aiqtoolkit_crewai-1.2.0rc4.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[aiq.components]
-aiq_crewai = aiq.plugins.crewai.register

aiqtoolkit_crewai-1.2.0rc4.dist-info/top_level.txt

@@ -1 +0,0 @@
-aiq