@treeseed/sdk 0.10.28 → 0.12.4

package/dist/sdk.js

@@ -3,6 +3,32 @@ import { normalizeAgentCliOptions } from "./cli-tools.js";
 import { ContentStore } from "./content-store.js";
 import { CloudflareD1AgentDatabase, MemoryAgentDatabase } from "./d1-store.js";
 import { ContentGraphRuntime } from "./graph.js";
+import {
+  createTreeDxClientFromAgentOptions,
+  LocalContentBackend,
+  LocalGraphBackend,
+  MissingTreeDxContentBackend,
+  resolveTreeDxOptions,
+  TreeDxContentBackend,
+  TreeDxGraphBackend,
+  TreeDxPortfolioResolver
+} from "./treedx-backends.js";
+import {
+  LocalGraphPort,
+  LocalRepositoryPort,
+  LocalRepositoryQueryPort,
+  TreeDxArtifactPort,
+  TreeDxExecPort,
+  TreeDxFederatedClient,
+  TreeDxFederatedPort,
+  TreeDxGraphAdapter,
+  TreeDxGraphPort,
+  TreeDxRegistryClient,
+  TreeDxRegistryPort,
+  TreeDxRepositoryPort,
+  TreeDxRepositoryQueryPort,
+  TreeDxClient as PublicTreeDxClient
+} from "./treedx/index.js";
 import { loadTreeseedPlugins } from "./platform/plugins.js";
 import { buildScopedModelRegistry, resolveModelDefinition } from "./model-registry.js";
 import { findDispatchCapability } from "./dispatch.js";
@@ -35,14 +61,22 @@ class AgentSdk {
   database;
   content;
   models;
+  ports;
+  treeDx;
+  localContentStore;
+  localGraphRuntime;
   graph;
   dispatchConfig;
   constructor(options = {}) {
-    const repoRoot = resolveSdkRepoRoot(options.repoRoot);
+    const rawTreeDxOptions = options.treeDx;
+    if (rawTreeDxOptions && !options.repoRoot && !options.models && !options.modelRegistry) {
+      throw new Error("AgentSdk TreeDX mode requires explicit models or modelRegistry.");
+    }
+    const repoRoot = options.repoRoot ? resolveSdkRepoRoot(options.repoRoot) : rawTreeDxOptions ? process.cwd() : resolveSdkRepoRoot();
     this.repoRoot = repoRoot;
     this.models = options.modelRegistry ?? buildScopedModelRegistry(repoRoot, options.models);
     this.database = options.database ?? new MemoryAgentDatabase();
-    this.content = new ContentStore(repoRoot, this.database, this.models);
+    this.localContentStore = new ContentStore(repoRoot, this.database, this.models);
     let plugins = options.plugins;
     if (!plugins) {
       try {
@@ -51,20 +85,137 @@ class AgentSdk {
         plugins = [];
       }
     }
-    this.graph = new ContentGraphRuntime(repoRoot, this.models, {
+    this.localGraphRuntime = new ContentGraphRuntime(repoRoot, this.models, {
       rankingProvider: options.graphRankingProvider,
       plugins
     });
+    const treeDxOptions = resolveTreeDxOptions(options.treeDx);
+    if (options.contentRepository?.adapter === "local") {
+      this.content = new LocalContentBackend(this.localContentStore);
+      this.graph = new LocalGraphBackend(this.localGraphRuntime);
+      this.ports = {
+        repository: new LocalRepositoryPort(this.localContentStore),
+        query: new LocalRepositoryQueryPort(this.localContentStore),
+        graph: new LocalGraphPort(this.localGraphRuntime)
+      };
+    } else if (treeDxOptions) {
+      const publicClient = rawTreeDxOptions?.client instanceof PublicTreeDxClient ? rawTreeDxOptions.client : new PublicTreeDxClient({
+        baseUrl: treeDxOptions.baseUrl,
+        token: treeDxOptions.token,
+        repoId: treeDxOptions.repoId,
+        fetch: treeDxOptions.fetchImpl
+      });
+      const publicFetch = (input, init) => {
+        const headers = init?.headers instanceof Headers ? Object.fromEntries(init.headers.entries()) : init?.headers;
+        return (publicClient.fetchImpl ?? fetch)(input, {
+          ...init,
+          headers
+        });
+      };
+      const client = createTreeDxClientFromAgentOptions({
+        ...treeDxOptions,
+        fetchImpl: treeDxOptions.fetchImpl ?? publicFetch
+      });
+      const resolver = new TreeDxPortfolioResolver({
+        client,
+        repoId: treeDxOptions.repoId,
+        ref: treeDxOptions.ref,
+        repositoryHints: treeDxOptions.repositoryHints
+      });
+      const publicGraph = new TreeDxGraphAdapter({
+        client: publicClient,
+        repoId: treeDxOptions.repoId,
+        defaultRef: treeDxOptions.ref
+      });
+      const publicRegistry = rawTreeDxOptions?.registryRouting ? new TreeDxRegistryClient(publicClient) : void 0;
+      const publicFederated = publicRegistry ? new TreeDxFederatedClient({
+        registry: publicRegistry,
+        token: treeDxOptions.token,
+        fetch: treeDxOptions.fetchImpl
+      }) : void 0;
+      const publicExec = new TreeDxExecPort(publicClient);
+      const publicArtifact = new TreeDxArtifactPort(publicClient);
+      this.content = new TreeDxContentBackend({
+        client,
+        repoRoot,
+        models: this.models,
+        resolver,
+        directRepoId: treeDxOptions.repoId,
+        ref: treeDxOptions.ref,
+        workspaceId: treeDxOptions.workspaceId,
+        contentPathMap: treeDxOptions.contentPathMap,
+        localLeaseStore: this.localContentStore
+      });
+      this.graph = new TreeDxGraphBackend({
+        client,
+        resolver,
+        localRuntime: this.localGraphRuntime,
+        directRepoId: treeDxOptions.repoId,
+        ref: treeDxOptions.ref
+      });
+      this.treeDx = {
+        client: publicClient,
+        graph: publicGraph,
+        registry: publicRegistry,
+        federated: publicFederated,
+        exec: publicExec,
+        artifact: publicArtifact
+      };
+      this.ports = {
+        repository: new TreeDxRepositoryPort(publicClient),
+        query: new TreeDxRepositoryQueryPort(publicClient),
+        graph: new TreeDxGraphPort(publicGraph),
+        registry: publicRegistry ? new TreeDxRegistryPort(publicRegistry) : void 0,
+        federated: publicFederated ? new TreeDxFederatedPort(publicFederated) : void 0,
+        exec: publicExec,
+        artifact: publicArtifact
+      };
+    } else if (rawTreeDxOptions?.client) {
+      const publicClient = rawTreeDxOptions.client instanceof PublicTreeDxClient ? rawTreeDxOptions.client : new PublicTreeDxClient(rawTreeDxOptions.client);
+      const publicGraph = new TreeDxGraphAdapter({
+        client: publicClient,
+        repoId: rawTreeDxOptions.repoId,
+        defaultRef: rawTreeDxOptions.ref
+      });
+      const publicExec = new TreeDxExecPort(publicClient);
+      const publicArtifact = new TreeDxArtifactPort(publicClient);
+      this.content = new MissingTreeDxContentBackend();
+      this.graph = new LocalGraphBackend(this.localGraphRuntime);
+      this.treeDx = {
+        client: publicClient,
+        graph: publicGraph,
+        exec: publicExec,
+        artifact: publicArtifact
+      };
+      this.ports = {
+        repository: new TreeDxRepositoryPort(publicClient),
+        query: new TreeDxRepositoryQueryPort(publicClient),
+        graph: new TreeDxGraphPort(publicGraph),
+        exec: publicExec,
+        artifact: publicArtifact
+      };
+    } else {
+      this.content = new MissingTreeDxContentBackend();
+      this.graph = new LocalGraphBackend(this.localGraphRuntime);
+      this.ports = {
+        repository: new LocalRepositoryPort(this.localContentStore),
+        query: new LocalRepositoryQueryPort(this.localContentStore),
+        graph: new LocalGraphPort(this.localGraphRuntime)
+      };
+    }
     this.dispatchConfig = options.dispatch;
   }
-  static createLocal(options) {
+  static createLocal(options = {}) {
     const repoRoot = resolveSdkRepoRoot(options.repoRoot);
     const d1 = new NodeSqliteD1Database(options.persistTo ?? options.databaseName ?? ".treeseed/generated/environments/local/site-data.sqlite");
     return new AgentSdk({
       repoRoot,
       database: new CloudflareD1AgentDatabase(d1),
       models: options.models,
-      modelRegistry: options.modelRegistry
+      modelRegistry: options.modelRegistry,
+      contentRepository: options.contentRepository,
+      treeDx: options.treeDx,
+      dispatch: options.dispatch
     });
   }
   async resolveDispatchToken(source) {
@@ -89,6 +240,27 @@ class AgentSdk {
         transport: "sdk"
       });
     }
+    if (this.content instanceof MissingTreeDxContentBackend) {
+      const input = request.input ?? {};
+      if (request.operation === "read" || request.operation === "get") {
+        const definition = resolveModelDefinition(input.model, this.models);
+        const payload = await this.localContentStore.get({
+          ...input,
+          model: definition.name
+        });
+        return this.envelope(definition.name, request.operation === "read" ? "read" : "get", payload);
+      }
+      if (request.operation === "search") {
+        const definition = resolveModelDefinition(input.model, this.models);
+        const payload = await this.localContentStore.search({
+          ...input,
+          model: definition.name
+        });
+        return this.envelope(definition.name, "search", payload, {
+          count: payload.length
+        });
+      }
+    }
     return executeSdkOperation(this, request.operation, request.input ?? {});
   }
   async dispatch(request) {
@@ -164,7 +336,7 @@ class AgentSdk {
   }
   async pick(request) {
     const definition = resolveModelDefinition(request.model, this.models);
-    const payload = definition.storage === "content" ? await this.content.pick({ ...request, model: definition.name }) : await this.database.pick({ ...request, model: definition.name });
+    const payload = definition.storage === "content" ? await this.localContentStore.pick({ ...request, model: definition.name }) : await this.database.pick({ ...request, model: definition.name });
     return this.envelope(definition.name, "pick", payload, {
       claimed: Boolean(payload.item)
     });
@@ -221,42 +393,10 @@ class AgentSdk {
     const payload = await this.database.closeWorkDay(request);
     return this.envelope("work_day", "update", payload);
   }
-  async createTask(request) {
-    const payload = await this.database.createTask(request);
-    return this.envelope("task", "create", payload);
-  }
-  async claimTask(request) {
-    const payload = await this.database.claimTask(request);
-    return this.envelope("task", "update", payload);
-  }
-  async recordTaskProgress(request) {
-    const payload = await this.database.recordTaskProgress(request);
-    return this.envelope("task", "update", payload);
-  }
-  async completeTask(request) {
-    const payload = await this.database.completeTask(request);
-    return this.envelope("task", "update", payload);
-  }
-  async failTask(request) {
-    const payload = await this.database.failTask(request);
-    return this.envelope("task", "update", payload);
-  }
-  async appendTaskEvent(request) {
-    const payload = await this.database.appendTaskEvent(request);
-    return this.envelope("task_event", "create", payload);
-  }
-  async searchTasks(request) {
-    const payload = await this.database.searchTasks(request);
-    return this.envelope("task", "search", payload, { count: payload.length });
-  }
   async createReport(request) {
     const payload = await this.database.createReport(request);
     return this.envelope("report", "create", payload);
   }
-  async getManagerContext(taskId) {
-    const payload = await this.database.getManagerContext(taskId);
-    return this.envelope("task", "get", payload);
-  }
   async getWorkPolicy(projectId, environment = "local") {
     const payload = await this.database.getWorkPolicy(projectId, environment);
     return this.envelope("work_day", "get", payload);
@@ -273,6 +413,26 @@ class AgentSdk {
     const payload = await this.database.listWorkdayRequests(projectId, environment, state);
     return this.envelope("workday_request", "search", payload, { count: payload.length });
   }
+  async createTask(request) {
+    const payload = await this.database.createTask(request);
+    return this.envelope("task", "create", payload);
+  }
+  async claimTask(request) {
+    const payload = await this.database.claimTask(request);
+    return this.envelope("task", "pick", payload);
+  }
+  async recordTaskProgress(request) {
+    const payload = await this.database.recordTaskProgress(request);
+    return this.envelope("task", "update", payload);
+  }
+  async completeTask(request) {
+    const payload = await this.database.completeTask(request);
+    return this.envelope("task", "update", payload);
+  }
+  async getManagerContext(taskId) {
+    const payload = await this.database.getManagerContext(taskId);
+    return this.envelope("task", "get", payload);
+  }
   async claimWorkdayManagerLease(request) {
     const payload = await this.database.claimWorkdayManagerLease(request);
     return this.envelope("workday_manager_lease", "claim", payload);
@@ -375,7 +535,7 @@ class AgentSdk {
   }
   async appendWorkstreamEvent(input) {
     const payload = await this.database.appendWorkstreamEvent(input);
-    return this.envelope("task_event", "create", payload);
+    return this.envelope("workstream_event", "create", payload);
   }
   async listReleases(projectId) {
     const payload = await this.database.listReleases(projectId);
@@ -423,43 +583,43 @@ class AgentSdk {
   }
   /** Advanced lexical graph primitive for file nodes. Prefer queryGraph() or buildContextPack() for AI-context retrieval. */
   searchFiles(query, options) {
-    return this.graph.searchFiles(query, options);
+    return this.localGraphRuntime.searchFiles(query, options);
   }
   /** Advanced lexical graph primitive for section nodes. Prefer queryGraph() or buildContextPack() for AI-context retrieval. */
   searchSections(query, options) {
-    return this.graph.searchSections(query, options);
+    return this.localGraphRuntime.searchSections(query, options);
   }
   /** Advanced lexical graph primitive for entity nodes. Prefer queryGraph() or buildContextPack() for AI-context retrieval. */
   searchEntities(query, options) {
-    return this.graph.searchEntities(query, options);
+    return this.localGraphRuntime.searchEntities(query, options);
   }
   /** Advanced graph primitive that returns one raw graph node by id. */
   getGraphNode(id) {
-    return this.graph.getNode(id);
+    return this.localGraphRuntime.getNode(id);
   }
   /** Advanced graph primitive for direct neighborhood inspection. Prefer queryGraph() for ranked retrieval. */
   getNeighbors(id, options) {
-    return this.graph.getNeighbors(id, options);
+    return this.localGraphRuntime.getNeighbors(id, options);
   }
   /** Advanced traversal primitive for direct reference walking. Prefer queryGraph() when you need ranking and ctx-aware behavior. */
   followReferences(id, options) {
-    return this.graph.followReferences(id, options);
+    return this.localGraphRuntime.followReferences(id, options);
   }
   /** Advanced graph primitive for incoming-link inspection. */
   getBacklinks(id, options) {
-    return this.graph.getBacklinks(id, options);
+    return this.localGraphRuntime.getBacklinks(id, options);
   }
   /** Advanced graph primitive for local relatedness. Prefer queryGraph() for the primary ranked graph workflow. */
   getRelated(id, options) {
-    return this.graph.getRelated(id, options);
+    return this.localGraphRuntime.getRelated(id, options);
   }
   /** Advanced traversal primitive for raw subgraph extraction. Prefer buildContextPack() when you need prompt-ready output. */
   getSubgraph(seedIds, options) {
-    return this.graph.getSubgraph(seedIds, options);
+    return this.localGraphRuntime.getSubgraph(seedIds, options);
   }
   /** Primary graph workflow helper. Resolves roots before ranking and traversal. */
   resolveSeeds(request) {
-    return this.graph.resolveSeeds(request);
+    return this.localGraphRuntime.resolveSeeds(request);
   }
   /** Primary graph workflow entrypoint for ranked graph retrieval. */
   queryGraph(request) {
@@ -475,11 +635,11 @@ class AgentSdk {
   }
   /** Primary graph workflow helper for resolving ids, paths, and anchors into graph nodes. */
   resolveReference(reference, options) {
-    return this.graph.resolveReference(reference, options);
+    return this.localGraphRuntime.resolveReference(reference, options);
   }
   /** Primary graph workflow helper for explaining why two nodes are connected. */
   explainReferenceChain(fromId, toId) {
-    return this.graph.explainReferenceChain(fromId, toId);
+    return this.localGraphRuntime.explainReferenceChain(fromId, toId);
   }
 }
 class ScopedAgentSdk {