File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed
Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change 22from crewai import Agent
33from crewai import Task
44from crewai import Crew , Process , LLM
5- from crewai .cli .constants import DEFAULT_LLM_MODEL
65from crewai_tools import MongoDBVectorSearchTool , MongoDBVectorSearchConfig
76from langchain_community .document_loaders import PyPDFLoader
87import time
5554 role = "AI Accuracy Researcher" ,
5655 goal = "Find and extract key information from a technical document" ,
5756 backstory = "You're specialized in analyzing technical content to extract insights and answers" ,
58- verbose = False ,
57+ verbose = True ,
5958 tools = [tool ],
60- llm = LLM (model = f "azure/{ DEFAULT_LLM_MODEL } " ),
59+ llm = LLM (model = "azure/gpt-4o" , seed = 12345 ),
6160)
6261research_task = Task (
6362 description = "Research information in a technical document" ,
6867 agents = [researcher ],
6968 tasks = [research_task ],
7069 process = Process .sequential ,
71- verbose = False ,
70+ verbose = True ,
7271)
7372
7473# Get the result and assert something about the results
7574print ("Running the crew..." )
7675result = crew .kickoff ()
7776text = result .raw .lower ()
78- assert "advancements" in text or "improvements " in text , text
77+ assert "limitations " in text , text
7978assert "GPT-4" in result .raw
You can’t perform that action at this time.
0 commit comments