Create gemini.py

mcp/gemini.py

@@ -0,0 +1,29 @@
+# mcp/gemini.py
+"""
+Lightweight Gemini-Pro helper (text in → text out).
+Requires env var GEMINI_KEY.
+"""
+
+import os, asyncio, google.generativeai as genai  # SDK  :contentReference[oaicite:1]{index=1}
+
+GEN_KEY = os.getenv("GEMINI_KEY")
+if GEN_KEY:
+    genai.configure(api_key=GEN_KEY)
+
+_MODEL = None
+def _model():
+    global _MODEL
+    if _MODEL is None:
+        _MODEL = genai.GenerativeModel("gemini-pro")  # 32 k ctx ­ :contentReference[oaicite:2]{index=2}
+    return _MODEL
+
+# ---------- public helpers ----------
+async def gemini_summarize(text: str, words: int = 150) -> str:
+    prompt = f"Summarize in ≤{words} words:\n{text[:12000]}"
+    rsp = await asyncio.to_thread(_model().generate_content, prompt)
+    return rsp.text
+
+async def gemini_qa(question: str, context: str = "") -> str:
+    prompt = f"Answer briefly.\nContext:\n{context[:10000]}\n\nQ: {question}\nA:"
+    rsp = await asyncio.to_thread(_model().generate_content, prompt)
+    return rsp.text