@tablestore/tablestore-agent-cli 0.2.11 → 0.4.0

package/dist/dashboard-runtime/dashboard/src/lib/scope-workspace.ts

@@ -0,0 +1,123 @@
+import type { MemoryScope } from "@/types/memory";
+import type { FileMemoryScope } from "@/types/file-memory";
+import type { Wiki, WikiBuildStatus } from "@/types/wiki";
+import type {
+  ChatScopeStorageMode,
+  ChatScopeWikiPolicy,
+} from "@/lib/chat-scope-entry";
+
+export type ScopeFeatureSegment =
+  "scope" | "units" | "search" | "dream" | "wiki";
+export type ScopeWorkspaceStatus =
+  | "loading"
+  | "draft"
+  | "building"
+  | "ready"
+  | "stale"
+  | "failed_with_fallback"
+  | "blocked";
+
+const NON_TERMINAL = new Set<WikiBuildStatus>([
+  "pending",
+  "processing",
+  "accepting_output",
+  "sealing",
+  "validating",
+  "indexing",
+  "publishing",
+]);
+
+export function parseExactScope(
+  params: URLSearchParams,
+): FileMemoryScope | null {
+  const values = ["appId", "tenantId", "agentId", "runId"].map(
+    (key) => params.get(key)?.trim() ?? "",
+  );
+  if (values.some((value) => !value || value === "*")) return null;
+  const [appId, tenantId, agentId, runId] = values;
+  return { appId, tenantId, agentId, runId };
+}
+
+export function toExactScope(scope: MemoryScope): FileMemoryScope | null {
+  return parseExactScope(
+    new URLSearchParams({
+      appId: scope.appId ?? "",
+      tenantId: scope.tenantId ?? "",
+      agentId: scope.agentId ?? "",
+      runId: scope.runId ?? "",
+    }),
+  );
+}
+
+function scopeParams(
+  scope: FileMemoryScope,
+  extra: Record<string, string> = {},
+) {
+  const params = new URLSearchParams({
+    appId: scope.appId,
+    tenantId: scope.tenantId,
+    agentId: scope.agentId,
+    runId: scope.runId,
+  });
+  for (const [key, value] of Object.entries(extra)) {
+    if (!["appId", "tenantId", "agentId", "runId"].includes(key)) {
+      params.set(key, value);
+    }
+  }
+  return params;
+}
+
+export function buildScopeFeatureHref(
+  store: string,
+  segment: ScopeFeatureSegment,
+  scope: FileMemoryScope,
+  extra: Record<string, string> = {},
+): string {
+  return `/memory/${encodeURIComponent(store)}/${segment}?${scopeParams(scope, extra)}`;
+}
+
+export function buildScopeWorkspaceHref(store: string, scope: FileMemoryScope) {
+  return buildScopeFeatureHref(store, "scope", scope);
+}
+
+export function buildScopeChatHref(
+  memoryStoreName: string,
+  storageMode: ChatScopeStorageMode,
+  scope: FileMemoryScope,
+  wiki: ChatScopeWikiPolicy,
+): string {
+  if (storageMode !== "ots" && storageMode !== "file+ots") {
+    throw new Error("Unsupported Chat Scope storage mode");
+  }
+  const params = new URLSearchParams({
+    memoryStoreName,
+    storageMode,
+    appId: scope.appId,
+    tenantId: scope.tenantId,
+    agentId: scope.agentId,
+    runId: scope.runId,
+    wiki,
+  });
+  return `/chat?${params}`;
+}
+
+export function deriveScopeWorkspaceStatus(input: {
+  loading?: boolean;
+  error?: unknown;
+  wiki?: Pick<Wiki, "status" | "hasPendingChanges"> & {
+    latestBuild?: { status: WikiBuildStatus };
+  };
+}): ScopeWorkspaceStatus {
+  if (input.loading) return "loading";
+  if (input.error || !input.wiki) return "blocked";
+  const latestStatus = input.wiki.latestBuild?.status;
+  if (latestStatus && NON_TERMINAL.has(latestStatus)) return "building";
+  if (latestStatus === "failed" || latestStatus === "conflict") {
+    return input.wiki.status === "ready" ? "failed_with_fallback" : "blocked";
+  }
+  if (input.wiki.status !== "ready") return "draft";
+  if (input.wiki.hasPendingChanges) {
+    return "stale";
+  }
+  return "ready";
+}

package/dist/dashboard-runtime/dashboard/src/lib/server/api-helpers.ts

@@ -44,6 +44,11 @@ const CODE_TO_HTTP_STATUS: Record<string, number> = {
   UNAUTHORIZED: 401,
   FORBIDDEN: 403,
   CONFLICT: 409,
+  PATH_EXISTS: 409,
+  LOCK_CONFLICT: 409,
+  QUOTA_EXCEEDED: 409,
+  READ_ONLY_STORE: 409,
+  PAYLOAD_TOO_LARGE: 413,
   QUOTA_EXHAUSTED: 429,
   FLOW_CONTROL: 429,
   INTERNAL_SERVER_ERROR: 500,
@@ -135,8 +140,15 @@ export interface ForwardableEnvelope {
   data: unknown;
   message: string;
   requestId?: string;
+  details?: ForwardableUpstreamErrorDetails;
 }
 
+type ForwardableUpstreamErrorDetails = {
+  upstreamCode?: string;
+  upstreamStatus?: number;
+  upstreamRequestId?: string;
+};
+
 /**
  * Forward a {@link StandardResponse} (or an ad-hoc envelope) produced by a
  * route handler as an HTTP response. Success envelopes become 200 with `data`
@@ -157,15 +169,24 @@ export function forwardResponse(
     message: resp.message,
   };
   if (resp.requestId) body.requestId = resp.requestId;
-
-  const status = resp.code === "SUCCESS" ? 200 : httpStatusFor(resp.code);
+  const details = normalizeUpstreamErrorDetails(resp.details);
+  if (details) body.details = details;
+
+  const upstreamStatus = details?.upstreamStatus;
+  const status =
+    resp.code === "SUCCESS"
+      ? 200
+      : isForwardableHttpStatus(upstreamStatus)
+        ? upstreamStatus
+        : httpStatusFor(resp.code);
   logApiResponse(
     resp.code === "SUCCESS" ? "api.response" : "api.service_error",
-    resp.message,
+    resp.code === "SUCCESS" ? "succeed" : "upstream service error",
     resp.code,
     status,
     resp.requestId,
     resp.code === "SUCCESS" ? undefined : "service",
+    details,
   );
   const response = NextResponse.json(body, { status });
   attachNoStoreHeader(response);
@@ -173,6 +194,33 @@ export function forwardResponse(
   return response;
 }
 
+function isForwardableHttpStatus(status: number | undefined): status is number {
+  return (
+    status !== undefined &&
+    Number.isInteger(status) &&
+    status >= 400 &&
+    status <= 599
+  );
+}
+
+function normalizeUpstreamErrorDetails(
+  details: unknown,
+): ForwardableUpstreamErrorDetails | undefined {
+  if (details === null || typeof details !== "object") return undefined;
+  const record = details as Record<string, unknown>;
+  const normalized: ForwardableUpstreamErrorDetails = {};
+  if (typeof record.upstreamCode === "string") {
+    normalized.upstreamCode = record.upstreamCode;
+  }
+  if (typeof record.upstreamStatus === "number") {
+    normalized.upstreamStatus = record.upstreamStatus;
+  }
+  if (typeof record.upstreamRequestId === "string") {
+    normalized.upstreamRequestId = record.upstreamRequestId;
+  }
+  return Object.keys(normalized).length > 0 ? normalized : undefined;
+}
+
 // ---------------------------------------------------------------------------
 // Error handling
 // ---------------------------------------------------------------------------

package/dist/dashboard-runtime/dashboard/src/lib/server/instance-connectivity.ts

@@ -0,0 +1,136 @@
+import type {
+  InstanceConnectivity,
+  InstanceConnectivityReason,
+} from "@/types/instance";
+
+const DEFAULT_TIMEOUT_MS = 3_000;
+const MAX_MESSAGE_LENGTH = 240;
+
+const DNS_CODES = new Set(["ENOTFOUND", "EAI_AGAIN", "EAI_FAIL"]);
+const CONNECTION_CODES = new Set([
+  "ECONNREFUSED",
+  "ECONNRESET",
+  "EHOSTUNREACH",
+  "ENETUNREACH",
+]);
+const TLS_CODES = new Set([
+  "CERT_HAS_EXPIRED",
+  "CERT_NOT_YET_VALID",
+  "DEPTH_ZERO_SELF_SIGNED_CERT",
+  "ERR_TLS_CERT_ALTNAME_INVALID",
+  "SELF_SIGNED_CERT_IN_CHAIN",
+  "UNABLE_TO_GET_ISSUER_CERT_LOCALLY",
+  "UNABLE_TO_VERIFY_LEAF_SIGNATURE",
+]);
+
+interface ProbeOptions {
+  fetchImpl?: typeof fetch;
+  timeoutMs?: number;
+  now?: () => number;
+}
+
+interface ErrorDetails {
+  reason: InstanceConnectivityReason;
+  message: string;
+}
+
+function safeMessage(value: unknown): string {
+  const message =
+    value instanceof Error && value.message.trim()
+      ? value.message
+      : "Network request failed";
+  return message.replace(/\s+/g, " ").slice(0, MAX_MESSAGE_LENGTH);
+}
+
+function errorDetails(error: unknown): ErrorDetails {
+  let current: unknown = error;
+  let deepestMessage = safeMessage(error);
+  const seen = new Set<unknown>();
+
+  while (current && typeof current === "object" && !seen.has(current)) {
+    seen.add(current);
+    const record = current as {
+      cause?: unknown;
+      code?: unknown;
+      message?: unknown;
+      name?: unknown;
+    };
+    if (typeof record.message === "string" && record.message.trim()) {
+      deepestMessage = record.message.replace(/\s+/g, " ");
+    }
+    if (record.name === "TimeoutError" || record.name === "AbortError") {
+      return { reason: "timeout", message: safeMessage(current) };
+    }
+    const code = typeof record.code === "string" ? record.code : "";
+    if (DNS_CODES.has(code)) {
+      return { reason: "dns", message: safeMessage(current) };
+    }
+    if (CONNECTION_CODES.has(code)) {
+      return { reason: "connection", message: safeMessage(current) };
+    }
+    if (TLS_CODES.has(code) || code.startsWith("ERR_TLS_")) {
+      return { reason: "tls", message: safeMessage(current) };
+    }
+    current = record.cause;
+  }
+
+  return {
+    reason: "network",
+    message: deepestMessage.replace(/\s+/g, " ").slice(0, MAX_MESSAGE_LENGTH),
+  };
+}
+
+export async function probeInstanceEndpoint(
+  endpoint: string,
+  options: ProbeOptions = {},
+): Promise<InstanceConnectivity> {
+  const now = options.now ?? Date.now;
+  const startedAt = now();
+  const checkedAt = () => new Date(now()).toISOString();
+
+  if (!endpoint.trim()) {
+    return { configured: false, reachable: null, checkedAt: checkedAt() };
+  }
+
+  let url: string;
+  try {
+    const parsed = new URL(endpoint);
+    if (parsed.protocol !== "http:" && parsed.protocol !== "https:") {
+      throw new TypeError("unsupported protocol");
+    }
+    url = parsed.toString();
+  } catch {
+    return {
+      configured: true,
+      reachable: false,
+      checkedAt: checkedAt(),
+      latencyMs: Math.max(0, now() - startedAt),
+      reason: "network",
+      message: "Invalid endpoint URL",
+    };
+  }
+
+  try {
+    const response = await (options.fetchImpl ?? fetch)(url, {
+      method: "HEAD",
+      signal: AbortSignal.timeout(options.timeoutMs ?? DEFAULT_TIMEOUT_MS),
+      redirect: "manual",
+      cache: "no-store",
+    });
+    return {
+      configured: true,
+      reachable: true,
+      checkedAt: checkedAt(),
+      latencyMs: Math.max(0, now() - startedAt),
+      httpStatus: response.status,
+    };
+  } catch (error) {
+    return {
+      configured: true,
+      reachable: false,
+      checkedAt: checkedAt(),
+      latencyMs: Math.max(0, now() - startedAt),
+      ...errorDetails(error),
+    };
+  }
+}

package/dist/dashboard-runtime/dashboard/src/lib/server/sdk-client.ts

@@ -18,6 +18,7 @@ import {
 import {
   KnowledgeBaseClient,
   MemoryClient,
+  WikiClient,
   inferRegionFromOtsEndpoint,
 } from "@tablestore/tablestore-agent-sdk";
 
@@ -57,6 +58,7 @@ export const TOKEN_SOURCE_COOKIE_NAME = "tablestore_token_source";
 const COOKIE_MAX_AGE_SECONDS = 60 * 60 * 24 * 30; // 30 days
 const RAW_TOKEN_SOURCE = "raw";
 const INSTANCE_TOKEN_SOURCE = "instance";
+const FORCE_ENV_VALUE = "1";
 
 const COOKIE_OPTIONS = {
   httpOnly: true,
@@ -138,6 +140,7 @@ export type CredentialSource = "cookie" | "instance" | "env";
 export interface ResolvedClients {
   kb: KnowledgeBaseClient;
   memory: MemoryClient;
+  wiki: WikiClient;
   config: AppConfig;
   source: CredentialSource;
   /** Trusted active project name for named-instance requests, else null. */
@@ -272,6 +275,10 @@ function configFromEnv(): Record<string, string> | null {
   };
 }
 
+function shouldForceEnvCredentials(): boolean {
+  return process.env.TABLESTORE_DASHBOARD_FORCE_ENV === FORCE_ENV_VALUE;
+}
+
 function configFromToken(token: string): Record<string, string> {
   // decodeToken throws on malformed input; callers translate to 401.
   return decodeToken(token);
@@ -521,12 +528,13 @@ function cacheSet(key: string, value: ResolvedClients): void {
  */
 export function resolveClients(request: NextRequest): ResolvedClients {
   const token = request.cookies.get(SESSION_COOKIE_NAME)?.value;
+  const forceEnv = shouldForceEnvCredentials();
 
-  let fields: Record<string, string> | null = null;
+  let fields: Record<string, string> | null = forceEnv ? configFromEnv() : null;
   let source: CredentialSource = "env";
   let instanceName: string | null = null;
 
-  if (token) {
+  if (!forceEnv && token) {
     try {
       const tokenFields = configFromToken(token);
       if (shouldUseCookieCredentials(request, tokenFields)) {
@@ -542,7 +550,7 @@ export function resolveClients(request: NextRequest): ResolvedClients {
     }
   }
 
-  if (!fields) {
+  if (!forceEnv && !fields) {
     const fromInstance = configFromActiveInstance();
     if (fromInstance) {
       fields = fromInstance.fields;
@@ -572,6 +580,7 @@ export function resolveClients(request: NextRequest): ResolvedClients {
   const resolved: ResolvedClients = {
     kb: new KnowledgeBaseClient(appConfig),
     memory: new MemoryClient(appConfig),
+    wiki: new WikiClient(appConfig),
     config: appConfig,
     source,
     instanceName,
@@ -591,6 +600,22 @@ export function inspectCredentials(request: NextRequest): {
   activeInstance: string | null;
   credentialFingerprint: string;
 } | null {
+  if (shouldForceEnvCredentials()) {
+    const fields = configFromEnv();
+    if (!fields) return null;
+    const region =
+      fields.region ||
+      inferRegionFromOtsEndpoint(fields.ots_endpoint) ||
+      DEFAULT_REGION;
+    return {
+      instanceName: fields.ots_instance_name,
+      region,
+      source: "env",
+      activeInstance: null,
+      credentialFingerprint: computeCredentialFingerprint(fields),
+    };
+  }
+
   const token = request.cookies.get(SESSION_COOKIE_NAME)?.value;
   if (token) {
     try {

package/dist/dashboard-runtime/dashboard/src/lib/subspace-history.ts

@@ -0,0 +1,151 @@
+export const MAX_RECENT_SUBSPACES = 10;
+
+const RECENT_STORAGE_KEY_PREFIX = "tsagent.kbSubspaces.v2";
+const ACTIVE_STORAGE_KEY_PREFIX = "tsagent.kbActiveSubspace.v2";
+
+type SubspaceStorage = Pick<Storage, "getItem" | "setItem" | "removeItem">;
+
+function normalizeSubspace(value: unknown): string | null {
+  if (typeof value !== "string") return null;
+  const normalized = value.trim();
+  return normalized || null;
+}
+
+export function getRecentSubspaceStorageKey(
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+): string {
+  return `${RECENT_STORAGE_KEY_PREFIX}:${JSON.stringify([
+    dataScopeKey,
+    knowledgeBaseName,
+  ])}`;
+}
+
+export function getActiveSubspaceStorageKey(
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+): string {
+  return `${ACTIVE_STORAGE_KEY_PREFIX}:${JSON.stringify([
+    dataScopeKey,
+    knowledgeBaseName,
+  ])}`;
+}
+
+export function readRecentSubspaces(
+  storage: Pick<SubspaceStorage, "getItem">,
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+): string[] {
+  try {
+    const raw = storage.getItem(
+      getRecentSubspaceStorageKey(dataScopeKey, knowledgeBaseName),
+    );
+    if (!raw) return [];
+    const parsed = JSON.parse(raw) as unknown;
+    if (!Array.isArray(parsed)) return [];
+
+    const seen = new Set<string>();
+    return parsed.reduce<string[]>((recent, item) => {
+      const subspace = normalizeSubspace(item);
+      if (
+        !subspace ||
+        seen.has(subspace) ||
+        recent.length >= MAX_RECENT_SUBSPACES
+      ) {
+        return recent;
+      }
+      seen.add(subspace);
+      recent.push(subspace);
+      return recent;
+    }, []);
+  } catch {
+    return [];
+  }
+}
+
+export function rememberRecentSubspace(
+  storage: Pick<SubspaceStorage, "getItem" | "setItem">,
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+  value: unknown,
+): string[] {
+  const subspace = normalizeSubspace(value);
+  const recent = readRecentSubspaces(storage, dataScopeKey, knowledgeBaseName);
+  if (!subspace) return recent;
+
+  const next = [subspace, ...recent.filter((item) => item !== subspace)].slice(
+    0,
+    MAX_RECENT_SUBSPACES,
+  );
+  try {
+    storage.setItem(
+      getRecentSubspaceStorageKey(dataScopeKey, knowledgeBaseName),
+      JSON.stringify(next),
+    );
+  } catch {
+    // Browser storage availability must not block navigation.
+  }
+  return next;
+}
+
+export function clearRecentSubspaces(
+  storage: Pick<SubspaceStorage, "removeItem">,
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+): void {
+  try {
+    storage.removeItem(
+      getRecentSubspaceStorageKey(dataScopeKey, knowledgeBaseName),
+    );
+  } catch {
+    // Browser storage availability must not block navigation.
+  }
+}
+
+export function readActiveSubspace(
+  storage: Pick<SubspaceStorage, "getItem">,
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+): string | null {
+  try {
+    return normalizeSubspace(
+      storage.getItem(
+        getActiveSubspaceStorageKey(dataScopeKey, knowledgeBaseName),
+      ),
+    );
+  } catch {
+    return null;
+  }
+}
+
+export function writeActiveSubspace(
+  storage: Pick<SubspaceStorage, "setItem">,
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+  value: unknown,
+): void {
+  const subspace = normalizeSubspace(value);
+  if (!subspace) return;
+  try {
+    storage.setItem(
+      getActiveSubspaceStorageKey(dataScopeKey, knowledgeBaseName),
+      subspace,
+    );
+  } catch {
+    // Browser storage availability must not block navigation.
+  }
+}
+
+export function clearActiveSubspace(
+  storage: Pick<SubspaceStorage, "removeItem">,
+  dataScopeKey: string,
+  knowledgeBaseName: string,
+): void {
+  try {
+    storage.removeItem(
+      getActiveSubspaceStorageKey(dataScopeKey, knowledgeBaseName),
+    );
+  } catch {
+    // Browser storage availability must not block navigation.
+  }
+}