jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add pgvecto_rs support and upgrade SQLAlchemy (#3833)

Browse source
This commit is contained in:
Jyong 2024-04-29 11:58:17 +08:00 • committed by GitHub
commit 3e9dbe3e0a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
26 changed files with 584 additions and 220 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/controllers/console/datasets/datasets.py
View file

@ -476,7 +476,7 @@ class DatasetRetrievalSettingApi(Resource):
@account_initialization_required
def get(self):
vector_type = current_app.config['VECTOR_STORE']
if vector_type == 'milvus' or vector_type == 'relyt':
if vector_type == 'milvus' or vector_type == 'pgvecto_rs' or vector_type == 'relyt':
return {
'retrieval_method': [
'semantic_search'