From e6c5e89f09637b8d9ebca6895d781663f12646d6 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 17 十二月 2024 16:43:55 +0800
Subject: [PATCH] dify 文档智能
---
app/api/knowledge.py | 39 ++++++++++++++++++++++++++-------------
1 files changed, 26 insertions(+), 13 deletions(-)
diff --git a/app/api/knowledge.py b/app/api/knowledge.py
index 625d591..c50464f 100644
--- a/app/api/knowledge.py
+++ b/app/api/knowledge.py
@@ -2,28 +2,41 @@
from fastapi import APIRouter, Depends, Query, HTTPException
from app.api import Response, get_current_user, ResponseList
-from app.models import RoleParameter, GroupModel, KnowledgeModel
+from app.models import klgParameter
from app.models.base_model import get_db
from app.models.user_model import UserModel
-from app.service.knowledge import get_knowledge_list
+from app.service.knowledge import get_knowledge_list, create_knowledge_service
from typing import Optional
knowledge_router = APIRouter()
@knowledge_router.get("/list", response_model=Response)
async def knowledge_list(
+ current: int,
+ pageSize: int,
+ keyword: str="",
+ status: str="",
+ knowledge: str="1",
+ location: str="",
current_user: UserModel = Depends(get_current_user),
- page: Optional[int] = Query(None, description="Page number", gt=0),
- num: Optional[int] = Query(18, description="Number of items per page", gt=0),
db=Depends(get_db)):
- getknowledgelist = await get_knowledge_list(db, current_user.id)
- if page is None:
- return Response(code=200, msg="", data={"rows": getknowledgelist["rows"][0:num]})
+ if current and not pageSize:
+ return ResponseList(code=400, msg="缂哄皯鍙傛暟")
+ getknowledgelist = await get_knowledge_list(db, current_user.id, keyword, pageSize, current, status, knowledge, location)
- start = (page - 1) * num
- end = start + num
+ return Response(code=200, msg="", data=getknowledgelist)
- if start >= len(getknowledgelist):
- raise HTTPException(status_code=404, detail="Page not found")
- print(getknowledgelist)
- return Response(code=200, msg="", data={"rows": getknowledgelist["rows"][start:end]})
\ No newline at end of file
+@knowledge_router.post("/create", response_model=Response)
+async def create_knowledge_api(klg: klgParameter, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
+ is_create = await create_knowledge_service(db, klg.id, klg.name, klg.description, klg.icon, klg.klgType, current_user.id)
+ if not is_create:
+ return Response(code=500, msg="role create failure", data={})
+ return Response(code=200, msg="role create success", data={})
+
+
+@knowledge_router.get("/update", response_model=Response)
+async def change_knowledge_api(knowledgeId: str, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
+ # is_create = await update_dialog_status_service(db, dialog.id, dialog.status)
+ # if not is_create:
+ # return Response(code=500, msg="dialog update failure", data={})
+ return Response(code=200, msg="dialog update success", data={})
\ No newline at end of file
--
Gitblit v1.8.0