andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 15:22:37 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: apply ruff linting fixes to tasks module

Browse Source
This commit is contained in:
Greyson LaLonde
2025-09-22 13:09:53 -04:00
committed by GitHub
parent 58413b663a
commit db5f565dea
2 changed files with 11 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/crewai/tasks/hallucination_guardrail.py
View File

@@ -6,7 +6,7 @@ Classes:
HallucinationGuardrail: Placeholder guardrail that validates task outputs. HallucinationGuardrail: Placeholder guardrail that validates task outputs.
""" """
from typing import Any, Optional, Tuple from typing import Any
from crewai.llm import LLM from crewai.llm import LLM
from crewai.tasks.task_output import TaskOutput from crewai.tasks.task_output import TaskOutput
@@ -48,7 +48,7 @@ class HallucinationGuardrail:
self, self,
context: str, context: str,
llm: LLM, llm: LLM,
threshold: Optional[float] = None, threshold: float | None = None,
tool_response: str = "", tool_response: str = "",
): ):
"""Initialize the HallucinationGuardrail placeholder. """Initialize the HallucinationGuardrail placeholder.
@@ -75,7 +75,7 @@ class HallucinationGuardrail:
"""Generate a description of this guardrail for event logging.""" """Generate a description of this guardrail for event logging."""
return "HallucinationGuardrail (no-op)" return "HallucinationGuardrail (no-op)"
def __call__(self, task_output: TaskOutput) -> Tuple[bool, Any]: def __call__(self, task_output: TaskOutput) -> tuple[bool, Any]:
"""Validate a task output against hallucination criteria. """Validate a task output against hallucination criteria.
In the open source, this method always returns that the output is valid. In the open source, this method always returns that the output is valid.

20
src/crewai/tasks/llm_guardrail.py
View File

@@ -1,4 +1,4 @@
from typing import Any, Tuple from typing import Any
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
@@ -53,7 +53,7 @@ class LLMGuardrail:
Guardrail: Guardrail:
{self.description} {self.description}
Your task: Your task:
- Confirm if the Task result complies with the guardrail. - Confirm if the Task result complies with the guardrail.
- If not, provide clear feedback explaining what is wrong (e.g., by how much it violates the rule, or what specific part fails). - If not, provide clear feedback explaining what is wrong (e.g., by how much it violates the rule, or what specific part fails).
@@ -61,11 +61,9 @@ class LLMGuardrail:
- If the Task result complies with the guardrail, saying that is valid - If the Task result complies with the guardrail, saying that is valid
""" """
result = agent.kickoff(query, response_format=LLMGuardrailResult) return agent.kickoff(query, response_format=LLMGuardrailResult)
return result def __call__(self, task_output: TaskOutput) -> tuple[bool, Any]:
def __call__(self, task_output: TaskOutput) -> Tuple[bool, Any]:
"""Validates the output of a task based on specified criteria. """Validates the output of a task based on specified criteria.
Args: Args:
@@ -79,13 +77,11 @@ class LLMGuardrail:
try: try:
result = self._validate_output(task_output) result = self._validate_output(task_output)
assert isinstance( if not isinstance(result.pydantic, LLMGuardrailResult):
result.pydantic, LLMGuardrailResult raise ValueError("The guardrail result is not a valid pydantic model")
), "The guardrail result is not a valid pydantic model"
if result.pydantic.valid: if result.pydantic.valid:
return True, task_output.raw return True, task_output.raw
else: return False, result.pydantic.feedback
return False, result.pydantic.feedback
except Exception as e: except Exception as e:
return False, f"Error while validating the task output: {str(e)}" return False, f"Error while validating the task output: {e!s}"