完善权限系统

bbit_ai/app/db/postgres/knowledge.py

@@ -0,0 +1,39 @@
+from config.pgDb import pg_pool
+
+
+def get_available_knowledge_bases(available_module: str):
+    with pg_pool.getConn() as conn:
+        with conn.cursor() as cursor:
+            cursor.execute(
+                """
+                SELECT id, name, description, is_active
+                FROM ai_knowledge
+                WHERE available_module::jsonb @> %s::jsonb
+                """,
+                (f'["{available_module}"]',),
+            )
+
+            return [
+                {
+                    "id": row[0],
+                    "name": row[1],
+                    "description": row[2],
+                    "is_active": row[3],
+                }
+                for row in cursor.fetchall()
+            ]
+
+
+def add_knowledge_base(name, description, user_id):
+    with pg_pool.getConn() as conn:
+        with conn.cursor() as cursor:
+            cursor.execute(
+                """
+                INSERT INTO ai_knowledge (name, description, created_by, created_at)
+                VALUES (%s, %s, %s, now())
+                RETURNING id
+                """,
+                (name, description, user_id),
+            )
+            new_id = cursor.fetchone()[0]  # 取返回的 id
+            return new_id