From f53cf838bd80618ea63a0a4795e9853f3996fdbf Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 9 Feb 2025 21:02:22 +0000 Subject: [PATCH] chore: address PR feedback for test functionality Co-Authored-By: Joe Moura --- src/crewai/crew.py | 1 + src/crewai/utilities/evaluators/crew_evaluator_handler.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 23f4a76f7..23aae3880 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -6,6 +6,7 @@ from concurrent.futures import Future from hashlib import md5 from typing import Any, Callable, Dict, List, Optional, Tuple, Union +from langchain_core.language_models.base import BaseLanguageModel from pydantic import ( UUID4, BaseModel, diff --git a/src/crewai/utilities/evaluators/crew_evaluator_handler.py b/src/crewai/utilities/evaluators/crew_evaluator_handler.py index 20052b685..9f2f3606d 100644 --- a/src/crewai/utilities/evaluators/crew_evaluator_handler.py +++ b/src/crewai/utilities/evaluators/crew_evaluator_handler.py @@ -2,7 +2,7 @@ import os from collections import defaultdict from typing import Any, Dict, List, Optional, Union -from langchain.base_language.base_language_model import BaseLanguageModel +from langchain_core.language_models.base import BaseLanguageModel from pydantic import BaseModel, Field, InstanceOf from rich.box import HEAVY_EDGE from rich.console import Console