feature(agent): Knowledge add retriever name&desc getter and stter (#2309)

Co-authored-by: cinjospeh <joseph.cjn@alibaba-inc.com>
This commit is contained in:
cinjoseph 2025-01-23 16:15:30 +08:00 committed by GitHub
parent e0081e6b0a
commit 8bb2a15f05
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 40 additions and 0 deletions

View File

@ -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."""

View File

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