Initial Commit
This commit is contained in:
BIN
app/core/__pycache__/config.cpython-313.pyc
Normal file
BIN
app/core/__pycache__/config.cpython-313.pyc
Normal file
Binary file not shown.
BIN
app/core/__pycache__/deepseek_client.cpython-313.pyc
Normal file
BIN
app/core/__pycache__/deepseek_client.cpython-313.pyc
Normal file
Binary file not shown.
10
app/core/config.py
Normal file
10
app/core/config.py
Normal file
@@ -0,0 +1,10 @@
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
# Load .env variables
|
||||
load_dotenv()
|
||||
|
||||
# Read environment variables with defaults
|
||||
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
||||
OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-3.5-turbo")
|
||||
OPENAI_API_BASE = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
|
||||
27
app/core/deepseek_client.py
Normal file
27
app/core/deepseek_client.py
Normal file
@@ -0,0 +1,27 @@
|
||||
# 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
|
||||
|
||||
# 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)
|
||||
|
||||
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
|
||||
)
|
||||
|
||||
return response.choices[0].message.content
|
||||
Reference in New Issue
Block a user