Add initial FastAPI application and Dockerfile for chatbot integration
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
from fastapi import FastAPI
|
||||
from openai import OpenAI
|
||||
import os
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
# DeepSeek OpenAI-compatible endpoint
|
||||
client = OpenAI(
|
||||
api_key=os.getenv("DEEPSEEK_API_KEY"),
|
||||
base_url="https://api.deepseek.com"
|
||||
)
|
||||
|
||||
@app.post("/v1/chat/completions")
|
||||
async def chat(data: dict):
|
||||
|
||||
messages = data["messages"]
|
||||
|
||||
repo_context = """
|
||||
This assistant has access to the repository.
|
||||
Help the user understand the codebase.
|
||||
This repo is building an chatbot, currently still in early stages. The main files are:
|
||||
- app.py: The main FastAPI application that defines the API endpoints and integrates with the Deep seek API.
|
||||
- agent.py: Contains the logic for the chatbot agent, including how it processes messages and generates responses.
|
||||
"""
|
||||
|
||||
messages.insert(0, {
|
||||
"role": "system",
|
||||
"content": repo_context
|
||||
})
|
||||
|
||||
response = client.chat.completions.create(
|
||||
model="deepseek-v4-flash",
|
||||
messages=messages,
|
||||
temperature=0.3
|
||||
)
|
||||
|
||||
return response.model_dump()
|
||||
Reference in New Issue
Block a user