andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: update typechecking

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-10-08 17:41:49 -03:00
parent 6d90ac3fc3
commit dc1902ff44
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/crewai/project/crew_base.py
View File

@@ -1,6 +1,6 @@
import inspect import inspect
from pathlib import Path from pathlib import Path
from typing import Any, Callable, Dict, Type, TypeVar from typing import Any, Callable, Dict, Type, TypeVar, cast
import yaml import yaml
from dotenv import load_dotenv from dotenv import load_dotenv
@@ -11,7 +11,7 @@ T = TypeVar("T", bound=Type[Any])
def CrewBase(cls: T) -> T: def CrewBase(cls: T) -> T:
class WrappedClass(cls): class WrappedClass(cls): # type: ignore
is_crew_class: bool = True # type: ignore is_crew_class: bool = True # type: ignore
# Get the directory of the class being decorated # Get the directory of the class being decorated
@@ -180,4 +180,4 @@ def CrewBase(cls: T) -> T:
callback_functions[callback]() for callback in callbacks callback_functions[callback]() for callback in callbacks
] ]
return WrappedClass return cast(T, WrappedClass)