From b0ef14c111394727f0a9ce93c9089daa5f24bf45 Mon Sep 17 00:00:00 2001 From: tam Date: Sat, 19 Apr 2025 17:48:54 +0800 Subject: [PATCH] fmt(knowledge): reformat retriever resource code - Adjust import order for better organization - Improve code readability with proper line breaks and indentation - Remove unnecessary empty lines --- .../src/dbgpt/agent/resource/knowledge.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py b/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py index 16f7ff622..82a1f3c43 100644 --- a/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py +++ b/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py @@ -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: