@tablestore/tablestore-agent-cli 0.3.1 → 0.4.0

package/dist/dashboard-runtime/dashboard/src/lib/memory-storage-mode.ts

@@ -5,7 +5,15 @@ export type StorageMode = "ots" | "file+ots" | "filemem";
 export type MemoryInputType = "short-term" | "file";
 export type MemoryStorageStructure = "structured" | "file";
 export type MemoryDetailSegment =
-  "units" | "search" | "messages" | "tasks" | "requests" | "dream" | "settings";
+  | "scope"
+  | "units"
+  | "search"
+  | "messages"
+  | "tasks"
+  | "requests"
+  | "dream"
+  | "wiki"
+  | "settings";
 export type MemoryCardShortcut =
   "longTerm" | "shortTerm" | "extract" | "consolidate" | "files";
 export type MemoryCardAction = "units" | "search" | "settings";
@@ -37,15 +45,21 @@ export const STORAGE_MODE_VALUES: StorageMode[] = [
 ];
 
 const STRUCTURED_DETAIL_SEGMENTS: MemoryDetailSegment[] = [
+  "scope",
   "units",
   "search",
   "messages",
   "tasks",
   "requests",
   "dream",
+  "wiki",
+  "settings",
+];
+const FILE_DETAIL_SEGMENTS: MemoryDetailSegment[] = [
+  "units",
+  "wiki",
   "settings",
 ];
-const FILE_DETAIL_SEGMENTS: MemoryDetailSegment[] = ["units", "settings"];
 const STRUCTURED_CARD_SHORTCUTS: MemoryCardShortcut[] = [
   "longTerm",
   "shortTerm",
@@ -63,9 +77,16 @@ const FILE_CARD_ACTIONS: MemoryCardAction[] = ["units", "settings"];
 export function normalizeStorageMode(
   value: string | undefined | null,
 ): StorageMode {
-  if (value === "filemem") return "filemem";
+  return parseKnownStorageMode(value) ?? "ots";
+}
+
+export function parseKnownStorageMode(
+  value: string | undefined | null,
+): StorageMode | null {
+  if (value === "ots") return "ots";
   if (value === "file+ots" || value === "oss+ots") return "file+ots";
-  return "ots";
+  if (value === "filemem") return "filemem";
+  return null;
 }
 
 export function memoryStorageSelectionForInput(

package/dist/dashboard-runtime/dashboard/src/lib/query-keys.ts

@@ -1,5 +1,6 @@
 import type { QueryClient, QueryKey } from "@tanstack/react-query";
 import type { FileMemoryScope } from "@/types/file-memory";
+import type { WikiScope } from "@/types/wiki";
 
 export const RUNTIME_CONTEXT_QUERY_KEY = ["runtime-context"] as const;
 export const PENDING_DATA_SCOPE_KEY = "pending";
@@ -64,6 +65,97 @@ export const fileMemoryQueryKeys = {
     ] as const,
 };
 
+export const wikiQueryKeys = {
+  root: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope | undefined,
+  ) =>
+    [
+      "wiki",
+      dataScopeKey,
+      memoryStoreName,
+      scope?.appId ?? "",
+      scope?.tenantId ?? "",
+      scope?.agentId ?? "",
+    ] as const,
+  detail: (dataScopeKey: string, memoryStoreName: string, scope: WikiScope) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "detail",
+    ] as const,
+  sources: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope,
+    limit?: number,
+  ) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "sources",
+      limit ?? null,
+    ] as const,
+  artifacts: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope,
+    kind: "skill" | "profile",
+    limit?: number,
+  ) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "artifacts",
+      kind,
+      limit ?? null,
+    ] as const,
+  source: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope,
+    path: string,
+  ) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "source",
+      path,
+    ] as const,
+  pages: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope,
+    limit?: number,
+  ) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "pages",
+      limit ?? null,
+    ] as const,
+  page: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope,
+    path: string,
+  ) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "page",
+      path,
+    ] as const,
+  search: (
+    dataScopeKey: string,
+    memoryStoreName: string,
+    scope: WikiScope,
+    query: string,
+    limit?: number,
+  ) =>
+    [
+      ...wikiQueryKeys.root(dataScopeKey, memoryStoreName, scope),
+      "search",
+      query,
+      limit ?? null,
+    ] as const,
+};
+
 export const queryKeys = {
   runtimeContext: RUNTIME_CONTEXT_QUERY_KEY,
   instances: (scope: string) => ["instances", scope] as const,
@@ -314,6 +406,7 @@ const INSTANCE_SCOPED_QUERY_ROOTS = new Set([
   "knowledge-bases",
   "chunks",
   "memory",
+  "wiki",
   "llm-providers",
   "llm-provider-models",
   "api-keys",

package/dist/dashboard-runtime/dashboard/src/lib/retrieval-security.ts

@@ -0,0 +1,54 @@
+export type RetrievalSecurityCode = "UNAUTHORIZED" | "FORBIDDEN";
+export type RetrievalSecuritySource = "memory" | "wiki";
+
+function publicMessage(
+  source: RetrievalSecuritySource,
+  code: RetrievalSecurityCode,
+): string {
+  const label = source === "memory" ? "Memory" : "Wiki";
+  return code === "UNAUTHORIZED"
+    ? `${label} authentication failed`
+    : `${label} access is forbidden`;
+}
+
+export class FatalRetrievalSecurityError extends Error {
+  readonly source: RetrievalSecuritySource;
+  readonly code: RetrievalSecurityCode;
+
+  constructor(source: RetrievalSecuritySource, code: RetrievalSecurityCode) {
+    super(publicMessage(source, code));
+    this.name = "FatalRetrievalSecurityError";
+    this.source = source;
+    this.code = code;
+  }
+}
+
+function errorRecord(value: unknown): Record<string, unknown> {
+  return value !== null && typeof value === "object"
+    ? (value as Record<string, unknown>)
+    : {};
+}
+
+function codeFromStatus(status: unknown): RetrievalSecurityCode | undefined {
+  if (status === 401) return "UNAUTHORIZED";
+  if (status === 403) return "FORBIDDEN";
+  return undefined;
+}
+
+export function retrievalSecurityCode(
+  value: unknown,
+): RetrievalSecurityCode | undefined {
+  if (value instanceof FatalRetrievalSecurityError) return value.code;
+
+  const record = errorRecord(value);
+  if (record.code === "UNAUTHORIZED" || record.code === "FORBIDDEN") {
+    return record.code;
+  }
+  const details = errorRecord(record.details);
+  return (
+    codeFromStatus(details.upstreamStatus) ??
+    codeFromStatus(record.upstreamStatus) ??
+    codeFromStatus(record.httpStatus) ??
+    codeFromStatus(record.status)
+  );
+}

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

@@ -0,0 +1,167 @@
+import type { MemoryScope } from "@/types/memory";
+
+export const SCOPE_LEVELS = ["app", "tenant", "agent", "run"] as const;
+
+export type ScopeLevel = (typeof SCOPE_LEVELS)[number];
+export type ScopeGranularity = "all" | ScopeLevel;
+
+export interface ScopeChild {
+  level: ScopeLevel;
+  value: string;
+}
+
+export interface ScopeBreadcrumbItem extends ScopeChild {
+  concrete: boolean;
+}
+
+export const SCOPE_FIELD_BY_LEVEL = {
+  app: "appId",
+  tenant: "tenantId",
+  agent: "agentId",
+  run: "runId",
+} as const satisfies Record<ScopeLevel, keyof MemoryScope>;
+
+const ALL_SCOPE: MemoryScope = {
+  appId: "*",
+  tenantId: "*",
+  agentId: "*",
+  runId: "*",
+};
+
+const LEVEL_BY_FIELD: Record<keyof MemoryScope, ScopeLevel> = {
+  appId: "app",
+  tenantId: "tenant",
+  agentId: "agent",
+  runId: "run",
+};
+
+export function normalizeHierarchicalScope(
+  scope: MemoryScope,
+): MemoryScope | null {
+  const appId = scope.appId?.trim();
+  if (!appId) return null;
+
+  const values = [
+    appId,
+    scope.tenantId?.trim() || "*",
+    scope.agentId?.trim() || "*",
+    scope.runId?.trim() || "*",
+  ];
+  let wildcardSeen = false;
+  const canonical = values.map((value) => {
+    if (wildcardSeen || value === "*") {
+      wildcardSeen = true;
+      return "*";
+    }
+    return value;
+  });
+
+  return {
+    appId: canonical[0],
+    tenantId: canonical[1],
+    agentId: canonical[2],
+    runId: canonical[3],
+  };
+}
+
+export function scopeGranularity(scope: MemoryScope): ScopeGranularity {
+  const canonical = normalizeHierarchicalScope(scope) ?? ALL_SCOPE;
+  if (canonical.appId === "*") return "all";
+  if (canonical.tenantId === "*") return "app";
+  if (canonical.agentId === "*") return "tenant";
+  if (canonical.runId === "*") return "agent";
+  return "run";
+}
+
+export function scopeAtLevel(
+  scope: MemoryScope,
+  level: ScopeGranularity,
+): MemoryScope {
+  if (level === "all") return { ...ALL_SCOPE };
+
+  const canonical = normalizeHierarchicalScope(scope) ?? ALL_SCOPE;
+  const selectedIndex = SCOPE_LEVELS.indexOf(level);
+  const result = { ...ALL_SCOPE };
+  for (let index = 0; index <= selectedIndex; index += 1) {
+    const field = SCOPE_FIELD_BY_LEVEL[SCOPE_LEVELS[index]];
+    const value = canonical[field];
+    if (!value || value === "*") break;
+    result[field] = value;
+  }
+  return result;
+}
+
+export function scopeWithSelection(
+  scope: MemoryScope,
+  field: keyof MemoryScope,
+  value: string,
+): MemoryScope {
+  const canonical = normalizeHierarchicalScope(scope) ?? { ...ALL_SCOPE };
+  const level = LEVEL_BY_FIELD[field];
+  const levelIndex = SCOPE_LEVELS.indexOf(level);
+  canonical[field] = value.trim() || "*";
+  for (let index = levelIndex + 1; index < SCOPE_LEVELS.length; index += 1) {
+    canonical[SCOPE_FIELD_BY_LEVEL[SCOPE_LEVELS[index]]] = "*";
+  }
+  return normalizeHierarchicalScope(canonical) ?? { ...ALL_SCOPE };
+}
+
+export function scopeChildren(
+  scopes: readonly MemoryScope[],
+  parent: MemoryScope,
+): ScopeChild[] {
+  const canonicalParent = normalizeHierarchicalScope(parent) ?? ALL_SCOPE;
+  const parentGranularity = scopeGranularity(canonicalParent);
+  const nextIndex =
+    parentGranularity === "all"
+      ? 0
+      : SCOPE_LEVELS.indexOf(parentGranularity) + 1;
+  if (nextIndex >= SCOPE_LEVELS.length) return [];
+
+  const nextLevel = SCOPE_LEVELS[nextIndex];
+  const nextField = SCOPE_FIELD_BY_LEVEL[nextLevel];
+  const values = new Set<string>();
+
+  for (const scope of scopes) {
+    const candidate = normalizeHierarchicalScope(scope);
+    if (!candidate) continue;
+
+    const matchesParent = SCOPE_LEVELS.slice(0, nextIndex).every((level) => {
+      const field = SCOPE_FIELD_BY_LEVEL[level];
+      return candidate[field] === canonicalParent[field];
+    });
+    const value = candidate[nextField];
+    if (matchesParent && value && value !== "*") values.add(value);
+  }
+
+  return [...values]
+    .sort((left, right) => left.localeCompare(right))
+    .map((value) => ({ level: nextLevel, value }));
+}
+
+export function scopeBreadcrumb(scope: MemoryScope): ScopeBreadcrumbItem[] {
+  const canonical = normalizeHierarchicalScope(scope) ?? ALL_SCOPE;
+  return SCOPE_LEVELS.map((level) => {
+    const value = canonical[SCOPE_FIELD_BY_LEVEL[level]] ?? "*";
+    return { level, value, concrete: value !== "*" };
+  });
+}
+
+export function filterScopeTuples(
+  scopes: readonly MemoryScope[],
+  query: string,
+): MemoryScope[] {
+  const terms = query.toLocaleLowerCase().split(/\s+/).filter(Boolean);
+  if (terms.length === 0) return [...scopes];
+
+  return scopes.filter((scope) => {
+    const canonical = normalizeHierarchicalScope(scope);
+    if (!canonical) return false;
+    const text = SCOPE_LEVELS.map(
+      (level) => canonical[SCOPE_FIELD_BY_LEVEL[level]],
+    )
+      .join(" ")
+      .toLocaleLowerCase();
+    return terms.every((term) => text.includes(term));
+  });
+}

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

@@ -140,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`
@@ -162,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);
@@ -178,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/sdk-client.ts

@@ -18,6 +18,7 @@ import {
 import {
   KnowledgeBaseClient,
   MemoryClient,
+  WikiClient,
   inferRegionFromOtsEndpoint,
 } from "@tablestore/tablestore-agent-sdk";
 
@@ -139,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. */
@@ -578,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,