Fix AI stream by returning stream.value directly from Server Actions.

createStreamableValue must be created and returned in the action itself; wrapping in { data } or a helper return caused production RSC serialization errors.

Co-authored-by: Cursor <cursoragent@cursor.com>

app/actions/bazi.ts

@@ -1,6 +1,7 @@
 "use server";
 
-import { streamAIResponse } from "@/lib/ai/stream";
+import { createStreamableValue } from "ai/rsc";
+import { pumpAIStream } from "@/lib/ai/stream";
 import {
   calculateBazi,
   formatBaziForPrompt,
@@ -16,7 +17,9 @@ export async function getBaziAnswer(
   const chart = calculateBazi(input);
   const chartText = formatBaziForPrompt(chart);
 
-  return streamAIResponse(
+  const stream = createStreamableValue<string>();
+  pumpAIStream(
+    stream,
     BAZI_SYSTEM_PROMPT,
     `【出生时空】
 出生地：${birthPlaceName}
@@ -27,4 +30,5 @@ ${chartText}
 【问事】
 ${question}`,
   );
+  return stream.value;
 }

app/actions/combined.ts

@@ -1,6 +1,7 @@
 "use server";
 
-import { streamAIResponse } from "@/lib/ai/stream";
+import { createStreamableValue } from "ai/rsc";
+import { pumpAIStream } from "@/lib/ai/stream";
 import {
   calculateBazi,
   formatBaziForPrompt,
@@ -69,7 +70,9 @@ export async function getCombinedAnswer(input: CombinedInput) {
     }
   }
 
-  return streamAIResponse(
+  const stream = createStreamableValue<string>();
+  pumpAIStream(
+    stream,
     COMBINED_SYSTEM_PROMPT,
     `【人和 · 八字排盘】
 出生地：${input.birthPlaceName}
@@ -81,4 +84,5 @@ ${hexagramText}
 【问事】
 ${input.question}`,
   );
+  return stream.value;
 }

app/actions/liuyao.ts

@@ -1,6 +1,7 @@
 "use server";
 
-import { streamAIResponse } from "@/lib/ai/stream";
+import { createStreamableValue } from "ai/rsc";
+import { pumpAIStream } from "@/lib/ai/stream";
 import {
   formatLiuyaoTimingForPrompt,
   getTimingInfoWithLongitude,
@@ -51,7 +52,9 @@ export async function getLiuyaoAnswer(input: LiuyaoInput) {
     guaDetailText = "";
   }
 
-  return streamAIResponse(
+  const stream = createStreamableValue<string>();
+  pumpAIStream(
+    stream,
     LIUYAO_SYSTEM_PROMPT,
     `${timingText}
 
@@ -63,4 +66,5 @@ ${input.question}
 
 ${guaDetailText}`,
   );
+  return stream.value;
 }

components/modes/bazi-form.tsx

@@ -70,18 +70,13 @@ export default function BaziForm() {
     setIsLoading(true);
     setShowAi(true);
     try {
-      const { data, error: apiError } = await getBaziAnswer(
+      const stream = await getBaziAnswer(
         input,
         question,
         location!.name,
       );
-      if (apiError) {
-        setError(apiError);
-        return;
-      }
-      if (data) {
       let ret = "";
-        for await (const delta of readStreamableValue(data)) {
+      for await (const delta of readStreamableValue(stream)) {
         if (typeof delta === "string" && delta.startsWith(ERROR_PREFIX)) {
           setError(delta.slice(ERROR_PREFIX.length));
           return;
@@ -89,7 +84,6 @@ export default function BaziForm() {
         ret += delta ?? "";
         setCompletion(ret);
       }
-      }
     } catch (err) {
       setError(err instanceof Error ? err.message : String(err));
     } finally {

components/modes/combined-form.tsx

@@ -109,7 +109,7 @@ export default function CombinedForm() {
     setShowAi(true);
 
     try {
-      const { data, error: apiError } = await getCombinedAnswer({
+      const stream = await getCombinedAnswer({
         birth: {
           date: birthDate,
           time: unknownHour ? "12:00" : birthTime,
@@ -133,13 +133,8 @@ export default function CombinedForm() {
           : undefined,
       });
 
-      if (apiError) {
-        setError(apiError);
-        return;
-      }
-      if (data) {
       let ret = "";
-        for await (const delta of readStreamableValue(data)) {
+      for await (const delta of readStreamableValue(stream)) {
         if (typeof delta === "string" && delta.startsWith(ERROR_PREFIX)) {
           setError(delta.slice(ERROR_PREFIX.length));
           return;
@@ -147,7 +142,6 @@ export default function CombinedForm() {
         ret += delta ?? "";
         setCompletion(ret);
       }
-      }
     } catch (e) {
       setError(e instanceof Error ? e.message : String(e));
     } finally {

components/modes/liuyao-form.tsx

@@ -72,7 +72,7 @@ export default function LiuyaoForm() {
     setShowAi(true);
 
     try {
-      const { data, error: apiError } = await getLiuyaoAnswer({
+      const stream = await getLiuyaoAnswer({
         question,
         calcDate,
         calcTime,
@@ -84,13 +84,8 @@ export default function LiuyaoForm() {
         guaChange: guaData!.result.guaChange,
       });
 
-      if (apiError) {
-        setError(apiError);
-        return;
-      }
-      if (data) {
       let ret = "";
-        for await (const delta of readStreamableValue(data)) {
+      for await (const delta of readStreamableValue(stream)) {
         if (typeof delta === "string" && delta.startsWith(ERROR_PREFIX)) {
           setError(delta.slice(ERROR_PREFIX.length));
           return;
@@ -98,7 +93,6 @@ export default function LiuyaoForm() {
         ret += delta ?? "";
         setCompletion(ret);
       }
-      }
     } catch (e) {
       setError(e instanceof Error ? e.message : String(e));
     } finally {

lib/ai/stream.ts

@@ -3,6 +3,8 @@ import { createOpenAI } from "@ai-sdk/openai";
 import { createStreamableValue } from "ai/rsc";
 import { ERROR_PREFIX } from "@/lib/constant";
 
+export type AIStream = ReturnType<typeof createStreamableValue<string>>;
+
 function getOpenAiBaseUrl(): string {
   return (
     process.env.OPENAI_BASE_URL ??
@@ -14,23 +16,27 @@ function getOpenAiBaseUrl(): string {
 const model =
   process.env.OPENAI_MODEL ?? "huihui_ai/gemma-4-abliterated:e4b";
 
-export async function streamAIResponse(
+export function assertOpenAiKey(): string {
-  system: string,
-  user: string,
-): Promise<{ data?: ReturnType<typeof createStreamableValue<string>>["value"]; error?: string }> {
   const apiKey = process.env.OPENAI_API_KEY?.trim();
   if (!apiKey) {
-    return { error: "未配置 OPENAI_API_KEY，请在 .env.local 或 Docker env_file 中设置" };
+    throw new Error(
+      "未配置 OPENAI_API_KEY，请在 .env.local 或 Docker env_file 中设置",
+    );
+  }
+  return apiKey;
 }
 
+/** 在 Server Action 内 createStreamableValue 后调用，不可从子函数返回 stream.value */
+export function pumpAIStream(
+  stream: AIStream,
+  system: string,
+  user: string,
+): void {
   const openai = createOpenAI({
-    apiKey,
+    apiKey: assertOpenAiKey(),
     baseURL: getOpenAiBaseUrl(),
   });
 
-  const stream = createStreamableValue<string>();
-
-  try {
   const { fullStream } = streamText({
     temperature: 0.5,
     model: openai(model),
@@ -80,11 +86,4 @@ export async function streamAIResponse(
     .finally(() => {
       done = true;
     });
-
-    return { data: stream.value };
-  } catch (err) {
-    stream.done();
-    const message = err instanceof Error ? err.message : String(err);
-    return { error: message };
-  }
 }