kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AI Agents content ideator team, prompt config

Browse Source
This commit is contained in:
ajaysi
2024-05-20 15:07:20 +05:30
parent b431bfcbd8
commit 8f2bf02b65
10 changed files with 172 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/ai_web_researcher/gpt_online_researcher.py
View File

@@ -67,12 +67,12 @@ def do_google_serp_search(search_keywords):
# Not failing, as tavily would do same and then GPT-V to search.
def do_tavily_ai_search(search_keywords):
def do_tavily_ai_search(search_keywords, max_results=10):
""" Common function to do Tavily AI web research."""
try:
# FIXME: Include the follow-up questions as blog FAQs.
logger.info(f"Doing Tavily AI search for: {search_keywords}")
t_results = get_tavilyai_results(search_keywords)
t_results = get_tavilyai_results(search_keywords, max_results)
t_titles = tavily_extract_information(t_results, 'titles')
t_answer = tavily_extract_information(t_results, 'answer')
return(t_results, t_titles, t_answer)