From fe4955f46667797c4eb32d1c7fe16b0bda798769 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Tue, 24 Jun 2025 19:09:56 +0800
Subject: [PATCH] 'commit'

---
 dsRag/Dao/KbDao.py                            |  13 ++
 dsRag/Dao/__pycache__/KbDao.cpython-310.pyc   | Bin 0 -> 756 bytes
 dsRag/Model/KbModel.py                        |  14 ++
 .../Model/__pycache__/KbModel.cpython-310.pyc | Bin 0 -> 768 bytes
 dsRag/Start.py                                | 188 ++++++++++++------
 5 files changed, 150 insertions(+), 65 deletions(-)
 create mode 100644 dsRag/Dao/KbDao.py
 create mode 100644 dsRag/Dao/__pycache__/KbDao.cpython-310.pyc
 create mode 100644 dsRag/Model/KbModel.py
 create mode 100644 dsRag/Model/__pycache__/KbModel.cpython-310.pyc

diff --git a/dsRag/Dao/KbDao.py b/dsRag/Dao/KbDao.py
new file mode 100644
index 00000000..45c73177
--- /dev/null
+++ b/dsRag/Dao/KbDao.py
@@ -0,0 +1,13 @@
+from typing import List, Optional
+from Model.KbModel import KbModel, KbFileModel
+from Util.MySQLUtil import init_mysql_pool
+
+class KbDao:
+    def __init__(self, mysql_pool):
+        self.mysql_pool = mysql_pool
+
+    async def create_kb(self, kb: KbModel) -> int:
+        # 实现创建知识库的数据库操作
+        pass
+    
+    # 其他CRUD方法实现...
\ No newline at end of file
diff --git a/dsRag/Dao/__pycache__/KbDao.cpython-310.pyc b/dsRag/Dao/__pycache__/KbDao.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..5b3956185250945c7d698a6a304d615efd47431d
GIT binary patch
literal 756
zcmZWn&2AGh5FUGflO}1zfk&vcmplMMRV6Aoqz#1#Az6{EP3)E>-rdx@K}yvFs1a|#
zL-aX#3tu_$3Y?g62t;C}nVp~YeBX>+_xoKy`Sw+O_{0EyP;e*#24|?vDH;We87P@x
zB^R7g%uP^+A}k{j`8Y6fnTUkJD^Q_|wxFU1E>aawAxnN@56HL+FN&sh>FZTn)KzXU
zlZ@w=wbI5Njpr{5qkWJucU)9OE6YuDYvih~&4M}{N{GQ3YIA}{gJ22-S4{COh(HB8
zRH2Tpk%@}8p@>z2JW>~p=jVC7;CN;)<deQYZD=R=>>ebJIE6<xK`PyUENFyAn`_&_
z{~o~};pf-M`Lmg7-q&`CpDSB`)Qc9qxB1Nsv1T4|wA#2-N<Wm035g)P>~63t?@Nv^
z=PtIoUEAtEmUvC@oz7B&FF5cNY>KX@k#S4p?O>nqS=+<Vbrx3Vt(MC<@os2!{By|f
zj(6aHtCK7tmQ`MADP32}vQ}$DeoxBVb#8VQ+b0a_;)0@T?T|uR#epGQo1lqMJ>e8m
z_B8OhN!D?3yIB?0jio(UdTg%m?|$^>LHPcbPP@UowlJg1&GgL$d6vfaeGc$EydO4w
TzGHn#>uP8}Lm09id%}MMaEYP;

literal 0
HcmV?d00001

diff --git a/dsRag/Model/KbModel.py b/dsRag/Model/KbModel.py
new file mode 100644
index 00000000..df15cf51
--- /dev/null
+++ b/dsRag/Model/KbModel.py
@@ -0,0 +1,14 @@
+from pydantic import BaseModel
+from typing import Optional
+
+class KbModel(BaseModel):
+    kb_name: str
+    short_name: str
+    is_delete: Optional[int] = 0
+
+class KbFileModel(BaseModel):
+    file_name: str
+    ext_name: str
+    kb_id: int
+    is_delete: Optional[int] = 0
+    state: Optional[int] = 0
\ No newline at end of file
diff --git a/dsRag/Model/__pycache__/KbModel.cpython-310.pyc b/dsRag/Model/__pycache__/KbModel.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..dfca195d348bc397deb2b44ff240952e37bbc307
GIT binary patch
literal 768
zcmZ{i!A{&T5QfKboH)yF*~4D?2+M(cs}(}1q6bu{s4AprB`dpjAtjq^Z6mSgBCb3K
zujDJIz5=JtIB3NIEcws;<HR0+lWsH`0FH<I^6C=<cp~$9q-YLtx*3WF%^Fy-k}-;t
zCRy>4Gx!Rc>+}wEYS>?1rs##v&}Y#Lo&SKT{D(Whl!fA`a^_p7O@l_Bt$p3KRkNhk
z*Mp-u#Od}?43tbmnfwI9O{&?fe<m|LxfCWd`A5KfieBJZS;$Wo@raOLFH~EtOc=PU
z&iWWzyeWjDb_$Cy-keTlkSZ=I6$VPJI(^d^jE747xv3i303CPUhFztqw(WdHbvvbO
zj+<GkHt@Rjk6=Ggg5-Jc%jdav7oEMv$9LQPHcOAud38CD)y-e3++S}jk*4V+$^ayj
zY{FvpeKg1T>icMnKqMTIq$L6}A|eve%dOAQ$M%Gslf`k}^qmRCFARHd(%kmgf<TtF
zj`ZTKgK&re@fDVewgn0OPj-i7NHQYXBC->*TR?rZOSv%$eedY2{`!)Q2?4E~VoT+E
pqpQ}}OPgROWPY=*+sjBkVu~-mwRe!SzM%)y_sDk$LpDw(;y<eVpqc;x

literal 0
HcmV?d00001

diff --git a/dsRag/Start.py b/dsRag/Start.py
index 1c452d94..34cf1968 100644
--- a/dsRag/Start.py
+++ b/dsRag/Start.py
@@ -1,82 +1,64 @@
-"""
-pip install fastapi uvicorn aiomysql
-"""
+from pathlib import Path
+
 import uvicorn
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-
-from Util.MySQLUtil import *
-
-"""
-API文档访问： http://localhost:8000/docs
-该实现包含以下功能：
-
-- 知识库(t_ai_kb)的增删改查接口
-- 知识库文件(t_ai_kb_files)的增删改查接口
-- 使用MySQLUtil.py中的连接池管理
-- 自动生成的Swagger文档
-"""
-
-app = FastAPI()
-
-# 知识库模型
-class KbModel(BaseModel):
-    kb_name: str
-    short_name: str
-    is_delete: Optional[int] = 0
-
-# 知识库文件模型
-class KbFileModel(BaseModel):
-    file_name: str
-    ext_name: str
-    kb_id: int
-    is_delete: Optional[int] = 0
-    state: Optional[int] = 0
-
-@app.on_event("startup")
-async def startup_event():
-    app.state.mysql_pool = await init_mysql_pool()
-
-@app.on_event("shutdown")
-async def shutdown_event():
-    app.state.mysql_pool.close()
-    await app.state.mysql_pool.wait_closed()
-
-# 知识库CRUD接口
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from pymysql.cursors import DictCursor
+
+from Dao.KbDao import KbDao
+from Model.KbModel import KbModel, KbFileModel
+from Util.MySQLUtil import init_mysql_pool
+from contextlib import asynccontextmanager
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # 启动时初始化数据库连接池
+    app.state.kb_dao = KbDao(await init_mysql_pool())
+    yield
+    # 关闭时清理资源
+    await app.state.kb_dao.mysql_pool.close()
+
+app = FastAPI(lifespan=lifespan)
+
+
 @app.post("/kb")
 async def create_kb(kb: KbModel):
-    async with app.state.mysql_pool.acquire() as conn:
-        async with conn.cursor() as cur:
-            await cur.execute(
-                """INSERT INTO t_ai_kb (kb_name, short_name, is_delete) 
-                VALUES (%s, %s, %s)""",
-                (kb.kb_name, kb.short_name, kb.is_delete)
-            )
-            await conn.commit()
-            return {"id": cur.lastrowid}
+    return await app.state.kb_dao.create_kb(kb)
 
 @app.get("/kb/{kb_id}")
 async def read_kb(kb_id: int):
+    """获取知识库详情
+    Args:
+        kb_id: 知识库ID
+    Returns:
+        返回知识库详细信息
+    """
     async with app.state.mysql_pool.acquire() as conn:
-        async with conn.cursor(DictCursor) as cur:
+        async with conn.cursor() as cur:
             await cur.execute("SELECT * FROM t_ai_kb WHERE id = %s", (kb_id,))
             result = await cur.fetchone()
             if not result:
-                raise HTTPException(status_code=404, detail="Knowledge base not found")
+                raise HTTPException(status_code=404, detail="知识库不存在")
             return result
 
-@app.put("/kb/{kb_id}")
+@app.post("/kb/update/{kb_id}")
 async def update_kb(kb_id: int, kb: KbModel):
+    """更新知识库信息
+    Args:
+        kb_id: 知识库ID
+        kb: 包含更新后知识库信息的对象
+    Returns:
+        返回更新结果
+    """
     async with app.state.mysql_pool.acquire() as conn:
         async with conn.cursor() as cur:
             await cur.execute(
                 """UPDATE t_ai_kb 
-                SET kb_name = %s, short_name = %s, is_delete = %s 
-                WHERE id = %s""",
+                SET kb_name=%s, short_name=%s, is_delete=%s 
+                WHERE id=%s""",
                 (kb.kb_name, kb.short_name, kb.is_delete, kb_id)
             )
             await conn.commit()
-            return {"message": "Knowledge base updated"}
+            return {"status": "success", "affected_rows": cur.rowcount}
 
 @app.delete("/kb/{kb_id}")
 async def delete_kb(kb_id: int):
@@ -87,8 +69,14 @@ async def delete_kb(kb_id: int):
             return {"message": "Knowledge base deleted"}
 
 # 知识库文件CRUD接口
-@app.post("/kb_files")
+@app.post("/kb_file")
 async def create_kb_file(file: KbFileModel):
+    """创建知识库文件
+    Args:
+        file: 包含文件名(file_name)、扩展名(ext_name)、关联知识库ID(kb_id)等信息的对象
+    Returns:
+        返回创建成功的文件ID
+    """
     async with app.state.mysql_pool.acquire() as conn:
         async with conn.cursor() as cur:
             await cur.execute(
@@ -110,20 +98,28 @@ async def read_kb_file(file_id: int):
                 raise HTTPException(status_code=404, detail="File not found")
             return result
 
-@app.put("/kb_files/{file_id}")
+# 知识库文件更新接口修改
+@app.post("/kb_files/update/{file_id}")
 async def update_kb_file(file_id: int, file: KbFileModel):
+    """更新知识库文件信息
+    Args:
+        file_id: 文件ID
+        file: 包含更新后文件信息的对象
+    Returns:
+        返回更新结果
+    """
     async with app.state.mysql_pool.acquire() as conn:
         async with conn.cursor() as cur:
             await cur.execute(
                 """UPDATE t_ai_kb_files 
-                SET file_name = %s, ext_name = %s, kb_id = %s, 
-                    is_delete = %s, state = %s 
-                WHERE id = %s""",
+                SET file_name=%s, ext_name=%s, kb_id=%s, 
+                    is_delete=%s, state=%s 
+                WHERE id=%s""",
                 (file.file_name, file.ext_name, file.kb_id, 
                  file.is_delete, file.state, file_id)
             )
             await conn.commit()
-            return {"message": "File updated"}
+            return {"status": "success", "affected_rows": cur.rowcount}
 
 @app.delete("/kb_files/{file_id}")
 async def delete_kb_file(file_id: int):
@@ -133,5 +129,67 @@ async def delete_kb_file(file_id: int):
             await conn.commit()
             return {"message": "File deleted"}
 
+# 允许的文件类型
+ALLOWED_EXTENSIONS = {'txt', 'pdf', 'docx', 'ppt', 'pptx'}
+MAX_FILE_SIZE = 100 * 1024 * 1024  # 100MB
+
+@app.post("/upload")
+async def upload_file(
+    kb_id: int, 
+    file: UploadFile = File(...)
+):
+    """
+    上传文件接口
+    Args:
+        kb_id: 关联的知识库ID
+        file: 上传的文件对象
+    Returns:
+        返回文件保存信息和数据库记录ID
+    """
+    # 检查文件类型
+    file_ext = Path(file.filename).suffix.lower()[1:]
+    if file_ext not in ALLOWED_EXTENSIONS:
+        raise HTTPException(
+            status_code=400, 
+            detail=f"不支持的文件类型: {file_ext}, 仅支持: {', '.join(ALLOWED_EXTENSIONS)}"
+        )
+
+    # 检查文件大小
+    file_size = 0
+    for chunk in file.file:
+        file_size += len(chunk)
+        if file_size > MAX_FILE_SIZE:
+            raise HTTPException(
+                status_code=400,
+                detail=f"文件大小超过限制: {MAX_FILE_SIZE/1024/1024}MB"
+            )
+    file.file.seek(0)
+
+    # 创建Upload目录
+    upload_dir = Path("Upload")
+    upload_dir.mkdir(exist_ok=True)
+
+    # 保存文件
+    file_path = upload_dir / file.filename
+    with open(file_path, "wb") as f:
+        f.write(await file.read())
+
+    # 记录到数据库
+    async with app.state.mysql_pool.acquire() as conn:
+        async with conn.cursor() as cur:
+            await cur.execute(
+                """INSERT INTO t_ai_kb_files 
+                (file_name, ext_name, kb_id, file_path, file_size) 
+                VALUES (%s, %s, %s, %s, %s)""",
+                (file.filename, file_ext, kb_id, str(file_path), file_size)
+            )
+            await conn.commit()
+            return {
+                "id": cur.lastrowid,
+                "filename": file.filename,
+                "size": file_size,
+                "path": str(file_path)
+            }
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)
\ No newline at end of file