Update src/ingestion.py

src/ingestion.py

@@ -135,9 +135,31 @@ def adaptive_fallback_toc(text: str, model_name: str = "gpt-4o"):
     """
     snippet = text[:7000]
 
+    # ✅ Read GenAI proxy credentials (same JSON used by QA)
+    creds_path = os.path.join(os.path.dirname(__file__), "GEN AI HUB PROXY.json")
+    base_url = ""
+
+    if os.path.exists(creds_path):
+        try:
+            with open(creds_path, "r") as f:
+                creds = json.load(f)
+                # Try all known locations for base URL
+                base_url = (
+                    creds.get("base_url")
+                    or creds.get("serviceurls", {}).get("AI_API_URL", "")
+                    or creds.get("AICORE_BASE_URL", "")
+                )
+        except Exception as e:
+            print(f"⚠️ Could not read GenAI proxy credentials: {e}")
+
+    if not base_url:
+        print("⚠️ Missing AI_API_URL or base_url in credentials — skipping fallback.")
+        return []
+
     try:
         print(f"⚙️ Invoking GenAI proxy for TOC inference using model: {model_name}")
-        proxy_client = get_proxy_client("gen-ai-hub")
+        proxy_client = get_proxy_client("gen-ai-hub", base_url=base_url)
+
         llm = ChatOpenAI(
             proxy_model_name=model_name,
             proxy_client=proxy_client,
@@ -155,7 +177,7 @@ def adaptive_fallback_toc(text: str, model_name: str = "gpt-4o"):
         """
 
         response = llm.invoke(prompt)
-        response_text = response.content if hasattr(response, "content") else str(response)
+        response_text = getattr(response, "content", str(response))
 
         # Extract clean TOC-like lines
         lines = [
@@ -172,6 +194,7 @@ def adaptive_fallback_toc(text: str, model_name: str = "gpt-4o"):
         print(f"⚠️ AI TOC fallback failed via GenAI proxy: {e}")
         return []
 
+
 # ==========================================================
 # 3B️⃣ UNIFIED WRAPPER (Heuristic + AI Hybrid)
 # ==========================================================