mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 12:21:08 +00:00
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:
parent
e0081e6b0a
commit
8bb2a15f05
@ -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."""
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user