Spaces:
Running
Running
File size: 8,749 Bytes
5d0b799 b4f8caf 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 a182e49 e7c7689 b4f8caf e7c7689 a182e49 5d0b799 e7c7689 5d0b799 e7c7689 a182e49 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 5d0b799 bd6bb03 e7c7689 5d0b799 e7c7689 5d0b799 e7c7689 b425bf1 e7c7689 bd6bb03 e7c7689 5d0b799 702a17d 5d0b799 7292b0f 5d0b799 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 |
import os
import gradio as gr
from crewai import Agent, Task, Crew
from langchain_groq import ChatGroq
from langchain_community.tools import DuckDuckGoSearchRun, DuckDuckGoSearchResults
from crewai_tools import tool, SeleniumScrapingTool, ScrapeWebsiteTool
# Define the DuckDuckGoSearch tool using the decorator for tool registration
@tool('DuckDuckGoSearch')
def search(search_query: str):
"""Search the web for information on a given topic."""
return DuckDuckGoSearchRun().run(search_query)
# Define the DuckDuckGoSearch tool
@tool('DuckDuckGoResults')
def search_results(search_query: str):
"""
Performs a web search using the DuckDuckGo search engine to gather and return a collection of search results.
This tool automates the retrieval of web-based information related to a specified query.
Parameters:
- search_query (str): The query string that specifies the information to be searched on the web. This should be a clear and concise expression of the user's information needs.
Returns:
- list: A list of dictionaries, where each dictionary represents a search result. Each dictionary includes at least the 'title' of the page and the 'url' linking to it. Additional information such as a brief summary or snippet from the page might also be included depending on the specifics of the DuckDuckGo API response.
Example:
- Input: search_results(""Generative AI in Telecom and Media")
- Output: [snippet: The telecommunications and media industry is at the forefront of integrating generative AI into their operations, viewing it as a catalyst for growth and innovation. Industry leaders are enthusiastic about its ability to not only enhance the current processes but also spearhead new innovations, create new opportunities, unlock new sources of ..., title: Generative AI in the telecom industry | Google Cloud Blog, link: https://cloud.google.com/blog/topics/telecommunications/generative-ai-in-the-telecom-industry], ...]
Use this tool to quickly gather a variety of sources on any topic without needing to manually search through a web browser. It can be especially useful in automated data gathering, research, and in contexts where multiple web sources need to be quickly consulted.
"""
return DuckDuckGoSearchResults().run(search_query)
# Define the WebScrapper tool
@tool('WebScrapper')
def web_scrapper(url: str):
"""
A tool designed to extract and read the content of a specified link.
It is capable of handling various types of web pages by making HTTP requests and parsing the received HTML content.
This tool can be particularly useful for web scraping tasks, data collection, or extracting specific information from websites.
"""
return ScrapeWebsiteTool(website_url=url)
def kickoff_crew(topic: str) -> dict:
"""Kickoff the research process for a given topic using CrewAI components."""
# Retrieve the API key from the environment variables
groq_api_key = os.environ.get("GROQ_API_KEY")
if not groq_api_key:
raise ValueError("API Key for Groq is not set in environment variables")
# Initialize the Groq large language model
groq_llm = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-70b-8192")
# Define Agents with Groq LLM
researcher = Agent(
role='Researcher',
goal='Collect detailed information on {topic}',
tools=[search, search_results, web_scrapper],
llm=groq_llm, # Assigning the Groq LLM here
backstory=(
"As a diligent researcher, you explore the depths of the internet to "
"unearth crucial information and insights on the assigned topics. "
"With a keen eye for detail and a commitment to accuracy, you meticulously document every source "
"and piece of data gathered. Your research is thorough, ensuring that no stone is left unturned. "
"This dedication not only enhances the quality of the information but also ensures "
"reliability and trustworthiness in your findings."
),
allow_delegation=False,
max_iter=5,
verbose=True, # Optional
)
editor = Agent(
role='Editor',
goal='Compile and refine the information into a comprehensive report on {topic}',
llm=groq_llm, # Assigning the Groq LLM here
backstory=(
"With a keen eye for detail and a strong command of language, you transform "
"raw data into polished, insightful reports that are both informative and engaging. "
"Your expertise in editing ensures that every report is not only thorough but also "
"clearly communicates the key findings in a manner that is accessible to all readers. "
"As an editor, your role is crucial in shaping the final presentation of data, making "
"complex information easy to understand and appealing to the audience."
),
allow_delegation=False,
max_iter=3,
verbose=True, # Optional
)
# Define Tasks
research_task = Task(
description=(
"Use DuckDuckGoSearch tool to gather initial information about {topic}. "
"Next, employ DuckDuckGoResults tool to gather full details of the insights from search results, reading them carefully and preparing detailed summaries on {topic}. "
"Utilize the WebScrapper tool to extract additional information and insights from all links or URLs that appear significant regarding {topic} after analyzing the snippets of the search results. "
"Compile your findings into an initial draft, ensuring to include all sources with their titles and links relevant to the topic. "
"Throughout this process, maintain a high standard of accuracy and ensure that no information is fabricated or misrepresented."
),
expected_output=(
"A draft report containing all relevant information about the topic and sources used. "
"The report should be well-structured, including an introduction, a detailed body with organized sections according to different aspects of the topic, and a conclusion. "
"Each section should cite sources accurately and provide a comprehensive overview of the findings."
),
agent=researcher
)
edit_task = Task(
description=(
"Review and refine the draft report produced by the research task. Organize the content methodically, "
"ensuring that the structure is logical and enhances the flow of information. Check all factual data for accuracy, "
"correct any discrepancies, and ensure that the information is current and well-supported by sources. "
"Enhance the readability of the report by improving language clarity, adjusting sentence structure, and ensuring consistency in tone. "
"Include a dedicated section that lists all sources used in the research_task. "
"Each source used in the analysis should be presented as a bullet point in the format: title: link "
"Ensure that all sources you include in the final report exist by scrapping them if necessary. "
"This section should be comprehensive, clearly formatted, and easy to navigate, providing full transparency on the references used."
),
expected_output=(
"A finalized comprehensive report on ## {topic} ##. The report should be polished, with a clear and engaging narrative "
"that accurately reflects the research findings. It should include an introduction, a detailed and extensive discussion section, a concise conclusion, "
"and a well-organized source list. The document should be free of grammatical errors and ready for publication or presentation."
),
agent=editor,
context=[research_task]
)
# Forming the Crew
crew = Crew(
agents=[researcher, editor],
tasks=[research_task, edit_task]
)
# Kick-off the research process
result = crew.kickoff(inputs={'topic': topic})
return result
def main():
"""Set up the Gradio interface for the CrewAI Research Tool."""
with gr.Blocks() as demo:
gr.Markdown("## CrewAI Research Tool")
topic_input = gr.Textbox(label="Enter Topic", placeholder="Type here...")
submit_button = gr.Button("Start Research")
output = gr.Markdown(label="Result")
submit_button.click(
fn=kickoff_crew,
inputs=topic_input,
outputs=output
)
# demo.launch(debug=True)
demo.queue(api_open=False, max_size=5).launch()
if __name__ == "__main__":
main()
|