@copilotkit/shared 1.54.1 → 1.55.0-next.7

package/src/utils/index.ts

@@ -6,6 +6,8 @@ export * from "./types";
 export * from "./random-id";
 export * from "./requests";
 
+import * as PartialJSON from "partial-json";
+
 /**
  * Safely parses a JSON string into an object
  * @param json The JSON string to parse
@@ -20,6 +22,39 @@ export function parseJson(json: string, fallback: any = "unset") {
   }
 }
 
+/**
+ * Parses a partial/incomplete JSON string, returning as much valid data as possible.
+ * Falls back to an empty object if parsing fails entirely.
+ */
+export function partialJSONParse(json: string) {
+  try {
+    const parsed = PartialJSON.parse(json);
+    if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
+      return parsed;
+    }
+    return {};
+  } catch (error) {
+    return {};
+  }
+}
+
+/**
+ * Returns an exponential backoff function suitable for Phoenix.js
+ * `reconnectAfterMs` and `rejoinAfterMs` options.
+ *
+ * @param baseMs  - Initial delay for the first retry attempt.
+ * @param maxMs   - Upper bound — delays are capped at this value.
+ *
+ * Phoenix calls the returned function with a 1-based `tries` count.
+ * The delay doubles on each attempt: baseMs, 2×baseMs, 4×baseMs, …, maxMs.
+ */
+export function phoenixExponentialBackoff(
+  baseMs: number,
+  maxMs: number,
+): (tries: number) => number {
+  return (tries: number) => Math.min(baseMs * 2 ** (tries - 1), maxMs);
+}
+
 /**
  * Maps an array of items to a new array, skipping items that throw errors during mapping
  * @param items The array to map
@@ -47,3 +82,26 @@ export function tryMap<TItem, TMapped>(
 export function isMacOS(): boolean {
   return /Mac|iMac|Macintosh/i.test(navigator.userAgent);
 }
+
+/**
+ * Safely parses a JSON string into a tool arguments object.
+ * Returns the parsed object only if it's a plain object (not an array, null, etc.).
+ * Falls back to an empty object for any non-object JSON value or parse failure.
+ */
+export function safeParseToolArgs(raw: string): Record<string, unknown> {
+  try {
+    const parsed = JSON.parse(raw);
+    if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
+      return parsed;
+    }
+    console.warn(
+      `[CopilotKit] Tool arguments parsed to non-object (${typeof parsed}), falling back to empty object`,
+    );
+    return {};
+  } catch {
+    console.warn(
+      "[CopilotKit] Failed to parse tool arguments, falling back to empty object",
+    );
+    return {};
+  }
+}

package/src/utils/types.ts

@@ -19,8 +19,29 @@ export interface AgentDescription {
   description: string;
 }
 
+export type RuntimeMode = "sse" | "intelligence";
+
+export const RUNTIME_MODE_SSE = "sse" as const;
+export const RUNTIME_MODE_INTELLIGENCE = "intelligence" as const;
+
+export interface IntelligenceRuntimeInfo {
+  wsUrl: string;
+}
+
+export type RuntimeLicenseStatus =
+  | "valid"
+  | "none"
+  | "expired"
+  | "expiring"
+  | "invalid"
+  | "unknown";
+
 export interface RuntimeInfo {
   version: string;
   agents: Record<string, AgentDescription>;
   audioFileTranscriptionEnabled: boolean;
+  mode: RuntimeMode;
+  intelligence?: IntelligenceRuntimeInfo;
+  a2uiEnabled?: boolean;
+  licenseStatus?: RuntimeLicenseStatus;
 }

package/src/utils.test.ts

@@ -0,0 +1,66 @@
+import { describe, test, expect, vi } from "vitest";
+import { partialJSONParse, safeParseToolArgs } from "./utils";
+
+describe("partialJSONParse", () => {
+  test("should parse a valid object", () => {
+    expect(partialJSONParse('{"key": "value"}')).toEqual({ key: "value" });
+  });
+
+  test("should return {} for a JSON string value", () => {
+    expect(partialJSONParse('"hello"')).toEqual({});
+  });
+
+  test("should return {} for a JSON number", () => {
+    expect(partialJSONParse("42")).toEqual({});
+  });
+
+  test("should return {} for a JSON boolean", () => {
+    expect(partialJSONParse("true")).toEqual({});
+  });
+
+  test("should return {} for a JSON null", () => {
+    expect(partialJSONParse("null")).toEqual({});
+  });
+
+  test("should return {} for a JSON array", () => {
+    expect(partialJSONParse("[1, 2, 3]")).toEqual({});
+  });
+
+  test("should return {} for unparseable input", () => {
+    expect(partialJSONParse("not-json")).toEqual({});
+  });
+
+  test("should return {} for an empty string", () => {
+    expect(partialJSONParse("")).toEqual({});
+  });
+});
+
+describe("safeParseToolArgs", () => {
+  test("should parse a valid JSON object string", () => {
+    expect(safeParseToolArgs('{"key": "value"}')).toEqual({ key: "value" });
+  });
+
+  test("should return {} for a JSON string value", () => {
+    expect(safeParseToolArgs('"hello"')).toEqual({});
+  });
+
+  test("should return {} for a JSON number", () => {
+    expect(safeParseToolArgs("42")).toEqual({});
+  });
+
+  test("should return {} for a JSON array", () => {
+    expect(safeParseToolArgs("[1, 2, 3]")).toEqual({});
+  });
+
+  test("should return {} for malformed JSON", () => {
+    expect(safeParseToolArgs("{broken")).toEqual({});
+  });
+
+  test("should return {} for a JSON null", () => {
+    expect(safeParseToolArgs("null")).toEqual({});
+  });
+
+  test("should return {} for a JSON boolean", () => {
+    expect(safeParseToolArgs("true")).toEqual({});
+  });
+});

package/tsconfig.json

@@ -2,8 +2,16 @@
   "extends": "../tsconfig/base.json",
   "include": ["."],
   "compilerOptions": {
+    "target": "es2017",
     "lib": ["es2017", "dom"],
     "resolveJsonModule": true
   },
-  "exclude": ["dist", "build", "node_modules", "**/*.test.ts", "**/*.test.tsx", "**/__tests__/*"]
+  "exclude": [
+    "dist",
+    "build",
+    "node_modules",
+    "**/*.test.ts",
+    "**/*.test.tsx",
+    "**/__tests__/*"
+  ]
 }

package/tsdown.config.ts

@@ -18,8 +18,14 @@ export default defineConfig([
     sourcemap: true,
     target: "es2018",
     outDir: "dist",
-    external: ["zod", "graphql", "uuid", "@ag-ui/core"],
-    codeSplitting: false,
+    external: [
+      "zod",
+      "graphql",
+      "uuid",
+      "@ag-ui/core",
+      "@ag-ui/client",
+      "partial-json",
+    ],
     outputOptions(options) {
       options.entryFileNames = "[name].umd.js";
       options.globals = {
@@ -27,8 +33,10 @@ export default defineConfig([
         graphql: "GraphQL",
         uuid: "UUID",
         "@ag-ui/core": "AgUICore",
+        "@ag-ui/client": "AgUIClient",
         "@segment/analytics-node": "SegmentAnalyticsNode",
         chalk: "chalk",
+        "partial-json": "PartialJSON",
       };
       return options;
     },