github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #29 from chuanzhubin/master 

Update my_openai_api.py 忽略客户端传来的未知字段
pull/825/head
jingyaogong 2024-09-15 17:34:57 +08:00 committed by GitHub
parent 64ae2050de 379394c5a4
commit dd8206324f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
my_openai_api.py
View File

@ -14,7 +14,7 @@ from flask_cors import CORS
from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformer
from sklearn.preprocessing import PolynomialFeatures from sklearn.preprocessing import PolynomialFeatures
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers import AutoTokenizer, AutoModelForCausalLM
from marshmallow import validate, Schema, fields from marshmallow import validate, Schema, fields, EXCLUDE
from pydantic import BaseModel from pydantic import BaseModel
warnings.filterwarnings('ignore', category=UserWarning) warnings.filterwarnings('ignore', category=UserWarning)
@ -165,6 +165,9 @@ class ChatMessageSchema(Schema):
class CreateChatCompletionSchema(Schema): class CreateChatCompletionSchema(Schema):
class Meta:
unknown = EXCLUDE # 忽略未知的字段
model = fields.Str(required=True, metadata={"example": "minimind"}) model = fields.Str(required=True, metadata={"example": "minimind"})
messages = fields.List( messages = fields.List(
fields.Nested(ChatMessageSchema), required=True, fields.Nested(ChatMessageSchema), required=True,