diff --git a/dbgpt/agent/resource/knowledge.py b/dbgpt/agent/resource/knowledge.py index ca651d012..24d47dae8 100644 --- a/dbgpt/agent/resource/knowledge.py +++ b/dbgpt/agent/resource/knowledge.py @@ -38,6 +38,16 @@ class RetrieverResource(Resource[ResourceParameters]): """Return the resource name.""" return self._name + @property + def retriever_name(self) -> str: + """Return the resource name.""" + return "" + + @property + def retriever_desc(self) -> str: + """Return the retriever desc.""" + return "" + @property def retriever(self) -> "BaseRetriever": """Return the retriever.""" diff --git a/dbgpt/serve/agent/resource/knowledge.py b/dbgpt/serve/agent/resource/knowledge.py index f74133e80..bc9c69741 100644 --- a/dbgpt/serve/agent/resource/knowledge.py +++ b/dbgpt/serve/agent/resource/knowledge.py @@ -73,6 +73,24 @@ class KnowledgeSpaceRetrieverResource(RetrieverResource): ) super().__init__(name, retriever=retriever) + knowledge_spaces = get_knowledge_spaces_info(id=space_name) + if knowledge_spaces is not None and len(knowledge_spaces) > 0: + self._retriever_name = knowledge_spaces[0].name + self._retriever_desc = knowledge_spaces[0].desc + else: + self._retriever_name = None + self._retriever_desc = None + + @property + def retriever_name(self) -> str: + """Return the resource name.""" + return self._retriever_name + + @property + def retriever_desc(self) -> str: + """Return the retriever desc.""" + return self._retriever_desc + @classmethod def resource_parameters_class( cls, **kwargs @@ -102,3 +120,15 @@ class KnowledgeSpaceRetrieverResource(RetrieverResource): ) return _DynamicKnowledgeSpaceLoadResourceParameters + + +def get_knowledge_spaces_info(**kwargs): + from dbgpt.app.knowledge.request.request import KnowledgeSpaceRequest + from dbgpt.app.knowledge.service import KnowledgeService + + knowledge_space_service = KnowledgeService() + knowledge_spaces = knowledge_space_service.get_knowledge_space( + KnowledgeSpaceRequest(**kwargs) + ) + + return knowledge_spaces