From d7e97ed509698905092a9aeb51f6725accde2fbb Mon Sep 17 00:00:00 2001 From: lorenzejay Date: Fri, 18 Apr 2025 18:17:46 -0700 Subject: [PATCH] Enhance KnowledgeConfig with field descriptions - Updated `results_limit` and `score_threshold` in `KnowledgeConfig` to use Pydantic's `Field` for improved documentation and clarity. - Added descriptions to both parameters to provide better context for their usage in knowledge retrieval configuration. --- src/crewai/knowledge/knowledge_config.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/crewai/knowledge/knowledge_config.py b/src/crewai/knowledge/knowledge_config.py index 3139e1722..69ff1107c 100644 --- a/src/crewai/knowledge/knowledge_config.py +++ b/src/crewai/knowledge/knowledge_config.py @@ -1,6 +1,9 @@ -from pydantic import BaseModel +from pydantic import BaseModel, Field class KnowledgeConfig(BaseModel): - results_limit: int = 3 - score_threshold: float = 0.35 + 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", + )