@dbx-tools/appkit-mastra 0.3.28 → 0.3.29

package/package.json

@@ -28,22 +28,21 @@
     "express": "^5.1.0",
     "pg": "^8.22.0",
     "zod": "^4.3.6",
-    "@dbx-tools/appkit": "0.3.28",
-    "@dbx-tools/core": "0.3.28",
-    "@dbx-tools/model": "0.3.28",
-    "@dbx-tools/shared-core": "0.3.28",
-    "@dbx-tools/databricks": "0.3.28",
-    "@dbx-tools/genie": "0.3.28",
-    "@dbx-tools/shared-genie": "0.3.28",
-    "@dbx-tools/shared-mastra": "0.3.28",
-    "@dbx-tools/shared-model": "0.3.28"
+    "@dbx-tools/appkit": "0.3.29",
+    "@dbx-tools/core": "0.3.29",
+    "@dbx-tools/model": "0.3.29",
+    "@dbx-tools/genie": "0.3.29",
+    "@dbx-tools/shared-core": "0.3.29",
+    "@dbx-tools/shared-mastra": "0.3.29",
+    "@dbx-tools/shared-genie": "0.3.29",
+    "@dbx-tools/shared-model": "0.3.29"
   },
   "main": "index.ts",
   "license": "UNLICENSED",
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.3.28",
+  "version": "0.3.29",
   "types": "index.ts",
   "type": "module",
   "exports": {

package/src/filesystems.ts

@@ -17,6 +17,17 @@ import { posix as path } from "node:path";
 import { getExecutionContext } from "@databricks/appkit";
 import { WorkspaceClient } from "@databricks/sdk-experimental";
 import { error, functionModule, hash, log } from "@dbx-tools/shared-core";
+import {
+  DirectoryNotEmptyError,
+  DirectoryNotFoundError,
+  FileExistsError,
+  FileNotFoundError,
+  IsDirectoryError,
+  MastraFilesystem,
+  NotDirectoryError,
+  PermissionError,
+  WorkspaceReadOnlyError,
+} from "@mastra/core/workspace";
 import type {
   CopyOptions,
   FileContent,
@@ -24,20 +35,11 @@ import type {
   FileStat,
   FilesystemInfo,
   ListOptions,
+  MastraFilesystemOptions,
   ProviderStatus,
   ReadOptions,
   RemoveOptions,
   WriteOptions,
-  DirectoryNotEmptyError,
-  DirectoryNotFoundError,
-  FileExistsError,
-  FileNotFoundError,
-  IsDirectoryError,
-  MastraFilesystem,
-  NotDirectoryError,
-  PermissionError,
-  WorkspaceReadOnlyError,
-  type MastraFilesystemOptions,
 } from "@mastra/core/workspace";
 
 /* ------------------------------ constants ------------------------------ */

package/src/genie.ts

@@ -50,7 +50,8 @@ import { z } from "zod";
 
 import type { MastraTools } from "./agents";
 import { chartPlannerRequestSchema, prepareChart } from "./chart";
-import type { MastraPluginConfig, MASTRA_USER_KEY, type User } from "./config";
+import { MASTRA_USER_KEY } from "./config";
+import type { MastraPluginConfig, User } from "./config";
 import { fetchStatementData } from "./statement";
 import { safeWrite } from "./writer";
 

package/src/memory.ts

@@ -29,9 +29,8 @@
  * @module
  */
 
-import { randomUUID } from "node:crypto";
 import { getUsernameWithApiLookup } from "@databricks/appkit";
-import { log } from "@dbx-tools/shared-core";
+import { hash, log } from "@dbx-tools/shared-core";
 import { fastembed } from "@mastra/fastembed";
 import { Memory } from "@mastra/memory";
 import { PgVector, PostgresStore } from "@mastra/pg";
@@ -252,7 +251,7 @@ export class MemoryBuilder {
  */
 function buildSharedPgVector(pool: Pool): PgVector {
   const vector = new PgVector({
-    id: `pg${randomUUID()}`,
+    id: `pg${hash.id()}`,
     // Keep the recall index out of `public`: on a Lakebase database the app
     // service principal has no CREATE on `public` (PG15+ locks it down), so a
     // default-schema PgVector fails on CREATE INDEX with "permission denied for
@@ -276,7 +275,7 @@ function buildSharedPgVector(pool: Pool): PgVector {
 /** Per-agent dedicated `PgVector` (rare; opt-in via object override). */
 function buildPgVector(setting: MastraMemoryConfigOverride): PgVector {
   return new PgVector(
-    withId(setting, `pg-vector__${randomUUID()}`) as ConstructorParameters<typeof PgVector>[0],
+    withId(setting, `pg-vector__${hash.id()}`) as ConstructorParameters<typeof PgVector>[0],
   );
 }
 

package/src/model.ts

@@ -25,7 +25,7 @@
 
 import { getExecutionContext } from "@databricks/appkit";
 import { classes, resolve } from "@dbx-tools/model";
-import { functionModule, log, net } from "@dbx-tools/shared-core";
+import { functionModule, json, log, net } from "@dbx-tools/shared-core";
 import { model } from "@dbx-tools/shared-model";
 import type { MastraModelConfig } from "@mastra/core/llm";
 import type { RequestContext } from "@mastra/core/request-context";
@@ -155,11 +155,13 @@ const setupFetchInterceptor = functionModule.memoize((): void => {
     if (rewritten !== init.body) {
       init = { ...init, body: rewritten };
     }
-    try {
-      logger.debug("POST", { url: url.toString(), body: JSON.parse(rewritten) });
-    } catch {
-      logger.debug("POST", { url: url.toString(), bodyType: "non-JSON" });
-    }
+    const parsed = json.parse<unknown>(rewritten);
+    logger.debug(
+      "POST",
+      parsed === undefined
+        ? { url: url.toString(), bodyType: "non-JSON" }
+        : { url: url.toString(), body: parsed },
+    );
     return original(input, init);
   }) as typeof globalThis.fetch;
 });

package/src/rest.ts

@@ -1,4 +1,5 @@
 import { appkit } from "@dbx-tools/appkit";
+import { json } from "@dbx-tools/shared-core";
 /**
  * Minimal authenticated Databricks REST helper. Pulls the workspace
  * host and a fresh bearer header off an OBO-scoped `WorkspaceClient`
@@ -58,10 +59,5 @@ export async function readResponseText(res: Response): Promise<string> {
 /** Parse a response body as JSON, returning `{}` on empty / invalid bodies. */
 export async function readResponseJson(res: Response): Promise<unknown> {
   const text = await readResponseText(res);
-  if (!text) return {};
-  try {
-    return JSON.parse(text);
-  } catch {
-    return {};
-  }
+  return json.parse<unknown>(text, {});
 }

package/src/server.ts

@@ -7,9 +7,8 @@
  * @module
  */
 
-import { randomUUID } from "node:crypto";
 import { getExecutionContext } from "@databricks/appkit";
-import { http, log, object, string, token } from "@dbx-tools/shared-core";
+import { hash, http, log, object, string, token } from "@dbx-tools/shared-core";
 import { feedback, thread } from "@dbx-tools/shared-mastra";
 import {
   MASTRA_RESOURCE_ID_KEY,
@@ -142,7 +141,7 @@ export class MastraServer extends MastraServerExpress {
     if (requestContext.get(MASTRA_REQUEST_ID_KEY)) return;
     const headerValue = req.headers["x-request-id"];
     const upstream = Array.isArray(headerValue) ? headerValue[0] : headerValue;
-    const requestId = upstream?.trim() || randomUUID();
+    const requestId = upstream?.trim() || hash.id();
     requestContext.set(MASTRA_REQUEST_ID_KEY, requestId);
     res.setHeader("X-Request-Id", requestId);
   }
@@ -218,7 +217,7 @@ export class MastraServer extends MastraServerExpress {
     );
     let sessionId = cookies[cookieName];
     if (!sessionId) {
-      sessionId = randomUUID();
+      sessionId = hash.id();
       res.cookie(cookieName, sessionId, {
         httpOnly: true,
         sameSite: "lax",

package/src/serving-sanitize.ts

@@ -1,5 +1,10 @@
-import { string } from "@dbx-tools/shared-core";
-import { type ChatMessage, type ChatRole, openaiChat } from "@dbx-tools/shared-model";
+import { json, string } from "@dbx-tools/shared-core";
+import {
+  type ChatMessage,
+  type ChatRole,
+  openaiChat,
+  openaiResponses,
+} from "@dbx-tools/shared-model";
 /**
  * Repairs Mastra / AI SDK message replays sent to Databricks Model
  * Serving before they hit the OpenAI-compatible `/chat/completions`
@@ -19,7 +24,9 @@ export interface ServingChatMessage extends ChatMessage {
   reasoning_content?: unknown;
 }
 
-const REASONING_PART_TYPES = new Set(["reasoning", "thinking", "redacted_thinking"]);
+// Shared with the Responses sanitize path so both wire surfaces agree on what
+// counts as a signed reasoning block.
+const REASONING_PART_TYPES = openaiResponses.REASONING_TYPES;
 
 /**
  * Parse, sanitize, and re-serialize a `/serving-endpoints/...` POST
@@ -27,12 +34,8 @@ const REASONING_PART_TYPES = new Set(["reasoning", "thinking", "redacted_thinkin
  * JSON or no rewrite was needed.
  */
 export function rewriteServingBody(body: string): string {
-  let parsed: Record<string, unknown>;
-  try {
-    parsed = JSON.parse(body) as Record<string, unknown>;
-  } catch {
-    return body;
-  }
+  const parsed = json.parseRecord(body);
+  if (!parsed) return body;
 
   // Runs regardless of `messages`: Databricks refuses to parse a body carrying
   // an unknown top-level field, so this failure is not specific to a transcript.