WIP - Code refactoring

This commit is contained in:
AjaySi
2024-03-28 23:22:58 +05:30
parent 3920186fc7
commit bedd0ac422
14 changed files with 17 additions and 16 deletions

View File

@@ -1,6 +1,6 @@
import sys
from ..gpt_providers.openai_chat_completion import openai_chatgpt
from ..gpt_providers.openai_text_gen import openai_chatgpt
from ..gpt_providers.gemini_pro_text import gemini_text_response
from loguru import logger

View File

@@ -56,7 +56,7 @@ def do_google_serp_search(search_keywords):
def do_tavily_ai_search(search_keywords, include_domains=None):
""" """
""" 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}")

View File

@@ -1,6 +1,6 @@
import sys
from ..gpt_providers.openai_chat_completion import openai_chatgpt
from ..gpt_providers.openai_text_gen import openai_chatgpt
from ..gpt_providers.gemini_pro_text import gemini_text_response
from loguru import logger

View File

@@ -81,7 +81,6 @@ def get_tavilyai_results(keywords, include_urls, search_depth="advanced"):
tavily_search_result = client.search(keywords, search_depth, include_answer=True)
print_result_table(tavily_search_result)
return(tavily_search_result)
except Exception as err:
logger.error(f"Failed to do Tavily Research: {err}")
@@ -147,7 +146,7 @@ def save_in_file(table_content):
file_path = os.environ.get('SEARCH_SAVE_FILE')
try:
# Save the content to the file
with open(file_path, "a+") as file:
with open(file_path, "a") as file:
file.write(table_content)
file.write("\n" * 3) # Add three newlines at the end
logger.info(f"Search content saved to {file_path}")