@milkdown/crepe 7.19.2 → 7.21.0

package/src/icons/chevron-right.ts

@@ -0,0 +1,15 @@
+export const chevronRightIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    fill="none"
+    viewBox="0 0 24 24"
+    stroke-width="1.5"
+    stroke="currentColor"
+  >
+    <path
+      stroke-linecap="round"
+      stroke-linejoin="round"
+      d="M8.25 4.5l7.5 7.5-7.5 7.5"
+    />
+  </svg>
+`

package/src/icons/code-block.ts

@@ -0,0 +1,12 @@
+export const codeBlockIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      d="M3 3h18a1 1 0 0 1 1 1v16a1 1 0 0 1-1 1H3a1 1 0 0 1-1-1V4a1 1 0 0 1 1-1zm1 2v14h16V5H4zm8 10h6v2h-6v-2zm-3.333-3L5.838 9.172l1.415-1.415L11.495 12l-4.242 4.243-1.415-1.415L8.667 12z"
+    />
+  </svg>
+`

package/src/icons/enter-key.ts

@@ -0,0 +1,13 @@
+export const enterKeyIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M9 16l-5-5l5-5l1.4 1.4L7.825 10H18V6h2v6H7.825l2.575 2.6z"
+    />
+  </svg>
+`

package/src/icons/grammar-check.ts

@@ -0,0 +1,13 @@
+export const grammarCheckIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M9.55 17.575L4.225 12.25l1.4-1.425L9.55 14.75l8.825-8.825l1.4 1.425z"
+    />
+  </svg>
+`

package/src/icons/index.ts

@@ -1,3 +1,4 @@
+export * from './ai'
 export * from './align-center'
 export * from './align-left'
 export * from './align-right'
@@ -8,13 +9,18 @@ export * from './caption'
 export * from './check-box-checked'
 export * from './check-box-unchecked'
 export * from './chevron-down'
+export * from './chevron-left'
+export * from './chevron-right'
 export * from './clear'
 export * from './code'
+export * from './code-block'
 export * from './confirm'
 export * from './copy'
 export * from './divider'
 export * from './drag-handle'
 export * from './edit'
+export * from './enter-key'
+export * from './grammar-check'
 export * from './h1'
 export * from './h2'
 export * from './h3'
@@ -24,15 +30,21 @@ export * from './h6'
 export * from './image'
 export * from './italic'
 export * from './link'
+export * from './longer'
 export * from './menu'
 export * from './ordered-list'
 export * from './plus'
 export * from './quote'
 export * from './remove'
+export * from './retry'
 export * from './search'
+export * from './send'
+export * from './send-prompt'
+export * from './shorter'
 export * from './strikethrough'
 export * from './table'
 export * from './text'
 export * from './todo-list'
+export * from './translate'
 export * from './functions'
 export * from './visibility-off'

package/src/icons/longer.ts

@@ -0,0 +1,13 @@
+export const longerIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M3 18v-2h13v2zm0-5v-2h18v2zm0-5V6h18v2zm15 13v-3h-3v-2h3v-3h2v3h3v2h-3v3z"
+    />
+  </svg>
+`

package/src/icons/retry.ts

@@ -0,0 +1,13 @@
+export const retryIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M12 20q-3.35 0-5.675-2.325T4 12q0-3.35 2.325-5.675T12 4q1.725 0 3.3.713T18 6.75V4h2v7h-7V9h4.2q-.8-1.4-2.187-2.2T12 6Q9.5 6 7.75 7.75T6 12q0 2.5 1.75 4.25T12 18q1.925 0 3.475-1.1T17.65 14h2.1q-.7 2.65-2.85 4.325T12 20"
+    />
+  </svg>
+`

package/src/icons/send-prompt.ts

@@ -0,0 +1,13 @@
+export const sendPromptIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M3 20v-6l8-2l-8-2V4l19 8z"
+    />
+  </svg>
+`

package/src/icons/send.ts

@@ -0,0 +1,13 @@
+export const sendIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M11 20V7.825l-5.6 5.6L4 12l8-8l8 8l-1.4 1.425l-5.6-5.6V20z"
+    />
+  </svg>
+`

package/src/icons/shorter.ts

@@ -0,0 +1,13 @@
+export const shorterIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="M5 13v-2h14v2z"
+    />
+  </svg>
+`

package/src/icons/translate.ts

@@ -0,0 +1,13 @@
+export const translateIcon = `
+  <svg
+    xmlns="http://www.w3.org/2000/svg"
+    width="24"
+    height="24"
+    viewBox="0 0 24 24"
+  >
+    <path
+      fill="currentColor"
+      d="m12.9 16l-2.55-2.5l.05-.05q1.45-1.6 2.262-3.117T13.95 7H16V5h-6V3H8v2H2v2h9.1q-.5 1.2-1.225 2.387T8 11.7q-.6-.6-1.187-1.412T5.85 8.6H3.85q.5 1.4 1.3 2.7T6.95 13.7L2.85 17.75L4.25 19.15L8 15.4L10.55 17.95zM18.5 22h-2L21 10h2l4.5 12h-2l-1.05-3h-4.9zm1.6-5h3.6l-1.8-5.1z"
+    />
+  </svg>
+`

package/src/llm-providers/anthropic/index.ts

@@ -0,0 +1,132 @@
+import type { AIPromptContext, AIProvider } from '../../feature/ai/types'
+
+import {
+  type BaseProviderConfig,
+  assertBrowserSafe,
+  buildDefaultUserMessage,
+  parseSSE,
+  readErrorBody,
+  resolveSystemPrompt,
+} from '../shared'
+
+const PROVIDER_NAME = 'milkdown/providers/anthropic'
+const DEFAULT_BASE_URL = 'https://api.anthropic.com'
+const DEFAULT_ANTHROPIC_VERSION = '2023-06-01'
+const DEFAULT_MAX_TOKENS = 4096
+
+export interface AnthropicMessage {
+  role: 'user' | 'assistant'
+  content: string
+}
+
+export interface AnthropicBuildMessagesResult {
+  /// Anthropic puts the system prompt in a top-level `system` field,
+  /// not in the messages array.
+  system?: string
+  messages: AnthropicMessage[]
+}
+
+export interface AnthropicProviderConfig extends BaseProviderConfig {
+  /// Anthropic's API requires `max_tokens`. Defaults to 4096.
+  maxTokens?: number
+
+  /// `anthropic-version` header. Defaults to `2023-06-01`.
+  anthropicVersion?: string
+
+  /// Customize the request payload. The default builder maps the
+  /// resolved system prompt to `system` and produces a single user
+  /// message containing the document, selection, and instruction.
+  buildMessages?: (
+    context: AIPromptContext,
+    defaults: { systemPrompt: string | null; userMessage: string }
+  ) => AnthropicBuildMessagesResult
+
+  /// Extra fields merged into the request body (e.g. `temperature`,
+  /// `top_p`, `metadata`, `tool_choice`).
+  body?: Record<string, unknown>
+}
+
+interface AnthropicStreamEvent {
+  type: string
+  delta?: {
+    type?: string
+    text?: string
+  }
+}
+
+/// Build an `AIProvider` backed by Anthropic's `/v1/messages` streaming
+/// endpoint. The returned function plugs directly into
+/// `Crepe.Feature.AI`'s `provider` field.
+export function createAnthropicProvider(
+  config: AnthropicProviderConfig
+): AIProvider {
+  assertBrowserSafe(config, PROVIDER_NAME)
+
+  return async function* anthropicProvider(context, signal) {
+    const systemPrompt = resolveSystemPrompt(config.systemPrompt)
+    const userMessage = buildDefaultUserMessage(context)
+
+    const built = config.buildMessages
+      ? config.buildMessages(context, { systemPrompt, userMessage })
+      : {
+          system: systemPrompt ?? undefined,
+          messages: [
+            { role: 'user' as const, content: userMessage },
+          ] satisfies AnthropicMessage[],
+        }
+
+    const baseURL = (config.baseURL ?? DEFAULT_BASE_URL).replace(/\/+$/, '')
+    const url = `${baseURL}/v1/messages`
+
+    const headers: Record<string, string> = {
+      'Content-Type': 'application/json',
+      'anthropic-version': config.anthropicVersion ?? DEFAULT_ANTHROPIC_VERSION,
+      ...(config.apiKey ? { 'x-api-key': config.apiKey } : {}),
+      // Required for direct browser → api.anthropic.com calls; harmless
+      // when proxying through a backend.
+      ...(config.dangerouslyAllowBrowser
+        ? { 'anthropic-dangerous-direct-browser-access': 'true' }
+        : {}),
+      ...config.headers,
+    }
+
+    const body = JSON.stringify({
+      model: config.model,
+      max_tokens: config.maxTokens ?? DEFAULT_MAX_TOKENS,
+      stream: true,
+      // Caller-supplied empty string is still a value; only `undefined`
+      // (i.e. resolveSystemPrompt returned `null`) means "omit".
+      ...(built.system !== undefined ? { system: built.system } : {}),
+      messages: built.messages,
+      ...config.body,
+    })
+
+    const response = await fetch(url, {
+      method: 'POST',
+      headers,
+      body,
+      signal,
+    })
+
+    if (!response.ok) {
+      throw await readErrorBody(response, PROVIDER_NAME)
+    }
+
+    for await (const payload of parseSSE(response, signal)) {
+      let event: AnthropicStreamEvent
+      try {
+        event = JSON.parse(payload) as AnthropicStreamEvent
+      } catch {
+        continue
+      }
+      if (
+        event.type === 'content_block_delta' &&
+        event.delta?.type === 'text_delta' &&
+        event.delta.text
+      ) {
+        yield event.delta.text
+      }
+      if (event.type === 'message_stop') return
+    }
+  }
+}

package/src/llm-providers/openai/index.ts

@@ -0,0 +1,109 @@
+import type { AIPromptContext, AIProvider } from '../../feature/ai/types'
+
+import {
+  type BaseProviderConfig,
+  assertBrowserSafe,
+  buildDefaultUserMessage,
+  parseSSE,
+  readErrorBody,
+  resolveSystemPrompt,
+} from '../shared'
+
+const PROVIDER_NAME = 'milkdown/providers/openai'
+const DEFAULT_BASE_URL = 'https://api.openai.com'
+
+export interface OpenAIChatMessage {
+  role: 'system' | 'user' | 'assistant'
+  content: string
+}
+
+export interface OpenAIProviderConfig extends BaseProviderConfig {
+  /// Customize the messages sent to `/v1/chat/completions`. The default
+  /// builder produces a system message (when not disabled) followed by
+  /// a structured user message containing the document, selection, and
+  /// instruction. The defaults are passed in so callers can wrap them
+  /// rather than re-deriving from scratch.
+  buildMessages?: (
+    context: AIPromptContext,
+    defaults: { systemPrompt: string | null; userMessage: string }
+  ) => OpenAIChatMessage[]
+
+  /// Extra fields merged into the request body (e.g. `temperature`,
+  /// `top_p`, `presence_penalty`, `response_format`).
+  body?: Record<string, unknown>
+}
+
+interface OpenAIStreamChunk {
+  choices?: Array<{
+    delta?: { content?: string }
+    finish_reason?: string | null
+  }>
+}
+
+/// Build an `AIProvider` backed by OpenAI's `/v1/chat/completions`
+/// streaming endpoint. The returned function plugs directly into
+/// `Crepe.Feature.AI`'s `provider` field.
+export function createOpenAIProvider(config: OpenAIProviderConfig): AIProvider {
+  assertBrowserSafe(config, PROVIDER_NAME)
+
+  return async function* openAIProvider(context, signal) {
+    const systemPrompt = resolveSystemPrompt(config.systemPrompt)
+    const userMessage = buildDefaultUserMessage(context)
+
+    const messages = config.buildMessages
+      ? config.buildMessages(context, { systemPrompt, userMessage })
+      : defaultMessages(systemPrompt, userMessage)
+
+    const baseURL = (config.baseURL ?? DEFAULT_BASE_URL).replace(/\/+$/, '')
+    const url = `${baseURL}/v1/chat/completions`
+
+    const headers: Record<string, string> = {
+      'Content-Type': 'application/json',
+      ...(config.apiKey ? { Authorization: `Bearer ${config.apiKey}` } : {}),
+      ...config.headers,
+    }
+
+    const body = JSON.stringify({
+      model: config.model,
+      stream: true,
+      messages,
+      ...config.body,
+    })
+
+    const response = await fetch(url, {
+      method: 'POST',
+      headers,
+      body,
+      signal,
+    })
+
+    if (!response.ok) {
+      throw await readErrorBody(response, PROVIDER_NAME)
+    }
+
+    for await (const payload of parseSSE(response, signal)) {
+      if (payload === '[DONE]') return
+      let event: OpenAIStreamChunk
+      try {
+        event = JSON.parse(payload) as OpenAIStreamChunk
+      } catch {
+        continue
+      }
+      const delta = event.choices?.[0]?.delta?.content
+      if (delta) yield delta
+    }
+  }
+}
+
+function defaultMessages(
+  systemPrompt: string | null,
+  userMessage: string
+): OpenAIChatMessage[] {
+  const messages: OpenAIChatMessage[] = []
+  // `null` means "omit"; an empty string is still a caller-provided
+  // value and should be sent as-is.
+  if (systemPrompt !== null)
+    messages.push({ role: 'system', content: systemPrompt })
+  messages.push({ role: 'user', content: userMessage })
+  return messages
+}