fix(core): Fix AWEL branch bug (#1640)

This commit is contained in:
Fangyin Cheng
2024-06-18 11:11:43 +08:00
committed by GitHub
parent 49b56b4576
commit ace169ac46
32 changed files with 870 additions and 481 deletions

View File

@@ -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."""

View File

@@ -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"),