From 72855344053f5078804fe90c14726e627be2b5f1 Mon Sep 17 00:00:00 2001
From: shenaowei <450702724@qq.com>
Date: Wed, 25 Feb 2026 01:24:57 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=83=AD=E7=82=B9=E6=94=B9=E8=BF=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 frontend/app/web-gold/src/api/forecast.js     |  75 ++++
 .../web-gold/src/views/trends/Forecast.vue    | 373 ++++++++++++------
 .../module/tik/dify/client/DifyClient.java    |  49 +++
 .../dify/controller/AppDifyController.java    |  11 +
 .../module/tik/dify/service/DifyService.java  |  10 +
 .../tik/dify/service/DifyServiceImpl.java     | 108 +++++
 .../tik/dify/vo/ForecastRewriteReqVO.java     |  29 ++
 .../module/tik/enums/AiModelTypeEnum.java     |   4 +
 8 files changed, 537 insertions(+), 122 deletions(-)
 create mode 100644 frontend/app/web-gold/src/api/forecast.js
 create mode 100644 yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/vo/ForecastRewriteReqVO.java

diff --git a/frontend/app/web-gold/src/api/forecast.js b/frontend/app/web-gold/src/api/forecast.js
new file mode 100644
index 0000000000..bf19faf032
--- /dev/null
+++ b/frontend/app/web-gold/src/api/forecast.js
@@ -0,0 +1,75 @@
+/**
+ * Forecast 文案改写 API
+ */
+import { fetchEventSource } from '@microsoft/fetch-event-source'
+import tokenManager from '@gold/utils/token-manager'
+import { API_BASE } from '@gold/config/api'
+
+const BASE_URL = `${API_BASE.APP_TIK}`
+
+/**
+ * 流式文案改写（SSE）
+ * @param {Object} options - 请求配置
+ * @param {number} options.agentId - 智能体ID
+ * @param {string} options.userText - 用户输入文案
+ * @param {number} [options.level] - 改写级别/强度
+ * @param {string} [options.modelType] - 模型类型：forecast_standard/forecast_meiju
+ * @param {AbortController} [options.ctrl] - 取消控制器
+ * @param {Function} options.onMessage - 消息回调
+ * @param {Function} [options.onError] - 错误回调
+ * @param {Function} [options.onClose] - 关闭回调
+ */
+export async function rewriteStream(options) {
+  const {
+    agentId,
+    userText,
+    level = 50,
+    modelType = 'forecast_standard',
+    ctrl,
+    onMessage,
+    onError,
+    onClose
+  } = options || {}
+
+  const token = tokenManager.getAccessToken()
+
+  return fetchEventSource(`${BASE_URL}/dify/forecast/rewrite`, {
+    method: 'post',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${token}`,
+      'tenant-id': import.meta.env?.VITE_TENANT_ID
+    },
+    openWhenHidden: true,
+    body: JSON.stringify({
+      agentId,
+      userText,
+      level,
+      modelType
+    }),
+    onmessage: (event) => {
+      if (typeof onMessage === 'function') {
+        try {
+          const data = JSON.parse(event.data)
+          // 解析 CommonResult 包装
+          const result = data.code === 0 ? data.data : data
+          onMessage(result)
+        } catch (e) {
+          console.error('解析 SSE 数据失败:', e)
+        }
+      }
+    },
+    onerror: (err) => {
+      if (typeof onError === 'function') {
+        onError(err)
+      }
+      throw err // 不重试
+    },
+    onclose: () => {
+      if (typeof onClose === 'function') {
+        onClose()
+      }
+    },
+    signal: ctrl ? ctrl.signal : undefined
+  })
+}
diff --git a/frontend/app/web-gold/src/views/trends/Forecast.vue b/frontend/app/web-gold/src/views/trends/Forecast.vue
index 66a6afda4b..f718dd4650 100644
--- a/frontend/app/web-gold/src/views/trends/Forecast.vue
+++ b/frontend/app/web-gold/src/views/trends/Forecast.vue
@@ -1,11 +1,10 @@
 <script setup>
-import { onMounted, reactive, ref } from 'vue'
+import { onMounted, reactive, ref, computed } from 'vue'
 import { message } from 'ant-design-vue'
 
-import { CommonService } from '@/api/common'
 import TikhubService, { InterfaceType, MethodType, ParamType } from '@/api/tikhub'
-import { UserPromptApi } from '@/api/userPrompt'
-import PromptSelector from '@/components/PromptSelector.vue'
+import { rewriteStream } from '@/api/forecast'
+import { getAgentList } from '@/api/agent'
 import { useUserStore } from '@/stores/user'
 import { getVoiceText } from '@gold/hooks/web/useVoiceText'
 
@@ -25,25 +24,42 @@ const generatedContent = ref('')
 // 搜索参数
 const searchParams = reactive({
   keyword: '',
-  offset: 0,
-  sort_type: '1',
-  publish_time: '7',
+  cursor: 0,
+  sort_type: '0',
+  publish_time: '0',
   filter_duration: '0',
-  content_type: '0'
+  content_type: '0',
+  search_id: '',
+  backtrace: ''
 })
 
 // 创作详情
 const topicDetails = reactive({
-  title: '',
   copywriting: '',
-  stylePrompt: '',
-  stylePromptId: null
+  selectedAgentId: null,
+  modelType: 'forecast_standard',
+  level: 50
 })
 
-// 提示词相关
-const allPrompts = ref([])
-const loadingPrompts = ref(false)
-const promptSearchKeyword = ref('')
+// 智能体列表
+const agentList = ref([])
+const loadingAgents = ref(false)
+const agentSearchKeyword = ref('')
+
+// 当前选中的智能体
+const selectedAgent = ref(null)
+
+// 过滤后的智能体列表
+const filteredAgentList = computed(() => {
+  if (!agentSearchKeyword.value.trim()) {
+    return agentList.value
+  }
+  const keyword = agentSearchKeyword.value.trim().toLowerCase()
+  return agentList.value.filter(agent =>
+    agent.name.toLowerCase().includes(keyword) ||
+    (agent.description && agent.description.toLowerCase().includes(keyword))
+  )
+})
 
 // 工具函数
 function formatNumber(num) {
@@ -80,39 +96,41 @@ async function copyContent() {
   }
 }
 
-// 提示词管理
-async function loadUserPrompts() {
-  if (!userStore.userId) {
-    console.warn('用户未登录，无法加载提示词')
-    return
-  }
-
-  loadingPrompts.value = true
+// 加载智能体列表
+async function loadAgentList() {
+  loadingAgents.value = true
   try {
-    const response = await UserPromptApi.getUserPromptPage({
-      pageNo: 1,
-      pageSize: 100,
-      status: 1
-    })
-
-    if (response?.data?.list) {
-      allPrompts.value = response.data.list
+    const res = await getAgentList()
+    if (res.code === 0 && res.data) {
+      agentList.value = res.data.map(item => ({
+        id: item.id,
+        agentId: item.agentId,
+        name: item.agentName,
+        description: item.description,
+        systemPrompt: item.systemPrompt,
+        avatar: item.icon,
+        categoryName: item.categoryName || '其他'
+      }))
+      // 默认选中第一个
+      if (agentList.value.length > 0 && !topicDetails.selectedAgentId) {
+        topicDetails.selectedAgentId = agentList.value[0].id
+        selectedAgent.value = agentList.value[0]
+      }
     }
   } catch (error) {
-    console.error('加载提示词失败:', error)
+    console.error('加载智能体列表失败:', error)
   } finally {
-    loadingPrompts.value = false
+    loadingAgents.value = false
   }
 }
 
-// 处理提示词选择
-function handlePromptChange(prompt) {
-  if (!prompt?.content) {
-    message.warning('提示词内容为空')
-    return
+// 处理智能体选择
+function handleAgentChange(agentId) {
+  const agent = agentList.value.find(a => a.id === agentId)
+  if (agent) {
+    selectedAgent.value = agent
+    topicDetails.selectedAgentId = agentId
   }
-  topicDetails.stylePromptId = prompt.id
-  topicDetails.stylePrompt = prompt.content
 }
 
 // 语音分析
@@ -145,16 +163,14 @@ async function analyzeVoice(audioUrl) {
   }
 }
 
+// 选择话题
+function handleSelectTopic(topic) {
+  selectedTopic.value = topic.id
+}
+
 // 创作处理
 async function handleCreate(topic) {
   selectedTopic.value = topic.id
-  topicDetails.title = topic.title
-
-  if (!topicDetails.stylePromptId && allPrompts.value.length > 0) {
-    const firstPrompt = allPrompts.value[0]
-    topicDetails.stylePromptId = firstPrompt.id
-    topicDetails.stylePrompt = firstPrompt.content || ''
-  }
 
   if (topic.audioUrl?.trim()) {
     await analyzeVoice(topic.audioUrl)
@@ -170,7 +186,7 @@ async function handleGenerate() {
     return
   }
 
-  if (!topicDetails.stylePrompt?.trim()) {
+  if (!selectedAgent.value?.id) {
     message.warning('请先选择文案风格')
     return
   }
@@ -179,12 +195,6 @@ async function handleGenerate() {
   generatedContent.value = ''
 
   try {
-    const requestData = {
-      audio_prompt: topicDetails.stylePrompt,
-      user_text: topicDetails.copywriting.trim(),
-      amplitude: 50
-    }
-
     const ctrl = new AbortController()
     let fullText = ''
     let completed = false
@@ -198,40 +208,27 @@ async function handleGenerate() {
         }
       }, 180000)
 
-      CommonService.callWorkflowStream({
-        data: requestData,
+      rewriteStream({
+        agentId: selectedAgent.value.id,
+        userText: topicDetails.copywriting.trim(),
+        level: topicDetails.level,
+        modelType: topicDetails.modelType,
         ctrl,
-        onMessage: (event) => {
-          if (completed || !event?.data) return
+        onMessage: (data) => {
+          if (completed) return
 
-          const dataStr = event.data.trim()
-
-          if (dataStr === '[DONE]') {
+          // 处理 Dify 响应格式
+          if (data.event === 'message' && data.content) {
+            fullText += data.content
+            generatedContent.value = fullText
+          } else if (data.event === 'done') {
             completed = true
             clearTimeout(timeout)
             resolve()
-            return
-          }
-
-          if (dataStr.startsWith('[TIMEOUT]')) {
+          } else if (data.event === 'error') {
             completed = true
             clearTimeout(timeout)
-            reject(new Error(dataStr.replace('[TIMEOUT]', '').trim() || '请求超时'))
-            return
-          }
-
-          try {
-            const obj = JSON.parse(event.data)
-            const piece = obj?.text || obj?.content || obj?.data || ''
-            if (piece) {
-              fullText += piece
-              generatedContent.value = fullText
-            }
-          } catch {
-            if (event.data && !event.data.startsWith('[')) {
-              fullText += event.data
-              generatedContent.value = fullText
-            }
+            reject(new Error(data.errorMessage || '生成失败'))
           }
         },
         onError: (err) => {
@@ -284,8 +281,13 @@ function extractCover(video) {
 
 function processSearchResults(response, startId = 1) {
   try {
-    currentCursor.value = response?.data?.cursor || null
-    const dataList = response?.data?.data || []
+    // 保存分页信息用于下次请求
+    const responseData = response?.data || {}
+    currentCursor.value = responseData.cursor || null
+    searchParams.search_id = responseData.search_id || ''
+    searchParams.backtrace = responseData.backtrace || ''
+
+    const dataList = responseData.data || []
 
     return dataList
       .map(el => el.aweme_info)
@@ -321,8 +323,11 @@ async function handleSearch() {
     return
   }
 
-  searchParams.offset = 0
+  // 重置分页参数
   searchParams.keyword = keyword
+  searchParams.cursor = 0
+  searchParams.search_id = ''
+  searchParams.backtrace = ''
   currentCursor.value = null
   hotTopics.value = []
   isLoading.value = true
@@ -330,11 +335,13 @@ async function handleSearch() {
   try {
     const urlParams = {
       keyword: searchParams.keyword,
-      offset: String(searchParams.offset),
-      sort_type: String(searchParams.sort_type),
-      publish_time: String(searchParams.publish_time),
+      cursor: searchParams.cursor,
+      sort_type: searchParams.sort_type,
+      publish_time: searchParams.publish_time,
       filter_duration: searchParams.filter_duration,
-      content_type: String(searchParams.content_type),
+      content_type: searchParams.content_type,
+      search_id: searchParams.search_id,
+      backtrace: searchParams.backtrace
     }
 
     const response = await TikhubService.postTikHup({
@@ -364,13 +371,7 @@ async function handleSearch() {
 
 // 初始化
 onMounted(() => {
-  if (userStore.userId) {
-    loadUserPrompts()
-  } else if (userStore.isLoggedIn) {
-    setTimeout(() => {
-      if (userStore.userId) loadUserPrompts()
-    }, 500)
-  }
+  loadAgentList()
 })
 </script>
 
@@ -413,26 +414,26 @@ onMounted(() => {
             <label>排序</label>
             <a-select v-model:value="searchParams.sort_type" size="small">
               <a-select-option value="0">综合</a-select-option>
-              <a-select-option value="1">最热</a-select-option>
-              <a-select-option value="2">最新</a-select-option>
+              <a-select-option value="1">最多点赞</a-select-option>
+              <a-select-option value="2">最新发布</a-select-option>
             </a-select>
           </div>
           <div class="filter-item">
             <label>时间</label>
             <a-select v-model:value="searchParams.publish_time" size="small">
               <a-select-option value="0">不限</a-select-option>
-              <a-select-option value="1">今天</a-select-option>
-              <a-select-option value="7">本周</a-select-option>
-              <a-select-option value="180">半年</a-select-option>
+              <a-select-option value="1">最近一天</a-select-option>
+              <a-select-option value="7">最近一周</a-select-option>
+              <a-select-option value="180">最近半年</a-select-option>
             </a-select>
           </div>
           <div class="filter-item">
             <label>时长</label>
             <a-select v-model:value="searchParams.filter_duration" size="small">
               <a-select-option value="0">不限</a-select-option>
-              <a-select-option value="0-1">&lt;1分钟</a-select-option>
+              <a-select-option value="0-1">1分钟以内</a-select-option>
               <a-select-option value="1-5">1-5分钟</a-select-option>
-              <a-select-option value="5-10000">&gt;5分钟</a-select-option>
+              <a-select-option value="5-10000">5分钟以上</a-select-option>
             </a-select>
           </div>
         </div>
@@ -464,7 +465,7 @@ onMounted(() => {
           <article
             v-for="topic in hotTopics"
             :key="topic.id"
-            @click="handleCreate(topic)"
+            @click="handleSelectTopic(topic)"
             class="topic-card"
             :class="{ active: selectedTopic === topic.id }"
           >
@@ -535,16 +536,6 @@ onMounted(() => {
       </header>
 
       <div class="editor-section">
-        <!-- 标题 -->
-        <div class="form-group">
-          <label class="form-label">标题</label>
-          <a-input
-            v-model:value="topicDetails.title"
-            placeholder="选择热点或输入标题"
-            class="form-input"
-          />
-        </div>
-
         <!-- 文案 -->
         <div class="form-group">
           <div class="form-label-row">
@@ -566,21 +557,75 @@ onMounted(() => {
         <!-- 风格选择 -->
         <div class="form-group">
           <label class="form-label">风格</label>
-          <PromptSelector
-            v-model="topicDetails.stylePromptId"
-            :prompts="allPrompts"
-            :loading="loadingPrompts"
-            :search-keyword="promptSearchKeyword"
-            @change="handlePromptChange"
-            @update:searchKeyword="promptSearchKeyword = $event"
-          />
+          <a-select
+            v-model:value="topicDetails.selectedAgentId"
+            placeholder="选择文案风格"
+            style="width: 100%"
+            :loading="loadingAgents"
+            show-search
+            :filter-option="false"
+            @change="handleAgentChange"
+            @search="agentSearchKeyword = $event"
+          >
+            <a-select-option
+              v-for="agent in filteredAgentList"
+              :key="agent.id"
+              :value="agent.id"
+            >
+              <div class="agent-option">
+                <span class="agent-option-name">{{ agent.name }}</span>
+                <span class="agent-option-category">{{ agent.categoryName }}</span>
+              </div>
+            </a-select-option>
+          </a-select>
+        </div>
+
+        <!-- 模型类型切换 -->
+        <div class="form-group">
+          <label class="form-label">模式</label>
+          <div class="model-type-switch">
+            <button
+              :class="['type-btn', { active: topicDetails.modelType === 'forecast_standard' }]"
+              @click="topicDetails.modelType = 'forecast_standard'"
+            >
+              标准版
+            </button>
+            <button
+              :class="['type-btn', { active: topicDetails.modelType === 'forecast_meiju' }]"
+              @click="topicDetails.modelType = 'forecast_meiju'"
+            >
+              美剧版
+            </button>
+          </div>
+        </div>
+
+        <!-- 改写幅度 -->
+        <div class="form-group">
+          <div class="form-label-row">
+            <label class="form-label">改写幅度</label>
+            <span class="level-value">{{ topicDetails.level }}%</span>
+          </div>
+          <div class="level-slider">
+            <input
+              type="range"
+              v-model.number="topicDetails.level"
+              min="10"
+              max="100"
+              step="10"
+              class="slider"
+            />
+            <div class="level-labels">
+              <span>保守</span>
+              <span>激进</span>
+            </div>
+          </div>
         </div>
 
         <!-- 生成按钮 -->
         <button
           class="generate-btn"
           :class="{ generating: isGenerating }"
-          :disabled="!topicDetails.copywriting?.trim() || !topicDetails.stylePromptId || isGenerating"
+          :disabled="!topicDetails.copywriting?.trim() || !topicDetails.selectedAgentId || isGenerating"
           @click="handleGenerate"
         >
           <span v-if="isGenerating" class="btn-loading">
@@ -1098,6 +1143,90 @@ onMounted(() => {
   50% { opacity: 0.5; transform: scale(0.8); }
 }
 
+// 模型类型切换
+.model-type-switch {
+  display: flex;
+  gap: 8px;
+}
+
+.type-btn {
+  flex: 1;
+  padding: 10px 16px;
+  font-size: 13px;
+  font-weight: 500;
+  color: @text-secondary;
+  background: @bg;
+  border: 1px solid @border;
+  border-radius: @radius-sm;
+  cursor: pointer;
+  transition: all 0.2s;
+
+  &:hover {
+    color: @text;
+    border-color: rgba(255, 255, 255, 0.15);
+  }
+
+  &.active {
+    color: #fff;
+    background: @primary;
+    border-color: @primary;
+  }
+}
+
+// 改写幅度滑块
+.level-value {
+  font-size: 13px;
+  font-weight: 600;
+  color: @primary;
+}
+
+.level-slider {
+  padding: 4px 0;
+}
+
+.slider {
+  width: 100%;
+  height: 6px;
+  -webkit-appearance: none;
+  appearance: none;
+  background: @bg;
+  border-radius: 3px;
+  outline: none;
+
+  &::-webkit-slider-thumb {
+    -webkit-appearance: none;
+    appearance: none;
+    width: 18px;
+    height: 18px;
+    background: @primary;
+    border-radius: 50%;
+    cursor: pointer;
+    transition: transform 0.15s, box-shadow 0.15s;
+
+    &:hover {
+      transform: scale(1.1);
+      box-shadow: 0 0 0 4px fade(@primary-color, 20%);
+    }
+  }
+
+  &::-moz-range-thumb {
+    width: 18px;
+    height: 18px;
+    background: @primary;
+    border: none;
+    border-radius: 50%;
+    cursor: pointer;
+  }
+}
+
+.level-labels {
+  display: flex;
+  justify-content: space-between;
+  margin-top: 6px;
+  font-size: 11px;
+  color: @text-secondary;
+}
+
 // 生成按钮
 .generate-btn {
   width: 100%;
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/client/DifyClient.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/client/DifyClient.java
index e32c4ccf5b..0c8751bc94 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/client/DifyClient.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/client/DifyClient.java
@@ -84,6 +84,55 @@ public class DifyClient {
                 .doOnError(e -> log.error("[chatStream] Dify 流式响应错误", e));
     }
 
+    /**
+     * 调用 Dify 聊天流式 API（支持自定义 inputs）
+     *
+     * @param apiKey         Dify API Key
+     * @param inputs         自定义输入参数
+     * @param content        用户输入
+     * @param conversationId 会话ID（可选）
+     * @return 流式响应
+     */
+    public Flux<DifyChatRespVO> chatStreamWithInputs(String apiKey, Map<String, Object> inputs, String content, String conversationId) {
+        String apiUrl = difyProperties.getApiUrl() + "/v1/chat-messages";
+
+        Map<String, Object> requestBody = new HashMap<>();
+        requestBody.put("inputs", inputs);
+        requestBody.put("query", content);
+        requestBody.put("response_mode", "streaming");
+        requestBody.put("conversation_id", conversationId != null ? conversationId : "");
+        requestBody.put("user", "user-" + System.currentTimeMillis());
+
+        AtomicReference<String> responseConversationId = new AtomicReference<>(conversationId);
+        StringBuilder fullContent = new StringBuilder();
+
+        WebClient webClient = WebClient.builder()
+                .baseUrl(apiUrl)
+                .defaultHeader("Authorization", "Bearer " + apiKey)
+                .defaultHeader("Content-Type", "application/json")
+                .build();
+
+        return webClient.post()
+                .bodyValue(requestBody)
+                .accept(MediaType.TEXT_EVENT_STREAM)
+                .retrieve()
+                .bodyToFlux(String.class)
+                .flatMap(event -> Mono.justOrEmpty(parseSSEEvent(event)))
+                .doOnNext(resp -> {
+                    if (resp.getConversationId() != null) {
+                        responseConversationId.set(resp.getConversationId());
+                    }
+                    if (resp.getContent() != null) {
+                        fullContent.append(resp.getContent());
+                    }
+                })
+                .doOnComplete(() -> {
+                    log.info("[chatStreamWithInputs] Dify 流式响应完成，会话ID: {}, 内容长度: {}",
+                            responseConversationId.get(), fullContent.length());
+                })
+                .doOnError(e -> log.error("[chatStreamWithInputs] Dify 流式响应错误", e));
+    }
+
     /**
      * 解析 SSE 事件
      */
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/controller/AppDifyController.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/controller/AppDifyController.java
index 22f4ee7994..d8dd446c46 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/controller/AppDifyController.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/controller/AppDifyController.java
@@ -5,6 +5,7 @@ import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
 import cn.iocoder.yudao.module.tik.dify.service.DifyService;
 import cn.iocoder.yudao.module.tik.dify.vo.DifyChatReqVO;
 import cn.iocoder.yudao.module.tik.dify.vo.DifyChatRespVO;
+import cn.iocoder.yudao.module.tik.dify.vo.ForecastRewriteReqVO;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.annotation.Resource;
@@ -39,4 +40,14 @@ public class AppDifyController {
                 .map(CommonResult::success);
     }
 
+    @PostMapping(value = "/forecast/rewrite", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    @Operation(summary = "Forecast 文案改写（流式）")
+    public Flux<CommonResult<DifyChatRespVO>> rewriteStream(@Valid @RequestBody ForecastRewriteReqVO reqVO) {
+        Long loginUserId = SecurityFrameworkUtils.getLoginUserId();
+        String userId = loginUserId != null ? loginUserId.toString() : "1"; // 默认用户ID
+
+        return difyService.rewriteStream(reqVO, userId)
+                .map(CommonResult::success);
+    }
+
 }
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyService.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyService.java
index 900e292bae..20f9a322a5 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyService.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyService.java
@@ -2,6 +2,7 @@ package cn.iocoder.yudao.module.tik.dify.service;
 
 import cn.iocoder.yudao.module.tik.dify.vo.DifyChatReqVO;
 import cn.iocoder.yudao.module.tik.dify.vo.DifyChatRespVO;
+import cn.iocoder.yudao.module.tik.dify.vo.ForecastRewriteReqVO;
 import reactor.core.publisher.Flux;
 
 /**
@@ -20,4 +21,13 @@ public interface DifyService {
      */
     Flux<DifyChatRespVO> chatStream(DifyChatReqVO reqVO, String userId);
 
+    /**
+     * Forecast 文案改写（流式）
+     *
+     * @param reqVO  请求参数
+     * @param userId 用户ID
+     * @return 流式响应
+     */
+    Flux<DifyChatRespVO> rewriteStream(ForecastRewriteReqVO reqVO, String userId);
+
 }
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java
index 6993df6dc6..87cead51db 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java
@@ -3,6 +3,7 @@ package cn.iocoder.yudao.module.tik.dify.service;
 import cn.iocoder.yudao.module.tik.dify.client.DifyClient;
 import cn.iocoder.yudao.module.tik.dify.vo.DifyChatReqVO;
 import cn.iocoder.yudao.module.tik.dify.vo.DifyChatRespVO;
+import cn.iocoder.yudao.module.tik.dify.vo.ForecastRewriteReqVO;
 import cn.iocoder.yudao.module.tik.enums.AiModelTypeEnum;
 import cn.iocoder.yudao.module.tik.enums.AiPlatformEnum;
 import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentDO;
@@ -16,6 +17,8 @@ import org.springframework.validation.annotation.Validated;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
+import java.util.HashMap;
+import java.util.Map;
 import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicReference;
 
@@ -133,9 +136,114 @@ public class DifyServiceImpl implements DifyService {
                 });
     }
 
+    @Override
+    public Flux<DifyChatRespVO> rewriteStream(ForecastRewriteReqVO reqVO, String userId) {
+        // 用于存储预扣记录ID
+        AtomicLong pendingRecordId = new AtomicLong();
+        // 用于存储会话ID
+        AtomicReference<String> conversationIdRef = new AtomicReference<>("");
+
+        return Mono.fromCallable(() -> {
+                    // 1. 获取智能体配置（通过 agentId 获取 systemPrompt）
+                    AiAgentDO agent = aiAgentService.getAiAgent(reqVO.getAgentId());
+                    if (agent == null) {
+                        throw new RuntimeException("智能体不存在");
+                    }
+
+                    // 2. 根据 modelType 获取对应的积分配置
+                    AiModelTypeEnum modelTypeEnum = "forecast_meiju".equals(reqVO.getModelType())
+                            ? AiModelTypeEnum.FORECAST_MEIJU
+                            : AiModelTypeEnum.FORECAST_STANDARD;
+                    AiModelConfigDO config = pointsService.getConfig(
+                            AiPlatformEnum.DIFY.getPlatform(),
+                            modelTypeEnum.getModelCode());
+
+                    // 3. 预检积分
+                    pointsService.checkPoints(userId, config.getConsumePoints());
+
+                    // 4. 创建预扣记录
+                    Long recordId = pointsService.createPendingDeduct(
+                            userId,
+                            config.getConsumePoints(),
+                            "forecast_rewrite",
+                            reqVO.getModelType()
+                    );
+                    pendingRecordId.set(recordId);
+
+                    // 5. 构建 inputs 参数（使用 agent 的 systemPrompt）
+                    Map<String, Object> inputs = new HashMap<>();
+                    inputs.put("sysPrompt", agent.getSystemPrompt());
+                    inputs.put("userText", reqVO.getUserText());
+                    inputs.put("level", reqVO.getLevel());
+
+                    // 6. 返回调用参数
+                    return new ForecastRewriteContext(inputs, config.getApiKey(), config.getConsumePoints());
+                })
+                .flatMapMany(context -> {
+                    // 7. 调用 Dify 流式 API
+                    return difyClient.chatStreamWithInputs(
+                                    context.apiKey(),
+                                    context.inputs(),
+                                    reqVO.getUserText(),
+                                    null
+                            )
+                            .doOnNext(resp -> {
+                                if (resp.getConversationId() != null) {
+                                    conversationIdRef.set(resp.getConversationId());
+                                }
+                            })
+                            // 8. 流结束时确认扣费
+                            .doOnComplete(() -> {
+                                if (pendingRecordId.get() > 0) {
+                                    try {
+                                        pointsService.confirmPendingDeduct(pendingRecordId.get());
+                                        log.info("[rewriteStream] 流结束，确认扣费，记录ID: {}", pendingRecordId.get());
+                                    } catch (Exception e) {
+                                        log.error("[rewriteStream] 确认扣费失败", e);
+                                    }
+                                }
+                            })
+                            // 9. 流出错时取消预扣
+                            .doOnError(e -> {
+                                if (pendingRecordId.get() > 0) {
+                                    try {
+                                        pointsService.cancelPendingDeduct(pendingRecordId.get());
+                                        log.info("[rewriteStream] 流出错，取消预扣，记录ID: {}", pendingRecordId.get());
+                                    } catch (Exception ex) {
+                                        log.error("[rewriteStream] 取消预扣失败", ex);
+                                    }
+                                }
+                            })
+                            // 10. 用户取消时确认扣费（已消费的部分）
+                            .doOnCancel(() -> {
+                                if (pendingRecordId.get() > 0) {
+                                    try {
+                                        pointsService.confirmPendingDeduct(pendingRecordId.get());
+                                        log.info("[rewriteStream] 用户取消，确认扣费，记录ID: {}", pendingRecordId.get());
+                                    } catch (Exception e) {
+                                        log.error("[rewriteStream] 用户取消后扣费失败", e);
+                                    }
+                                }
+                            });
+                })
+                // 11. 在最后添加 done 事件
+                .concatWith(Mono.defer(() -> {
+                    return Mono.just(DifyChatRespVO.done(conversationIdRef.get(), null));
+                }))
+                .onErrorResume(e -> {
+                    log.error("[rewriteStream] Forecast 文案改写异常", e);
+                    return Flux.just(DifyChatRespVO.error(e.getMessage()));
+                });
+    }
+
     /**
      * Dify 聊天上下文
      */
     private record DifyChatContext(String systemPrompt, String apiKey, Integer consumePoints) {}
 
+    /**
+     * Forecast 改写上下文
+     */
+    private record ForecastRewriteContext(Map<String, Object> inputs, String apiKey, Integer consumePoints) {}
+
 }
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/vo/ForecastRewriteReqVO.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/vo/ForecastRewriteReqVO.java
new file mode 100644
index 0000000000..48a6426ece
--- /dev/null
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/vo/ForecastRewriteReqVO.java
@@ -0,0 +1,29 @@
+package cn.iocoder.yudao.module.tik.dify.vo;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import jakarta.validation.constraints.NotEmpty;
+import jakarta.validation.constraints.NotNull;
+import lombok.Data;
+
+/**
+ * Forecast 文案改写请求 VO
+ */
+@Schema(description = "Forecast 文案改写请求")
+@Data
+public class ForecastRewriteReqVO {
+
+    @Schema(description = "智能体ID", requiredMode = Schema.RequiredMode.REQUIRED)
+    @NotNull(message = "智能体ID不能为空")
+    private Long agentId;
+
+    @Schema(description = "用户输入文案", requiredMode = Schema.RequiredMode.REQUIRED)
+    @NotEmpty(message = "用户文案不能为空")
+    private String userText;
+
+    @Schema(description = "改写级别/强度", example = "50")
+    private Integer level = 50;
+
+    @Schema(description = "模型类型：forecast_standard-标准版 forecast_meiju-美剧版", example = "forecast_standard")
+    private String modelType = "forecast_standard";
+
+}
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/enums/AiModelTypeEnum.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/enums/AiModelTypeEnum.java
index bd4a1ae9f0..18d3c4901d 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/enums/AiModelTypeEnum.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/enums/AiModelTypeEnum.java
@@ -20,6 +20,10 @@ public enum AiModelTypeEnum implements ArrayValuable<String> {
     DIFY_WRITING_PRO("writing_pro", "Pro深度版", AiPlatformEnum.DIFY, "text"),
     DIFY_WRITING_STANDARD("writing_standard", "标准版", AiPlatformEnum.DIFY, "text"),
 
+    // ========== Forecast 文案改写 ==========
+    FORECAST_STANDARD("forecast_standard", "文案改写-标准版", AiPlatformEnum.DIFY, "text"),
+    FORECAST_MEIJU("forecast_meiju", "文案改写-Pro版", AiPlatformEnum.DIFY, "text"),
+
     // ========== 数字人模型 ==========
     DIGITAL_HUMAN_LATENTSYNC("latentsync", "LatentSync", AiPlatformEnum.DIGITAL_HUMAN, "video"),
     DIGITAL_HUMAN_KLING("kling", "可灵", AiPlatformEnum.DIGITAL_HUMAN, "video"),