mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-07 18:27:47 +08:00
dayuan.jiang
efd29e6599
The canvas now fills the window and the chat floats on its right. draw.io is served from our own origin (public/drawio, downloaded at build time and checked against the release's SHA-256), so the app can drive the editor directly: - every AI change becomes a version card with a thumbnail; compare, restore or undo it, and Ctrl+Z on the canvas takes an AI change back in one step - shapes the AI just changed are highlighted on the canvas - selected shapes appear as a chip and are sent to the model - the app's own canvas toolbar, shape library and page tabs replace draw.io's chrome - new start screen, session rename and switcher, open .drawio files - settings split into Models & keys, Appearance, Advanced and About - Bedrock in the model settings also takes an API key instead of access keys - phones switch between canvas and chat - dark mode keeps the diagram on a white sheet and no longer flashes With an external draw.io (NEXT_PUBLIC_DRAWIO_BASE_URL) the app falls back to draw.io's own toolbar and the features above that need same origin are off. The chat logic moved from the 1500-line chat panel into a chat engine plus small UI components; zustand holds canvas, versions, settings and UI state.
38 lines
1.5 KiB
TypeScript
38 lines
1.5 KiB
TypeScript
import type { getSelectedAIConfig } from "@/hooks/use-model-config"
|
|
|
|
type AIConfig = ReturnType<typeof getSelectedAIConfig>
|
|
|
|
/**
|
|
* Request headers for /api/chat: the access code, the user's own provider
|
|
* credentials (when a client-side model is selected) and per-request options.
|
|
*/
|
|
export function buildChatHeaders(
|
|
config: AIConfig,
|
|
options: { minimalStyle: boolean; maxOutputTokens: string },
|
|
): Record<string, string> {
|
|
const headers: Record<string, string> = {
|
|
"x-access-code": config.accessCode,
|
|
}
|
|
const set = (name: string, value: string | undefined) => {
|
|
if (value) headers[name] = value
|
|
}
|
|
if (config.aiProvider) {
|
|
set("x-ai-provider", config.aiProvider)
|
|
set("x-ai-base-url", config.aiBaseUrl)
|
|
set("x-ai-api-key", config.aiApiKey)
|
|
set("x-ai-model", config.aiModel)
|
|
// AWS Bedrock credentials
|
|
set("x-aws-access-key-id", config.awsAccessKeyId)
|
|
set("x-aws-secret-access-key", config.awsSecretAccessKey)
|
|
set("x-aws-region", config.awsRegion)
|
|
set("x-aws-session-token", config.awsSessionToken)
|
|
// Vertex AI credentials (Express Mode)
|
|
set("x-vertex-api-key", config.vertexApiKey)
|
|
}
|
|
// Selected model ID for server model lookup (apiKeyEnv/baseUrlEnv)
|
|
set("x-selected-model-id", config.selectedModelId)
|
|
if (options.minimalStyle) headers["x-minimal-style"] = "true"
|
|
set("x-max-output-tokens", options.maxOutputTokens)
|
|
return headers
|
|
}
|