GiantAnalytics commited on
Commit
f658306
·
verified ·
1 Parent(s): 0ad67ed

Update phase1_agents.py

Browse files
Files changed (1) hide show
  1. phase1_agents.py +3 -3
phase1_agents.py CHANGED
@@ -33,7 +33,7 @@ company_search_agent = Agent(
33
  def search_company(company_name: str) -> dict:
34
  """ Searches for detailed company information using web search. """
35
  query = f"Find details for {company_name}, including its official website, mission, services, and AI-related initiatives. Include sources."
36
- response = company_search_agent.print_response(query)
37
  return {"company_name": company_name, "details": response.content}
38
 
39
 
@@ -55,7 +55,7 @@ def scrape_website(url):
55
  response = firecrawl_agent.print_response(f"Extract business details from {url}, including mission, services, and AI-related information.")
56
  except LiveError:
57
  # If Live() is already active, print a simple log instead
58
- response = firecrawl_agent.print_response(f"Extract business details from {url}, including mission, services, and AI-related information.")
59
 
60
  return response.content
61
 
@@ -75,7 +75,7 @@ text_processing_agent = Agent(
75
 
76
  def process_company_description(text: str) -> dict:
77
  """ Summarizes the user-provided company description. """
78
- response = text_processing_agent.print_response(f"Summarize the following description: {text}. Focus on mission, key services, industry, and AI potential.")
79
  return {"user_description": text, "summary": response.content}
80
 
81
 
 
33
  def search_company(company_name: str) -> dict:
34
  """ Searches for detailed company information using web search. """
35
  query = f"Find details for {company_name}, including its official website, mission, services, and AI-related initiatives. Include sources."
36
+ response = company_search_agent.run(query)
37
  return {"company_name": company_name, "details": response.content}
38
 
39
 
 
55
  response = firecrawl_agent.print_response(f"Extract business details from {url}, including mission, services, and AI-related information.")
56
  except LiveError:
57
  # If Live() is already active, print a simple log instead
58
+ response = firecrawl_agent.run(f"Extract business details from {url}, including mission, services, and AI-related information.")
59
 
60
  return response.content
61
 
 
75
 
76
  def process_company_description(text: str) -> dict:
77
  """ Summarizes the user-provided company description. """
78
+ response = text_processing_agent.run(f"Summarize the following description: {text}. Focus on mission, key services, industry, and AI potential.")
79
  return {"user_description": text, "summary": response.content}
80
 
81