对话流实现文件上传

2025-11-06 16:43:28 +08:00
parent d9d62e22de
commit 0bb4853d54
35 changed files with 1748 additions and 575 deletions
--- a/schoolNewsServ/ai/src/main/resources/mapper/AiAgentConfigMapper.xml
+++ b/schoolNewsServ/ai/src/main/resources/mapper/AiAgentConfigMapper.xml
@@ -8,12 +8,7 @@
        <result column="name" property="name" jdbcType="VARCHAR"/>
        <result column="avatar" property="avatar" jdbcType="VARCHAR"/>
        <result column="description" property="description" jdbcType="VARCHAR"/>
-        <result column="system_prompt" property="systemPrompt" jdbcType="LONGVARCHAR"/>
-        <result column="model_name" property="modelName" jdbcType="VARCHAR"/>
-        <result column="model_provider" property="modelProvider" jdbcType="VARCHAR"/>
-        <result column="temperature" property="temperature" jdbcType="DECIMAL"/>
-        <result column="max_tokens" property="maxTokens" jdbcType="INTEGER"/>
-        <result column="top_p" property="topP" jdbcType="DECIMAL"/>
+        <result column="connect_internet" property="connectInternet" jdbcType="INTEGER"/>
        <result column="dify_app_id" property="difyAppId" jdbcType="VARCHAR"/>
        <result column="dify_api_key" property="difyApiKey" jdbcType="VARCHAR"/>
        <result column="status" property="status" jdbcType="INTEGER"/>
@@ -27,8 +22,7 @@

    <!-- 基础字段 -->
    <sql id="Base_Column_List">
-        id, name, avatar, description, system_prompt, model_name, model_provider, 
-        temperature, max_tokens, top_p, dify_app_id, dify_api_key, status, 
+        id, name, avatar, description, connect_internet, dify_app_id, dify_api_key, status, 
        creator, updater, create_time, update_time, delete_time, deleted
    </sql>

@@ -39,12 +33,6 @@
            <if test="name != null and name != ''">
                AND name LIKE CONCAT('%', #{name}, '%')
            </if>
-            <if test="modelName != null and modelName != ''">
-                AND model_name = #{modelName}
-            </if>
-            <if test="modelProvider != null and modelProvider != ''">
-                AND model_provider = #{modelProvider}
-            </if>
            <if test="status != null">
                AND status = #{status}
            </if>
@@ -54,12 +42,10 @@
    <!-- 插入智能体配置 -->
    <insert id="insertAgentConfig" parameterType="org.xyzh.common.dto.ai.TbAiAgentConfig">
        INSERT INTO tb_ai_agent_config (
-            id, name, avatar, description, system_prompt, model_name, model_provider,
-            temperature, max_tokens, top_p, dify_app_id, dify_api_key, status,
+            id, name, avatar, description, connect_internet, dify_app_id, dify_api_key, status,
            creator, updater, create_time, update_time, deleted
        ) VALUES (
-            #{id}, #{name}, #{avatar}, #{description}, #{systemPrompt}, #{modelName}, #{modelProvider},
-            #{temperature}, #{maxTokens}, #{topP}, #{difyAppId}, #{difyApiKey}, #{status},
+            #{id}, #{name}, #{avatar}, #{description}, #{connectInternet}, #{difyAppId}, #{difyApiKey}, #{status},
            #{creator}, #{updater}, #{createTime}, #{updateTime}, #{deleted}
        )
    </insert>
@@ -71,12 +57,7 @@
            <if test="name != null and name != ''">name = #{name},</if>
            <if test="avatar != null">avatar = #{avatar},</if>
            <if test="description != null">description = #{description},</if>
-            <if test="systemPrompt != null">system_prompt = #{systemPrompt},</if>
-            <if test="modelName != null">model_name = #{modelName},</if>
-            <if test="modelProvider != null">model_provider = #{modelProvider},</if>
-            <if test="temperature != null">temperature = #{temperature},</if>
-            <if test="maxTokens != null">max_tokens = #{maxTokens},</if>
-            <if test="topP != null">top_p = #{topP},</if>
+            <if test="connectInternet != null">connect_internet = #{connectInternet},</if>
            <if test="difyAppId != null">dify_app_id = #{difyAppId},</if>
            <if test="difyApiKey != null">dify_api_key = #{difyApiKey},</if>
            <if test="status != null">status = #{status},</if>
@@ -116,9 +97,6 @@
            <if test="filter.status != null">
                AND status = #{filter.status}
            </if>
-            <if test="filter.modelProvider != null and filter.modelProvider != ''">
-                AND model_provider = #{filter.modelProvider}
-            </if>
        </if>
        ORDER BY create_time DESC
    </select>
@@ -136,9 +114,6 @@
            <if test="filter.status != null">
                AND status = #{filter.status}
            </if>
-            <if test="filter.modelProvider != null and filter.modelProvider != ''">
-                AND model_provider = #{filter.modelProvider}
-            </if>
        </if>
        ORDER BY create_time DESC
        LIMIT #{pageParam.offset}, #{pageParam.pageSize}
@@ -156,9 +131,6 @@
            <if test="filter.status != null">
                AND status = #{filter.status}
            </if>
-            <if test="filter.modelProvider != null and filter.modelProvider != ''">
-                AND model_provider = #{filter.modelProvider}
-            </if>
        </if>
    </select>

--- a/schoolNewsServ/ai/src/main/resources/mapper/AiUploadFileMapper.xml
+++ b/schoolNewsServ/ai/src/main/resources/mapper/AiUploadFileMapper.xml
@@ -8,6 +8,8 @@
        <result column="user_id" property="userID" jdbcType="VARCHAR"/>
        <result column="knowledge_id" property="knowledgeId" jdbcType="VARCHAR"/>
        <result column="conversation_id" property="conversationID" jdbcType="VARCHAR"/>
+        <result column="message_id" property="messageID" jdbcType="VARCHAR"/>
+        <result column="sys_file_id" property="sysFileId" jdbcType="VARCHAR"/>
        <result column="file_name" property="fileName" jdbcType="VARCHAR"/>
        <result column="file_path" property="filePath" jdbcType="VARCHAR"/>
        <result column="file_size" property="fileSize" jdbcType="BIGINT"/>
@@ -16,12 +18,10 @@
        <result column="extracted_text" property="extractedText" jdbcType="LONGVARCHAR"/>
        <result column="dify_document_id" property="difyDocumentId" jdbcType="VARCHAR"/>
        <result column="dify_batch_id" property="difyBatchId" jdbcType="VARCHAR"/>
-        <result column="vector_status" property="vectorStatus" jdbcType="INTEGER"/>
+        <result column="dify_upload_file_id" property="difyUploadFileId" jdbcType="VARCHAR"/>
        <result column="chunk_count" property="chunkCount" jdbcType="INTEGER"/>
        <result column="status" property="status" jdbcType="INTEGER"/>
        <result column="error_message" property="errorMessage" jdbcType="VARCHAR"/>
-        <result column="creator" property="creator" jdbcType="VARCHAR"/>
-        <result column="updater" property="updater" jdbcType="VARCHAR"/>
        <result column="create_time" property="createTime" jdbcType="TIMESTAMP"/>
        <result column="update_time" property="updateTime" jdbcType="TIMESTAMP"/>
        <result column="delete_time" property="deleteTime" jdbcType="TIMESTAMP"/>
@@ -30,9 +30,9 @@

    <!-- 基础字段 -->
    <sql id="Base_Column_List">
-        id, user_id, knowledge_id, conversation_id, file_name, file_path, file_size, 
-        file_type, mime_type, extracted_text, dify_document_id, dify_batch_id, 
-        vector_status, chunk_count, status, error_message, creator, updater, 
+        id, user_id, knowledge_id, conversation_id, message_id, sys_file_id, file_name, file_path, file_size, 
+        file_type, mime_type, extracted_text, dify_document_id, dify_batch_id, dify_upload_file_id,
+        chunk_count, status, error_message, 
        create_time, update_time, delete_time, deleted
    </sql>

@@ -56,9 +56,6 @@
                <if test="filter.fileType != null and filter.fileType != ''">
                    AND file_type = #{filter.fileType}
                </if>
-                <if test="filter.vectorStatus != null">
-                    AND vector_status = #{filter.vectorStatus}
-                </if>
                <if test="filter.status != null">
                    AND status = #{filter.status}
                </if>
@@ -69,14 +66,14 @@
    <!-- insertUploadFile（插入文件记录） -->
    <insert id="insertUploadFile" parameterType="org.xyzh.common.dto.ai.TbAiUploadFile">
        INSERT INTO tb_ai_upload_file (
-            id, user_id, knowledge_id, conversation_id, file_name, file_path, file_size,
-            file_type, mime_type, extracted_text, dify_document_id, dify_batch_id,
-            vector_status, chunk_count, status, error_message, creator, updater,
+            id, user_id, knowledge_id, conversation_id, message_id, sys_file_id, file_name, file_path, file_size,
+            file_type, mime_type, extracted_text, dify_document_id, dify_batch_id, dify_upload_file_id,
+            chunk_count, status, error_message,
            create_time, update_time, deleted
        ) VALUES (
-            #{ID}, #{userID}, #{knowledgeId}, #{conversationID}, #{fileName}, #{filePath}, #{fileSize},
-            #{fileType}, #{mimeType}, #{extractedText}, #{difyDocumentId}, #{difyBatchId},
-            #{vectorStatus}, #{chunkCount}, #{status}, #{errorMessage}, #{creator}, #{updater},
+            #{ID}, #{userID}, #{knowledgeId}, #{conversationID}, #{messageID}, #{sysFileId}, #{fileName}, #{filePath}, #{fileSize},
+            #{fileType}, #{mimeType}, #{extractedText}, #{difyDocumentId}, #{difyBatchId}, #{difyUploadFileId},
+            #{chunkCount}, #{status}, #{errorMessage},
            #{createTime}, #{updateTime}, #{deleted}
        )
    </insert>
@@ -88,6 +85,8 @@
            <if test="userID != null">user_id = #{userID},</if>
            <if test="knowledgeId != null">knowledge_id = #{knowledgeId},</if>
            <if test="conversationID != null">conversation_id = #{conversationID},</if>
+            <if test="messageID != null">message_id = #{messageID},</if>
+            <if test="sysFileId != null">sys_file_id = #{sysFileId},</if>
            <if test="fileName != null">file_name = #{fileName},</if>
            <if test="filePath != null">file_path = #{filePath},</if>
            <if test="fileSize != null">file_size = #{fileSize},</if>
@@ -96,11 +95,10 @@
            <if test="extractedText != null">extracted_text = #{extractedText},</if>
            <if test="difyDocumentId != null">dify_document_id = #{difyDocumentId},</if>
            <if test="difyBatchId != null">dify_batch_id = #{difyBatchId},</if>
-            <if test="vectorStatus != null">vector_status = #{vectorStatus},</if>
+            <if test="difyUploadFileId != null">dify_upload_file_id = #{difyUploadFileId},</if>
            <if test="chunkCount != null">chunk_count = #{chunkCount},</if>
            <if test="status != null">status = #{status},</if>
            <if test="errorMessage != null">error_message = #{errorMessage},</if>
-            <if test="updater != null">updater = #{updater},</if>
            <if test="updateTime != null">update_time = #{updateTime},</if>
        </set>
        WHERE id = #{ID} AND deleted = 0
@@ -110,8 +108,7 @@
    <update id="deleteUploadFile" parameterType="org.xyzh.common.dto.ai.TbAiUploadFile">
        UPDATE tb_ai_upload_file
        SET deleted = 1,
-            delete_time = NOW(),
-            updater = #{updater}
+            delete_time = NOW()
        WHERE id = #{ID} AND deleted = 0
    </update>

@@ -186,4 +183,31 @@
        ORDER BY create_time DESC
    </select>

+    <!-- selectFilesByMessageId（根据消息ID查询关联的文件列表） -->
+    <select id="selectFilesByMessageId" resultMap="BaseResultMap">
+        SELECT
+        <include refid="Base_Column_List"/>
+        FROM tb_ai_upload_file
+        WHERE message_id = #{messageId}
+          AND deleted = 0
+        ORDER BY create_time ASC
+    </select>
+
+    <!-- batchInsertUploadFiles（批量插入文件记录） -->
+    <insert id="batchInsertUploadFiles" parameterType="java.util.List">
+        INSERT INTO tb_ai_upload_file (
+            id, user_id, knowledge_id, conversation_id, message_id, sys_file_id, file_name, file_path, file_size,
+            file_type, mime_type, dify_document_id, dify_batch_id, dify_upload_file_id,
+            chunk_count, status, create_time, update_time, deleted
+        ) VALUES
+        <foreach collection="files" item="file" separator=",">
+            (
+                #{file.ID}, #{file.userID}, #{file.knowledgeId}, #{file.conversationID}, #{file.messageID},
+                #{file.sysFileId}, #{file.fileName}, #{file.filePath}, #{file.fileSize},
+                #{file.fileType}, #{file.mimeType}, #{file.difyDocumentId}, #{file.difyBatchId}, #{file.difyUploadFileId},
+                #{file.chunkCount}, #{file.status}, #{file.createTime}, #{file.updateTime}, #{file.deleted}
+            )
+        </foreach>
+    </insert>
+
 </mapper>