Agents team, better prompts, WIP

2024-05-12 19:13:02 +05:30
parent f9b9204349
commit 6a95b96973
13 changed files with 392 additions and 90 deletions
--- a/lib/ai_web_researcher/gpt_online_researcher.py
+++ b/lib/ai_web_researcher/gpt_online_researcher.py
@@ -74,7 +74,8 @@ def do_tavily_ai_search(search_keywords):
        logger.info(f"Doing Tavily AI search for: {search_keywords}")
        t_results = get_tavilyai_results(search_keywords)
        t_titles = tavily_extract_information(t_results, 'titles')
-        return(t_results, t_titles)
+        t_answer = tavily_extract_information(t_results, 'answer')
+        return(t_results, t_titles, t_answer)
    except Exception as err:
        logger.error(f"Failed to do Tavily AI Search: {err}")