Browse Source
Update with basemodel
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
eqa_insert.py
|
|
@ -25,7 +25,7 @@ class EnhancedQAInsertConfig(BaseModel): |
|
|
""" |
|
|
""" |
|
|
# config for text_splitter |
|
|
# config for text_splitter |
|
|
type: Optional[str] = 'RecursiveCharacter' |
|
|
type: Optional[str] = 'RecursiveCharacter' |
|
|
chunk_size: int = 300 |
|
|
|
|
|
|
|
|
chunk_size: Optional[int] = 300 |
|
|
splitter_kwargs: Optional[Dict[str, Any]] = {} |
|
|
splitter_kwargs: Optional[Dict[str, Any]] = {} |
|
|
# config for sentence_embedding |
|
|
# config for sentence_embedding |
|
|
model: Optional[str] = 'all-MiniLM-L6-v2' |
|
|
model: Optional[str] = 'all-MiniLM-L6-v2' |
|
|
|
