mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 20:26:15 +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 the resource name."""
|
||||||
return self._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
|
@property
|
||||||
def retriever(self) -> "BaseRetriever":
|
def retriever(self) -> "BaseRetriever":
|
||||||
"""Return the retriever."""
|
"""Return the retriever."""
|
||||||
|
@ -73,6 +73,24 @@ class KnowledgeSpaceRetrieverResource(RetrieverResource):
|
|||||||
)
|
)
|
||||||
super().__init__(name, retriever=retriever)
|
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
|
@classmethod
|
||||||
def resource_parameters_class(
|
def resource_parameters_class(
|
||||||
cls, **kwargs
|
cls, **kwargs
|
||||||
@ -102,3 +120,15 @@ class KnowledgeSpaceRetrieverResource(RetrieverResource):
|
|||||||
)
|
)
|
||||||
|
|
||||||
return _DynamicKnowledgeSpaceLoadResourceParameters
|
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