feat(rule): add rule api (#46)

Author: ch-liuzhide

server/api/rule/router.py

@@ -0,0 +1,64 @@
+from core.auth import get_tenant
+from core.log import logger
+from core.plugin_manager import PluginManager
+from core.response import ResponseModel
+from fastapi import APIRouter, Depends, HTTPException, Path
+from whiskerrag_types.model import Tenant, GlobalRule
+
+router = APIRouter(
+    prefix="/api/rule",
+    tags=["rule"],
+    responses={404: {"description": "Not found"}},
+    dependencies=[Depends(get_tenant)],
+)
+
+
+@router.get("/global", operation_id="get_global_rule", response_model_by_alias=False)
+async def get_global_rule(
+    tenant: Tenant = Depends(get_tenant),
+) -> ResponseModel[GlobalRule]:
+    try:
+        db_engine = PluginManager().dbPlugin
+        rule = await db_engine.get_tenant_rule(tenant.tenant_id)
+        if not rule:
+            logger.error(
+                "[get_tenant_rule][tenant rule not exists],tenant_id={}".format(
+                    tenant.tenant_id
+                )
+            )
+            raise HTTPException(status_code=404, detail="knowledge base not exists")
+        return ResponseModel(data=rule, success=True)
+    except HTTPException as e:
+        raise e
+    except Exception as e:
+        logger.error(f"[get_tenant_rule][error] error={str(e)}")
+        raise HTTPException(status_code=500, detail="Failed to get tenant global rule")
+
+
+@router.get(
+    "/space/{space_id}/rule",
+    operation_id="get_space_rule",
+    response_model_by_alias=False,
+)
+async def get_space_rule(
+    space_id: str = Path(..., description="knowledge base id"),
+    tenant: Tenant = Depends(get_tenant),
+) -> ResponseModel[GlobalRule]:
+    try:
+        db_engine = PluginManager().dbPlugin
+        rule = await db_engine.get_space_rule(tenant.tenant_id, space_id)
+        if not rule:
+            logger.error(
+                "[get_space_rule][space rule not exists],tenant_id={}".format(
+                    tenant.tenant_id
+                )
+            )
+            raise HTTPException(
+                status_code=404, detail="knowledge base rule not exists"
+            )
+        return ResponseModel(data=rule, success=True)
+    except HTTPException as e:
+        raise e
+    except Exception as e:
+        logger.error(f"[get_space_rule][error] error={str(e)}")
+        raise HTTPException(status_code=500, detail="Failed to get knowledge base rule")

server/plugins/db_engine/supabase_client.py

@@ -20,6 +20,9 @@
     Space,
     KnowledgeSourceEnum,
     TaskStatus,
+    GlobalRule,
+    SpaceRule,
+    Wiki,
 )
 from whiskerrag_types.model.page import QueryParams
 from whiskerrag_utils import RegisterTypeEnum, get_register
@@ -562,3 +565,24 @@ async def retrieve(
         params: RetrievalRequest,
     ) -> List[RetrievalChunk]:
         pass
+
+    # =================== rule ===================
+    async def create_rule(
+        self, tenant_id: str, rule: Union[GlobalRule, SpaceRule]
+    ) -> None:
+        pass
+
+    async def update_rule(
+        self, tenant_id: str, rule_id: str, rule: Union[GlobalRule, SpaceRule]
+    ) -> Union[GlobalRule, SpaceRule]:
+        pass
+
+    async def get_tenant_rule(self, tenant_id: str) -> Optional[str]:
+        return
+
+    async def get_space_rule(self, space_id: str) -> Optional[str]:
+        pass
+
+    # =================== wiki ===================
+    async def create_wiki(self, wiki: Wiki) -> None:
+        pass