fix (deepseek client): fix deepseek api base model api
This commit is contained in:
Binary file not shown.
@@ -1,27 +1,35 @@
|
||||
# app/services/openai_service.py
|
||||
import openai
|
||||
from openai import OpenAI
|
||||
from app.core.config import OPENAI_API_KEY, OPENAI_MODEL
|
||||
from app.core.config import OPENAI_API_BASE
|
||||
import openai
|
||||
from app.core.config import OPENAI_API_KEY, OPENAI_MODEL, OPENAI_API_BASE
|
||||
|
||||
# Initialize OpenAI client with DeepSeek's API
|
||||
DEEPSEEK_API_BASE = "https://api.deepseek.com/v1"
|
||||
DEEPSEEK_MODEL = "deepseek-chat"
|
||||
|
||||
# Set OpenAI API key from the environment
|
||||
openai.api_key = OPENAI_API_KEY
|
||||
openai.api_base = OPENAI_API_BASE
|
||||
|
||||
print(openai.api_key)
|
||||
print(OPENAI_MODEL)
|
||||
print(OPENAI_API_BASE)
|
||||
openai.api_base = DEEPSEEK_API_BASE
|
||||
|
||||
async def chat_with_openai(messages: list):
|
||||
# Use the model from environment variable or fallback to default
|
||||
model = OPENAI_MODEL
|
||||
client = OpenAI(api_key=openai.api_key, base_url=openai.api_base)
|
||||
|
||||
response = client.chat.completions.create(
|
||||
model="deepseek-chat", # Or the model you want
|
||||
messages=messages, # Update this according to the new API syntax
|
||||
max_tokens=100, # Example parameter
|
||||
stream=False
|
||||
client = OpenAI(
|
||||
api_key=openai.api_key,
|
||||
base_url=DEEPSEEK_API_BASE
|
||||
)
|
||||
|
||||
return response.choices[0].message.content
|
||||
|
||||
try:
|
||||
response = client.chat.completions.create(
|
||||
model=DEEPSEEK_MODEL,
|
||||
messages=messages,
|
||||
max_tokens=1000,
|
||||
temperature=0.7,
|
||||
stream=False
|
||||
)
|
||||
|
||||
if response.choices and len(response.choices) > 0:
|
||||
return response.choices[0].message.content
|
||||
return "No response from the model"
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error calling DeepSeek API: {str(e)}")
|
||||
raise
|
||||
|
||||
Reference in New Issue
Block a user