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

chore: remove redundant casting flask app config into dict (#2564)

Bowen Liang преди 1 година
родител
ревизия
e352a8ed1b
променени са 3 файла, в които са добавени 5 реда и са изтрити 6 реда
  1. 1 2
      api/commands.py
  2. 2 2
      api/core/rag/datasource/keyword/keyword_factory.py
  3. 2 2
      api/core/rag/datasource/vdb/vector_factory.py

+ 1 - 2
api/commands.py

@@ -1,7 +1,6 @@
 import base64
 import json
 import secrets
-from typing import cast
 
 import click
 from flask import current_app
@@ -132,7 +131,7 @@ def vdb_migrate():
     """
     click.echo(click.style('Start migrate vector db.', fg='green'))
     create_count = 0
-    config = cast(dict, current_app.config)
+    config = current_app.config
     vector_type = config.get('VECTOR_STORE')
     page = 1
     while True:

+ 2 - 2
api/core/rag/datasource/keyword/keyword_factory.py

@@ -1,4 +1,4 @@
-from typing import Any, cast
+from typing import Any
 
 from flask import current_app
 
@@ -14,7 +14,7 @@ class Keyword:
         self._keyword_processor = self._init_keyword()
 
     def _init_keyword(self) -> BaseKeyword:
-        config = cast(dict, current_app.config)
+        config = current_app.config
         keyword_type = config.get('KEYWORD_STORE')
 
         if not keyword_type:

+ 2 - 2
api/core/rag/datasource/vdb/vector_factory.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, cast
+from typing import Any
 
 from flask import current_app
 
@@ -23,7 +23,7 @@ class Vector:
         self._vector_processor = self._init_vector()
 
     def _init_vector(self) -> BaseVector:
-        config = cast(dict, current_app.config)
+        config = current_app.config
         vector_type = config.get('VECTOR_STORE')
 
         if self._dataset.index_struct_dict: