基于 shiy-temp分支修改

2025-06-04 11:18:52 +08:00
parent 87ad3376fe
commit af33de09db
36 changed files with 6293 additions and 105 deletions
@@ -0,0 +1,68 @@
+import { type ActionFunctionArgs } from '@remix-run/node';
+import { difyClient } from '../services/dify-client.server';
+import { getSessionInfo } from '../utils/session.server';
+
+export async function action({ request }: ActionFunctionArgs) {
+    if (request.method !== 'POST') {
+        return new Response('Method not allowed', { status: 405 });
+    }
+
+    try {
+        const { user } = await getSessionInfo(request);
+        const body = await request.json();
+
+        const {
+            inputs,
+            query,
+            files,
+            conversation_id: conversationId,
+            response_mode: responseMode,
+        } = body;
+
+        // ('🚀 Chat Messages API - User:', user, 'Query:', query?.substring(0, 100));
+
+        const response = await difyClient.createChatMessage(
+            inputs,
+            query,
+            user,
+            responseMode,
+            conversationId,
+            files
+        );
+
+        // 对于流式响应，直接返回流
+        if (responseMode === 'streaming') {
+            return new Response(response.body, {
+                status: response.status,
+                headers: {
+                    'Content-Type': 'text/event-stream',
+                    'Cache-Control': 'no-cache',
+                    'Connection': 'keep-alive',
+                    'Access-Control-Allow-Origin': '*',
+                    'Access-Control-Allow-Methods': 'POST',
+                    'Access-Control-Allow-Headers': 'Content-Type',
+                },
+            });
+        }
+
+        // 对于非流式响应，返回JSON
+        return new Response(JSON.stringify(response), {
+            status: 200,
+            headers: {
+                'Content-Type': 'application/json',
+            },
+        });
+
+    } catch (error: any) {
+        // console.error('❌ Chat Messages API - Error:', error);
+        return new Response(
+            JSON.stringify({ error: error.message || 'Failed to send message' }),
+            {
+                status: 500,
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+            }
+        );
+    }
+}