From f5966fe69e306d01dc2ace01e61538c662393838 Mon Sep 17 00:00:00 2001 From: Ben Date: Sun, 15 Sep 2024 07:37:24 +0000 Subject: [PATCH] =?UTF-8?q?Update=20my=5Fopenai=5Fapi.py=20=E5=BF=BD?= =?UTF-8?q?=E7=95=A5=E5=AE=A2=E6=88=B7=E7=AB=AF=E4=BC=A0=E6=9D=A5=E7=9A=84?= =?UTF-8?q?=E6=9C=AA=E7=9F=A5=E5=AD=97=E6=AE=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 像dify这样的客户端,会传入一个用户序列号, 需要兼容一下,避免报错. ``` request.json={'model': 'minimind-small-T', 'stream': True, 'temperature': 0.7, 'messages': [{'role': 'user', 'content': '1'}], 'user': 'cdc1f28b-4bb3-4326-ac4b-1cf78e0a6a3b'} ``` --- my_openai_api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/my_openai_api.py b/my_openai_api.py index 098c36f..3d170d7 100644 --- a/my_openai_api.py +++ b/my_openai_api.py @@ -14,7 +14,7 @@ from flask_cors import CORS from sentence_transformers import SentenceTransformer from sklearn.preprocessing import PolynomialFeatures from transformers import AutoTokenizer, AutoModelForCausalLM -from marshmallow import validate, Schema, fields +from marshmallow import validate, Schema, fields, EXCLUDE from pydantic import BaseModel warnings.filterwarnings('ignore', category=UserWarning) @@ -165,6 +165,9 @@ class ChatMessageSchema(Schema): class CreateChatCompletionSchema(Schema): + class Meta: + unknown = EXCLUDE # 忽略未知的字段 + model = fields.Str(required=True, metadata={"example": "minimind"}) messages = fields.List( fields.Nested(ChatMessageSchema), required=True,