首页知识库对话-rg

zhaoqingang

2025-03-18 282a631b9ceee9a634ee1d93751a5254ed37ccef

 app/models/dialog_model.py

@@ -1,3 +1,4 @@
import json
from datetime import datetime
from typing import Optional

@@ -23,6 +24,8 @@
    dialog_type = Column(String(1))            #  平台
    # agent_id = Column(String(36))
    mode = Column(String(36))
    parameters = Column(Text)
    kb_ids = Column(String(128))

    def get_id(self):
        return str(self.id)
@@ -41,6 +44,9 @@
            # 'agentId': self.agent_id,
            'mode': self.mode,
        }

    def get_kb_ids(self):
        return json.loads(self.kb_ids) if self.kb_ids else []


class ConversationModel(Base):
@@ -82,6 +88,8 @@
    id: str
    status: Optional[str] = "1"
    icon: Optional[str] = ""
    name: Optional[str] = ""
    description: Optional[str] = None


class dialogList(BaseModel):