Update main.py

main.py

@@ -99,7 +99,7 @@ class Word_BERT(nn.Module):
         ly_transfer = self.ly_transfer(pool)
         return out,cancer,transfer,ly_transfer
 
-def getChat(text: str, userid: int):
+def getChat(text):
     # while True:
     #     if True:
     # text = input("user:")
@@ -180,34 +180,19 @@ model.eval()
 Allhistory = {}
 print('初始化完成')
 
-# if __name__ == '__main__':
-#     # getChat("测试一下", 0)
-#     # main()
-#     uvicorn.run(app='main:app', host="localhost",
-#                 port=7860, reload=False)
-#     # testFunc()
-
-
-class Items1(BaseModel):
-    context: str
-    userid: int
-    # must: bool
-
-
 import time
 
 lastReplyTime = 0
 
 
 @app.post("/")
-async def get_Chat(item1: Items1):
+async def get_Chat(context):
     global lastReplyTime
     tempReplyTime = int(time.time() * 1000)
     # if tempReplyTime % 10 == 0 or item1.must == True or tempReplyTime - lastReplyTime < 30000:
     # if item1.must == True:
     #     lastReplyTime = tempReplyTime
-    result = getChat(
-        item1.context, item1.userid)
+    result = getChat(context)
     return {"res": result}
 
 @app.get("/hello")