新增对reload信号的处理,实现对systemctl reload的响应

This commit is contained in:
高子兴 2024-10-26 15:02:43 +08:00
parent d661cad5ca
commit 97eef02a92

32
main.py
View File

@ -3,14 +3,36 @@ from fastapi.responses import StreamingResponse, HTMLResponse
from typing import List
import base64
import openai
import signal
app = FastAPI()
with open("key", "r") as k:
key = k.read()
key = ""
prompt = ""
def get_key():
with open("key", "r") as f:
k = f.read()
return k
def get_prompt():
with open("prompt", "r", encoding="utf-8") as f:
p = f.read()
return p
def load_config():
global key, prompt
key = get_key()
prompt = get_prompt()
signal.signal(signal.SIGHUP, load_config())
load_config()
client = openai.OpenAI(api_key=key,
base_url="https://open.bigmodel.cn/api/paas/v4/")
with open("prompt", "r", encoding="utf-8") as p:
prompt = p.read()
# WebSocket连接管理器
@ -67,7 +89,7 @@ async def predict(file: UploadFile = File(...)):
"content": [
{
"type": "text",
"text": prompt,
"text": get_prompt(),
},
{
"type": "image_url",