From 54237c997451a8076a6b3f679ba620951273f527 Mon Sep 17 00:00:00 2001 From: Braelyn Boynton Date: Thu, 9 May 2024 13:15:12 -0700 Subject: [PATCH] track task evaluator --- src/crewai/utilities/evaluators/task_evaluator.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/src/crewai/utilities/evaluators/task_evaluator.py b/src/crewai/utilities/evaluators/task_evaluator.py index f5bb5b1bf..171326592 100644 --- a/src/crewai/utilities/evaluators/task_evaluator.py +++ b/src/crewai/utilities/evaluators/task_evaluator.py @@ -5,6 +5,17 @@ from pydantic import BaseModel, Field from crewai.utilities import Converter from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser +agentops = None +try: + import agentops + from agentops import track_agent +except ImportError: + + def track_agent(name): + def noop(f): + return f + + return noop class Entity(BaseModel): @@ -26,6 +37,7 @@ class TaskEvaluation(BaseModel): ) +@track_agent(name="Task Evaluator") class TaskEvaluator: def __init__(self, original_agent): self.llm = original_agent.llm