SwatGarg commited on
Commit
c95b5a3
·
verified ·
1 Parent(s): 9ff48a8

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +5 -3
agents.py CHANGED
@@ -4,6 +4,8 @@ import os
4
  from dotenv import load_dotenv
5
  from tools.image_generation_tools import ImageGenerationTools
6
  from tools.web_search_tools import WebSearchTools
 
 
7
  load_dotenv()
8
  secret_token = os.getenv("content")
9
 
@@ -40,7 +42,7 @@ class ContentAgents:
40
  7. Images to support the content
41
  """,
42
  llm=self.llm,
43
- tools=[web_search_tools.search_internet, web_search_tools.scrape_and_summarize_website, web_search_tools.bing_image_search, web_search_tools.download_images],
44
  allow_delegation=False,
45
  verbose=True)
46
 
@@ -52,7 +54,7 @@ class ContentAgents:
52
  the Planner, you do web searches on the topic {topic} in order to find compelling images as well as provide precise and up-to-date
53
  domain knowledge. Your work is the basis for the Content Writer to write an article on this topic.""",
54
  llm=self.llm,
55
- tools=[web_search_tools.search_internet, web_search_tools.scrape_and_summarize_website, web_search_tools.bing_image_search, web_search_tools.download_images],
56
  allow_delegation=False,
57
  verbose=True)
58
 
@@ -78,6 +80,6 @@ class ContentAgents:
78
  are opinions as opposed to objective statements. You avoid plagiarism, you provide references for your citations, and mention
79
  attribution to the images/information coming from others.""",
80
  llm=self.llm,
81
- tools=[web_search_tools.search_internet, web_search_tools.scrape_and_summarize_website, web_search_tools.bing_image_search, web_search_tools.download_images],
82
  allow_delegation=False,
83
  verbose=True)
 
4
  from dotenv import load_dotenv
5
  from tools.image_generation_tools import ImageGenerationTools
6
  from tools.web_search_tools import WebSearchTools
7
+ from web_scraper_tool import web_scraper_tool
8
+ from image_search_tool import image_search_tool
9
  load_dotenv()
10
  secret_token = os.getenv("content")
11
 
 
42
  7. Images to support the content
43
  """,
44
  llm=self.llm,
45
+ tools=[web_scraper_tool],
46
  allow_delegation=False,
47
  verbose=True)
48
 
 
54
  the Planner, you do web searches on the topic {topic} in order to find compelling images as well as provide precise and up-to-date
55
  domain knowledge. Your work is the basis for the Content Writer to write an article on this topic.""",
56
  llm=self.llm,
57
+ tools=[web_scraper_tool],
58
  allow_delegation=False,
59
  verbose=True)
60
 
 
80
  are opinions as opposed to objective statements. You avoid plagiarism, you provide references for your citations, and mention
81
  attribution to the images/information coming from others.""",
82
  llm=self.llm,
83
+ tools=[image_search_tool],
84
  allow_delegation=False,
85
  verbose=True)