Update tools.py

tools.py

@@ -1697,21 +1697,21 @@ def _call_gemini_label(prompt: str, api_key: str) -> str:
         return f'[Gemini: {e}]'
 
 
-def _call_deepseek_label(prompt: str, api_key: str,
-                         model: str = 'deepseek-chat') -> str:
-    """DeepSeek API (OpenAI-compatible) — free $5 credit on signup."""
+def _call_ollama_label(prompt: str, base_url: str = 'http://localhost:11434',
+                       model: str = 'llama3') -> str:
+    """Ollama local inference API."""
     try:
+        url = f'{base_url.rstrip("/")}/api/chat'
         r = _httpx.post(
-            'https://api.deepseek.com/v1/chat/completions',
-            headers={'Authorization': f'Bearer {api_key}', 'Content-Type': 'application/json'},
+            url,
             json={'model': model, 'messages': [{'role': 'user', 'content': prompt}],
-                  'max_tokens': 20, 'temperature': 0.2},
-            timeout=25.0,
+                  'stream': False},
+            timeout=30.0,
         )
         r.raise_for_status()
-        return r.json()['choices'][0]['message']['content'].strip()
+        return r.json()['message']['content'].strip()[:80]
     except Exception as e:
-        return f'[DeepSeek: {e}]'
+        return f'[Ollama: {e}]'
 
 
 def _label_prompt(titles: List[str]) -> str:
@@ -1731,10 +1731,10 @@ def label_clusters_3llm(
     embeddings: np.ndarray,
     mistral_api_key: str = '',
     gemini_api_key: str = '',
-    deepseek_api_key: str = '',
+    ollama_url: str = 'http://localhost:11434',
     max_clusters: int = 30,
 ) -> pd.DataFrame:
-    """Label each cluster using Mistral + Gemini + DeepSeek (all free/affordable APIs).
+    """Label each cluster using Mistral + Gemini + Ollama.
 
     For each cluster:
     1. Selects 3 most representative paper titles (closest to centroid).
@@ -1749,7 +1749,7 @@ def label_clusters_3llm(
         embeddings        : SPECTER2 embeddings aligned with cluster_df rows.
         mistral_api_key   : Mistral API key (free tier).
         gemini_api_key    : Google AI Studio API key (free tier).
-        deepseek_api_key  : DeepSeek API key (free $5 credit — deepseek.com).
+        ollama_url        : URL of the local Ollama instance.
         max_clusters      : Cap API calls to this many clusters.
 
     Returns:
@@ -1761,10 +1761,10 @@ def label_clusters_3llm(
         summary['label_mistral'] = ''
     if 'label_gemini' not in summary.columns:
         summary['label_gemini'] = ''
-    if 'label_hf' not in summary.columns:
-        summary['label_hf'] = ''
+    if 'label_ollama' not in summary.columns:
+        summary['label_ollama'] = ''
 
-    has_any_key = any([mistral_api_key.strip(), gemini_api_key.strip(), deepseek_api_key.strip()])
+    has_any_key = any([mistral_api_key.strip(), gemini_api_key.strip(), ollama_url.strip()])
 
     for idx, row in summary.iterrows():
         if idx >= max_clusters:
@@ -1801,10 +1801,10 @@ def label_clusters_3llm(
             candidates.append(gl)
             _time.sleep(0.2)
 
-        if deepseek_api_key.strip():
-            dl = _call_deepseek_label(prompt, deepseek_api_key)
-            summary.at[idx, 'label_deepseek'] = dl
-            candidates.append(dl)
+        if ollama_url.strip():
+            ol = _call_ollama_label(prompt, ollama_url)
+            summary.at[idx, 'label_ollama'] = ol
+            candidates.append(ol)
             _time.sleep(0.2)
 
         summary.at[idx, 'label'] = _majority_label(candidates) if candidates else 'Research Cluster'
@@ -1833,8 +1833,8 @@ Panel A (Mistral) said:
 {mistral_response}
 Panel B (Gemini) said:
 {gemini_response}
-Panel C (DeepSeek) said:
-{deepseek_response}
+Panel C (Ollama) said:
+{ollama_response}
 Your task:
 1. Identify the 2-3 points ALL panels AGREE on (consensus insights)
 2. Identify where they DIVERGE and explain which view is most defensible
@@ -1864,7 +1864,7 @@ def _call_mistral(prompt: str, api_key: str, model: str = 'mistral-large-latest'
         return f'[Mistral unavailable: {e}]'
 
 
-def _call_gemini(prompt: str, api_key: str, model: str = 'gemini-2.0-flash') -> str:
+def _call_gemini(prompt: str, api_key: str, model: str = 'gemini-1.5-flash') -> str:
     try:
         url = (f'https://generativelanguage.googleapis.com/v1beta/models/'
                f'{model}:generateContent?key={api_key}')
@@ -1883,25 +1883,25 @@ def _call_gemini(prompt: str, api_key: str, model: str = 'gemini-2.0-flash') ->
         return f'[Gemini unavailable: {e}]'
 
 
-def _call_deepseek(
+def _call_ollama(
     prompt: str,
-    api_key: str,
-    model: str = 'deepseek-chat',
+    base_url: str = 'http://localhost:11434',
+    model: str = 'llama3',
 ) -> str:
-    """DeepSeek API (OpenAI-compatible endpoint). Uses deepseek-chat (DeepSeek-V3)."""
+    """Ollama API (Local). Uses llama3 by default."""
     try:
+        url = f'{base_url.rstrip("/")}/api/chat'
         r = _httpx.post(
-            'https://api.deepseek.com/v1/chat/completions',
-            headers={'Authorization': f'Bearer {api_key}', 'Content-Type': 'application/json'},
+            url,
             json={'model': model, 'messages': [{'role': 'user', 'content': prompt}],
-                  'max_tokens': 500, 'temperature': 0.4},
-            timeout=30.0,
+                  'stream': False},
+            timeout=120.0,
         )
         r.raise_for_status()
-        return r.json()['choices'][0]['message']['content'].strip()
+        return r.json()['message']['content'].strip()
     except Exception as e:
-        logger.error(f'DeepSeek call failed: {e}')
-        return f'[DeepSeek unavailable: {e}]'
+        logger.error(f'Ollama call failed: {e}')
+        return f'[Ollama unavailable: {e}]'
 
 
 def run_agentic_council(
@@ -1909,13 +1909,13 @@ def run_agentic_council(
     topic_df: Optional[pd.DataFrame],
     mistral_api_key: str = '',
     gemini_api_key: str = '',
-    deepseek_api_key: str = '',
+    ollama_url: str = 'http://localhost:11434',
     anthropic_api_key: str = '',   # kept for backward compat — not used
 ) -> Dict[str, str]:
-    """Run 4-stage agentic council: Mistral → Gemini → DeepSeek panels + DeepSeek synthesis.
+    """Run 4-stage agentic council: Mistral → Gemini → Ollama panels + Ollama synthesis.
 
     Returns dict with keys:
-        'findings_summary', 'mistral', 'gemini', 'deepseek', 'synthesis'
+        'findings_summary', 'mistral', 'gemini', 'ollama', 'synthesis'
     """
     gap = taxonomy_map.get('gap_analysis', {})
     novel_themes = taxonomy_map.get('novel_themes', [])[:5]
@@ -1963,31 +1963,23 @@ def run_agentic_council(
         else '[Gemini API key not provided]'
     )
 
-    # ── Stage 3: DeepSeek panel ──
-    logger.info('Council: calling DeepSeek (Panel C)…')
-    deepseek_resp = (
-        _call_deepseek(council_prompt, deepseek_api_key)
-        if deepseek_api_key.strip()
-        else '[DeepSeek API key not provided]'
-    )
+    # ── Stage 3: Ollama panel ──
+    logger.info('Council: calling Ollama (Panel C)…')
+    ollama_resp = _call_ollama(council_prompt, ollama_url)
 
-    # ── Stage 4: DeepSeek synthesis judge ──
+    # ── Stage 4: Ollama synthesis judge ──
     synthesis_prompt = SYNTHESIS_PROMPT_TEMPLATE.format(
         mistral_response=mistral_resp,
         gemini_response=gemini_resp,
-        deepseek_response=deepseek_resp,
-    )
-    logger.info('Council: calling DeepSeek (synthesis judge)…')
-    synthesis_resp = (
-        _call_deepseek(synthesis_prompt, deepseek_api_key, model='deepseek-reasoner')
-        if deepseek_api_key.strip()
-        else '[DeepSeek API key not provided — synthesis skipped]'
+        ollama_response=ollama_resp,
     )
+    logger.info('Council: calling Ollama (synthesis judge)…')
+    synthesis_resp = _call_ollama(synthesis_prompt, ollama_url, model='llama3')
 
     return {
         'findings_summary': findings,
         'mistral': mistral_resp,
         'gemini': gemini_resp,
-        'deepseek': deepseek_resp,
+        'ollama': ollama_resp,
         'synthesis': synthesis_resp,
     }