Update app.py

app.py

@@ -16,30 +16,45 @@ from gpt_researcher import GPTResearcher
 # -------------------------
 st.set_page_config(layout="wide", page_title="GPT Researcher")
 
-# Providers & models — set safe defaults to avoid `o1-preview`
+# Base providers & defaults
 os.environ.setdefault("LLM_PROVIDER", "openai")
 os.environ.setdefault("EMBEDDING_PROVIDER", "openai")
 os.environ.setdefault("EMBEDDING_MODEL", "text-embedding-3-small")
-os.environ.setdefault("STRATEGIC_LLM", "gpt-4o")
-os.environ.setdefault("SMART_LLM", "gpt-4o-mini")
-# Compatibility aliases some versions of gpt_researcher read
-os.environ.setdefault("STRATEGIC_MODEL", os.environ["STRATEGIC_LLM"])
-os.environ.setdefault("SMART_MODEL", os.environ["SMART_LLM"])
-os.environ.setdefault("STRATEGY_LLM", os.environ["STRATEGIC_LLM"])
-os.environ.setdefault("STRATEGY_MODEL", os.environ["STRATEGIC_LLM"])
+
+# IMPORTANT: gpt_researcher expects "<provider>:<model>" for SMART_LLM / FAST_LLM / STRATEGIC_LLM
+_provider = os.environ.get("LLM_PROVIDER", "openai")
+_default_strategic = "gpt-4o"
+_default_smart = "gpt-4o-mini"
+
+# Seed all the variants some releases look for
+def _seed_llm_env(strategic_model: str, smart_model: str, provider: str = _provider):
+    strategic = f"{provider}:{strategic_model}"
+    smart = f"{provider}:{smart_model}"
+    # Required (newer versions check these):
+    os.environ["STRATEGIC_LLM"] = strategic
+    os.environ["SMART_LLM"] = smart
+    os.environ["FAST_LLM"] = smart  # alias some builds use
+
+    # Back-compat aliases some releases read:
+    os.environ["STRATEGY_LLM"] = strategic
+    os.environ["STRATEGIC_MODEL"] = strategic_model
+    os.environ["SMART_MODEL"] = smart_model
+
+    # Embeddings (some builds accept both split and combined)
+    os.environ["EMBEDDING"] = f"{os.environ.get('EMBEDDING_PROVIDER','openai')}:{os.environ.get('EMBEDDING_MODEL','text-embedding-3-small')}"
+
+_seed_llm_env(_default_strategic, _default_smart)
 
 # Allow asyncio.run inside Streamlit
 nest_asyncio.apply()
 
+
 # -------------------------
 # Small helpers
 # -------------------------
 def _apply_model_env(strategic_model: str, smart_model: str):
-    """Apply model choices to environment for gpt_researcher."""
-    for k in ("STRATEGIC_LLM", "STRATEGIC_MODEL", "STRATEGY_LLM", "STRATEGY_MODEL"):
-        os.environ[k] = strategic_model
-    for k in ("SMART_LLM", "SMART_MODEL"):
-        os.environ[k] = smart_model
+    """Apply model choices in the provider-qualified format required by gpt_researcher."""
+    _seed_llm_env(strategic_model, smart_model, provider=os.environ.get("LLM_PROVIDER", "openai"))
 
 def _clean_logs(text: str) -> str:
     """Optionally hide noisy lines about unavailable models, keep everything else."""
@@ -185,7 +200,7 @@ report_type = st.sidebar.selectbox(
     help="Choose the format of the final report.",
 )
 
-# Model choices (so you never hit `o1-preview`)
+# Model choices (ensure we never hit `o1-preview`)
 with st.sidebar.expander("Model Settings", expanded=False):
     strategic_choice = st.selectbox(
         "Strategic model",
@@ -238,7 +253,7 @@ if run_clicked:
         # Retriever back-end (Tavily)
         os.environ["RETRIEVER"] = "tavily"
 
-        # Apply model selections so gpt_researcher never tries `o1-preview`
+        # Apply model selections so gpt_researcher gets "<provider>:<model>"
         _apply_model_env(strategic_choice, smart_choice)
 
         # Decide the report source