limcheekin commited on
Commit
df03c96
1 Parent(s): b92d070

bug: fixed warning 'schema_extra' has been renamed to 'json_schema_extra'

Browse files
Files changed (1) hide show
  1. open/text/embeddings/server/app.py +5 -5
open/text/embeddings/server/app.py CHANGED
@@ -42,7 +42,7 @@ class CreateEmbeddingRequest(BaseModel):
42
  user: Optional[str]
43
 
44
  class Config:
45
- schema_extra = {
46
  "example": {
47
  "input": "The food was delicious and the waiter...",
48
  }
@@ -82,12 +82,12 @@ def _create_embedding(
82
  encode_kwargs=encode_kwargs)
83
  elif model_name.startswith("BAAI/bge-") and model_name.endswith("-en"):
84
  embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
85
- query_instruction=BGE_EN_QUERY_INSTRUCTION,
86
- encode_kwargs=encode_kwargs)
87
  elif model_name.startswith("BAAI/bge-") and model_name.endswith("-zh"):
88
  embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
89
- query_instruction=BGE_ZH_QUERY_INSTRUCTION,
90
- encode_kwargs=encode_kwargs)
91
  else:
92
  embeddings = HuggingFaceEmbeddings(
93
  model_name=model_name, encode_kwargs=encode_kwargs)
 
42
  user: Optional[str]
43
 
44
  class Config:
45
+ json_schema_extra = {
46
  "example": {
47
  "input": "The food was delicious and the waiter...",
48
  }
 
82
  encode_kwargs=encode_kwargs)
83
  elif model_name.startswith("BAAI/bge-") and model_name.endswith("-en"):
84
  embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
85
+ query_instruction=BGE_EN_QUERY_INSTRUCTION,
86
+ encode_kwargs=encode_kwargs)
87
  elif model_name.startswith("BAAI/bge-") and model_name.endswith("-zh"):
88
  embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
89
+ query_instruction=BGE_ZH_QUERY_INSTRUCTION,
90
+ encode_kwargs=encode_kwargs)
91
  else:
92
  embeddings = HuggingFaceEmbeddings(
93
  model_name=model_name, encode_kwargs=encode_kwargs)