Spaces:
Sleeping
Sleeping
| from typing import List, Dict, Any, Optional | |
| from crewai import Agent | |
| from crewai_tools import BraveSearchTool, ScrapeWebsiteTool | |
| from tools import ContentAnalyzerTool, RateLimitedToolWrapper, TavilySearchTool, SearchRotationTool | |
| def create_researcher_agent(llm=None, verbose=True) -> Agent: | |
| """ | |
| Creates a researcher agent responsible for query refinement and web search. | |
| Args: | |
| llm: Language model to use for the agent | |
| verbose: Whether to log agent activity | |
| Returns: | |
| Configured researcher agent | |
| """ | |
| # Initialize search tools | |
| brave_search_tool = BraveSearchTool( | |
| n_results=5, | |
| save_file=False | |
| ) | |
| # Initialize Tavily search tool | |
| # Requires a TAVILY_API_KEY in environment variables | |
| tavily_search_tool = TavilySearchTool( | |
| max_results=5, | |
| search_depth="basic", | |
| timeout=15 # Increase timeout for more reliable results | |
| ) | |
| # Add minimal rate limiting to avoid API throttling | |
| # Set delay to 0 to disable rate limiting completely | |
| rate_limited_brave_search = RateLimitedToolWrapper(tool=brave_search_tool, delay=0) | |
| rate_limited_tavily_search = RateLimitedToolWrapper(tool=tavily_search_tool, delay=0) | |
| # Create the search rotation tool | |
| search_rotation_tool = SearchRotationTool( | |
| search_tools=[rate_limited_brave_search, rate_limited_tavily_search], | |
| max_searches_per_query=5 # Limit to 5 searches per query as requested | |
| ) | |
| return Agent( | |
| role="Research Specialist", | |
| goal="Discover accurate and relevant information from the web", | |
| backstory=( | |
| "You are an expert web researcher with a talent for crafting effective search queries " | |
| "and finding high-quality information on any topic. Your goal is to find the most " | |
| "relevant and factual information to answer user questions. You have access to multiple " | |
| "search engines and know how to efficiently use them within the search limits." | |
| ), | |
| # Use the search rotation tool | |
| tools=[search_rotation_tool], | |
| verbose=verbose, | |
| allow_delegation=True, | |
| memory=True, | |
| llm=llm | |
| ) | |
| def create_analyst_agent(llm=None, verbose=True) -> Agent: | |
| """ | |
| Creates an analyst agent responsible for content analysis and evaluation. | |
| Args: | |
| llm: Language model to use for the agent | |
| verbose: Whether to log agent activity | |
| Returns: | |
| Configured analyst agent | |
| """ | |
| # Initialize tools | |
| scrape_tool = ScrapeWebsiteTool() | |
| content_analyzer = ContentAnalyzerTool() | |
| return Agent( | |
| role="Content Analyst", | |
| goal="Analyze web content for relevance, factuality, and quality", | |
| backstory=( | |
| "You are a discerning content analyst with a keen eye for detail and a strong " | |
| "commitment to factual accuracy. You excel at evaluating information and filtering " | |
| "out irrelevant or potentially misleading content. Your expertise helps ensure that " | |
| "only the most reliable information is presented." | |
| ), | |
| tools=[scrape_tool, content_analyzer], | |
| verbose=verbose, | |
| allow_delegation=True, | |
| memory=True, | |
| llm=llm | |
| ) | |
| def create_writer_agent(llm=None, verbose=True) -> Agent: | |
| """ | |
| Creates a writer agent responsible for synthesizing information into coherent responses. | |
| Args: | |
| llm: Language model to use for the agent | |
| verbose: Whether to log agent activity | |
| Returns: | |
| Configured writer agent | |
| """ | |
| return Agent( | |
| role="Research Writer", | |
| goal="Create informative, factual, and well-cited responses to research queries", | |
| backstory=( | |
| "You are a skilled writer specializing in creating clear, concise, and informative " | |
| "responses based on research findings. You have a talent for synthesizing information " | |
| "from multiple sources and presenting it in a coherent and readable format, always with " | |
| "proper citations. You prioritize factual accuracy and clarity in your writing." | |
| ), | |
| verbose=verbose, | |
| allow_delegation=True, | |
| memory=True, | |
| llm=llm | |
| ) |