mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
82647358b2
* initial fix on delegation tools
* fixing tests for delegations and coding
* Refactor prepare tool and adding initial add images logic
* supporting image tool
* fixing linter
* fix linter
* Making sure multimodal feature support i18n
* fix linter and types
* mixxing translations
* fix types and linter
* Revert "fixing linter"
This reverts commit 2eda5fdeed.
* fix linters
* test
* fix
* fix
* fix linter
* fix
* ignore
* type improvements
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import json
|
|
import os
|
|
from typing import Dict, Optional, Union
|
|
|
|
from pydantic import BaseModel, Field, PrivateAttr, model_validator
|
|
|
|
|
|
class I18N(BaseModel):
|
|
_prompts: Dict[str, Dict[str, str]] = PrivateAttr()
|
|
prompt_file: Optional[str] = Field(
|
|
default=None,
|
|
description="Path to the prompt_file file to load",
|
|
)
|
|
|
|
@model_validator(mode="after")
|
|
def load_prompts(self) -> "I18N":
|
|
"""Load prompts from a JSON file."""
|
|
try:
|
|
if self.prompt_file:
|
|
with open(self.prompt_file, "r", encoding="utf-8") as f:
|
|
self._prompts = json.load(f)
|
|
else:
|
|
dir_path = os.path.dirname(os.path.realpath(__file__))
|
|
prompts_path = os.path.join(dir_path, "../translations/en.json")
|
|
|
|
with open(prompts_path, "r", encoding="utf-8") as f:
|
|
self._prompts = json.load(f)
|
|
except FileNotFoundError:
|
|
raise Exception(f"Prompt file '{self.prompt_file}' not found.")
|
|
except json.JSONDecodeError:
|
|
raise Exception("Error decoding JSON from the prompts file.")
|
|
|
|
if not self._prompts:
|
|
self._prompts = {}
|
|
|
|
return self
|
|
|
|
def slice(self, slice: str) -> str:
|
|
return self.retrieve("slices", slice)
|
|
|
|
def errors(self, error: str) -> str:
|
|
return self.retrieve("errors", error)
|
|
|
|
def tools(self, tool: str) -> Union[str, Dict[str, str]]:
|
|
return self.retrieve("tools", tool)
|
|
|
|
def retrieve(self, kind, key) -> str:
|
|
try:
|
|
return self._prompts[kind][key]
|
|
except Exception as _:
|
|
raise Exception(f"Prompt for '{kind}':'{key}' not found.")
|