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

…2309)

Co-authored-by: cinjospeh <[email protected]>

dbgpt/agent/resource/knowledge.py

@@ -38,6 +38,16 @@ def name(self) -> str:
         """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."""

dbgpt/serve/agent/resource/knowledge.py

@@ -73,6 +73,24 @@ def __init__(self, name: str, space_name: str, context: Optional[dict] = None):
         )
         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 _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