From 9683aeeafa2f1067ef061b34124a1c362df07e5e Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 03 四月 2025 14:10:13 +0800
Subject: [PATCH] rg配置修改

---
 app/models/knowledge_model.py |   51 ++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 48 insertions(+), 3 deletions(-)

diff --git a/app/models/knowledge_model.py b/app/models/knowledge_model.py
index 30aca5d..a7f0494 100644
--- a/app/models/knowledge_model.py
+++ b/app/models/knowledge_model.py
@@ -24,12 +24,13 @@
     documents = Column(Integer, default=0)                    # 鏂囨。
     icon = Column(String(128))                    # 鏂囨。
     knowledge_type = Column(String(1))                  # 鐭ヨ瘑搴撳钩鍙�
+    embd_id = Column(String(128))
 
 
     def get_id(self):
         return str(self.id)
 
-    def to_json(self):
+    def to_json(self, user_id=""):
         return {
             'id': self.id,
             'name': self.name,
@@ -40,7 +41,9 @@
             'description': self.description,
             'status': self.status,
             'documents': self.documents if self.documents else 0,
-            'icon': self.icon
+            'icon': self.icon,
+            'embd_id': self.embd_id,
+            'link': True if (self.tenant_id == user_id or user_id == "admin" ) and self.permission == "team" else False,
         }
     def __repr__(self):
         return '<Knowledge name:%r url:%r>\n' % (self.name, self.id)
@@ -51,4 +54,46 @@
     name: str
     description: Optional[str] = ""
     icon: str
-    klgType: str
\ No newline at end of file
+    klgType: str
+
+
+
+class klgIcon(BaseModel):
+    id: str
+    icon: str
+
+
+
+class KnowledgeUserModel(Base):
+    __tablename__ = 'knowledgebase_user'
+
+    id = Column(Integer, primary_key=True)  # id
+    kb_id = Column(String(36))
+    user_id = Column(Integer)
+    create_date = Column(DateTime, default=datetime.now())             # 鍒涘缓鏃堕棿
+    update_date = Column(DateTime, default=datetime.now(), onupdate=datetime.now())             # 鏇存柊鏃堕棿
+    status = Column(Integer)  # 鐘舵��
+
+
+
+    def get_id(self):
+        return str(self.id)
+
+    def to_json(self, user_id=""):
+        return {
+            'id': self.id,
+            'kb_id': self.kb_id,
+            'create_time': self.create_date.strftime('%Y-%m-%d %H:%M:%S'),
+            'update_time': self.update_date.strftime('%Y-%m-%d %H:%M:%S'),
+            'user_id': self.user_id,
+            'status': self.status,
+            'owner': True if str(self.user_id) == user_id else False,
+        }
+    def __repr__(self):
+        return '<Knowledge name:%r url:%r>\n' % (self.name, self.id)
+
+
+
+class KlgOwner(BaseModel):
+    knowledgeId: str
+    UserId: int
\ No newline at end of file

--
Gitblit v1.8.0