mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
24ed8a2549
* fix: fix crewai-tools cli command * feat: add crewai train CLI command * feat: add the tests * fix: fix typing hinting issue on code * fix: test.yml * fix: fix test * fix: removed fix since it didnt changed the test
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import inspect
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
from dotenv import load_dotenv
|
|
from pydantic import ConfigDict
|
|
|
|
load_dotenv()
|
|
|
|
|
|
def CrewBase(cls):
|
|
class WrappedClass(cls):
|
|
model_config = ConfigDict(arbitrary_types_allowed=True)
|
|
is_crew_class: bool = True # type: ignore
|
|
|
|
base_directory = None
|
|
for frame_info in inspect.stack():
|
|
if "site-packages" not in frame_info.filename:
|
|
base_directory = Path(frame_info.filename).parent.resolve()
|
|
break
|
|
|
|
if base_directory is None:
|
|
raise Exception(
|
|
"Unable to dynamically determine the project's base directory, you must run it from the project's root directory."
|
|
)
|
|
|
|
original_agents_config_path = getattr(
|
|
cls, "agents_config", "config/agents.yaml"
|
|
)
|
|
original_tasks_config_path = getattr(cls, "tasks_config", "config/tasks.yaml")
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.agents_config = self.load_yaml(
|
|
os.path.join(self.base_directory, self.original_agents_config_path)
|
|
)
|
|
self.tasks_config = self.load_yaml(
|
|
os.path.join(self.base_directory, self.original_tasks_config_path)
|
|
)
|
|
|
|
@staticmethod
|
|
def load_yaml(config_path: str):
|
|
with open(config_path, "r") as file:
|
|
# parsedContent = YamlParser.parse(file) # type: ignore # Argument 1 to "parse" has incompatible type "TextIOWrapper"; expected "YamlParser"
|
|
return yaml.safe_load(file)
|
|
|
|
return WrappedClass
|