Skip to content

Commit f1de603

Browse files
committed
update test
1 parent 9b373a1 commit f1de603

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

crewai-tools/test_mongodb_vector_search_tool.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from crewai import Agent
33
from crewai import Task
44
from crewai import Crew, Process, LLM
5+
from crewai.cli.constants import DEFAULT_LLM_MODEL
56
from crewai_tools import MongoDBVectorSearchTool, MongoDBVectorSearchConfig
67
from langchain_community.document_loaders import PyPDFLoader
78
import time
@@ -56,7 +57,7 @@
5657
backstory="You're specialized in analyzing technical content to extract insights and answers",
5758
verbose=False,
5859
tools=[tool],
59-
llm=LLM(model="azure/gpt-4o-mini"),
60+
llm=LLM(model=f"azure/{DEFAULT_LLM_MODEL}"),
6061
)
6162
research_task = Task(
6263
description="Research information in a technical document",

0 commit comments

Comments
 (0)