Fix issue #2233: Add support for task-specific knowledge

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-26 04:03:20 +00:00
parent 34d2993456
commit b524855e22
4 changed files with 69 additions and 0 deletions

View File

@@ -207,6 +207,17 @@ class Agent(BaseAgent):
if memory.strip() != "":
task_prompt += self.i18n.slice("memory").format(memory=memory)
# Check if the task has knowledge first
if hasattr(task, 'knowledge') and task.knowledge:
task_knowledge_snippets = task.knowledge.query([task.prompt()])
if task_knowledge_snippets:
task_knowledge_context = extract_knowledge_context(
task_knowledge_snippets
)
if task_knowledge_context:
task_prompt += task_knowledge_context
# Then check agent's knowledge
if self.knowledge:
agent_knowledge_snippets = self.knowledge.query([task.prompt()])
if agent_knowledge_snippets:
@@ -216,6 +227,7 @@ class Agent(BaseAgent):
if agent_knowledge_context:
task_prompt += agent_knowledge_context
# Finally check crew's knowledge
if self.crew:
knowledge_snippets = self.crew.query_knowledge([task.prompt()])
if knowledge_snippets:

View File

@@ -0,0 +1,3 @@
from crewai.knowledge.knowledge import Knowledge
__all__ = ["Knowledge"]

View File

@@ -32,6 +32,7 @@ from pydantic import (
from pydantic_core import PydanticCustomError
from crewai.agents.agent_builder.base_agent import BaseAgent
from crewai.knowledge import Knowledge
from crewai.tasks.guardrail_result import GuardrailResult
from crewai.tasks.output_format import OutputFormat
from crewai.tasks.task_output import TaskOutput
@@ -144,6 +145,10 @@ class Task(BaseModel):
end_time: Optional[datetime.datetime] = Field(
default=None, description="End time of the task execution"
)
knowledge: Optional[Knowledge] = Field(
default=None,
description="Knowledge sources for the task. This knowledge will be used by the agent when executing the task.",
)
@field_validator("guardrail")
@classmethod