Update openai_client.py

openai_client.py

@@ -1,17 +1,18 @@
 import os
-import google.generativeai as genai
+from google import genai
 
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+# Initialize Gemini client using the correct environment variable
+api_key = os.getenv("GEMINI_API_KEY")
+if not api_key:
+    raise ValueError("Missing GEMINI_API_KEY environment variable")
 
-# ✅ Use the correct model name for v1
-model = genai.GenerativeModel("gemini-pro")  # no "models/" prefix needed
+client = genai.Client(api_key=api_key)
 
-def ask_gpt(messages):
-    try:
-        prompt = "\n".join(
-            [f"{m['role'].capitalize()}: {m['content']}" for m in messages]
-        )
-        response = model.generate_content(prompt)
-        return response.text.strip()
-    except Exception as e:
-        return f"⚠️ Gemini Error: {str(e)}"
+# Default model (same as used in the full system)
+model_name = "gemini-2.0-flash-thinking-exp"
+
+def ask_gpt(messages: list[dict]) -> str:
+    prompt_parts = "\n".join([msg["content"] for msg in messages])
+    chat = client.chats.create(model=model_name)
+    resp = chat.send_message(prompt_parts)
+    return resp.text