nolo-cli 0.1.23 → 0.1.29

package/server/handlers/agentRun/inheritedDialogPrompt.ts

@@ -0,0 +1,111 @@
+import { estimateTokenCount as defaultEstimateTokenCount } from "../../../ai/context/tokenUtils";
+
+const DIALOG_REFERENCE_MESSAGE_LIMIT = 20;
+const DIALOG_REFERENCE_SNIPPET_CHARS = 1200;
+
+type DialogRecord = Record<string, unknown>;
+
+type FetchConversationMessages = (args: {
+  dialogId: string;
+  limit: number;
+}) => Promise<unknown[]>;
+
+type BuildInheritedDialogPromptBlockOptions = {
+  dialogKey?: string;
+  parentDialogId?: string;
+  getDialogRecord(dialogKey: string): Promise<DialogRecord | null>;
+  fetchConversationMessages: FetchConversationMessages;
+  estimateTokenCount?: typeof defaultEstimateTokenCount;
+};
+
+const dialogReferenceContentToText = (content: unknown): string => {
+  if (typeof content === "string") return content;
+  if (Array.isArray(content)) {
+    return content
+      .map((part) => {
+        if (part && typeof part === "object" && "text" in part) {
+          return String((part as { text?: unknown }).text ?? "");
+        }
+        return JSON.stringify(part);
+      })
+      .join("\n");
+  }
+  if (content == null) return "";
+  return JSON.stringify(content);
+};
+
+const clipDialogReferenceText = (
+  value: string,
+  max = DIALOG_REFERENCE_SNIPPET_CHARS,
+): string => {
+  const trimmed = value.trim();
+  if (trimmed.length <= max) return trimmed;
+  return `${trimmed.slice(0, max)}\n...[truncated ${trimmed.length - max} chars]`;
+};
+
+const asRecord = (value: unknown): Record<string, unknown> =>
+  value && typeof value === "object" ? (value as Record<string, unknown>) : {};
+
+export async function buildInheritedDialogPromptBlock({
+  dialogKey: rawDialogKey,
+  parentDialogId,
+  getDialogRecord,
+  fetchConversationMessages,
+  estimateTokenCount = defaultEstimateTokenCount,
+}: BuildInheritedDialogPromptBlockOptions): Promise<string | undefined> {
+  const dialogKey = rawDialogKey?.trim();
+  if (!dialogKey?.startsWith("dialog-")) return undefined;
+
+  const dialogRecord = await getDialogRecord(dialogKey);
+  const dialogId =
+    parentDialogId?.trim() ||
+    (typeof dialogRecord?.id === "string" ? dialogRecord.id : "") ||
+    dialogKey.split("-").at(-1) ||
+    "";
+
+  let recentMessages: unknown[] = [];
+  if (dialogId) {
+    try {
+      const newestFirst = await fetchConversationMessages({
+        dialogId,
+        limit: DIALOG_REFERENCE_MESSAGE_LIMIT,
+      });
+      recentMessages = Array.isArray(newestFirst) ? [...newestFirst].reverse() : [];
+    } catch (_) {}
+  }
+
+  const sections = [
+    "Conversation Reference:",
+    `- dialogKey: ${dialogKey}`,
+    `- title: ${
+      typeof dialogRecord?.title === "string" ? dialogRecord.title : dialogKey
+    }`,
+    dialogRecord?.checkpointSummary
+      ? `- checkpoint: ${String(dialogRecord.checkpointSummary)}`
+      : "",
+    dialogRecord?.passiveSummary
+      ? `- passive summary: ${String(dialogRecord.passiveSummary)}`
+      : "",
+    dialogRecord?.proactiveSummary
+      ? `- proactive summary: ${String(dialogRecord.proactiveSummary)}`
+      : "",
+    `- coverage: recent ${recentMessages.length} message(s) are included below. For exact old wording, early-history details, or tool evidence not shown here, call searchDialogMessages({ dialogKey: "${dialogKey}", query }).`,
+    recentMessages.length
+      ? [
+          "Recent transcript:",
+          ...recentMessages.map((message) => {
+            const record = asRecord(message);
+            return `- ${String(record.role ?? "unknown")} ${String(
+              record.id ?? "",
+            )}: ${clipDialogReferenceText(
+              dialogReferenceContentToText(record.content),
+            )}`;
+          }),
+        ].join("\n")
+      : "",
+  ]
+    .filter(Boolean)
+    .join("\n");
+
+  return `${sections}\n- token estimate: ${estimateTokenCount(sections)}`;
+}

package/server/handlers/agentRun/localRuntimeBootstrap.ts

@@ -0,0 +1,69 @@
+const LOCAL_RUNTIME_HOSTS = new Set(["127.0.0.1", "localhost", "::1"]);
+
+export function isLocalEndpoint(endpoint: string): boolean {
+  try {
+    const hostname = new URL(endpoint).hostname.toLowerCase();
+    return LOCAL_RUNTIME_HOSTS.has(hostname);
+  } catch {
+    return false;
+  }
+}
+
+export function shouldEnsureLocalModelRuntime(agentConfig: any): boolean {
+  if (agentConfig?.skipLocalRuntimeEnsure === true) return false;
+
+  const endpoint =
+    typeof agentConfig?.customProviderUrl === "string"
+      ? agentConfig.customProviderUrl.trim()
+      : "";
+
+  return !!endpoint && isLocalEndpoint(endpoint);
+}
+
+async function isLocalProviderAlreadyHealthy(endpoint: string): Promise<boolean> {
+  try {
+    const url = new URL(endpoint);
+    const response = await fetch(`${url.origin}/health`, {
+      signal: AbortSignal.timeout(3000),
+    });
+    return response.ok;
+  } catch {
+    return false;
+  }
+}
+
+export function createLocalRuntimeUnavailableError(endpoint: string): Error & {
+  reason: "local_provider_unavailable";
+  status: 409;
+} {
+  const origin = new URL(endpoint).origin;
+  return Object.assign(
+    new Error(
+      `Local provider is not running at ${origin}. Start it from the desktop runtime controls and retry.`
+    ),
+    {
+      reason: "local_provider_unavailable" as const,
+      status: 409 as const,
+    }
+  );
+}
+
+export function isLocalRuntimeUnavailableError(error: unknown): error is Error & {
+  reason: "local_provider_unavailable";
+  status: 409;
+} {
+  return (
+    typeof error === "object" &&
+    error !== null &&
+    (error as any).reason === "local_provider_unavailable" &&
+    (error as any).status === 409
+  );
+}
+
+export async function ensureLocalModelRuntimeIfNeeded(agentConfig: any): Promise<void> {
+  if (!shouldEnsureLocalModelRuntime(agentConfig)) return;
+  const endpoint = agentConfig.customProviderUrl.trim();
+  if (await isLocalProviderAlreadyHealthy(endpoint)) return;
+
+  throw createLocalRuntimeUnavailableError(endpoint);
+}