| import os | |
| from google import genai | |
| def test_gemini(): | |
| api_key = os.environ.get("GEMINI_API_KEY", "") | |
| print(f"GEMINI_API_KEY from env: '{api_key[:5]}...' (len={len(api_key)})") | |
| try: | |
| # Initialize client (it will look at GEMINI_API_KEY automatically) | |
| client = genai.Client() | |
| # Test basic embed content | |
| response = client.models.embed_content( | |
| model="text-embedding-004", | |
| contents="Hello World" | |
| ) | |
| print("Success! Generated embedding:") | |
| print(f"Vector length: {len(response.embeddings[0].values)}") | |
| except Exception as e: | |
| print(f"Failed to generate embedding with default client: {e}") | |
| if __name__ == "__main__": | |
| test_gemini() | |