github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 23:28:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fmt(knowledge): reformat retriever resource code

Browse Source 
- Adjust import order for better organization
- Improve code readability with proper line breaks and indentation
- Remove unnecessary empty lines
This commit is contained in:
tam 2025-04-19 17:48:54 +08:00
parent fbfc972908
commit b0ef14c111
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py
View File

@ -7,9 +7,9 @@ import cachetools
from dbgpt._private.config import Config
from dbgpt.core import Chunk
from dbgpt.util.cache_utils import cached
from dbgpt.rag.retriever.rerank import RerankEmbeddingsRanker
from dbgpt.rag.embedding.embedding_factory import RerankEmbeddingFactory
from dbgpt.rag.retriever.rerank import RerankEmbeddingsRanker
from dbgpt.util.cache_utils import cached
from .base import Resource, ResourceParameters, ResourceType
@ -19,6 +19,7 @@ if TYPE_CHECKING:
CFG = Config()
@dataclasses.dataclass
class RetrieverResourceParameters(ResourceParameters):
"""Retriever resource parameters."""
@ -37,11 +38,11 @@ class RetrieverResource(Resource[ResourceParameters]):
self._name = name
self._retriever = retriever
app_config = CFG.SYSTEM_APP.config.configs.get("app_config")
rerank_embeddings = RerankEmbeddingFactory.get_instance(
CFG.SYSTEM_APP
).create()
rerank_embeddings = RerankEmbeddingFactory.get_instance(CFG.SYSTEM_APP).create()
self.need_rerank = bool(app_config.models.rerankers)
self.reranker = RerankEmbeddingsRanker(rerank_embeddings, topk=app_config.rag.rerank_top_k)
self.reranker = RerankEmbeddingsRanker(
rerank_embeddings, topk=app_config.rag.rerank_top_k
)
@property
def name(self) -> str: