Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 03:32:23 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

feat: use throwing exceptions instead of program exit and update the default the topk value.

Browse Source
This commit is contained in:
hwzhuhao 2024-10-30 09:58:47 +08:00
parent eb87e690ed
commit 719caccb72
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py
View File

@ -104,8 +104,8 @@ class OceanBaseVector(BaseVector):
val = int(row[6])
vals.append(val)
if len(vals) == 0:
print("ob_vector_memory_limit_percentage not found in parameters.")
exit(1)
logger.exception("ob_vector_memory_limit_percentage not found in parameters.")
raise Exception("ob_vector_memory_limit_percentage not found in parameters.")
if any(val == 0 for val in vals):
try:
self._client.perform_raw_text_sql("ALTER SYSTEM SET ob_vector_memory_limit_percentage = 30")
@ -157,7 +157,7 @@ class OceanBaseVector(BaseVector):
if ef_search != self._hnsw_ef_search:
self._client.set_ob_hnsw_ef_search(ef_search)
self._hnsw_ef_search = ef_search
topk = kwargs.get("top_k", 10)
topk = kwargs.get("top_k", 4)
cur = self._client.ann_search(
table_name=self._collection_name,
vec_column_name="vector",