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

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

Browse Source
This commit is contained in:
Bowen Liang 2024-02-27 09:39:26 +08:00 committed by GitHub
parent e55225e2bc
commit e352a8ed1b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/commands.py
View File

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

4
api/core/rag/datasource/keyword/keyword_factory.py
View File

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

4
api/core/rag/datasource/vdb/vector_factory.py
View File

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