more fixes

This commit is contained in:
Lorenze Jay
2024-11-20 14:02:12 -08:00
parent 8564f5551f
commit 38c0d61b11
3 changed files with 4 additions and 3 deletions

View File

@@ -6,7 +6,7 @@ from pydantic import BaseModel, ConfigDict, Field
from crewai.knowledge.source.base_knowledge_source import BaseKnowledgeSource
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
from crewai.utilities.logger import Logger
from crewai.utilities.constants import DEFAULT_SCORE_THRESHOLD
os.environ["TOKENIZERS_PARALLELISM"] = "false" # removes logging from fastembed
@@ -42,6 +42,6 @@ class Knowledge(BaseModel):
query,
limit,
filter={"preference": preference} if preference else None,
score_threshold=0.35,
score_threshold=DEFAULT_SCORE_THRESHOLD,
)
return results

View File

@@ -20,7 +20,7 @@ class BaseKnowledgeSource(BaseModel, ABC):
metadata: Dict[str, Any] = Field(default_factory=dict)
@abstractmethod
def load_content(self):
def load_content(self) -> Dict[Any, str]:
"""Load and preprocess content from the source."""
pass

View File

@@ -1,2 +1,3 @@
TRAINING_DATA_FILE = "training_data.pkl"
TRAINED_AGENTS_DATA_FILE = "trained_agents_data.pkl"
DEFAULT_SCORE_THRESHOLD = 0.35