26 lines
694 B
Python
26 lines
694 B
Python
import os
|
|
import openai
|
|
from openai import AsyncOpenAI
|
|
|
|
# Set environment variables
|
|
os.environ["OPENAI_API_KEY"] = "sk-55f6e57f1d834b0e93ceaf98cc2cb715"
|
|
os.environ["OPENAI_API_BASE"] = "https://api.deepseek.com/v1"
|
|
|
|
client = AsyncOpenAI(
|
|
api_key=os.environ["OPENAI_API_KEY"],
|
|
base_url=os.environ["OPENAI_API_BASE"]
|
|
)
|
|
|
|
async def test():
|
|
try:
|
|
response = await client.chat.completions.create(
|
|
model="deepseek-chat",
|
|
messages=[{"role": "user", "content": "Hello"}],
|
|
max_tokens=10
|
|
)
|
|
print("Success:", response.choices[0].message.content)
|
|
except Exception as e:
|
|
print("Error:", e)
|
|
|
|
import asyncio
|
|
asyncio.run(test()) |