提交api

main.py

@@ -1,7 +1,55 @@
+import os
 from fastapi import FastAPI
+import requests
+
+api_key = os.environ.get('api_key')
+initial_prompt = "You are a helpful assistant."
+API_URL = "https://api.openai.com/v1/chat/completions"
 
 app = FastAPI()
 
+
 @app.get("/")
 def read_root():
-    return {"Hello": "World!"}
+    return {"Hello": "World!"}
+
+
+@app.post("/api/chat")
+def chat(input, history: list):
+    history.append(construct_user(input))
+    res = get_response(initial_prompt, history)
+    return res
+
+
+def get_response(system_prompt, history):
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    history = [construct_system(system_prompt), *history]
+
+    payload = {
+        "model": "gpt-3.5-turbo",
+        "messages": history,  # [{"role": "user", "content": f"{inputs}"}],
+    }
+
+    response = requests.post(API_URL, headers=headers,
+                             json=payload, stream=True, timeout=timeout)
+    return response
+
+
+def construct_user(text):
+    return construct_text("user", text)
+
+
+def construct_system(text):
+    return construct_text("system", text)
+
+
+def construct_assistant(text):
+    return construct_text("assistant", text)
+
+
+def construct_text(role, text):
+    return {"role": role, "content": text}

requirements.txt

@@ -1,4 +1,3 @@
 fastapi==0.74.*
 requests==2.27.*
-sentencepiece==0.1.*
 uvicorn[standard]==0.17.*