SmartInc-API / gemini_client.py
yoursdvniel's picture
Update gemini_client.py
dc0b52b verified
raw
history blame contribute delete
493 Bytes
import os
from google import genai
client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
def ask_gpt(messages):
try:
prompt = "\n".join(
[f"{m['role'].capitalize()}: {m['content']}" for m in messages]
)
response = client.models.generate_content(
model="gemini-2.5-flash",
contents=prompt,
)
return (response.text or "").strip()
except Exception as e:
return f"⚠️ Gemini Error: {str(e)}"