|
|
|
@ -318,6 +318,8 @@ def create_qdrant_indexes():
|
|
|
|
|
|
|
|
|
|
|
|
page += 1
|
|
|
|
page += 1
|
|
|
|
for dataset in datasets:
|
|
|
|
for dataset in datasets:
|
|
|
|
|
|
|
|
if dataset.index_struct_dict:
|
|
|
|
|
|
|
|
if dataset.index_struct_dict['type'] != 'qdrant':
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
click.echo('Create dataset qdrant index: {}'.format(dataset.id))
|
|
|
|
click.echo('Create dataset qdrant index: {}'.format(dataset.id))
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
@ -351,13 +353,13 @@ def create_qdrant_indexes():
|
|
|
|
embeddings=embeddings
|
|
|
|
embeddings=embeddings
|
|
|
|
)
|
|
|
|
)
|
|
|
|
if index:
|
|
|
|
if index:
|
|
|
|
|
|
|
|
index.create_qdrant_dataset(dataset)
|
|
|
|
index_struct = {
|
|
|
|
index_struct = {
|
|
|
|
"type": 'qdrant',
|
|
|
|
"type": 'qdrant',
|
|
|
|
"vector_store": {"class_prefix": dataset.index_struct_dict['vector_store']['class_prefix']}
|
|
|
|
"vector_store": {"class_prefix": dataset.index_struct_dict['vector_store']['class_prefix']}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
dataset.index_struct = json.dumps(index_struct)
|
|
|
|
dataset.index_struct = json.dumps(index_struct)
|
|
|
|
db.session.commit()
|
|
|
|
db.session.commit()
|
|
|
|
index.create_qdrant_dataset(dataset)
|
|
|
|
|
|
|
|
create_count += 1
|
|
|
|
create_count += 1
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
click.echo('passed.')
|
|
|
|
click.echo('passed.')
|
|
|
|
|