TRIEM_AI / debug_2_flash.py
rzvn1's picture
Initial commit of TRIEM AI
3017127
Raw
History Blame Contribute Delete
402 Bytes
import os
from dotenv import load_dotenv
from google import genai
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
client = genai.Client(api_key=api_key)
print("Testing gemini-2.0-flash...")
try:
response = client.models.generate_content(
model="gemini-2.0-flash",
contents="Hello"
)
print(f"Success: {response.text}")
except Exception as e:
print(f"Failed: {e}")