@falai/agent 3.2.4 → 3.3.0

package/src/providers/ZaiProvider.ts

@@ -0,0 +1,67 @@
+/**
+ * Z.ai Coding Plan.
+ *
+ * The flat-rate plan hosts GLM on an Anthropic-compatible endpoint with
+ * Bearer auth and its own thinking dialect — all of that lives in
+ * `@providerkit/core` (adapter + preset); this file is the constructor, the
+ * family default for cheap high-volume work.
+ *
+ * Dialect notes (measured 2026-09-13, see the core preset for detail):
+ * model ids are BARE (`glm-5.3-flash`, not `z-ai/glm-5.3-flash`), and
+ * "no thinking" must be an explicit marker — silence means the model default,
+ * which is thinking ON.
+ */
+
+import { createZaiCodingProvider } from "@providerkit/core";
+
+import type { ProviderCapabilities } from "../types/ai.js";
+import { ProviderAdapter, type RequestConfig } from "./ProviderAdapter.js";
+
+export interface ZaiProviderOptions {
+  /** Z.ai Coding Plan key */
+  apiKey: string;
+  /** Model to use. Defaults to the fleet workhorse `glm-5.3-flash`. */
+  model?: string;
+  /** Backup models to try if the primary fails (default: []) */
+  backupModels?: string[];
+  /** Any endpoint speaking the Anthropic Messages dialect with the plan's
+   *  auth. Defaults to the coding endpoint. */
+  baseUrl?: string;
+  /** Request defaults sent with every call */
+  config?: RequestConfig;
+  /** Idle-stream deadline and retry budget */
+  retryConfig?: { timeout?: number; retries?: number };
+  /** Replacement `fetch`, for tests that script the wire. */
+  fetchImpl?: typeof fetch;
+}
+
+export class ZaiProvider extends ProviderAdapter {
+  public readonly name = "zai";
+  public readonly capabilities: ProviderCapabilities = {
+    supportsTools: true,
+    supportsNativeJsonSchema: false,
+    supportsStreaming: true,
+    supportsStreamingToolCalls: true,
+    supportsPromptCaching: true,
+  };
+
+  constructor(options: ZaiProviderOptions) {
+    if (!options.apiKey) throw new Error("Z.ai Coding Plan API key is required");
+
+    const model = options.model ?? "glm-5.3-flash";
+    super({
+      provider: createZaiCodingProvider({
+        apiKey: options.apiKey,
+        model,
+        ...(options.baseUrl ? { baseUrl: options.baseUrl } : {}),
+        ...(options.config?.maxTokens ? { maxTokens: options.config.maxTokens } : {}),
+        ...(options.config?.effort ? { effort: options.config.effort } : {}),
+        ...(options.fetchImpl ? { fetchImpl: options.fetchImpl } : {}),
+      }),
+      model,
+      ...(options.config ? { defaults: options.config } : {}),
+      ...(options.backupModels ? { backupModels: options.backupModels } : {}),
+      ...(options.retryConfig ? { retryConfig: options.retryConfig } : {}),
+    });
+  }
+}

package/src/providers/index.ts

@@ -33,3 +33,6 @@ export type { OpenAICompatibleOptions } from "./GenericOpenAICompatibleProvider.
 
 export { ProviderAdapter, resolveRetryConfig } from "./ProviderAdapter.js";
 export type { ProviderAdapterInit, RequestConfig, RetryConfig } from "./ProviderAdapter.js";
+
+export { ZaiProvider } from "./ZaiProvider.js";
+export type { ZaiProviderOptions } from "./ZaiProvider.js";

package/src/utils/index.ts

@@ -24,6 +24,7 @@ export {
   completeCurrentFlow,
   isFlowCompletedThisSession,
   mergeCollected,
+  dropUndeclaredFields,
   sessionStepToData,
   sessionDataToStep,
 } from "./session.js";

package/src/utils/session.ts

@@ -3,6 +3,7 @@ import type { SessionState } from "../types/session.js";
 import type { History } from "../types/history.js";
 import type { Directive } from "../types/flow.js";
 import type { CollectedStateData } from "../types/persistence.js";
+import type { StructuredSchema } from "../types/schema.js";
 import { logger } from "./logger.js";
 
 /**
@@ -288,6 +289,29 @@ export function mergeCollected<TData = Record<string, unknown>>(
   };
 }
 
+/**
+ * Keep only the fields the agent schema declares. A model can extract a key
+ * nobody declared (e.g. `battery_health`); it is dropped with one warning
+ * instead of reaching session data. No schema, or no properties, keeps all.
+ */
+export function dropUndeclaredFields<TData>(
+  data: Partial<TData>,
+  schema: StructuredSchema | undefined
+): Partial<TData> {
+  const properties = schema?.properties;
+  if (!properties) return data;
+
+  const dropped = Object.keys(data).filter((key) => !(key in properties));
+  if (dropped.length === 0) return data;
+
+  logger.warn(
+    `[Agent] Dropped fields not declared in agent schema: ${dropped.join(", ")}`
+  );
+  return Object.fromEntries(
+    Object.entries(data).filter(([key]) => key in properties)
+  ) as Partial<TData>;
+}
+
 /**
  * Helper to convert SessionState to persistence-friendly format
  * Used when saving to database