27 lines
963 B
Python
27 lines
963 B
Python
import os
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
print('Testing environment variables:')
|
|
print(f'OPENAI_API_KEY: {os.getenv("OPENAI_API_KEY")[:20]}...' if os.getenv('OPENAI_API_KEY') else 'OPENAI_API_KEY not set')
|
|
print(f'OPENAI_API_BASE: {os.getenv("OPENAI_API_BASE")}')
|
|
print(f'LLM_BINDING_API_KEY: {os.getenv("LLM_BINDING_API_KEY")[:20]}...' if os.getenv('LLM_BINDING_API_KEY') else 'LLM_BINDING_API_KEY not set')
|
|
print(f'LLM_BINDING_HOST: {os.getenv("LLM_BINDING_HOST")}')
|
|
|
|
# Test the OpenAI client with the current configuration
|
|
from openai import OpenAI
|
|
|
|
client = OpenAI(
|
|
api_key=os.getenv('OPENAI_API_KEY'),
|
|
base_url=os.getenv('OPENAI_API_BASE')
|
|
)
|
|
|
|
try:
|
|
response = client.chat.completions.create(
|
|
model='deepseek-chat',
|
|
messages=[{'role': 'user', 'content': 'Hello, test'}],
|
|
max_tokens=10
|
|
)
|
|
print(f'Success! Response: {response.choices[0].message.content}')
|
|
except Exception as e:
|
|
print(f'Error: {e}') |