Alwrity - WIP - main_config

lib/ai_web_researcher/gpt_competitor_analysis.py

@@ -1,8 +1,5 @@
 import sys
 
-from ..gpt_providers.openai_text_gen import openai_chatgpt
-from ..gpt_providers.gemini_pro_text import gemini_text_response
-
 from loguru import logger
 logger.remove()
 logger.add(sys.stdout,
@@ -10,14 +7,13 @@ logger.add(sys.stdout,
         format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
     )
 
+from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
+
 
 def summarize_competitor_content(research_content, gpt_providers="openai"):
     """Combine the given online research and gpt blog content"""
 
-    prompt = f""" Web page content: {research_content} """
-
-    if 'gemini' in gpt_providers:
-        prompt = f"""You are a helpful assistant writing a research report about a company. I will provide you with company details. 
+    prompt = f"""You are a helpful assistant writing a research report about a company. I will provide you with company details. 
         Summarize the given company details into multiple paragraphs. 
         Be extremely concise, professional, and factual as possible. 
         The first paragraph should be an introduction and summary of the company. 
@@ -25,17 +21,10 @@ def summarize_competitor_content(research_content, gpt_providers="openai"):
         The third paragraph should be on their pricing model.
         Include a conclusion, summarizing your research about the given company details.
         Company details: '{research_content}'"""
-        try:
-            response = gemini_text_response(prompt)
-            return response
-        except Exception as err:
-            logger.error(f"Failed to get response from gemini: {err}")
-            raise err
-    elif 'openai' in gpt_providers:
-        try:
-            logger.info("Calling OpenAI LLM.")
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            logger.error(f"failed to get response from Openai: {err}")
-            raise err
+    
+    try:
+        response = gemini_text_response(prompt)
+        return response
+    except Exception as err:
+        logger.error(f"Failed to get response from LLM: {err}")
+        raise err