35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import os
|
|
from openai import OpenAI
|
|
from dotenv import load_dotenv
|
|
|
|
# Load env
|
|
load_dotenv()
|
|
|
|
key = os.getenv("GOOGLE_API_KEY")
|
|
print(f"Checking GOOGLE_API_KEY...")
|
|
if not key:
|
|
print("❌ GOOGLE_API_KEY not found in environment or .env file.")
|
|
exit(1)
|
|
|
|
print(f"✅ Key found: {key[:4]}...{key[-4:]}")
|
|
|
|
client = OpenAI(
|
|
api_key=key,
|
|
base_url="https://generativelanguage.googleapis.com/v1beta/openai/"
|
|
)
|
|
|
|
print("Attempting to generate embedding for 'Hello World'...")
|
|
try:
|
|
resp = client.embeddings.create(
|
|
model="text-embedding-004",
|
|
input="Hello world"
|
|
)
|
|
print("✅ Embedding Success! The API Key is valid and the model is accessible.")
|
|
print(f"Embedding vector length: {len(resp.data[0].embedding)}")
|
|
except Exception as e:
|
|
print(f"❌ Embedding Failed: {e}")
|
|
print("\nTroubleshooting:")
|
|
print("1. Ensure the API Key is from Google AI Studio (aistudio.google.com).")
|
|
print("2. Ensure the 'Generative Language API' is enabled in Google Cloud Console if using a GCP project.")
|
|
print("3. Verify you have not exceeded your quota.")
|