@picoflow/ezgraph 0.0.4

package/dist/types/src/testing/scripted-gateway.d.ts

@@ -0,0 +1,45 @@
+import { AIMessage, type BaseMessage } from '@langchain/core/messages';
+import type { ZodType } from 'zod';
+import { type GraphLlmConfig } from '../framework/graph-llm.config.js';
+import type { LlmCallOptions, LlmGateway, LlmGatewayTool, LlmGatewayToolCall } from '../framework/llm-gateway.js';
+import { type ModelResult, type TokenUsage } from '../framework/token-usage.js';
+export declare const SCRIPTED_MODEL: GraphLlmConfig;
+export declare const SCRIPTED_TOKEN_USAGE: TokenUsage;
+export type ScriptedCall = Readonly<{
+    method: 'structured' | 'generate' | 'respond' | 'agent' | 'toolCall';
+    systemPrompt: string;
+    history: readonly BaseMessage[];
+    tools: readonly string[];
+    llmConfig?: GraphLlmConfig | undefined;
+    callOptions?: LlmCallOptions | undefined;
+}>;
+export declare class ScriptedGateway implements LlmGateway {
+    private readonly turns;
+    private consumed;
+    readonly calls: ScriptedCall[];
+    text(text: string): this;
+    callsTool(name: string, args?: Record<string, unknown>, id?: string): this;
+    callsTools(calls: readonly {
+        name: string;
+        args?: Record<string, unknown>;
+        id?: string;
+    }[]): this;
+    structuredResult(value: Record<string, unknown>): this;
+    toolDecision(call: LlmGatewayToolCall | undefined): this;
+    empty(finishReason?: string): this;
+    fail(error: Error | string): this;
+    repeat(times: number, declare: (gateway: this) => this): this;
+    get drained(): boolean;
+    get remaining(): number;
+    structured<T extends Record<string, unknown>>(_schema: ZodType<T>, systemPrompt: string, input: string, name: string, llmConfig?: GraphLlmConfig, history?: readonly BaseMessage[], callOptions?: LlmCallOptions): Promise<ModelResult<T>>;
+    generate(systemPrompt: string, userPrompt: string, llmConfig?: GraphLlmConfig, callOptions?: LlmCallOptions): Promise<ModelResult<string>>;
+    respond(systemPrompt: string, history: readonly BaseMessage[], llmConfig?: GraphLlmConfig, callOptions?: LlmCallOptions): Promise<ModelResult<string>>;
+    agent(systemPrompt: string, history: readonly BaseMessage[], tools: readonly LlmGatewayTool[], llmConfig?: GraphLlmConfig, callOptions?: LlmCallOptions): Promise<ModelResult<AIMessage>>;
+    toolCall(systemPrompt: string, userPrompt: string, tools: readonly LlmGatewayTool[], llmConfig?: GraphLlmConfig, history?: readonly BaseMessage[], callOptions?: LlmCallOptions): Promise<ModelResult<LlmGatewayToolCall | undefined>>;
+    private record;
+    private takeText;
+    private take;
+    private mismatch;
+    private result;
+}
+export declare function scriptedGateway(): ScriptedGateway;

package/dist/types/src/testing/turn-harness.d.ts

@@ -0,0 +1,35 @@
+import { GraphEngine, type GraphConstructor, type RunResult } from '../framework/graph-engine.js';
+import type { GraphState } from '../framework/graph-state.js';
+import type { LlmGateway } from '../framework/llm-gateway.js';
+import type { SessionLogs } from '../framework/session-log.js';
+import { SessionManager, type SessionDocument } from '../framework/persistence/session-manager.js';
+export type HarnessTurn<State extends GraphState> = Readonly<{
+    status: number;
+    session: string;
+    response: string;
+    completed: boolean;
+    code?: string;
+    body: RunResult['body'];
+    contentType?: 'application/json';
+    document?: SessionDocument<State>;
+    state?: State;
+    currentNode?: State['currentNode'];
+    warnings: SessionLogs['warnings'];
+    errors: SessionLogs['errors'];
+}>;
+export type TurnHarness<State extends GraphState> = Readonly<{
+    session: string;
+    engine: GraphEngine;
+    send: (userMessage?: string, options?: Readonly<{
+        config?: Record<string, unknown>;
+        signal?: AbortSignal;
+    }>) => Promise<HarnessTurn<State>>;
+    document: () => Promise<SessionDocument<State> | undefined>;
+    close: () => Promise<void>;
+}>;
+export declare function createTurnHarness<State extends GraphState = GraphState>(options: Readonly<{
+    graph: GraphConstructor;
+    gateway?: LlmGateway;
+    sessionManager?: SessionManager;
+    sessionId?: string;
+}>): TurnHarness<State>;

package/model-catalog.schema.json

@@ -0,0 +1,57 @@
+{
+  "$schema": "https://json-schema.org/draft/2020-12/schema",
+  "$id": "https://ezgraph.dev/schemas/model-catalog-v1.json",
+  "title": "EZGraph model catalog",
+  "type": "object",
+  "additionalProperties": false,
+  "required": ["version", "models"],
+  "properties": {
+    "$schema": { "type": "string", "minLength": 1 },
+    "version": { "const": 1 },
+    "profiles": {
+      "type": "object",
+      "additionalProperties": { "$ref": "#/$defs/profile" }
+    },
+    "models": {
+      "type": "object",
+      "additionalProperties": { "$ref": "#/$defs/model" }
+    }
+  },
+  "$defs": {
+    "profile": {
+      "type": "object",
+      "additionalProperties": false,
+      "required": ["provider", "family", "paramsSchema", "parameterMappings"],
+      "properties": {
+        "provider": {
+          "type": "string",
+          "pattern": "^[a-z][a-z0-9_-]*$"
+        },
+        "family": { "enum": ["chat", "reasoning", "thinking"] },
+        "paramsSchema": {
+          "type": "object",
+          "description": "JSON Schema for the complete params object."
+        },
+        "parameterMappings": {
+          "type": "object",
+          "description": "Parameter-to-LangChain paths; null denotes EZGraph policy.",
+          "additionalProperties": {
+            "oneOf": [
+              { "type": "string", "minLength": 1 },
+              { "type": "null" }
+            ]
+          }
+        }
+      }
+    },
+    "model": {
+      "type": "object",
+      "additionalProperties": false,
+      "required": ["profile"],
+      "properties": {
+        "profile": { "type": "string", "minLength": 1 },
+        "providerModel": { "type": "string", "minLength": 1 }
+      }
+    }
+  }
+}

package/package.json

@@ -0,0 +1,64 @@
+{
+  "name": "@picoflow/ezgraph",
+  "version": "0.0.4",
+  "description": "Provider-neutral graph primitives built on LangGraph and LangChain.",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/types/npmlib/src/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/types/npmlib/src/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    },
+    "./testing": {
+      "types": "./dist/types/npmlib/src/testing.d.ts",
+      "import": "./dist/testing.js",
+      "default": "./dist/testing.js"
+    },
+    "./model-catalog.schema.json": "./model-catalog.schema.json"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE.md",
+    "model-catalog.schema.json"
+  ],
+  "engines": {
+    "node": ">=22.5"
+  },
+  "license": "SEE LICENSE IN LICENSE.md",
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@anthropic-ai/sdk": "0.115.0",
+    "@azure/cosmos": "4.10.0",
+    "@azure/identity": "4.13.1",
+    "@google/genai": "2.14.0",
+    "dotenv": "17.4.2",
+    "@langchain/core": "1.2.3",
+    "@langchain/langgraph": "1.4.8",
+    "langchain": "1.5.4",
+    "mongodb": "6.21.0",
+    "openai": "7.1.0",
+    "reflect-metadata": "0.2.2",
+    "zod": "4.4.3"
+  },
+  "peerDependencies": {
+    "@langchain/anthropic": ">=1.5.2",
+    "@langchain/google": ">=0.2.1",
+    "@langchain/openai": ">=1.5.5"
+  },
+  "peerDependenciesMeta": {
+    "@langchain/anthropic": {
+      "optional": true
+    },
+    "@langchain/google": {
+      "optional": true
+    },
+    "@langchain/openai": {
+      "optional": true
+    }
+  }
+}