AI finance TA writer, long form writer, WIP

2024-05-16 22:46:12 +05:30
parent e3b51e2713
commit 8bd4436414
3 changed files with 72 additions and 71 deletions
--- a/lib/ai_web_researcher/tavily_ai_search.py
+++ b/lib/ai_web_researcher/tavily_ai_search.py
@@ -85,12 +85,14 @@ def get_tavilyai_results(keywords):
        if include_urls:
            tavily_search_result = client.search(keywords, 
                    search_depth="advanced", 
-                    include_answer=True, 
+                    include_answer=True,
+                    max_results=10,
                    include_domains=include_urls)
        else:
            tavily_search_result = client.search(keywords, 
                    search_depth = "advanced", 
-                    include_answer=True)
+                    include_answer=True,
+                    max_results=10)

        print_result_table(tavily_search_result)
        return(tavily_search_result)