File size: 623 Bytes
2883afc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os
import google.generativeai as genai
def get_gemini_api_key():
return os.getenv("GEMINI_API_KEY")
def run_gemini_chat(prompt, context=None):
api_key = get_gemini_api_key()
if not api_key:
return "Error: Gemini API key not set. Please add it to your environment variables."
genai.configure(api_key=api_key)
model = genai.GenerativeModel("gemini-pro")
chat = model.start_chat()
if context:
prompt = f"{context}\n\n{prompt}"
try:
response = chat.send_message(prompt)
return response.text
except Exception as e:
return f"Gemini Error: {e}"
|