Merge pull request #29 from chuanzhubin/master
Update my_openai_api.py 忽略客户端传来的未知字段
This commit is contained in:
commit
30b498f0c3
@ -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,
|
||||
|
Loading…
x
Reference in New Issue
Block a user