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

6
lib/ai_web_researcher/tavily_ai_search.py
View File

@@ -49,7 +49,7 @@ from tenacity import retry, stop_after_attempt, wait_random_exponential
@retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
def get_tavilyai_results(keywords):
def get_tavilyai_results(keywords, max_results=10):
"""
Get Tavily AI search results based on specified keywords and options.
@@ -86,13 +86,13 @@ def get_tavilyai_results(keywords):
tavily_search_result = client.search(keywords,
search_depth="advanced",
include_answer=True,
max_results=10,
max_results=max_results,
include_domains=include_urls)
else:
tavily_search_result = client.search(keywords,
search_depth = "advanced",
include_answer=True,
max_results=10)
max_results=max_results)
print_result_table(tavily_search_result)
return(tavily_search_result)