structured-json-agent 1.0.1 → 1.1.0

package/README.md

@@ -2,58 +2,72 @@
 
 A typed and extensible TypeScript library for creating and running Iterative AI Agents that guarantee structured JSON output.
 
-This library orchestrates a **Generator ↔ Reviewer** cycle to ensure that the output from Large Language Models (LLMs) strictly adheres to a defined JSON Schema.
+This library orchestrates a **Generator ↔ Reviewer** cycle to ensure that the output from Large Language Models (LLMs) strictly adheres to a defined Zod Schema.
 
 ## Features
 
-*   **Guaranteed JSON Output**: Enforces strict adherence to JSON Schemas (Draft-07+).
+*   **Guaranteed JSON Output**: Enforces strict adherence to Zod Schemas.
+*   **Multi-Provider Support**: Built-in adapters for **OpenAI**, **Google GenAI (Gemini)**, and **Anthropic (Claude)**.
+*   **Structured Outputs**: Leverages native structured output capabilities of providers (e.g., OpenAI Structured Outputs, Anthropic Beta) when available.
 *   **Iterative Self-Correction**: Automatically detects validation errors and feeds them back to a "Reviewer" model to fix the output.
-*   **Type-Safe**: Built with TypeScript for full type inference and safety.
-*   **Model Agnostic**: Compatible with OpenAI by default, but extensible for other providers.
-*   **Production Ready**: Includes typed errors, extensive validation, and a clean API.
+*   **Type-Safe**: Built with TypeScript and Zod for full type inference and safety.
+*   **Flexible Configuration**: Mix and match different providers for generation and review (e.g., generate with GPT-4o, review with Claude 3.5 Sonnet).
 
 ## Installation
 
 ```bash
-npm install structured-json-agent
+npm install structured-json-agent zod openai @anthropic-ai/sdk @google/genai
 ```
 
+> Note: Install the SDKs for the providers you intend to use.
+
 ## Usage
 
 ### 1. Import and Configure
 
 ```typescript
 import { StructuredAgent } from "structured-json-agent";
+import { z } from "zod";
+import OpenAI from "openai";
+import Anthropic from "@anthropic-ai/sdk";
+
+// 1. Define your Schemas using Zod
+const inputSchema = z.object({
+  topic: z.string(),
+  depth: z.enum(["basic", "advanced"])
+});
 
-// Define your Schemas
-const inputSchema = {
-  type: "object",
-  properties: {
-    topic: { type: "string" },
-    depth: { type: "string", enum: ["basic", "advanced"] }
-  },
-  required: ["topic", "depth"]
-};
-
-const outputSchema = {
-  type: "object",
-  properties: {
-    title: { type: "string" },
-    keyPoints: { type: "array", items: { type: "string" } },
-    summary: { type: "string" }
-  },
-  required: ["title", "keyPoints", "summary"]
-};
+const outputSchema = z.object({
+  title: z.string(),
+  keyPoints: z.array(z.string()),
+  summary: z.string()
+});
+
+// 2. Initialize Provider Instances
+const openAiInstance = new OpenAI({
+  apiKey: process.env.OPENAI_API_KEY,
+});
 
-// Initialize the Agent
+const anthropicInstance = new Anthropic({
+  apiKey: process.env.ANTHROPIC_API_KEY,
+});
+
+// 3. Initialize the Agent, you can use the same instance for generator and reviewer
 const agent = new StructuredAgent({
-  openAiApiKey: process.env.OPENAI_API_KEY!,
-  generatorModel: "gpt-4-turbo",
-  reviewerModel: "gpt-3.5-turbo", // Can be a faster/cheaper model for simple fixes
+  // Generator Configuration
+  generator: {
+    llmService: openAiInstance, // Inject the instance directly
+    model: "gpt-5-nano",       // Specify the model
+  },
+  // Reviewer Configuration (Optional but recommended)
+  reviewer: {
+    llmService: anthropicInstance,
+    model: "claude-sonnet-4-5",
+  },
+  // Schemas & Prompt
   inputSchema,
   outputSchema,
   systemPrompt: "You are an expert summarizer. Create a structured summary based on the topic.",
-  maxIterations: 3 // Optional: Max correction attempts (default: 5)
 });
 ```
 
@@ -68,7 +82,7 @@ async function main() {
     });
 
     console.log("Result:", result);
-    // Output is guaranteed to match outputSchema
+    // Result is typed as inferred from outputSchema
   } catch (error) {
     console.error("Agent failed:", error);
   }
@@ -79,47 +93,44 @@ main();
 
 ## How It Works
 
-1.  **Validation**: The input JSON is validated against the `inputSchema`.
-2.  **Generation**: The `generatorModel` creates an initial response based on the system prompt and input.
+1.  **Validation**: The input JSON is validated against the `inputSchema` (Zod).
+2.  **Generation**: The `generator` model creates an initial response based on the system prompt and input.
+    *   If the provider supports native Structured Outputs (like OpenAI or Anthropic), it is used to maximize reliability.
 3.  **Verification Loop**:
     *   The response is parsed and validated against `outputSchema`.
     *   **If Valid**: The result is returned immediately.
-    *   **If Invalid**: The `reviewerModel` is invoked with the invalid JSON, the specific validation errors, and the expected schema. It attempts to fix the JSON.
+    *   **If Invalid**: The `reviewer` model (or generator if no reviewer is set) is invoked with the invalid JSON and specific validation errors. It attempts to fix the output.
 4.  **Convergence**: This cycle repeats until a valid JSON is produced or `maxIterations` is reached.
 
 ## API Reference
 
-### `StructuredAgent` Config
+### `AgentConfig`
+
+Configuration object passed to `new StructuredAgent(config)`.
 
 | Property | Type | Description |
-|Col |Col |Col |
-| `openAiApiKey` | `string` | Your OpenAI API Key. |
-| `generatorModel` | `string` | Model ID for the initial generation (e.g., `gpt-4`). |
-| `reviewerModel` | `string` | Model ID for the review/correction phase. |
-| `inputSchema` | `object` | JSON Schema for validating the input. |
-| `outputSchema` | `object` | JSON Schema for the expected output. |
+| :--- | :--- | :--- |
+| `generator` | `LLMConfig` | Configuration for the generation model. |
+| `reviewer` | `LLMConfig?` | Configuration for the reviewer model (optional). |
+| `inputSchema` | `ZodSchema` | Zod Schema for validating the input. |
+| `outputSchema` | `ZodSchema` | Zod Schema for the expected output. |
 | `systemPrompt` | `string` | Core instructions for the agent. |
 | `maxIterations` | `number?` | Max retries for correction. Default: 5. |
-| `modelConfig` | `ModelConfig?` | Optional parameters (temperature, etc.). |
-| `llmService` | `ILLMService?` | Optional custom LLM service implementation. |
 
-### Error Handling
+### `LLMConfig`
 
-The library exports specific error classes for handling failures:
-
-*   `InvalidInputSchemaError`: Input schema is invalid.
-*   `InvalidOutputSchemaError`: Output schema is invalid.
-*   `SchemaValidationError`: Input data does not match the schema.
-*   `MaxIterationsExceededError`: The agent could not produce valid JSON within the limit.
-*   `LLMExecutionError`: Failure in communicating with the LLM provider.
+| Property | Type | Description |
+| :--- | :--- | :--- |
+| `llmService` | `OpenAI \| GoogleGenAI \| Anthropic \| ILLMService` | The provider instance or custom service. |
+| `model` | `string` | Model ID (e.g., `gpt-4o`, `claude-3-5-sonnet`). |
+| `config` | `ModelConfig?` | Optional parameters (temperature, max_tokens, etc.). |
 
 ## Architecture
 
 The project is structured by domain:
 
-*   `src/agent`: Core orchestration logic.
-*   `src/schemas`: Validation logic using AJV.
-*   `src/llm`: Interface and implementation for LLM providers.
+*   `src/agent`: Core orchestration logic (`StructuredAgent`).
+*   `src/schemas`: Validation logic using **Zod**.
+*   `src/llm`: Adapters and Factory for LLM providers (`OpenAI`, `Google`, `Anthropic`).
 *   `src/errors`: Custom error definitions.
 *   `src/types`: Shared interfaces.
-

package/dist/agent/index.d.ts

@@ -1,11 +1,29 @@
 import { AgentConfig } from "../types/index.js";
+/**
+ * StructuredAgent is a class that implements a structured agent for generating and reviewing JSON outputs.
+ * It uses a language model to generate initial responses and, optionally, a reviewer model to validate and improve those responses.
+ *
+ * @param config - The configuration object for the agent.
+ */
 export declare class StructuredAgent {
     private schemaValidator;
-    private llmService;
     private inputValidator;
     private outputValidator;
     private config;
+    private generatorService;
+    private reviewerService?;
+    /**
+     * Creates an instance of StructuredAgent.
+     *
+     * @param config - The configuration object for the agent.
+     */
     constructor(config: AgentConfig);
+    /**
+     * Runs the agent with the given input JSON.
+     *
+     * @param inputJson - The input JSON to process.
+     * @returns A promise that resolves to the processed JSON.
+     */
     run(inputJson: unknown): Promise<unknown>;
     private generateInitialResponse;
     private reviewResponse;

package/dist/agent/index.js

@@ -1,17 +1,28 @@
+import { zodToJsonSchema } from "zod-to-json-schema";
 import { SchemaValidator } from "../schemas/validator.js";
-import { OpenAILLMService } from "../llm/index.js";
+import { LLMFactory } from "../llm/index.js";
 import { InvalidInputSchemaError, InvalidOutputSchemaError, SchemaValidationError, MaxIterationsExceededError, LLMExecutionError, } from "../errors/index.js";
+/**
+ * StructuredAgent is a class that implements a structured agent for generating and reviewing JSON outputs.
+ * It uses a language model to generate initial responses and, optionally, a reviewer model to validate and improve those responses.
+ *
+ * @param config - The configuration object for the agent.
+ */
 export class StructuredAgent {
     schemaValidator;
-    llmService;
     inputValidator;
     outputValidator;
     config;
+    generatorService;
+    reviewerService;
+    /**
+     * Creates an instance of StructuredAgent.
+     *
+     * @param config - The configuration object for the agent.
+     */
     constructor(config) {
         this.config = config;
         this.schemaValidator = new SchemaValidator();
-        // Use provided LLM service or default to OpenAI
-        this.llmService = config.llmService || new OpenAILLMService(config.openAiApiKey);
         try {
             this.inputValidator = this.schemaValidator.compile(config.inputSchema);
         }
@@ -24,29 +35,35 @@ export class StructuredAgent {
         catch (e) {
             throw new InvalidOutputSchemaError("Failed to compile output schema", e);
         }
+        this.generatorService = LLMFactory.create(config.generator.llmService);
+        if (config.reviewer) {
+            this.reviewerService = LLMFactory.create(config.reviewer.llmService);
+        }
     }
+    /**
+     * Runs the agent with the given input JSON.
+     *
+     * @param inputJson - The input JSON to process.
+     * @returns A promise that resolves to the processed JSON.
+     */
     async run(inputJson) {
-        // 1. Validate Input
         try {
             this.schemaValidator.validate(this.inputValidator, inputJson);
         }
         catch (error) {
             if (error instanceof SchemaValidationError) {
-                // Enhance error message if needed, but the original is good
                 throw error;
             }
             throw error;
         }
-        const maxIterations = this.config.maxIterations ?? 5;
+        const maxIterations = this.config.maxIterations ?? 1;
         const history = [];
-        // 2. Initial Generation
         let currentJson = await this.generateInitialResponse(inputJson);
         history.push({ step: "generation", result: currentJson });
         let validationResult = this.validateOutput(currentJson);
         if (validationResult.valid) {
             return currentJson;
         }
-        // 3. Review Loop
         for (let i = 0; i < maxIterations; i++) {
             try {
                 currentJson = await this.reviewResponse(currentJson, validationResult.errors || [], inputJson // Context might be needed
@@ -58,16 +75,9 @@ export class StructuredAgent {
                 }
             }
             catch (error) {
-                // If LLM fails or parsing fails during review, we record it and continue?
-                // Or throw? The requirement says "return exclusively a valid JSON".
-                // If we can't continue, we should probably throw or let the loop hit max iterations.
-                // For now, let's treat execution errors as fatal or part of the attempt?
-                // If it's a parsing error, it's a validation error.
-                // If it's an API error, it's an LLMExecutionError.
                 if (error instanceof LLMExecutionError) {
                     throw error;
                 }
-                // If JSON parse error in reviewResponse, it might be caught there.
             }
         }
         throw new MaxIterationsExceededError(`Failed to generate valid JSON after ${maxIterations} review iterations`, history);
@@ -83,10 +93,17 @@ export class StructuredAgent {
                 content: JSON.stringify(input),
             },
         ];
-        const responseText = await this.llmService.complete(messages, this.config.generatorModel, this.config.modelConfig);
+        const responseText = await this.generatorService.complete({
+            messages,
+            model: this.config.generator.model,
+            config: this.config.generator.config,
+            outputFormat: this.outputValidator
+        });
         return this.parseJson(responseText);
     }
     async reviewResponse(invalidJson, errors, originalInput) {
+        const reviewerService = this.reviewerService || this.generatorService;
+        const reviewerConfig = this.config.reviewer || this.config.generator;
         const messages = [
             {
                 role: "system",
@@ -105,12 +122,17 @@ Validation Errors:
 ${errors.join("\n")}
 
 Expected Output Schema:
-${JSON.stringify(this.config.outputSchema)}
+${JSON.stringify(zodToJsonSchema(this.config.outputSchema))}
 
 Please correct the JSON.`,
             },
         ];
-        const responseText = await this.llmService.complete(messages, this.config.reviewerModel, this.config.modelConfig);
+        const responseText = await reviewerService.complete({
+            messages,
+            model: reviewerConfig.model,
+            config: reviewerConfig.config,
+            outputFormat: this.outputValidator
+        });
         return this.parseJson(responseText);
     }
     validateOutput(data) {
@@ -131,19 +153,16 @@ Please correct the JSON.`,
             return JSON.parse(text);
         }
         catch (e) {
-            // If parsing fails, we return the text (as string) or null?
-            // But the flow expects unknown (object).
-            // If we return the text, the schema validation will likely fail (unless schema allows string).
-            // This allows the reviewer to see the malformed JSON text.
             return text;
         }
     }
     buildSystemPrompt() {
+        const jsonSchema = zodToJsonSchema(this.config.outputSchema);
         return `${this.config.systemPrompt}
 
 IMPORTANT: You must output strict JSON only.
 The output must adhere to the following JSON Schema:
-${JSON.stringify(this.config.outputSchema)}
+${JSON.stringify(jsonSchema)}
 `;
     }
 }

package/dist/agent/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAA4B,MAAM,iBAAiB,CAAC;AAI7E,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,EACrB,0BAA0B,EAC1B,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAE5B,MAAM,OAAO,eAAe;IAClB,eAAe,CAAkB;IACjC,UAAU,CAAc;IACxB,cAAc,CAAmB;IACjC,eAAe,CAAmB;IAClC,MAAM,CAAc;IAE5B,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE7C,gDAAgD;QAChD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,gBAAgB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAEjF,IAAI,CAAC;YACH,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,uBAAuB,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,wBAAwB,CAAC,iCAAiC,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,GAAG,CAAC,SAAkB;QACjC,oBAAoB;QACpB,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,qBAAqB,EAAE,CAAC;gBAC3C,4DAA4D;gBAC5D,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;QACrD,MAAM,OAAO,GAAc,EAAE,CAAC;QAE9B,wBAAwB;QACxB,IAAI,WAAW,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,SAAS,CAAC,CAAC;QAChE,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QAE1D,IAAI,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACxD,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;YAC3B,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,iBAAiB;QACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,IAAI,CAAC;gBACH,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CACrC,WAAW,EACX,gBAAgB,CAAC,MAAM,IAAI,EAAE,EAC7B,SAAS,CAAC,0BAA0B;iBACrC,CAAC;gBACF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBAE/D,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;gBACpD,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;oBAC3B,OAAO,WAAW,CAAC;gBACrB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,0EAA0E;gBAC1E,oEAAoE;gBACpE,qFAAqF;gBACrF,yEAAyE;gBACzE,oDAAoD;gBACpD,mDAAmD;gBACnD,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;oBACvC,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,mEAAmE;YACrE,CAAC;QACH,CAAC;QAED,MAAM,IAAI,0BAA0B,CAClC,uCAAuC,aAAa,oBAAoB,EACxE,OAAO,CACR,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,KAAc;QAClD,MAAM,QAAQ,GAAkB;YAC9B;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE;aAClC;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;aAC/B;SACF,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CACjD,QAAQ,EACR,IAAI,CAAC,MAAM,CAAC,cAAc,EAC1B,IAAI,CAAC,MAAM,CAAC,WAAW,CACxB,CAAC;QAEF,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,WAAoB,EACpB,MAAgB,EAChB,aAAsB;QAEtB,MAAM,QAAQ,GAAkB;YAC9B;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE;;gCAEe;aACzB;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,2BAA2B,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;;;EAGvE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;;;EAG3B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;;;EAGjB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;;yBAEjB;aAClB;SACF,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CACjD,QAAQ,EACR,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,IAAI,CAAC,MAAM,CAAC,WAAW,CACxB,CAAC;QAEF,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC;IAEO,cAAc,CAAC,IAAa;QAClC,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YAC1D,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,qBAAqB,EAAE,CAAC;gBAC3C,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,KAAK,CAAC,MAAe,CAAC,CAAC;gBACjF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YACrD,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,0BAA0B,CAAC,EAAE,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,IAAY;QAC5B,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,4DAA4D;YAC5D,yCAAyC;YACzC,+FAA+F;YAC/F,2DAA2D;YAC3D,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY;;;;EAIpC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;CACzC,CAAC;IACA,CAAC;CACF"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAe,UAAU,EAAe,MAAM,iBAAiB,CAAC;AAIvE,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,EACrB,0BAA0B,EAC1B,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAE5B;;;;;GAKG;AACH,MAAM,OAAO,eAAe;IAClB,eAAe,CAAkB;IACjC,cAAc,CAAY;IAC1B,eAAe,CAAY;IAC3B,MAAM,CAAc;IACpB,gBAAgB,CAAc;IAC9B,eAAe,CAAe;IAEtC;;;;OAIG;IACH,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE7C,IAAI,CAAC;YACH,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACzE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,uBAAuB,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,wBAAwB,CAAC,iCAAiC,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACvE,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,GAAG,CAAC,SAAkB;QAEjC,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,qBAAqB,EAAE,CAAC;gBAC3C,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;QACrD,MAAM,OAAO,GAAc,EAAE,CAAC;QAE9B,IAAI,WAAW,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,SAAS,CAAC,CAAC;QAChE,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QAE1D,IAAI,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACxD,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;YAC3B,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,IAAI,CAAC;gBACH,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CACrC,WAAW,EACX,gBAAgB,CAAC,MAAM,IAAI,EAAE,EAC7B,SAAS,CAAC,0BAA0B;iBACrC,CAAC;gBACF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBAE/D,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;gBACpD,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;oBAC3B,OAAO,WAAW,CAAC;gBACrB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;oBACvC,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,0BAA0B,CAClC,uCAAuC,aAAa,oBAAoB,EACxE,OAAO,CACR,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,KAAc;QAClD,MAAM,QAAQ,GAAkB;YAC9B;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE;aAClC;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;aAC/B;SACF,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC;YACxD,QAAQ;YACR,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK;YAClC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM;YACpC,YAAY,EAAE,IAAI,CAAC,eAAe;SACnC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,WAAoB,EACpB,MAAgB,EAChB,aAAsB;QAEtB,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,gBAAgB,CAAC;QACtE,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QAErE,MAAM,QAAQ,GAAkB;YAC9B;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE;;gCAEe;aACzB;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,2BAA2B,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;;;EAGvE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;;;EAG3B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;;;EAGjB,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;;yBAElC;aAClB;SACF,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC;YAClD,QAAQ;YACR,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,MAAM,EAAE,cAAc,CAAC,MAAM;YAC7B,YAAY,EAAE,IAAI,CAAC,eAAe;SACnC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC;IAEO,cAAc,CAAC,IAAa;QAClC,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YAC1D,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,qBAAqB,EAAE,CAAC;gBAC3C,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,KAAK,CAAC,MAAe,CAAC,CAAC;gBACjF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YACrD,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,0BAA0B,CAAC,EAAE,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,IAAY;QAC5B,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC7D,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY;;;;EAIpC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;CAC3B,CAAC;IACA,CAAC;CACF"}

package/dist/llm/adapters/anthropic.d.ts

@@ -0,0 +1,14 @@
+import Anthropic from "@anthropic-ai/sdk";
+import { ZodSchema } from "zod";
+import { ILLMService, ChatMessage } from "../types.js";
+import { ModelConfig } from "../../types/index.js";
+export declare class AnthropicAdapter implements ILLMService {
+    private client;
+    constructor(client: Anthropic);
+    complete(params: {
+        messages: ChatMessage[];
+        model: string;
+        config?: ModelConfig;
+        outputFormat?: ZodSchema;
+    }): Promise<string>;
+}

package/dist/llm/adapters/anthropic.js

@@ -0,0 +1,46 @@
+import { zodToJsonSchema } from "zod-to-json-schema";
+import { LLMExecutionError } from "../../errors/index.js";
+export class AnthropicAdapter {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+    async complete(params) {
+        try {
+            const systemMessage = params.messages.find((m) => m.role === "system");
+            const conversationMessages = params.messages
+                .filter((m) => m.role !== "system")
+                .map((m) => ({
+                role: m.role,
+                content: m.content,
+            }));
+            const requestOptions = {
+                model: params.model,
+                messages: conversationMessages,
+                system: systemMessage?.content,
+                max_tokens: params.config?.max_tokens ?? 1024,
+                temperature: params.config?.temperature ?? 0.7,
+                top_p: params.config?.top_p,
+            };
+            const headers = {};
+            if (params.outputFormat) {
+                requestOptions.output_format = {
+                    type: "json_schema",
+                    schema: zodToJsonSchema(params.outputFormat)
+                };
+                // Enable beta feature
+                headers["anthropic-beta"] = "structured-outputs-2025-11-13";
+            }
+            const response = await this.client.messages.create(requestOptions, { headers });
+            const content = response.content[0];
+            if (content.type !== "text") {
+                throw new LLMExecutionError("Received non-text response from Anthropic");
+            }
+            return content.text;
+        }
+        catch (error) {
+            throw new LLMExecutionError("Failed to execute Anthropic request", error);
+        }
+    }
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/llm/adapters/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/llm/adapters/anthropic.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1D,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAY;IAE1B,YAAY,MAAiB;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,QAAQ,CAAC,MAKrB;QACC,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACvE,MAAM,oBAAoB,GAAG,MAAM,CAAC,QAAQ;iBACzC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACX,IAAI,EAAE,CAAC,CAAC,IAA4B;gBACpC,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC,CAAC;YAEN,MAAM,cAAc,GAAQ;gBAC1B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,QAAQ,EAAE,oBAAoB;gBAC9B,MAAM,EAAE,aAAa,EAAE,OAAO;gBAC9B,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,IAAI,IAAI;gBAC7C,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,IAAI,GAAG;gBAC9C,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK;aAC5B,CAAC;YAEF,MAAM,OAAO,GAA2B,EAAE,CAAC;YAE3C,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBACxB,cAAc,CAAC,aAAa,GAAG;oBAC7B,IAAI,EAAE,aAAa;oBACnB,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC;iBAC7C,CAAC;gBACF,sBAAsB;gBACtB,OAAO,CAAC,gBAAgB,CAAC,GAAG,+BAA+B,CAAC;YAC9D,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAChD,cAAc,EACd,EAAE,OAAO,EAAE,CACZ,CAAC;YAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC3B,MAAM,IAAI,iBAAiB,CAAC,2CAA2C,CAAC,CAAC;YAC5E,CAAC;YAED,OAAO,OAAO,CAAC,IAAI,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;CACF"}

package/dist/llm/adapters/google.d.ts

@@ -0,0 +1,14 @@
+import { GoogleGenAI } from "@google/genai";
+import { ZodSchema } from "zod";
+import { ILLMService, ChatMessage } from "../types.js";
+import { ModelConfig } from "../../types/index.js";
+export declare class GoogleGenAIAdapter implements ILLMService {
+    private client;
+    constructor(client: GoogleGenAI);
+    complete(params: {
+        messages: ChatMessage[];
+        model: string;
+        config?: ModelConfig;
+        outputFormat?: ZodSchema;
+    }): Promise<string>;
+}

package/dist/llm/adapters/google.js

@@ -0,0 +1,52 @@
+import { LLMExecutionError } from "../../errors/index.js";
+import { zodToJsonSchema } from "zod-to-json-schema";
+export class GoogleGenAIAdapter {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+    async complete(params) {
+        try {
+            // Extract system instruction if present
+            const systemMessage = params.messages.find((m) => m.role === "system");
+            // Filter out system message for the history
+            const history = params.messages
+                .filter((m) => m.role !== "system")
+                .map((m) => ({
+                role: m.role === "assistant" ? "model" : "user",
+                parts: [{ text: m.content }],
+            }));
+            // New SDK uses client.models.generateContent
+            // System instruction is part of the config
+            const generationConfig = {
+                temperature: params.config?.temperature,
+                topP: params.config?.top_p,
+                maxOutputTokens: params.config?.max_tokens,
+                responseMimeType: "application/json",
+            };
+            const result = await this.client.models.generateContent({
+                model: params.model,
+                contents: history,
+                config: {
+                    ...generationConfig,
+                    responseJsonSchema: params.outputFormat ? zodToJsonSchema(params.outputFormat) : undefined,
+                    systemInstruction: systemMessage ? { parts: [{ text: systemMessage.content }] } : undefined
+                }
+            });
+            const text = result.text || result.candidates?.[0]?.content?.parts?.[0]?.text;
+            if (!text) {
+                if (result.candidates && result.candidates.length > 0) {
+                    const part = result.candidates[0].content?.parts?.[0];
+                    if (part && 'text' in part)
+                        return part.text;
+                }
+                throw new LLMExecutionError("Received empty response from Google GenAI");
+            }
+            return text;
+        }
+        catch (error) {
+            throw new LLMExecutionError("Failed to execute Google GenAI request", error);
+        }
+    }
+}
+//# sourceMappingURL=google.js.map

package/dist/llm/adapters/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../../src/llm/adapters/google.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,MAAM,OAAO,kBAAkB;IACrB,MAAM,CAAc;IAE5B,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,QAAQ,CAAC,MAKrB;QACC,IAAI,CAAC;YACH,wCAAwC;YACxC,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YAEvE,4CAA4C;YAC5C,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ;iBAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACX,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;gBAC/C,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;aAC7B,CAAC,CAAC,CAAC;YAEN,6CAA6C;YAC7C,2CAA2C;YAE3C,MAAM,gBAAgB,GAAG;gBACvB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW;gBACvC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK;gBAC1B,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU;gBAC1C,gBAAgB,EAAE,kBAAkB;aACrC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;gBACtD,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE;oBACN,GAAG,gBAAgB;oBACnB,kBAAkB,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;oBAC1F,iBAAiB,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;iBAC5F;aACF,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAE9E,IAAI,CAAC,IAAI,EAAE,CAAC;gBACT,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrD,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;oBACtD,IAAI,IAAI,IAAI,MAAM,IAAI,IAAI;wBAAE,OAAO,IAAI,CAAC,IAAc,CAAC;gBAC1D,CAAC;gBACD,MAAM,IAAI,iBAAiB,CAAC,2CAA2C,CAAC,CAAC;YAC5E,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;CACF"}

package/dist/llm/adapters/openai.d.ts

@@ -0,0 +1,14 @@
+import OpenAI from "openai";
+import { ZodSchema } from "zod";
+import { ILLMService, ChatMessage } from "../types.js";
+import { ModelConfig } from "../../types/index.js";
+export declare class OpenAIAdapter implements ILLMService {
+    private client;
+    constructor(client: OpenAI);
+    complete(params: {
+        messages: ChatMessage[];
+        model: string;
+        config?: ModelConfig;
+        outputFormat?: ZodSchema;
+    }): Promise<string>;
+}

package/dist/llm/adapters/openai.js

@@ -0,0 +1,42 @@
+import { zodResponseFormat } from "openai/helpers/zod";
+import { LLMExecutionError } from "../../errors/index.js";
+export class OpenAIAdapter {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+    async complete(params) {
+        try {
+            const createParams = {
+                messages: params.messages,
+                model: params.model,
+                response_format: { type: "json_object" },
+                temperature: params.config?.temperature ?? 1,
+                top_p: params.config?.top_p,
+                max_tokens: params.config?.max_tokens,
+                presence_penalty: params.config?.presence_penalty,
+                frequency_penalty: params.config?.frequency_penalty,
+            };
+            if (params.outputFormat) {
+                createParams.response_format = zodResponseFormat(params.outputFormat, "response");
+            }
+            const response = await this.client.chat.completions.create(createParams);
+            const message = response.choices[0]?.message;
+            const content = message?.content;
+            if (message?.refusal) {
+                throw new LLMExecutionError(`Model refused to generate response: ${message.refusal}`);
+            }
+            if (!content) {
+                throw new LLMExecutionError("Received empty response from LLM");
+            }
+            return content;
+        }
+        catch (error) {
+            if (error instanceof LLMExecutionError) {
+                throw error;
+            }
+            throw new LLMExecutionError("Failed to execute OpenAI request", error);
+        }
+    }
+}
+//# sourceMappingURL=openai.js.map

package/dist/llm/adapters/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/llm/adapters/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAIvD,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAG1D,MAAM,OAAO,aAAa;IAChB,MAAM,CAAS;IAEvB,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,QAAQ,CAAC,MAKrB;QACC,IAAI,CAAC;YACH,MAAM,YAAY,GAA+B;gBAC/C,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,eAAe,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE;gBACxC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,IAAI,CAAC;gBAC5C,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK;gBAC3B,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU;gBACrC,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,gBAAgB;gBACjD,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB;aACpD,CAAA;YAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBACxB,YAAY,CAAC,eAAe,GAAG,iBAAiB,CAAC,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAEzE,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;YAC7C,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC;YAEjC,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;gBACrB,MAAM,IAAI,iBAAiB,CAAC,uCAAuC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACxF,CAAC;YAED,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,iBAAiB,CAAC,kCAAkC,CAAC,CAAC;YAClE,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;gBACvC,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,iBAAiB,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;CACF"}

package/dist/llm/factory.d.ts

@@ -0,0 +1,12 @@
+import OpenAI from "openai";
+import { GoogleGenAI } from "@google/genai";
+import Anthropic from "@anthropic-ai/sdk";
+import { ILLMService } from "./types.js";
+export type LLMInstance = OpenAI | GoogleGenAI | Anthropic | ILLMService;
+export declare class LLMFactory {
+    static create(instance: LLMInstance): ILLMService;
+    private static isILLMService;
+    private static isOpenAI;
+    private static isAnthropic;
+    private static isGoogleGenAI;
+}

package/dist/llm/factory.js

@@ -0,0 +1,49 @@
+import OpenAI from "openai";
+import { GoogleGenAI } from "@google/genai";
+import Anthropic from "@anthropic-ai/sdk";
+import { OpenAIAdapter } from "./adapters/openai.js";
+import { GoogleGenAIAdapter } from "./adapters/google.js";
+import { AnthropicAdapter } from "./adapters/anthropic.js";
+export class LLMFactory {
+    static create(instance) {
+        if (this.isILLMService(instance)) {
+            return instance;
+        }
+        if (instance instanceof OpenAI) {
+            return new OpenAIAdapter(instance);
+        }
+        if (instance instanceof Anthropic) {
+            return new AnthropicAdapter(instance);
+        }
+        if (instance instanceof GoogleGenAI) {
+            return new GoogleGenAIAdapter(instance);
+        }
+        // Fallback check based on properties if instanceof fails (e.g. different versions or mocked)
+        if (this.isOpenAI(instance)) {
+            return new OpenAIAdapter(instance);
+        }
+        if (this.isAnthropic(instance)) {
+            return new AnthropicAdapter(instance);
+        }
+        if (this.isGoogleGenAI(instance)) {
+            return new GoogleGenAIAdapter(instance);
+        }
+        throw new Error("Unknown LLM instance type provided.");
+    }
+    static isILLMService(instance) {
+        return typeof instance.complete === "function";
+    }
+    static isOpenAI(instance) {
+        return instance?.chat?.completions?.create !== undefined;
+    }
+    static isAnthropic(instance) {
+        return instance?.messages?.create !== undefined;
+    }
+    static isGoogleGenAI(instance) {
+        // New SDK check
+        return instance?.models?.generateContent !== undefined ||
+            // Old SDK check just in case
+            instance?.getGenerativeModel !== undefined;
+    }
+}
+//# sourceMappingURL=factory.js.map

package/dist/llm/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../src/llm/factory.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAE1C,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAI3D,MAAM,OAAO,UAAU;IACd,MAAM,CAAC,MAAM,CAAC,QAAqB;QACxC,IAAI,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,QAAQ,YAAY,MAAM,EAAE,CAAC;YAC/B,OAAO,IAAI,aAAa,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,QAAQ,YAAY,SAAS,EAAE,CAAC;YAClC,OAAO,IAAI,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,QAAQ,YAAY,WAAW,EAAE,CAAC;YACpC,OAAO,IAAI,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAED,6FAA6F;QAC7F,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,OAAO,IAAI,aAAa,CAAC,QAAkB,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,gBAAgB,CAAC,QAAqB,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,OAAO,IAAI,kBAAkB,CAAC,QAAuB,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,QAAa;QACxC,OAAO,OAAO,QAAQ,CAAC,QAAQ,KAAK,UAAU,CAAC;IACjD,CAAC;IAEO,MAAM,CAAC,QAAQ,CAAC,QAAa;QACnC,OAAO,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,KAAK,SAAS,CAAC;IAC3D,CAAC;IAEO,MAAM,CAAC,WAAW,CAAC,QAAa;QACtC,OAAO,QAAQ,EAAE,QAAQ,EAAE,MAAM,KAAK,SAAS,CAAC;IAClD,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,QAAa;QACxC,gBAAgB;QAChB,OAAO,QAAQ,EAAE,MAAM,EAAE,eAAe,KAAK,SAAS;YAC/C,6BAA6B;YAC7B,QAAQ,EAAE,kBAAkB,KAAK,SAAS,CAAC;IACpD,CAAC;CACF"}

package/dist/llm/index.d.ts

@@ -1,2 +1,5 @@
 export * from "./types.js";
-export * from "./openai.js";
+export * from "./factory.js";
+export * from "./adapters/openai.js";
+export * from "./adapters/google.js";
+export * from "./adapters/anthropic.js";

package/dist/llm/index.js

@@ -1,3 +1,6 @@
 export * from "./types.js";
-export * from "./openai.js";
+export * from "./factory.js";
+export * from "./adapters/openai.js";
+export * from "./adapters/google.js";
+export * from "./adapters/anthropic.js";
 //# sourceMappingURL=index.js.map

package/dist/llm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/llm/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,aAAa,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/llm/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,cAAc,CAAC;AAC7B,cAAc,sBAAsB,CAAC;AACrC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC"}

package/dist/llm/types.d.ts

@@ -1,8 +1,14 @@
+import { ZodSchema } from "zod";
 import { ModelConfig } from "../types/index.js";
 export interface ChatMessage {
     role: "system" | "user" | "assistant";
     content: string;
 }
 export interface ILLMService {
-    complete(messages: ChatMessage[], model: string, config?: ModelConfig): Promise<string>;
+    complete(params: {
+        messages: ChatMessage[];
+        model: string;
+        config?: ModelConfig;
+        outputFormat?: ZodSchema;
+    }): Promise<string>;
 }

package/dist/schemas/validator.d.ts

@@ -1,20 +1,17 @@
-import type { ValidateFunction } from "ajv";
+import { ZodSchema } from "zod";
 export declare class SchemaValidator {
-    private ajv;
-    constructor();
     /**
      * Compiles a schema and returns a validation function.
-     * Throws InvalidInputSchemaError if the schema is invalid.
+     * For Zod, the schema itself is the validator.
      */
-    compile(schema: object): ValidateFunction;
+    compile(schema: ZodSchema): ZodSchema;
     /**
      * Validates data against a compiled schema validator.
-     * Returns true if valid.
      * Throws SchemaValidationError if invalid.
      */
-    validate(validator: ValidateFunction, data: unknown): void;
+    validate(validator: ZodSchema, data: unknown): void;
     /**
-     * Helper to format AJV errors into a readable string
+     * Helper to format Zod errors into a readable string
      */
-    formatErrors(errors: any[]): string;
+    formatErrors(errors: unknown[]): string;
 }

package/dist/schemas/validator.js

@@ -1,39 +1,32 @@
-import Ajv from "ajv";
-import { SchemaValidationError, InvalidInputSchemaError } from "../errors/index.js";
+import { SchemaValidationError } from "../errors/index.js";
 export class SchemaValidator {
-    ajv;
-    constructor() {
-        // @ts-ignore: Ajv import structure compatibility
-        this.ajv = new Ajv({ allErrors: true, strict: false });
-    }
     /**
      * Compiles a schema and returns a validation function.
-     * Throws InvalidInputSchemaError if the schema is invalid.
+     * For Zod, the schema itself is the validator.
      */
     compile(schema) {
-        try {
-            return this.ajv.compile(schema);
-        }
-        catch (error) {
-            throw new InvalidInputSchemaError("Invalid JSON Schema provided", error);
-        }
+        return schema;
     }
     /**
      * Validates data against a compiled schema validator.
-     * Returns true if valid.
      * Throws SchemaValidationError if invalid.
      */
     validate(validator, data) {
-        const valid = validator(data);
-        if (!valid) {
-            throw new SchemaValidationError("Data validation failed", validator.errors || []);
+        const result = validator.safeParse(data);
+        if (!result.success) {
+            throw new SchemaValidationError("Data validation failed", result.error.errors);
         }
     }
     /**
-     * Helper to format AJV errors into a readable string
+     * Helper to format Zod errors into a readable string
      */
     formatErrors(errors) {
-        return this.ajv.errorsText(errors);
+        return errors
+            .map((e) => {
+            const path = e.path.join(".");
+            return path ? `${path}: ${e.message}` : e.message;
+        })
+            .join("; ");
     }
 }
 //# sourceMappingURL=validator.js.map

package/dist/schemas/validator.js.map

@@ -1 +1 @@
-{"version":3,"file":"validator.js","sourceRoot":"","sources":["../../src/schemas/validator.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,KAAK,CAAC;AAEtB,OAAO,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAEpF,MAAM,OAAO,eAAe;IAClB,GAAG,CAAM;IAEjB;QACE,iDAAiD;QACjD,IAAI,CAAC,GAAG,GAAG,IAAI,GAAG,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACzD,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,MAAc;QAC3B,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,uBAAuB,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED;;;;OAIG;IACI,QAAQ,CAAC,SAA2B,EAAE,IAAa;QACxD,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,qBAAqB,CAC7B,wBAAwB,EACxB,SAAS,CAAC,MAAM,IAAI,EAAE,CACvB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACI,YAAY,CAAC,MAAa;QAC/B,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;CACF"}
+{"version":3,"file":"validator.js","sourceRoot":"","sources":["../../src/schemas/validator.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,MAAM,OAAO,eAAe;IAC1B;;;OAGG;IACI,OAAO,CAAC,MAAiB;QAC9B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACI,QAAQ,CAAC,SAAoB,EAAE,IAAa;QACjD,MAAM,MAAM,GAAG,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,IAAI,qBAAqB,CAC7B,wBAAwB,EACxB,MAAM,CAAC,KAAK,CAAC,MAAM,CACpB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACI,YAAY,CAAC,MAAiB;QACnC,OAAQ,MAAgB;aACrB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACpD,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;CACF"}

package/dist/types/index.d.ts

@@ -1,4 +1,8 @@
+import OpenAI from "openai";
 import { ILLMService } from "../llm/types.js";
+import { GoogleGenAI } from "@google/genai";
+import { ZodSchema } from "zod";
+import Anthropic from "@anthropic-ai/sdk";
 export interface ModelConfig {
     temperature?: number;
     top_p?: number;
@@ -6,16 +10,18 @@ export interface ModelConfig {
     presence_penalty?: number;
     frequency_penalty?: number;
 }
+export interface LLMInputInstance {
+    llmService: OpenAI | GoogleGenAI | Anthropic | ILLMService;
+    model: string;
+    config?: ModelConfig;
+}
 export interface AgentConfig {
-    openAiApiKey: string;
-    generatorModel: string;
-    reviewerModel: string;
-    inputSchema: object;
-    outputSchema: object;
+    generator: LLMInputInstance;
+    reviewer?: LLMInputInstance;
+    inputSchema: ZodSchema;
+    outputSchema: ZodSchema;
     systemPrompt: string;
-    modelConfig?: ModelConfig;
     maxIterations?: number;
-    llmService?: ILLMService;
 }
 export interface AgentRunOptions {
     input: unknown;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "structured-json-agent",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "description": "A typed and extensible library for creating and running Iterative AI Agents that generate structured JSON output.",
   "type": "module",
   "main": "./dist/index.js",
@@ -29,19 +29,23 @@
   "bugs": {
     "url": "https://github.com/thiagoaramizo/structured-json-agent/issues"
   },
-  "homepage": "https://github.com/thiagoaramizo/structured-json-agent#readme",
+  "homepage": "https://thiagoaramizo.github.io/structured-json-agent",
   "author": "",
   "license": "ISC",
   "devDependencies": {
     "@types/jest": "^30.0.0",
     "@types/node": "^20.0.0",
+    "dotenv": "^17.2.3",
     "jest": "^30.2.0",
     "ts-jest": "^29.4.6",
     "typescript": "^5.0.0"
   },
   "dependencies": {
-    "ajv": "^8.12.0",
-    "openai": "^4.0.0"
+    "@anthropic-ai/sdk": "^0.71.2",
+    "@google/genai": "^1.38.0",
+    "openai": "^4.0.0",
+    "zod": "^3.25.76",
+    "zod-to-json-schema": "^3.25.1"
   },
   "engines": {
     "node": ">=18.0.0"

package/dist/llm/openai.d.ts

@@ -1,7 +0,0 @@
-import { ILLMService, ChatMessage } from "./types.js";
-import { ModelConfig } from "../types/index.js";
-export declare class OpenAILLMService implements ILLMService {
-    private client;
-    constructor(apiKey: string);
-    complete(messages: ChatMessage[], model: string, config?: ModelConfig): Promise<string>;
-}

package/dist/llm/openai.js

@@ -1,33 +0,0 @@
-import OpenAI from "openai";
-import { LLMExecutionError } from "../errors/index.js";
-export class OpenAILLMService {
-    client;
-    constructor(apiKey) {
-        this.client = new OpenAI({
-            apiKey: apiKey,
-        });
-    }
-    async complete(messages, model, config) {
-        try {
-            const response = await this.client.chat.completions.create({
-                messages: messages,
-                model: model,
-                response_format: { type: "json_object" }, // Force JSON mode
-                temperature: config?.temperature ?? 0.7,
-                top_p: config?.top_p,
-                max_tokens: config?.max_tokens,
-                presence_penalty: config?.presence_penalty,
-                frequency_penalty: config?.frequency_penalty,
-            });
-            const content = response.choices[0]?.message?.content;
-            if (!content) {
-                throw new LLMExecutionError("Received empty response from LLM");
-            }
-            return content;
-        }
-        catch (error) {
-            throw new LLMExecutionError("Failed to execute LLM request", error);
-        }
-    }
-}
-//# sourceMappingURL=openai.js.map

package/dist/llm/openai.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/llm/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAG5B,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAS;IAEvB,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,QAAQ,CACnB,QAAuB,EACvB,KAAa,EACb,MAAoB;QAEpB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACzD,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE,KAAK;gBACZ,eAAe,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,kBAAkB;gBAC5D,WAAW,EAAE,MAAM,EAAE,WAAW,IAAI,GAAG;gBACvC,KAAK,EAAE,MAAM,EAAE,KAAK;gBACpB,UAAU,EAAE,MAAM,EAAE,UAAU;gBAC9B,gBAAgB,EAAE,MAAM,EAAE,gBAAgB;gBAC1C,iBAAiB,EAAE,MAAM,EAAE,iBAAiB;aAC7C,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC;YACtD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,iBAAiB,CAAC,kCAAkC,CAAC,CAAC;YAClE,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,iBAAiB,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;CACF"}