feat: add GetModelAPI for retrieving model information in knowledge base

Author: liuruibin

apps/knowledge/api/knowledge.py

@@ -248,3 +248,21 @@ def get_request():
 
 class EmbeddingAPI(SyncWebAPI):
     pass
+
+
+class GetModelAPI(SyncWebAPI):
+    @staticmethod
+    def get_parameters():
+        return [
+            OpenApiParameter(
+                name="workspace_id",
+                description="工作空间id",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            ),
+        ]
+
+    @staticmethod
+    def get_response():
+        return DefaultResultSerializer

apps/knowledge/urls.py

@@ -7,6 +7,7 @@
     path('workspace/<str:workspace_id>/knowledge', views.KnowledgeView.as_view()),
     path('workspace/<str:workspace_id>/knowledge/base', views.KnowledgeBaseView.as_view()),
     path('workspace/<str:workspace_id>/knowledge/web', views.KnowledgeWebView.as_view()),
+    path('workspace/<str:workspace_id>/knowledge/model', views.KnowledgeView.Model.as_view()),
     path('workspace/<str:workspace_id>/knowledge/<str:knowledge_id>', views.KnowledgeView.Operate.as_view()),
     path('workspace/<str:workspace_id>/knowledge/<str:knowledge_id>/sync', views.KnowledgeView.SyncWeb.as_view()),
     path('workspace/<str:workspace_id>/knowledge/<str:knowledge_id>/generate_related', views.KnowledgeView.GenerateRelated.as_view()),

apps/knowledge/views/knowledge.py

@@ -8,8 +8,10 @@
 from common.constants.permission_constants import PermissionConstants
 from common.result import result
 from knowledge.api.knowledge import KnowledgeBaseCreateAPI, KnowledgeWebCreateAPI, KnowledgeTreeReadAPI, \
-    KnowledgeEditAPI, KnowledgeReadAPI, KnowledgePageAPI, SyncWebAPI, GenerateRelatedAPI, HitTestAPI, EmbeddingAPI
+    KnowledgeEditAPI, KnowledgeReadAPI, KnowledgePageAPI, SyncWebAPI, GenerateRelatedAPI, HitTestAPI, EmbeddingAPI, \
+    GetModelAPI
 from knowledge.serializers.knowledge import KnowledgeSerializer
+from models_provider.serializers.model_serializer import ModelSerializer
 
 
 class KnowledgeView(APIView):
@@ -193,11 +195,33 @@ class GenerateRelated(APIView):
             responses=GenerateRelatedAPI.get_response(),
             tags=[_('Knowledge Base')]  # type: ignore
         )
+        @has_permissions(PermissionConstants.KNOWLEDGE_EDIT.get_workspace_permission())
         def put(self, request: Request, workspace_id: str, knowledge_id: str):
             return result.success(KnowledgeSerializer.Operate(
                 data={'knowledge_id': knowledge_id, 'workspace_id': workspace_id, 'user_id': request.user.id}
             ).generate_related(request.data))
 
+    class Model(APIView):
+        authentication_classes = [TokenAuth]
+
+        @extend_schema(
+            methods=['GET'],
+            summary=_('Get model for knowledge base'),
+            description=_('Get model for knowledge base'),
+            operation_id=_('Get model for knowledge base'),  # type: ignore
+            parameters=GetModelAPI.get_parameters(),
+            responses=GetModelAPI.get_response(),
+            tags=[_('Knowledge Base')]  # type: ignore
+        )
+        @has_permissions(PermissionConstants.KNOWLEDGE_EDIT.get_workspace_permission())
+        def get(self, request: Request, workspace_id: str):
+            return result.success(ModelSerializer.Query(
+                data={
+                    'workspace_id': workspace_id,
+                    'model_type': 'LLM'
+                }
+            ).list(workspace_id, True))
+
 
 class KnowledgeBaseView(APIView):
     authentication_classes = [TokenAuth]