25 lines
No EOL
587 B
Python
25 lines
No EOL
587 B
Python
from google import genai
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
print(f"API Key set: {bool(os.getenv('GOOGLE_API_KEY'))}")
|
|
|
|
try:
|
|
# Initialize client with API key
|
|
api_key = os.getenv("GOOGLE_API_KEY")
|
|
client = genai.Client(api_key=api_key)
|
|
|
|
# Test connection
|
|
response = client.models.generate_content(
|
|
model='gemini-1.5-pro',
|
|
contents='Test the API connection'
|
|
)
|
|
|
|
print("API connection test successful!")
|
|
print(f"Response: {response.text}")
|
|
|
|
except Exception as e:
|
|
print(f"API error: {e}") |