@allbluecn/web-app 0.6.3 → 0.7.0

package/src/server/ai-gateway/adapter.ts

@@ -0,0 +1,255 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { streamToText, type AnyTextAdapter, type StreamChunk } from "@tanstack/ai"
+import type { UnifiedAiStream } from "./unified-router"
+import { createOpenRouterText } from "@tanstack/ai-openrouter"
+import { openaiText } from "@tanstack/ai-openai"
+import { openaiCompatibleText } from "@tanstack/ai-openai/compatible"
+import { anthropicText } from "@tanstack/ai-anthropic"
+import { geminiText } from "@tanstack/ai-gemini"
+import { groqText } from "@tanstack/ai-groq"
+import { grokText } from "@tanstack/ai-grok"
+import { createOllamaChat } from "@tanstack/ai-ollama"
+import { prisma } from "@allbluecn/database"
+import type { AiConfig } from "./types"
+import { getPorts } from "./runtime"
+
+const PLACEHOLDER_KEYS = new Set(["sk-your-api-key-here", "", "sk-or-your-api-key-here"])
+
+const configCache = new Map<string, { data: AiConfig; at: number }>()
+const CACHE_TTL = 30_000
+
+function isValidKey(key: string | null | undefined): key is string {
+  return !!key && !PLACEHOLDER_KEYS.has(key)
+}
+
+function resolveEnvConfig(): { apiKey: string; model: string; baseUrl?: string } | null {
+  const openrouter = process.env.OPENROUTER_API_KEY
+  if (isValidKey(openrouter)) {
+    return {
+      apiKey: openrouter,
+      model: process.env.AI_MODEL || "openai/gpt-4o-mini",
+      ...(process.env.AI_BASE_URL ? { baseUrl: process.env.AI_BASE_URL } : {}),
+    }
+  }
+  const openai = process.env.OPENAI_API_KEY
+  if (isValidKey(openai)) {
+    return {
+      apiKey: openai,
+      model: process.env.AI_MODEL || "openai/gpt-4o-mini",
+      ...(process.env.AI_BASE_URL ? { baseUrl: process.env.AI_BASE_URL } : {}),
+    }
+  }
+  return null
+}
+
+export async function resolveAiConfig(userId?: string): Promise<AiConfig | null> {
+  if (userId) {
+    const cached = configCache.get(userId)
+    if (cached && Date.now() - cached.at < CACHE_TTL) return cached.data
+
+    const dbUser = await prisma.user.findUnique({
+      where: { id: userId },
+      select: { aiApiKey: true, aiProvider: true, aiModel: true },
+    })
+    if (dbUser && isValidKey(dbUser.aiApiKey)) {
+      const config: AiConfig = {
+        provider: (dbUser.aiProvider as "openrouter" | "openai") ?? "openrouter",
+        apiKey: dbUser.aiApiKey as string,
+        model: dbUser.aiModel || "openai/gpt-4o-mini",
+        source: "user",
+      }
+      configCache.set(userId, { data: config, at: Date.now() })
+      return config
+    }
+  }
+  const env = resolveEnvConfig()
+  if (!env) return null
+  return {
+    provider: "openrouter",
+    apiKey: env.apiKey,
+    model: env.model,
+    ...(env.baseUrl ? { baseUrl: env.baseUrl } : {}),
+    source: "env",
+  }
+}
+
+export async function hasAiKey(userId?: string): Promise<boolean> {
+  if (userId) {
+    // 新表：用户已启用服务商且 apiKey 有效
+    const cfg = await prisma.aiProviderConfig.findFirst({
+      where: { userId, enabled: true },
+      select: { apiKey: true },
+    })
+    if (cfg?.apiKey && isValidKey(getPorts().keyCipher.decrypt(cfg.apiKey))) return true
+  }
+  // 旧字段 / env 兜底
+  return (await resolveAiConfig(userId)) !== null
+}
+
+export function createAdapter(config: AiConfig) {
+  return createOpenRouterText(
+    config.model as Parameters<typeof createOpenRouterText>[0],
+    config.apiKey,
+    {
+      ...(config.baseUrl
+        ? { serverURL: config.baseUrl.replace(/\/chat\/completions\/?$/, "") }
+        : {}),
+      ...(process.env.VITE_APP_URL ? { httpReferer: process.env.VITE_APP_URL } : {}),
+      appTitle: "AllBlue",
+    },
+  )
+}
+
+/**
+ * 多 provider adapter 工厂：根据 providerId 从 DB 查 adapterKind，
+ * 再选择对应 TanStack AI adapter。统一返回 AnyTextAdapter 便于 chat() 接收。
+ */
+export async function createProviderAdapter(
+  providerId: string,
+  opts: { model: string; apiKey?: string | null; baseUrl?: string | null },
+): Promise<AnyTextAdapter> {
+  const provider = await prisma.aiProvider.findUnique({
+    where: { providerId },
+    select: { adapterKind: true, defaultBaseUrl: true, isLocal: true, models: true },
+  })
+  if (!provider) {
+    throw new Error(`未知的 AI 服务商：${providerId}`)
+  }
+
+  const baseUrl = opts.baseUrl || provider.defaultBaseUrl || undefined
+  const kind = provider.adapterKind
+
+  let adapter: AnyTextAdapter
+  switch (kind) {
+    case "openai":
+      adapter = openaiText(opts.model as never, {
+        ...(opts.apiKey ? { apiKey: opts.apiKey } : {}),
+        ...(baseUrl ? { baseURL: baseUrl } : {}),
+      }) as unknown as AnyTextAdapter
+      break
+    case "anthropic":
+      adapter = anthropicText(opts.model as never, {
+        ...(opts.apiKey ? { apiKey: opts.apiKey } : {}),
+        ...(baseUrl ? { baseURL: baseUrl } : {}),
+      }) as unknown as AnyTextAdapter
+      break
+    case "gemini":
+      adapter = geminiText(opts.model as never, {
+        ...(opts.apiKey ? { apiKey: opts.apiKey } : {}),
+        // @google/genai 改用 httpOptions.baseUrl（旧顶层 baseURL/baseUrl 已移除）
+        ...(baseUrl ? { httpOptions: { baseUrl } } : {}),
+      }) as unknown as AnyTextAdapter
+      break
+    case "groq":
+      adapter = groqText(opts.model as never, {
+        ...(opts.apiKey ? { apiKey: opts.apiKey } : {}),
+        ...(baseUrl ? { baseURL: baseUrl } : {}),
+      }) as unknown as AnyTextAdapter
+      break
+    case "grok":
+      adapter = grokText(opts.model as never, {
+        ...(opts.apiKey ? { apiKey: opts.apiKey } : {}),
+        ...(baseUrl ? { baseURL: baseUrl } : {}),
+      }) as unknown as AnyTextAdapter
+      break
+    case "openrouter":
+      adapter = createOpenRouterText(
+        opts.model as Parameters<typeof createOpenRouterText>[0],
+        opts.apiKey ?? "",
+        {
+          ...(baseUrl ? { serverURL: baseUrl } : {}),
+          ...(process.env.VITE_APP_URL ? { httpReferer: process.env.VITE_APP_URL } : {}),
+          appTitle: "AllBlue",
+        },
+      ) as unknown as AnyTextAdapter
+      break
+    case "ollama":
+      adapter = createOllamaChat(
+        opts.model as never,
+        baseUrl ?? undefined,
+      ) as unknown as AnyTextAdapter
+      break
+    case "openai-compatible": {
+      const finalBaseUrl = baseUrl ?? provider.defaultBaseUrl
+      if (!finalBaseUrl) {
+        throw new Error(`provider ${providerId} 缺少 baseUrl`)
+      }
+      adapter = openaiCompatibleText(opts.model, {
+        name: providerId,
+        baseURL: finalBaseUrl,
+        apiKey: opts.apiKey ?? "",
+      }) as unknown as AnyTextAdapter
+      break
+    }
+    default:
+      throw new Error(`不支持的 adapter 类型：${kind}`)
+  }
+  return adapter
+}
+export async function runChatStream(
+  messages: Array<{ role: "user" | "assistant" | "tool"; content: string }>,
+  options?: {
+    systemPrompts?: string[]
+    userId?: string
+    workspaceId?: string
+    model?: string
+  },
+): Promise<UnifiedAiStream> {
+  if (!options?.userId) {
+    throw new Error("userId required for AI calls")
+  }
+  const { resolveUnifiedAiStream } = await import("./unified-router")
+  return resolveUnifiedAiStream({
+    userId: options.userId,
+    messages,
+    model: options.model,
+    workspaceId: options.workspaceId,
+    systemPrompts: options.systemPrompts,
+  })
+}
+
+export async function runChat(
+  messages: Array<{ role: "user" | "assistant" | "tool"; content: string }>,
+  options?: {
+    systemPrompts?: string[]
+    modelOptions?: Record<string, unknown>
+    userId?: string
+    workspaceId?: string
+    model?: string
+  },
+): Promise<string> {
+  const stream = await runChatStream(messages, options)
+  return streamToText(stream)
+}
+
+export async function runChatWithModel(
+  messages: Array<{ role: "user" | "assistant" | "tool"; content: string }>,
+  options: {
+    systemPrompts?: string[]
+    userId?: string
+    model?: string
+    workspaceId?: string
+  },
+): Promise<string> {
+  const stream = await runChatStream(messages, {
+    ...options,
+    model: options.model,
+  })
+  return streamToText(stream)
+}

package/src/server/ai-gateway/chat-handler.ts

@@ -0,0 +1,238 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { type StreamChunk } from "@tanstack/ai"
+import { resolveUnifiedAiRoute } from "./unified-router"
+import { getPorts } from "./runtime"
+import { z } from "zod"
+import type { AgentToolKey } from "./tools/select"
+
+function mergeChatOverrides(overrides?: ChatRequestOverrides): Partial<z.infer<typeof chatBodySchema>> {
+  if (!overrides) return {}
+  const persona = overrides.agent?.persona
+  return {
+    ...(persona && { systemPrompts: [persona] }),
+    ...(overrides.allowedTools && { allowedTools: overrides.allowedTools }),
+    ...(overrides.knowledgeBaseIds && { knowledgeBaseIds: overrides.knowledgeBaseIds }),
+    ...(overrides.temperature != null && { temperature: overrides.temperature }),
+  }
+}
+
+const TIME_SENSITIVE_KEYWORDS = [
+  "日期", "时间", "今天", "明天", "昨天", "星期", "年", "月", "日",
+  "新闻", "最新", "实时", "天气", "现在",
+  "当前", "今年", "本月", "上周", "下周",
+  "data", "news", "current", "today", "weather",
+  "update", "latest", "recent",
+]
+
+function needsWebSearch(text: string): boolean {
+  return TIME_SENSITIVE_KEYWORDS.some(kw => text.includes(kw))
+}
+
+export interface ChatRequestOverrides {
+  agent?: { persona: string; knowledgeBaseIds?: string[]; allowedTools?: AgentToolKey[]; temperature?: number }
+  allowedTools?: AgentToolKey[]
+  knowledgeBaseIds?: string[]
+  temperature?: number
+}
+
+export interface HandleChatOptions {
+  body?: Record<string, unknown>
+  overrides?: ChatRequestOverrides
+}
+
+const chatBodySchema = z.object({
+  messages: z.array(
+    z
+      .object({
+        role: z.enum(["user", "assistant", "tool"]),
+        content: z.string().optional(),
+        parts: z.array(z.any()).optional(),
+      })
+      .passthrough(),
+  ).default([]),
+  systemPrompts: z.array(z.string()).default([]),
+  allowedTools: z.array(z.string()).default([]),
+  knowledgeBaseIds: z.array(z.string()).default([]),
+  temperature: z.number().optional(),
+  webSearchEnabled: z.boolean().default(true),
+  model: z.string().optional(),
+  workspaceId: z.string().optional(),
+  threadId: z.string().optional(),
+  runId: z.string().optional(),
+  resume: z.array(z.any()).optional(),
+})
+
+function extractText(msg: {
+  content?: unknown
+  parts?: unknown
+}): string {
+  if (typeof msg.content === "string") return msg.content
+  if (Array.isArray(msg.parts)) {
+    return msg.parts
+      .filter(
+        (p: any) => p?.type === "text" && typeof p?.content === "string",
+      )
+      .map((p: any) => p.content as string)
+      .join("")
+  }
+  return ""
+}
+
+export async function* handleChatRequest(
+  request: Request,
+  options?: HandleChatOptions,
+): AsyncIterable<StreamChunk> {
+  const session = await getPorts().auth.getSession(request)
+  if (!session?.user?.id) {
+    throw new Error("Unauthorized")
+  }
+
+  const raw = await (options?.body ?? request.json())
+  const body = { ...raw, ...(raw.forwardedProps ?? raw.data ?? {}) }
+  const merged = { ...mergeChatOverrides(options?.overrides), ...body }
+  if (options?.overrides?.agent?.persona) {
+    const existingPrompts = (merged as any).systemPrompts ?? []
+    merged.systemPrompts = [...[options.overrides.agent.persona], ...existingPrompts]
+  }
+  const parsed = chatBodySchema.safeParse(merged)
+  if (!parsed.success) {
+    const issue = parsed.error.issues[0]
+    const path = issue?.path?.join(".") ?? "?"
+    throw new Error(
+      `Invalid chat request body [${path}]: ${issue?.message ?? "validation failed"}`,
+    )
+  }
+  const { messages, systemPrompts, allowedTools, knowledgeBaseIds, temperature, webSearchEnabled, model, workspaceId, threadId, runId, resume } = parsed.data
+
+  const lastUserMsg = [...messages].reverse().find((m) => m.role === "user")
+  let citations: Array<{ id: string; number: number; text?: string; sources: Array<{ id: string; title: string; url: string; snippet?: string; type: string }> }> = []
+  if (webSearchEnabled && lastUserMsg) {
+    try {
+      const today = new Date()
+      const dateStr = `${today.getFullYear()}年${today.getMonth() + 1}月${today.getDate()}日`
+      const searchQuery = extractText(lastUserMsg)
+      const searchResult = await getPorts().webSearch.search(searchQuery, 30)
+      const lines = [`【系统当前日期：${dateStr}】`]
+      if (searchResult.results.length > 0) {
+        lines.push("【联网搜索结果 - 这是真实的实时数据，你必须用此信息回答，严禁使用你的训练数据】")
+        searchResult.results.forEach((r, i) => {
+          lines.push(`[${i + 1}] ${r.title}\n    ${r.content}\n    来源: ${r.url}`)
+        })
+        citations = searchResult.results.map((r, i) => ({
+          id: `web-${i}`,
+          number: i + 1,
+          text: r.content,
+          sources: [{
+            id: r.url,
+            title: r.title,
+            url: r.url,
+            snippet: r.content,
+            type: 'web',
+          }],
+        }))
+      }
+      lines.push(`【当前真实日期是 ${dateStr}。你必须使用系统日期和搜索结果回答，绝对不允许使用你的训练数据中的日期。直接告诉用户今天的日期。】`)
+      systemPrompts.push(lines.join("\n\n"))
+      console.log(`[AI] auto web search injected: ${searchResult.results.length} results, date=${dateStr}`)
+    } catch (e) {
+      console.warn("[AI] auto web search failed:", e)
+    }
+  }
+
+  if (citations.length > 0) {
+    yield { type: 'CUSTOM', name: 'citations', value: citations } as StreamChunk
+  }
+
+  const route = await resolveUnifiedAiRoute({
+    userId: session.user.id,
+    messages: messages as Array<{ role: 'user' | 'assistant' | 'tool'; content: string }>,
+    model,
+    workspaceId,
+    systemPrompts,
+    allowedTools,
+    knowledgeBaseIds,
+    temperature,
+    webSearchEnabled,
+    threadId,
+    runId,
+    resume,
+    signal: request.signal,
+    request,
+  }, request)
+
+  if (route.kind === 'remote') {
+    yield* forwardToSaas(route, request)
+    return
+  }
+
+  yield* route.stream
+}
+
+async function* forwardToSaas(
+  route: { kind: 'remote'; saasUrl: string; token: string; model?: string },
+  request: Request,
+): AsyncIterable<StreamChunk> {
+  const body = await request.json()
+  const fwdBody = {
+    ...body,
+    ...(body.forwardedProps ?? body.data ?? {}),
+    model: route.model ?? body.model,
+  }
+
+  const res = await fetch(`${route.saasUrl}/api/chat`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${route.token}`,
+    },
+    body: JSON.stringify(fwdBody),
+    signal: request.signal,
+  })
+  if (!res.ok || !res.body) {
+    throw new Error(`SaaS 托管 AI 转发失败: ${res.status}`)
+  }
+
+  const decoder = new TextDecoder()
+  let buffer = ''
+
+  const reader = res.body.getReader()
+  try {
+    while (true) {
+      const { done, value } = await reader.read()
+      if (done) break
+      buffer += decoder.decode(value, { stream: true })
+
+      const parts = buffer.split('\n\n')
+      buffer = parts.pop() ?? ''
+
+      for (const part of parts) {
+        if (!part.startsWith('data: ')) continue
+        const raw = part.slice(6)
+        if (!raw || raw === '[DONE]') continue
+        try {
+          yield JSON.parse(raw) as StreamChunk
+        } catch {
+          continue
+        }
+      }
+    }
+  } finally {
+    reader.releaseLock()
+  }
+}

package/src/server/ai-gateway/chat-stream-builder.ts

@@ -0,0 +1,91 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import type { StreamChunk } from '@tanstack/ai'
+import { chat } from '@tanstack/ai'
+import { prisma } from '@allbluecn/database'
+import { baseServerTools, serverTools } from './tools/server'
+import { buildProviderWebSearchTools } from './provider-tools'
+import { createPrismaPersistence } from './persistence'
+import { loggerMiddleware, rateLimitMiddleware } from './middlewares'
+import { withPersistence } from '@tanstack/ai-persistence'
+import type { AppContext } from './types'
+import { selectTools } from './tools/select'
+import { deleteProject } from './tools/delete-project'
+
+interface BuildChatStreamParams {
+  adapter: any
+  userId: string
+  messages: any[]
+  systemPrompts?: string[]
+  webSearchEnabled?: boolean
+  allowedTools?: string[]
+  knowledgeBaseIds?: string[]
+  temperature?: number
+  threadId?: string
+  runId?: string
+  resume?: any[]
+  providerConfigKind?: string
+}
+
+export function buildChatStreamWithTools(
+  params: BuildChatStreamParams,
+): AsyncIterable<StreamChunk> {
+  const nativeSearchTools = buildProviderWebSearchTools(
+    params.providerConfigKind,
+  ) as any[]
+
+  let basePool: any[]
+  if (nativeSearchTools.length > 0) {
+    basePool = [...baseServerTools, ...nativeSearchTools, deleteProject]
+  } else {
+    basePool = [...serverTools, deleteProject]
+  }
+
+  const tools = selectTools(basePool, params.allowedTools)
+
+  const persistence = createPrismaPersistence(params.userId)
+  const context: AppContext = {
+    userId: params.userId,
+    db: prisma,
+    knowledgeBaseIds: params.knowledgeBaseIds,
+  }
+
+  const modelOptions: Record<string, unknown> = {}
+  if (params.temperature != null) {
+    modelOptions.temperature = params.temperature
+  }
+
+  return chat({
+    adapter: params.adapter,
+    messages: params.messages,
+    ...(params.systemPrompts?.length
+      ? { systemPrompts: params.systemPrompts }
+      : {}),
+    tools,
+    ...(Object.keys(modelOptions).length > 0 ? { modelOptions } : {}),
+    threadId: params.threadId,
+    runId: params.runId,
+    resume: params.resume,
+    context,
+    middleware: [
+      loggerMiddleware,
+      rateLimitMiddleware,
+      withPersistence(persistence),
+    ] as any,
+  }) as AsyncIterable<StreamChunk>
+}