feat: chat

Author: ulleo

backend/alembic/versions/029_modify_chat.py

@@ -0,0 +1,54 @@
+"""029_modify_chat
+
+Revision ID: 77d4c39ec22f
+Revises: e96b16d3daab
+Create Date: 2025-07-17 17:05:13.392973
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = '77d4c39ec22f'
+down_revision = 'e96b16d3daab'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('chat', sa.Column('oid', sa.BigInteger(), nullable=True))
+    op.execute('update chat set oid = 1')
+    op.alter_column('chat', 'create_time',
+               existing_type=postgresql.TIMESTAMP(timezone=True),
+               type_=sa.DateTime(),
+               existing_nullable=True)
+    op.alter_column('chat_record', 'create_time',
+               existing_type=postgresql.TIMESTAMP(timezone=True),
+               type_=sa.DateTime(),
+               existing_nullable=True)
+    op.alter_column('chat_record', 'finish_time',
+               existing_type=postgresql.TIMESTAMP(timezone=True),
+               type_=sa.DateTime(),
+               existing_nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('chat_record', 'finish_time',
+               existing_type=sa.DateTime(),
+               type_=postgresql.TIMESTAMP(timezone=True),
+               existing_nullable=True)
+    op.alter_column('chat_record', 'create_time',
+               existing_type=sa.DateTime(),
+               type_=postgresql.TIMESTAMP(timezone=True),
+               existing_nullable=True)
+    op.alter_column('chat', 'create_time',
+               existing_type=sa.DateTime(),
+               type_=postgresql.TIMESTAMP(timezone=True),
+               existing_nullable=True)
+    op.drop_column('chat', 'oid')
+    # ### end Alembic commands ###

backend/apps/chat/models/chat_model.py

@@ -17,7 +17,8 @@
 class Chat(SQLModel, table=True):
     __tablename__ = "chat"
     id: Optional[int] = Field(sa_column=Column(BigInteger, Identity(always=True), primary_key=True))
-    create_time: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=True))
+    oid: Optional[int] = Field(sa_column=Column(BigInteger, nullable=True, default=1))
+    create_time: datetime = Field(sa_column=Column(DateTime(timezone=False), nullable=True))
     create_by: int = Field(sa_column=Column(BigInteger, nullable=True))
     brief: str = Field(max_length=64, nullable=True)
     chat_type: str = Field(max_length=20, default="chat")  # chat, datasource
@@ -31,8 +32,8 @@ class ChatRecord(SQLModel, table=True):
     chat_id: int = Field(sa_column=Column(BigInteger, nullable=False))
     ai_modal_id: Optional[int] = Field(sa_column=Column(BigInteger))
     first_chat: bool = Field(sa_column=Column(Boolean, nullable=True, default=False))
-    create_time: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=True))
-    finish_time: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=True))
+    create_time: datetime = Field(sa_column=Column(DateTime(timezone=False), nullable=True))
+    finish_time: datetime = Field(sa_column=Column(DateTime(timezone=False), nullable=True))
     create_by: int = Field(sa_column=Column(BigInteger, nullable=True))
     datasource: int = Field(sa_column=Column(BigInteger, nullable=True))
     engine_type: str = Field(max_length=64)
@@ -66,6 +67,7 @@ class ChatRecord(SQLModel, table=True):
     analysis_record_id: int = Field(sa_column=Column(BigInteger, nullable=True))
     predict_record_id: int = Field(sa_column=Column(BigInteger, nullable=True))
 
+
 class CreateChat(BaseModel):
     id: int = None
     question: str = None

backend/template.yaml

@@ -132,8 +132,8 @@ template:
       你是一个数据分析师，你的任务是根据给定的数据进行数据预测，我将以json格式给你一组数据，你帮我预测之后的数据（一段可以展示趋势的数据，至少2个周期），用json格式返回。
       ```json
       
-      无法预测或者不支持预测的数据请直接返回："抱歉，该数据无法进行预测。(有原因则返回无法预测的原因)"
-      不需要返回原有数据，直接返回预测的部份
+      无法预测或者不支持预测的数据请直接返回(不需要返回JSON格式)："抱歉，该数据无法进行预测。(有原因则返回无法预测的原因)"
+      如果可以预测，则不需要返回原有数据，直接返回预测的部份
 
     user: |
       ### 请使用 i18n: {lang} 对应的语言输出你的结果

frontend/package.json

@@ -26,6 +26,7 @@
     "element-plus": "^2.10.1",
     "element-plus-secondary": "^1.0.0",
     "element-resize-detector": "^1.2.4",
+    "github-markdown-css": "^5.8.1",
     "highlight.js": "^11.11.1",
     "html2canvas": "^1.4.1",
     "lodash": "^4.17.21",

frontend/src/assets/svg/icon_replace_outlined.svg

@@ -84,7 +84,8 @@
     "data_predict": "数据分析",
     "chat_search": "搜索",
     "thinking": "思考中",
-    "thinking_step": "思考过程"
+    "thinking_step": "思考过程",
+    "ask_again": "重新生成"
   },
   "ds": {
     "title": "数据源",

frontend/src/assets/svg/icon_screen_outlined.svg

@@ -2,25 +2,23 @@ import MarkdownIt from 'markdown-it'
 import hljs from 'highlight.js'
 
 const md = new MarkdownIt({
-  html: true,
-  linkify: true,
+  html: false,
+  linkify: false,
+  breaks: true,
+  typographer: true,
+  quotes: '“”‘’',
   highlight: (str: any, lang: any): string => {
     if (lang && hljs.getLanguage(lang)) {
       try {
-        return `
-          <pre>
-          <code class="hljs">
-          ${hljs.highlight(str, { language: lang, ignoreIllegals: true }).value}
-          </code>
-          </pre>
-            `
+        return `<pre class="hljs">
+                  <code>${hljs.highlight(str, { language: lang, ignoreIllegals: true }).value}</code>
+                </pre>`
       } catch (e) {
         console.error(e)
-        return str
       }
     }
 
-    return '<pre><code class="hljs">' + md.utils.escapeHtml(str) + '</code></pre>'
+    return '<pre class="hljs"><code>' + md.utils.escapeHtml(str) + '</code></pre>'
   },
 })
 

frontend/src/assets/svg/icon_start_outlined.svg

@@ -53,7 +53,6 @@ function clickQuestion(question: string): void {
 
 <style scoped lang="less">
 .recommend-questions {
-  padding: 8px;
   font-size: 14px;
   font-weight: 500;
   line-height: 22px;