Bläddra i källkod

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

tags/0.5.7
Bowen Liang 1 år sedan
förälder
incheckning
e352a8ed1b
Inget konto är kopplat till bidragsgivarens mejladress

+ 1
- 2
api/commands.py Visa fil

@@ -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 Visa fil

@@ -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 Visa fil

@@ -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:

Laddar…
Avbryt
Spara