iris-chatbot 0.2.4

package/template/src/lib/types.ts

@@ -0,0 +1,309 @@
+export type BuiltInProvider = "openai" | "anthropic" | "google";
+export type ConnectionKind = "builtin" | "openai_compatible" | "ollama";
+
+export type ConnectionHeader = {
+  key: string;
+  value: string;
+};
+
+export type ModelConnection = {
+  id: string;
+  name: string;
+  kind: ConnectionKind;
+  provider?: BuiltInProvider;
+  baseUrl?: string;
+  apiKey?: string;
+  headers?: ConnectionHeader[];
+  supportsTools?: boolean;
+  enabled: boolean;
+  createdAt: number;
+  updatedAt: number;
+};
+
+// Backwards-compat alias for older UI code; runtime now uses connection ids.
+export type Provider = BuiltInProvider;
+
+export type MessageNode = {
+  id: string;
+  conversationId: string;
+  parentId: string | null;
+  role: "system" | "user" | "assistant";
+  content: string;
+  createdAt: number;
+  provider?: string;
+  model?: string;
+};
+
+export type Thread = {
+  id: string;
+  conversationId: string;
+  headMessageId: string | null;
+  title: string;
+  forkedFromMessageId: string | null;
+  updatedAt: number;
+};
+
+export type ApprovalMode =
+  | "always_confirm_writes"
+  | "confirm_risky_only"
+  | "trusted_auto";
+
+export type SafetyProfile = "strict" | "balanced" | "auto";
+
+export type WebSearchBackend = "no_key" | "hybrid";
+
+export type LocalToolsSettings = {
+  enabled: boolean;
+  approvalMode: ApprovalMode;
+  safetyProfile: SafetyProfile;
+  allowedRoots: string[];
+  enableNotes: boolean;
+  enableApps: boolean;
+  enableNumbers: boolean;
+  enableWeb: boolean;
+  enableMusic: boolean;
+  enableCalendar: boolean;
+  enableMail: boolean;
+  enableWorkflow: boolean;
+  enableSystem: boolean;
+  webSearchBackend: WebSearchBackend;
+  dryRun: boolean;
+};
+
+export const DEFAULT_LOCAL_TOOL_ROOTS = [
+  "/Users/evanalexander/zenith",
+  "~/Desktop",
+  "~/Documents",
+  "~/Downloads",
+] as const;
+
+export const DEFAULT_LOCAL_TOOLS_SETTINGS: LocalToolsSettings = {
+  enabled: true,
+  approvalMode: "trusted_auto",
+  safetyProfile: "balanced",
+  allowedRoots: [...DEFAULT_LOCAL_TOOL_ROOTS],
+  enableNotes: true,
+  enableApps: true,
+  enableNumbers: false,
+  enableWeb: false,
+  enableMusic: true,
+  enableCalendar: false,
+  enableMail: false,
+  enableWorkflow: true,
+  enableSystem: true,
+  webSearchBackend: "no_key",
+  dryRun: false,
+};
+
+export type MemoryKind = "profile" | "preference" | "person_alias" | "music_alias" | "note";
+
+export type MemoryScope = "global" | "conversation";
+
+export type MemorySource = "auto" | "explicit" | "manual";
+
+export type MemoryEntry = {
+  id: string;
+  kind: MemoryKind;
+  scope: MemoryScope;
+  conversationId?: string;
+  key: string;
+  value: string;
+  normalizedKey: string;
+  source: MemorySource;
+  confidence: number;
+  createdAt: number;
+  updatedAt: number;
+  lastUsedAt: number;
+};
+
+export type MemorySettings = {
+  enabled: boolean;
+  autoCapture: boolean;
+  toolInfluence: boolean;
+  extractionMode: "conservative";
+};
+
+export const DEFAULT_MEMORY_SETTINGS: MemorySettings = {
+  enabled: true,
+  autoCapture: true,
+  toolInfluence: true,
+  extractionMode: "conservative",
+};
+
+export type MemoryPersonAlias = {
+  alias: string;
+  target: string;
+};
+
+export type MemoryMusicAlias = {
+  alias: string;
+  query: string;
+  title?: string;
+  artist?: string;
+};
+
+export type MemoryContextPayload = {
+  enabled: boolean;
+  aliases: {
+    people: MemoryPersonAlias[];
+    music: MemoryMusicAlias[];
+  };
+};
+
+export type Settings = {
+  id: "settings";
+  // Legacy fields kept for one migration cycle.
+  openaiKey?: string;
+  anthropicKey?: string;
+  geminiKey?: string;
+  defaultProvider: Provider;
+  defaultModel: string;
+  connections: ModelConnection[];
+  defaultConnectionId: string;
+  defaultModelByConnection: Record<string, string>;
+  showExtendedOpenAIModels?: boolean;
+  enableWebSources: boolean;
+  accentColor?: string;
+  font?: "ibm" | "manrope" | "sora" | "space" | "poppins";
+  theme?: "dark" | "light";
+  localTools?: LocalToolsSettings;
+  memory?: MemorySettings;
+};
+
+export type ChatConnectionPayload = {
+  id: string;
+  name: string;
+  kind: ConnectionKind;
+  provider?: BuiltInProvider;
+  baseUrl?: string;
+  apiKey?: string;
+  headers?: ConnectionHeader[];
+  supportsTools?: boolean;
+};
+
+export type ToolExecutionStage = "call" | "progress" | "result";
+
+export type ToolEvent = {
+  id: string;
+  conversationId: string;
+  threadId: string;
+  assistantMessageId: string;
+  toolCallId: string;
+  toolName: string;
+  stage: ToolExecutionStage;
+  message?: string;
+  payloadJson?: string;
+  createdAt: number;
+};
+
+export type ToolApprovalStatus = "requested" | "approved" | "denied" | "timeout";
+
+export type ToolApproval = {
+  id: string;
+  conversationId: string;
+  threadId: string;
+  assistantMessageId: string;
+  toolCallId: string;
+  toolName: string;
+  argsJson?: string;
+  reason?: string;
+  status: ToolApprovalStatus;
+  requestedAt: number;
+  resolvedAt?: number;
+};
+
+export type ChatMessageInput = {
+  role: "user" | "assistant" | "system";
+  content: string;
+};
+
+export type ChatCitationSource = {
+  url: string;
+  title?: string;
+};
+
+export type ChatRequest = {
+  // Legacy fallback for one migration cycle.
+  provider?: Provider;
+  model: string;
+  apiKey?: string;
+  connection?: ChatConnectionPayload;
+  system?: string;
+  messages: ChatMessageInput[];
+  stream: boolean;
+  localTools?: LocalToolsSettings;
+  memory?: MemoryContextPayload;
+  enableWebSources?: boolean;
+  meta?: {
+    threadId?: string;
+    conversationId?: string;
+    assistantMessageId?: string;
+  };
+};
+
+export type ToolCallEvent = {
+  type: "tool_call";
+  callId: string;
+  name: string;
+  args?: Record<string, unknown>;
+};
+
+export type ToolProgressEvent = {
+  type: "tool_progress";
+  callId: string;
+  name: string;
+  status: "started" | "running" | "warning" | "info" | "completed";
+  message: string;
+};
+
+export type ToolResultEvent = {
+  type: "tool_result";
+  callId: string;
+  name: string;
+  ok: boolean;
+  result: unknown;
+  meta?: {
+    retryCount?: number;
+    workflowStepId?: string;
+  };
+};
+
+export type ApprovalRequestEvent = {
+  type: "approval_requested";
+  approvalId: string;
+  callId: string;
+  name: string;
+  args?: Record<string, unknown>;
+  reason: string;
+};
+
+export type ApprovalResolvedEvent = {
+  type: "approval_resolved";
+  approvalId: string;
+  callId: string;
+  decision: "approve" | "deny" | "timeout";
+  message?: string;
+};
+
+export type SourcesEvent = {
+  type: "sources";
+  items: ChatCitationSource[];
+};
+
+export type ThinkingEvent = {
+  type: "thinking";
+  status: "started" | "update" | "done";
+  summary?: string;
+};
+
+export type ChatStreamChunk =
+  | { type: "token"; value?: string }
+  | { type: "done" }
+  | { type: "error"; error?: string }
+  | ThinkingEvent
+  | ToolCallEvent
+  | ToolProgressEvent
+  | ToolResultEvent
+  | ApprovalRequestEvent
+  | ApprovalResolvedEvent
+  | SourcesEvent;

package/template/src/lib/utils.ts

@@ -0,0 +1,108 @@
+import type { ChatCitationSource, MessageNode } from "./types";
+
+const SOURCES_MARKER_PREFIX = "\n\n[[ZENITH_SOURCES:";
+const SOURCES_MARKER_SUFFIX = "]]";
+
+function normalizeCitationSources(input: ChatCitationSource[]): ChatCitationSource[] {
+  if (!Array.isArray(input) || input.length === 0) {
+    return [];
+  }
+
+  const deduped = new Map<string, ChatCitationSource>();
+  for (const source of input) {
+    if (!source || typeof source.url !== "string") {
+      continue;
+    }
+    const url = source.url.trim();
+    if (!url) {
+      continue;
+    }
+    const title =
+      typeof source.title === "string" && source.title.trim()
+        ? source.title.trim()
+        : undefined;
+    const existing = deduped.get(url);
+    if (!existing || (!existing.title && title)) {
+      deduped.set(url, { url, title });
+    }
+  }
+  return [...deduped.values()];
+}
+
+export function splitContentAndSources(content: string): {
+  content: string;
+  sources: ChatCitationSource[];
+} {
+  if (!content) {
+    return { content: "", sources: [] };
+  }
+
+  const start = content.lastIndexOf(SOURCES_MARKER_PREFIX);
+  if (start === -1) {
+    return { content, sources: [] };
+  }
+
+  const markerEnd = content.length;
+  if (!content.endsWith(SOURCES_MARKER_SUFFIX)) {
+    return { content, sources: [] };
+  }
+
+  const encodedStart = start + SOURCES_MARKER_PREFIX.length;
+  const encodedEnd = markerEnd - SOURCES_MARKER_SUFFIX.length;
+  if (encodedStart > encodedEnd) {
+    return { content, sources: [] };
+  }
+
+  const encoded = content.slice(encodedStart, encodedEnd);
+  try {
+    const parsed = JSON.parse(decodeURIComponent(encoded)) as unknown;
+    if (!Array.isArray(parsed)) {
+      return { content, sources: [] };
+    }
+    const sources = normalizeCitationSources(parsed as ChatCitationSource[]);
+    const plainContent = content.slice(0, start).replace(/\s+$/, "");
+    return { content: plainContent, sources };
+  } catch {
+    return { content, sources: [] };
+  }
+}
+
+export function embedSourcesInContent(content: string, sources: ChatCitationSource[]): string {
+  const split = splitContentAndSources(content);
+  const normalizedSources = normalizeCitationSources(sources);
+  if (normalizedSources.length === 0) {
+    return split.content;
+  }
+  const encoded = encodeURIComponent(JSON.stringify(normalizedSources));
+  return `${split.content}${SOURCES_MARKER_PREFIX}${encoded}${SOURCES_MARKER_SUFFIX}`;
+}
+
+export function deriveTitle(text: string) {
+  const clean = text.replace(/\s+/g, " ").trim();
+  if (!clean) return "New chat";
+  return clean.length > 42 ? `${clean.slice(0, 42)}...` : clean;
+}
+
+export function buildPath(headId: string | null, map: Map<string, MessageNode>) {
+  const path: MessageNode[] = [];
+  let currentId = headId;
+  while (currentId) {
+    const node = map.get(currentId);
+    if (!node) break;
+    path.push(node);
+    currentId = node.parentId;
+  }
+  return path.reverse();
+}
+
+export function buildPathIds(headId: string | null, map: Map<string, MessageNode>) {
+  const ids = new Set<string>();
+  let currentId = headId;
+  while (currentId) {
+    const node = map.get(currentId);
+    if (!node) break;
+    ids.add(node.id);
+    currentId = node.parentId;
+  }
+  return ids;
+}

package/template/tsconfig.json

@@ -0,0 +1,34 @@
+{
+  "compilerOptions": {
+    "target": "ES2017",
+    "lib": ["dom", "dom.iterable", "esnext"],
+    "allowJs": true,
+    "skipLibCheck": true,
+    "strict": true,
+    "noEmit": true,
+    "esModuleInterop": true,
+    "module": "esnext",
+    "moduleResolution": "bundler",
+    "resolveJsonModule": true,
+    "isolatedModules": true,
+    "jsx": "react-jsx",
+    "incremental": true,
+    "plugins": [
+      {
+        "name": "next"
+      }
+    ],
+    "paths": {
+      "@/*": ["./src/*"]
+    }
+  },
+  "include": [
+    "next-env.d.ts",
+    "**/*.ts",
+    "**/*.tsx",
+    ".next/types/**/*.ts",
+    ".next/dev/types/**/*.ts",
+    "**/*.mts"
+  ],
+  "exclude": ["node_modules"]
+}