소스 검색

feat(api/core/rag/datasource/vdb/analyticdb/analyticdb_vector.py): Checking config before init analyticdb (#7050)

-LAN- 8 달 전
부모
커밋
df8f8c9a2c
1개의 변경된 파일17개의 추가작업 그리고 1개의 파일을 삭제
  1. 17 1
      api/core/rag/datasource/vdb/analyticdb/analyticdb_vector.py

+ 17 - 1
api/core/rag/datasource/vdb/analyticdb/analyticdb_vector.py

@@ -322,7 +322,23 @@ class AnalyticdbVectorFactory(AbstractVectorFactory):
                 self.gen_index_struct_dict(VectorType.ANALYTICDB, collection_name)
             )
 
-        # TODO handle optional params
+        # handle optional params
+        if dify_config.ANALYTICDB_KEY_ID is None:
+            raise ValueError("ANALYTICDB_KEY_ID should not be None")
+        if dify_config.ANALYTICDB_KEY_SECRET is None:
+            raise ValueError("ANALYTICDB_KEY_SECRET should not be None")
+        if dify_config.ANALYTICDB_REGION_ID is None:
+            raise ValueError("ANALYTICDB_REGION_ID should not be None")
+        if dify_config.ANALYTICDB_INSTANCE_ID is None:
+            raise ValueError("ANALYTICDB_INSTANCE_ID should not be None")
+        if dify_config.ANALYTICDB_ACCOUNT is None:
+            raise ValueError("ANALYTICDB_ACCOUNT should not be None")
+        if dify_config.ANALYTICDB_PASSWORD is None:
+            raise ValueError("ANALYTICDB_PASSWORD should not be None")
+        if dify_config.ANALYTICDB_NAMESPACE is None:
+            raise ValueError("ANALYTICDB_NAMESPACE should not be None")
+        if dify_config.ANALYTICDB_NAMESPACE_PASSWORD is None:
+            raise ValueError("ANALYTICDB_NAMESPACE_PASSWORD should not be None")
         return AnalyticdbVector(
             collection_name,
             AnalyticdbConfig(