WIP- AI-Writer and Web research, Working on packaging.

lib/ai_web_researcher/google_serp_search.py

@@ -304,7 +304,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}")

lib/ai_web_researcher/google_trends_researcher.py

@@ -482,7 +482,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, "w") 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}")

lib/ai_web_researcher/metaphor_basic_neural_web_search.py

@@ -205,7 +205,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}")

lib/ai_web_researcher/tavily_ai_search.py

@@ -73,7 +73,7 @@ def get_tavilyai_results(keywords, include_urls, search_depth="advanced"):
         client = TavilyClient(api_key=api_key)
     except Exception as err:
         logger.error(f"Failed to create Tavily client. Check TAVILY_API_KEY: {err}")
-        exit(1)
+    
     try:
         if include_urls:
             tavily_search_result = client.search(keywords, search_depth, include_answer=True, include_domains=include_urls)
@@ -147,7 +147,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}")