diff --git a/docs/concepts/knowledge.mdx b/docs/concepts/knowledge.mdx index 6d3f6c167..305df8d73 100644 --- a/docs/concepts/knowledge.mdx +++ b/docs/concepts/knowledge.mdx @@ -163,7 +163,7 @@ You can configure the knowledge configuration for the crew or agent. ```python Code from crewai.knowledge.knowledge_config import KnowledgeConfig -knowledge_config = KnowledgeConfig(limit=10, score_threshold=0.5) +knowledge_config = KnowledgeConfig(results_limit=10, score_threshold=0.5) agent = Agent( ... @@ -172,7 +172,7 @@ agent = Agent( ``` - limit: is the number of relevant documents to return. Default is 3. + results_limit: is the number of relevant documents to return. Default is 3. score_threshold: is the minimum score for a document to be considered relevant. Default is 0.35. diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 600dd5d4e..7c9696f6d 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -1135,11 +1135,11 @@ class Crew(BaseModel): return result def query_knowledge( - self, query: List[str], limit: int = 3, score_threshold: float = 0.35 + self, query: List[str], results_limit: int = 3, score_threshold: float = 0.35 ) -> Union[List[Dict[str, Any]], None]: if self.knowledge: return self.knowledge.query( - query, limit=limit, score_threshold=score_threshold + query, results_limit=results_limit, score_threshold=score_threshold ) return None diff --git a/src/crewai/knowledge/knowledge.py b/src/crewai/knowledge/knowledge.py index 8cd8af2aa..824325d12 100644 --- a/src/crewai/knowledge/knowledge.py +++ b/src/crewai/knowledge/knowledge.py @@ -44,7 +44,7 @@ class Knowledge(BaseModel): self._add_sources() def query( - self, query: List[str], limit: int = 3, score_threshold: float = 0.35 + self, query: List[str], results_limit: int = 3, score_threshold: float = 0.35 ) -> List[Dict[str, Any]]: """ Query across all knowledge sources to find the most relevant information. @@ -58,7 +58,7 @@ class Knowledge(BaseModel): results = self.storage.search( query, - limit, + limit=results_limit, score_threshold=score_threshold, ) return results diff --git a/src/crewai/knowledge/knowledge_config.py b/src/crewai/knowledge/knowledge_config.py index 434d52f1e..3139e1722 100644 --- a/src/crewai/knowledge/knowledge_config.py +++ b/src/crewai/knowledge/knowledge_config.py @@ -2,5 +2,5 @@ from pydantic import BaseModel class KnowledgeConfig(BaseModel): - limit: int = 3 + results_limit: int = 3 score_threshold: float = 0.35