File size: 661 Bytes
5d6a9b2 68308d3 41b4bf3 5d6a9b2 68308d3 d9678d2 bed6b6d 68308d3 98a619d 68308d3 bed6b6d 68308d3 98a619d 68308d3 1a9430f 68308d3 14b0947 68308d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import os
import google.generativeai as genai
import asyncio
# Load API key from environment variable
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
if not GOOGLE_API_KEY:
raise ValueError("Google API key not set in the environment variables.")
# Configure Google AI model
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel('gemini-pro')
# Function to query AI model
async def query_ai_model(prompt: str) -> str:
try:
response = model.generate_content(prompt)
return response.text
except Exception as e:
print(f"Error querying AI model: {e}")
return "Sorry, I couldn't process your request."
|