AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-11 18:59:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: better handle error (#11265)

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong 2024-12-03 09:15:38 +08:00 committed by GitHub
parent a86f1eca79
commit e686f12317
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -104,8 +104,7 @@ class OceanBaseVector(BaseVector):
val = int(row[6]) val = int(row[6])
vals.append(val) vals.append(val)
if len(vals) == 0: if len(vals) == 0:
print("ob_vector_memory_limit_percentage not found in parameters.") raise ValueError("ob_vector_memory_limit_percentage not found in parameters.")
exit(1)
if any(val == 0 for val in vals): if any(val == 0 for val in vals):
try: try:
self._client.perform_raw_text_sql("ALTER SYSTEM SET ob_vector_memory_limit_percentage = 30") self._client.perform_raw_text_sql("ALTER SYSTEM SET ob_vector_memory_limit_percentage = 30")
@ -200,10 +199,10 @@ class OceanBaseVectorFactory(AbstractVectorFactory):
return OceanBaseVector( return OceanBaseVector(
collection_name, collection_name,
OceanBaseVectorConfig( OceanBaseVectorConfig(
host=dify_config.OCEANBASE_VECTOR_HOST, host=dify_config.OCEANBASE_VECTOR_HOST or "",
port=dify_config.OCEANBASE_VECTOR_PORT, port=dify_config.OCEANBASE_VECTOR_PORT or 0,
user=dify_config.OCEANBASE_VECTOR_USER, user=dify_config.OCEANBASE_VECTOR_USER or "",
password=(dify_config.OCEANBASE_VECTOR_PASSWORD or ""), password=(dify_config.OCEANBASE_VECTOR_PASSWORD or ""),
database=dify_config.OCEANBASE_VECTOR_DATABASE, database=dify_config.OCEANBASE_VECTOR_DATABASE or "",
), ),
) )