mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-16 22:51:24 +00:00
fix(core): Fix AWEL branch bug (#1640)
This commit is contained in:
@@ -17,7 +17,7 @@ class CrossEncoderRerankEmbeddings(BaseModel, RerankEmbeddings):
|
||||
|
||||
client: Any #: :meta private:
|
||||
model_name: str = "BAAI/bge-reranker-base"
|
||||
max_length: int = None # type: ignore
|
||||
max_length: Optional[int] = None
|
||||
"""Max length for input sequences. Longer sequences will be truncated. If None, max
|
||||
length of the model will be used"""
|
||||
"""Model name to use."""
|
||||
|
@@ -29,9 +29,10 @@ class EmbeddingRetrieverOperator(RetrieverOperator[Union[str, List[str]], List[C
|
||||
parameters=[
|
||||
Parameter.build_from(
|
||||
_("Storage Index Store"),
|
||||
"vector_store_connector",
|
||||
"index_store",
|
||||
IndexStoreBase,
|
||||
description=_("The vector store connector."),
|
||||
alias=["vector_store_connector"],
|
||||
),
|
||||
Parameter.build_from(
|
||||
_("Top K"),
|
||||
@@ -128,9 +129,10 @@ class EmbeddingAssemblerOperator(AssemblerOperator[Knowledge, List[Chunk]]):
|
||||
parameters=[
|
||||
Parameter.build_from(
|
||||
_("Vector Store Connector"),
|
||||
"vector_store_connector",
|
||||
"index_store",
|
||||
IndexStoreBase,
|
||||
description=_("The vector store connector."),
|
||||
alias=["vector_store_connector"],
|
||||
),
|
||||
Parameter.build_from(
|
||||
_("Chunk Parameters"),
|
||||
|
Reference in New Issue
Block a user