mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
17 lines
539 B
Python
17 lines
539 B
Python
from pydantic import BaseModel, Field
|
|
|
|
|
|
class KnowledgeConfig(BaseModel):
|
|
"""Configuration for knowledge retrieval.
|
|
|
|
Args:
|
|
results_limit (int): The number of relevant documents to return.
|
|
score_threshold (float): The minimum score for a document to be considered relevant.
|
|
"""
|
|
|
|
results_limit: int = Field(default=3, description="The number of results to return")
|
|
score_threshold: float = Field(
|
|
default=0.35,
|
|
description="The minimum score for a result to be considered relevant",
|
|
)
|