nvidia-nat-crewai 1.2.0rc5__py3-none-any.whl → 1.2.0rc7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
15
15
  limitations under the License.
16
16
  -->
17
17
 
18
- ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
18
+ ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
19
19
 
20
20
  # NVIDIA NeMo Agent Toolkit Subpackage
21
21
  This is a subpackage for CrewAI integration in NeMo Agent toolkit.
@@ -23,15 +23,15 @@ from typing import Any
23
23
  import litellm
24
24
  from crewai.tools import tool_usage
25
25
 
26
- from aiq.builder.context import AIQContext
27
- from aiq.builder.framework_enum import LLMFrameworkEnum
28
- from aiq.data_models.intermediate_step import IntermediateStepPayload
29
- from aiq.data_models.intermediate_step import IntermediateStepType
30
- from aiq.data_models.intermediate_step import StreamEventData
31
- from aiq.data_models.intermediate_step import TraceMetadata
32
- from aiq.data_models.intermediate_step import UsageInfo
33
- from aiq.profiler.callbacks.base_callback_class import BaseProfilerCallback
34
- from aiq.profiler.callbacks.token_usage_base_model import TokenUsageBaseModel
26
+ from nat.builder.context import Context
27
+ from nat.builder.framework_enum import LLMFrameworkEnum
28
+ from nat.data_models.intermediate_step import IntermediateStepPayload
29
+ from nat.data_models.intermediate_step import IntermediateStepType
30
+ from nat.data_models.intermediate_step import StreamEventData
31
+ from nat.data_models.intermediate_step import TraceMetadata
32
+ from nat.data_models.intermediate_step import UsageInfo
33
+ from nat.profiler.callbacks.base_callback_class import BaseProfilerCallback
34
+ from nat.profiler.callbacks.token_usage_base_model import TokenUsageBaseModel
35
35
 
36
36
  logger = logging.getLogger(__name__)
37
37
 
@@ -42,14 +42,14 @@ class CrewAIProfilerHandler(BaseProfilerCallback):
42
42
  - ToolUsage._use
43
43
  - LLM Calls
44
44
  to collect usage statistics (tokens, inputs, outputs, time intervals, etc.)
45
- and store them in AIQ Toolkit's usage_stats queue for subsequent analysis.
45
+ and store them in NAT's usage_stats queue for subsequent analysis.
46
46
  """
47
47
 
48
48
  def __init__(self) -> None:
49
49
  super().__init__()
50
50
  self._lock = threading.Lock()
51
51
  self.last_call_ts = time.time()
52
- self.step_manager = AIQContext.get().intermediate_step_manager
52
+ self.step_manager = Context.get().intermediate_step_manager
53
53
 
54
54
  # Original references to CrewAI methods (for uninstrumenting if needed)
55
55
  self._original_tool_use = None
@@ -15,13 +15,13 @@
15
15
 
16
16
  import os
17
17
 
18
- from aiq.builder.builder import Builder
19
- from aiq.builder.framework_enum import LLMFrameworkEnum
20
- from aiq.cli.register_workflow import register_llm_client
21
- from aiq.data_models.retry_mixin import RetryMixin
22
- from aiq.llm.nim_llm import NIMModelConfig
23
- from aiq.llm.openai_llm import OpenAIModelConfig
24
- from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
18
+ from nat.builder.builder import Builder
19
+ from nat.builder.framework_enum import LLMFrameworkEnum
20
+ from nat.cli.register_workflow import register_llm_client
21
+ from nat.data_models.retry_mixin import RetryMixin
22
+ from nat.llm.nim_llm import NIMModelConfig
23
+ from nat.llm.openai_llm import OpenAIModelConfig
24
+ from nat.utils.exception_handlers.automatic_retries import patch_with_retry
25
25
 
26
26
 
27
27
  @register_llm_client(config_type=NIMModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
@@ -15,10 +15,10 @@
15
15
 
16
16
  import asyncio
17
17
 
18
- from aiq.builder.builder import Builder
19
- from aiq.builder.framework_enum import LLMFrameworkEnum
20
- from aiq.builder.function import Function
21
- from aiq.cli.register_workflow import register_tool_wrapper
18
+ from nat.builder.builder import Builder
19
+ from nat.builder.framework_enum import LLMFrameworkEnum
20
+ from nat.builder.function import Function
21
+ from nat.cli.register_workflow import register_tool_wrapper
22
22
 
23
23
 
24
24
  @register_tool_wrapper(wrapper_type=LLMFrameworkEnum.CREWAI)
@@ -1,12 +1,12 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat-crewai
3
- Version: 1.2.0rc5
4
- Summary: Subpackage for CrewAI integration in AIQtoolkit
3
+ Version: 1.2.0rc7
4
+ Summary: Subpackage for CrewAI integration in NeMo Agent toolkit
5
5
  Keywords: ai,rag,agents
6
6
  Classifier: Programming Language :: Python
7
7
  Requires-Python: <3.13,>=3.11
8
8
  Description-Content-Type: text/markdown
9
- Requires-Dist: nvidia-nat~=1.2
9
+ Requires-Dist: nvidia-nat==v1.2.0-rc7
10
10
  Requires-Dist: crewai~=0.95.0
11
11
 
12
12
  <!--
@@ -26,7 +26,7 @@ See the License for the specific language governing permissions and
26
26
  limitations under the License.
27
27
  -->
28
28
 
29
- ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
29
+ ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
30
30
 
31
31
  # NVIDIA NeMo Agent Toolkit Subpackage
32
32
  This is a subpackage for CrewAI integration in NeMo Agent toolkit.
@@ -0,0 +1,11 @@
1
+ nat/meta/pypi.md,sha256=T68FnThRzDGFf1LR8u-okM-r11-skSnKqSyI6HOktQY,1107
2
+ nat/plugins/crewai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
+ nat/plugins/crewai/crewai_callback_handler.py,sha256=LDOctDQC9qdba1SVGoVkceCOSYuDj_mnl3HCuq2nIuQ,8382
4
+ nat/plugins/crewai/llm.py,sha256=_ykUOFJCc9czWPGQZIfQerfqH_cGZfjZ09U297I4xuo,3000
5
+ nat/plugins/crewai/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
6
+ nat/plugins/crewai/tool_wrapper.py,sha256=BNKEPQQCLKtXNzGDAKBLCdmGJXe9lBOVI1hObha8hoI,1569
7
+ nvidia_nat_crewai-1.2.0rc7.dist-info/METADATA,sha256=aCerBDyxQ_vr9jlQf_ruHFVOpDow-RC8FZTcDWa-o1A,1442
8
+ nvidia_nat_crewai-1.2.0rc7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
+ nvidia_nat_crewai-1.2.0rc7.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
10
+ nvidia_nat_crewai-1.2.0rc7.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
11
+ nvidia_nat_crewai-1.2.0rc7.dist-info/RECORD,,
@@ -0,0 +1,2 @@
1
+ [nat.components]
2
+ nat_crewai = nat.plugins.crewai.register
@@ -1,11 +0,0 @@
1
- aiq/meta/pypi.md,sha256=jzCckU9TWDD1lXqll_BxfrHjhtyCxxjaF88aeQ5SG_I,1118
2
- aiq/plugins/crewai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- aiq/plugins/crewai/crewai_callback_handler.py,sha256=mFNWmtDsOUVsa6LKw7BBjBl7lsytAEaYKZfPgl9yY1o,8396
4
- aiq/plugins/crewai/llm.py,sha256=FsNt5TFYNpPU_FxB0dW0Cq_ZkviwJOmvivSYcVPtPvg,3000
5
- aiq/plugins/crewai/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
6
- aiq/plugins/crewai/tool_wrapper.py,sha256=LuL5VHHxOXf-CBd5E6kA6OsUz2N_aGin_r5KF59srL0,1569
7
- nvidia_nat_crewai-1.2.0rc5.dist-info/METADATA,sha256=5cL-H8xIqu9Os4Mi32kYoTaxwdjOftos32P6p1tPba8,1438
8
- nvidia_nat_crewai-1.2.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
- nvidia_nat_crewai-1.2.0rc5.dist-info/entry_points.txt,sha256=gY-k52LFDNWYEcSYiFs78LRKWeGijnGCkwcOroG4T3E,58
10
- nvidia_nat_crewai-1.2.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
11
- nvidia_nat_crewai-1.2.0rc5.dist-info/RECORD,,
@@ -1,2 +0,0 @@
1
- [aiq.components]
2
- aiq_crewai = aiq.plugins.crewai.register
@@ -1 +0,0 @@
1
- aiq
File without changes
File without changes