Merge 719caccb72 into 5ff02b469f

feat: use throwing exceptions instead of program exit and update the default the topk value.
2024-11-16 11:42:29 +08:00 · 2024-11-15 08:51:11 +08:00 · 2024-10-30 09:58:47 +08:00
1 changed files with 3 additions and 3 deletions
--- a/api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py
+++ b/api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py
@ -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",
Author	SHA1	Message	Date
zhuhao	2f5f17882e	Merge `719caccb72` into `5ff02b469f`	2024-11-15 08:51:11 +08:00
hwzhuhao	719caccb72	feat: use throwing exceptions instead of program exit and update the default the topk value.	2024-10-30 09:58:47 +08:00