pull/21617/head
Bowen Liang 11 months ago
parent d6e60e6ce9
commit bf969f4f01

@ -112,13 +112,13 @@ class OceanBaseVector(BaseVector):
params=DEFAULT_OCEANBASE_HNSW_BUILD_PARAM, params=DEFAULT_OCEANBASE_HNSW_BUILD_PARAM,
) )
fts_idxs: Optional[list[FtsIndexParam]] fts_idxs: Optional[list[FtsIndexParam]] = None
if self._hybrid_search_enabled: if self._hybrid_search_enabled:
fts_idxs = [FtsIndexParam(index_name="fulltext_index_for_col_text", fts_idxs = [
field_names=["text"], FtsIndexParam(
parser_type=FtsParser.IK)] index_name="fulltext_index_for_col_text", field_names=["text"], parser_type=FtsParser.IK
else: )
fts_idxs = None ]
self._client.create_table_with_index_params( self._client.create_table_with_index_params(
table_name=self._collection_name, table_name=self._collection_name,
@ -158,12 +158,10 @@ class OceanBaseVector(BaseVector):
batch_ids = ids[i : i + batch_size] batch_ids = ids[i : i + batch_size]
batch_docs = documents[i : i + batch_size] batch_docs = documents[i : i + batch_size]
batch_embs = embeddings[i : i + batch_size] batch_embs = embeddings[i : i + batch_size]
batch_data = [{ batch_data = [
"id": id, {"id": id, "vector": emb, "text": doc.page_content, "metadata": doc.metadata}
"vector": emb, for id, doc, emb in zip(batch_ids, batch_docs, batch_embs)
"text": doc.page_content, ]
"metadata": doc.metadata
} for id, doc, emb in zip(batch_ids, batch_docs, batch_embs)]
self._client.insert( self._client.insert(
table_name=self._collection_name, table_name=self._collection_name,

Loading…
Cancel
Save