Преглед на файлове

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(