WIP - Code refactoring

AJaySi · Mar 31, 2024 · 1833a85 · 1833a85
1 parent bedd0ac
commit 1833a85
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/lib/ai_web_researcher/google_serp_search.py b/lib/ai_web_researcher/google_serp_search.py
@@ -50,11 +50,8 @@
            )
 
 
-#from tenacity import retry, stop_after_attempt, wait_random_exponential
-#@retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
-
-
-#FIXME: Accept language, country and time frame to search for.
+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 google_search(query):
     """
     Perform a Google search for the given query.

diff --git a/lib/ai_writers/keywords_to_blog.py b/lib/ai_writers/keywords_to_blog.py
@@ -39,9 +39,12 @@ def write_blog_from_keywords(search_keywords, url=None):
 
     logger.info(f"Researching and Writing Blog on keywords: {search_keywords}")
     # Call on the got-researcher, tavily apis for this. Do google search for organic competition.
-    google_search_result, g_titles = do_google_serp_search(search_keywords)
-    example_blog_titles.append(g_titles)
-    blog_markdown_str = write_blog_google_serp(search_keywords, google_search_result)
+    try:
+        google_search_result, g_titles = do_google_serp_search(search_keywords)
+        example_blog_titles.append(g_titles)
+        blog_markdown_str = write_blog_google_serp(search_keywords, google_search_result)
+    except Exception as err:
+        logger.error(f"Failed in Google web research: {err}")
     # logger.info/check the final blog content.
     logger.info(f"######### Blog content Google SERP research: ###########\n\n{blog_markdown_str}\n\n")
 

diff --git a/lib/gpt_providers/openai_text_gen.py b/lib/gpt_providers/openai_text_gen.py
@@ -15,7 +15,7 @@
 
 
 @retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
-def openai_chatgpt(prompt, model="gpt-3.5-turbo-0125", temperature=0.2, max_tokens=4096, top_p=0.9, n=1):
+def openai_chatgpt(prompt, model="gpt-3.5-turbo-0125", temperature=0.7, max_tokens=4096, top_p=0.5, n=1):
     """
     Wrapper function for OpenAI's ChatGPT completion.