diff --git a/main.py b/main.py index 9239202..b91395f 100644 --- a/main.py +++ b/main.py @@ -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",