nvidia-nat-crewai 1.2.0rc5__py3-none-any.whl → 1.2.0rc6__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.
- {aiq → nat}/meta/pypi.md +1 -1
- {aiq → nat}/plugins/crewai/crewai_callback_handler.py +11 -11
- {aiq → nat}/plugins/crewai/llm.py +7 -7
- {aiq → nat}/plugins/crewai/tool_wrapper.py +4 -4
- {nvidia_nat_crewai-1.2.0rc5.dist-info → nvidia_nat_crewai-1.2.0rc6.dist-info}/METADATA +4 -4
- nvidia_nat_crewai-1.2.0rc6.dist-info/RECORD +11 -0
- nvidia_nat_crewai-1.2.0rc6.dist-info/entry_points.txt +2 -0
- nvidia_nat_crewai-1.2.0rc6.dist-info/top_level.txt +1 -0
- nvidia_nat_crewai-1.2.0rc5.dist-info/RECORD +0 -11
- nvidia_nat_crewai-1.2.0rc5.dist-info/entry_points.txt +0 -2
- nvidia_nat_crewai-1.2.0rc5.dist-info/top_level.txt +0 -1
- {aiq → nat}/plugins/crewai/__init__.py +0 -0
- {aiq → nat}/plugins/crewai/register.py +0 -0
- {nvidia_nat_crewai-1.2.0rc5.dist-info → nvidia_nat_crewai-1.2.0rc6.dist-info}/WHEEL +0 -0
{aiq → nat}/meta/pypi.md
RENAMED
@@ -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
|
-

|
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
|
27
|
-
from
|
28
|
-
from
|
29
|
-
from
|
30
|
-
from
|
31
|
-
from
|
32
|
-
from
|
33
|
-
from
|
34
|
-
from
|
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
|
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 =
|
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
|
19
|
-
from
|
20
|
-
from
|
21
|
-
from
|
22
|
-
from
|
23
|
-
from
|
24
|
-
from
|
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
|
19
|
-
from
|
20
|
-
from
|
21
|
-
from
|
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.
|
4
|
-
Summary: Subpackage for CrewAI integration in
|
3
|
+
Version: 1.2.0rc6
|
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
|
9
|
+
Requires-Dist: nvidia-nat==v1.2.0-rc6
|
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
|
-

|
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.0rc6.dist-info/METADATA,sha256=0cJxv5QXjMEeGXQsShLx55FHK9IEGw_eI3pAFiDyD4I,1442
|
8
|
+
nvidia_nat_crewai-1.2.0rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
9
|
+
nvidia_nat_crewai-1.2.0rc6.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
|
10
|
+
nvidia_nat_crewai-1.2.0rc6.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
|
11
|
+
nvidia_nat_crewai-1.2.0rc6.dist-info/RECORD,,
|
@@ -0,0 +1 @@
|
|
1
|
+
nat
|
@@ -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 +0,0 @@
|
|
1
|
-
aiq
|
File without changes
|
File without changes
|
File without changes
|