File size: 1,500 Bytes
3cdddc1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | #!/usr/bin/env python3
"""Quick test script for Azure OpenAI"""
import os
import sys
from langchain_openai import AzureChatOpenAI
# Get environment variables
api_key = os.getenv("AZURE_OPENAI_API_KEY")
endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
api_version = os.getenv("OPENAI_API_VERSION", "2024-12-01-preview")
model = os.getenv("OPENAI_MODEL", "gpt-4.1")
temperature = float(os.getenv("AZURE_OPENAI_TEMPERATURE", "0.1"))
max_tokens = int(os.getenv("AZURE_OPENAI_MAX_TOKENS", "8192"))
timeout = int(os.getenv("AZURE_OPENAI_TIMEOUT", "600"))
if not api_key or not endpoint:
print("❌ Error: AZURE_OPENAI_API_KEY and AZURE_OPENAI_ENDPOINT are required")
sys.exit(1)
print("🔍 Testing Azure OpenAI connection...")
print(f" Endpoint: {endpoint}")
print(f" Model: {model}")
print(f" API Version: {api_version}")
try:
llm = AzureChatOpenAI(
model=model,
azure_deployment=model,
api_key=api_key,
temperature=temperature,
max_tokens=max_tokens,
timeout=timeout,
max_retries=1,
azure_endpoint=endpoint,
api_version=api_version
)
print("✅ Client initialized successfully")
print("📤 Sending test message...")
response = llm.invoke([{"role": "user", "content": "Say 'Hello' in one word."}])
print(f"✅ Success! Response: {response.content}")
print("🎉 Azure OpenAI is working correctly!")
except Exception as e:
print(f"❌ Error: {e}")
sys.exit(1)
|