@effect/ai 0.26.0 → 0.27.0

package/dist/esm/LanguageModel.js

@@ -0,0 +1,572 @@
+/**
+ * The `LanguageModel` module provides AI text generation capabilities with tool
+ * calling support.
+ *
+ * This module offers a comprehensive interface for interacting with large
+ * language models, supporting both streaming and non-streaming text generation,
+ * structured output generation, and tool calling functionality. It provides a
+ * unified API that can be implemented by different AI providers while
+ * maintaining type safety and effect management.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect } from "effect"
+ *
+ * // Basic text generation
+ * const program = Effect.gen(function* () {
+ *   const response = yield* LanguageModel.generateText({
+ *     prompt: "Explain quantum computing"
+ *   })
+ *
+ *   console.log(response.text)
+ *
+ *   return response
+ * })
+ * ```
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect, Schema } from "effect"
+ *
+ * // Structured output generation
+ * const ContactSchema = Schema.Struct({
+ *   name: Schema.String,
+ *   email: Schema.String
+ * })
+ *
+ * const extractContact = Effect.gen(function* () {
+ *   const response = yield* LanguageModel.generateObject({
+ *     prompt: "Extract contact: John Doe, john@example.com",
+ *     schema: ContactSchema
+ *   })
+ *
+ *   return response.value
+ * })
+ * ```
+ *
+ * @since 1.0.0
+ */
+import * as Chunk from "effect/Chunk";
+import * as Context from "effect/Context";
+import * as Effect from "effect/Effect";
+import * as Option from "effect/Option";
+import * as ParseResult from "effect/ParseResult";
+import * as Predicate from "effect/Predicate";
+import * as Schema from "effect/Schema";
+import * as Stream from "effect/Stream";
+import * as AiError from "./AiError.js";
+import { defaultIdGenerator, IdGenerator } from "./IdGenerator.js";
+import * as Prompt from "./Prompt.js";
+import * as Response from "./Response.js";
+import { CurrentSpanTransformer } from "./Telemetry.js";
+import * as Toolkit from "./Toolkit.js";
+// =============================================================================
+// Service Definition
+// =============================================================================
+/**
+ * The `LanguageModel` service tag for dependency injection.
+ *
+ * This tag provides access to language model functionality throughout your
+ * application, enabling text generation, streaming, and structured output
+ * capabilities.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect } from "effect"
+ *
+ * const useLanguageModel = Effect.gen(function* () {
+ *   const model = yield* LanguageModel
+ *   const response = yield* model.generateText({
+ *     prompt: "What is machine learning?"
+ *   })
+ *   return response.text
+ * })
+ * ```
+ *
+ * @since 1.0.0
+ * @category Context
+ */
+export class LanguageModel extends /*#__PURE__*/Context.Tag("@effect/ai/LanguageModel")() {}
+/**
+ * Response class for text generation operations.
+ *
+ * Contains the generated content and provides convenient accessors for
+ * extracting different types of response parts like text, tool calls, and usage
+ * information.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect } from "effect"
+ *
+ * const program = Effect.gen(function* () {
+ *   const response = yield* LanguageModel.generateText({
+ *     prompt: "Explain photosynthesis"
+ *   })
+ *
+ *   console.log(response.text) // Generated text content
+ *   console.log(response.finishReason) // "stop", "length", etc.
+ *   console.log(response.usage) // Usage information
+ *
+ *   return response
+ * })
+ * ```
+ *
+ * @since 1.0.0
+ * @category Models
+ */
+export class GenerateTextResponse {
+  content;
+  constructor(content) {
+    this.content = content;
+  }
+  /**
+   * Extracts and concatenates all text parts from the response.
+   */
+  get text() {
+    const text = [];
+    for (const part of this.content) {
+      if (part.type === "text") {
+        text.push(part.text);
+      }
+    }
+    return text.join("");
+  }
+  /**
+   * Returns all reasoning parts from the response.
+   */
+  get reasoning() {
+    return this.content.filter(part => part.type === "reasoning");
+  }
+  /**
+   * Extracts and concatenates all reasoning text, or undefined if none exists.
+   */
+  get reasoningText() {
+    const text = [];
+    for (const part of this.content) {
+      if (part.type === "reasoning") {
+        text.push(part.text);
+      }
+    }
+    return text.length === 0 ? undefined : text.join("");
+  }
+  /**
+   * Returns all tool call parts from the response.
+   */
+  get toolCalls() {
+    return this.content.filter(part => part.type === "tool-call");
+  }
+  /**
+   * Returns all tool result parts from the response.
+   */
+  get toolResults() {
+    return this.content.filter(part => part.type === "tool-result");
+  }
+  /**
+   * The reason why text generation finished.
+   */
+  get finishReason() {
+    const finishPart = this.content.find(part => part.type === "finish");
+    return Predicate.isUndefined(finishPart) ? "unknown" : finishPart.reason;
+  }
+  /**
+   * Token usage statistics for the generation request.
+   */
+  get usage() {
+    const finishPart = this.content.find(part => part.type === "finish");
+    if (Predicate.isUndefined(finishPart)) {
+      return new Response.Usage({
+        inputTokens: undefined,
+        outputTokens: undefined,
+        totalTokens: undefined,
+        reasoningTokens: undefined,
+        cachedInputTokens: undefined
+      });
+    }
+    return finishPart.usage;
+  }
+}
+/**
+ * Response class for structured object generation operations.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect, Schema } from "effect"
+ *
+ * const UserSchema = Schema.Struct({
+ *   name: Schema.String,
+ *   email: Schema.String
+ * })
+ *
+ * const program = Effect.gen(function* () {
+ *   const response = yield* LanguageModel.generateObject({
+ *     prompt: "Create user: John Doe, john@example.com",
+ *     schema: UserSchema
+ *   })
+ *
+ *   console.log(response.value) // { name: "John Doe", email: "john@example.com" }
+ *   console.log(response.text) // Raw generated text
+ *
+ *   return response.value
+ * })
+ * ```
+ *
+ * @since 1.0.0
+ * @category Models
+ */
+export class GenerateObjectResponse extends GenerateTextResponse {
+  /**
+   * The parsed structured object that conforms to the provided schema.
+   */
+  value;
+  constructor(value, content) {
+    super(content);
+    this.value = value;
+  }
+}
+/**
+ * Creates a LanguageModel service from provider-specific implementations.
+ *
+ * This constructor takes provider-specific implementations for text generation
+ * and streaming text generation and returns a LanguageModel service.
+ *
+ * @since 1.0.0
+ * @category Constructors
+ */
+export const make = /*#__PURE__*/Effect.fnUntraced(function* (params) {
+  const parentSpanTransformer = yield* Effect.serviceOption(CurrentSpanTransformer);
+  const getSpanTransformer = Effect.serviceOption(CurrentSpanTransformer).pipe(Effect.map(Option.orElse(() => parentSpanTransformer)));
+  const idGenerator = yield* Effect.serviceOption(IdGenerator).pipe(Effect.map(Option.getOrElse(() => defaultIdGenerator)));
+  const generateText = options => Effect.useSpan("LanguageModel.generateText", {
+    captureStackTrace: false,
+    attributes: {
+      concurrency: options.concurrency,
+      toolChoice: options.toolChoice
+    }
+  }, Effect.fnUntraced(function* (span) {
+    const spanTransformer = yield* getSpanTransformer;
+    const providerOptions = {
+      prompt: Prompt.make(options.prompt),
+      tools: [],
+      toolChoice: "none",
+      responseFormat: {
+        type: "text"
+      },
+      span
+    };
+    const content = yield* generateContent(options, providerOptions);
+    applySpanTransformer(spanTransformer, content, providerOptions);
+    return new GenerateTextResponse(content);
+  }, Effect.catchTag("ParseError", error => AiError.MalformedOutput.fromParseError({
+    module: "LanguageModel",
+    method: "generateText",
+    error
+  })), (effect, span) => Effect.withParentSpan(effect, span), Effect.provideService(IdGenerator, idGenerator)));
+  const generateObject = options => {
+    const schema = options.schema;
+    const objectName = getObjectName(options.objectName, schema);
+    return Effect.useSpan("LanguageModel.generateObject", {
+      captureStackTrace: false,
+      attributes: {
+        objectName,
+        concurrency: options.concurrency,
+        toolChoice: options.toolChoice
+      }
+    }, Effect.fnUntraced(function* (span) {
+      const spanTransformer = yield* getSpanTransformer;
+      const providerOptions = {
+        prompt: Prompt.make(options.prompt),
+        tools: [],
+        toolChoice: "none",
+        responseFormat: {
+          type: "json",
+          objectName,
+          schema
+        },
+        span
+      };
+      const content = yield* generateContent(options, providerOptions);
+      applySpanTransformer(spanTransformer, content, providerOptions);
+      const value = yield* resolveStructuredOutput(content, schema);
+      return new GenerateObjectResponse(value, content);
+    }, Effect.catchTag("ParseError", error => AiError.MalformedOutput.fromParseError({
+      module: "LanguageModel",
+      method: "generateText",
+      error
+    })), (effect, span) => Effect.withParentSpan(effect, span), Effect.provideService(IdGenerator, idGenerator)));
+  };
+  const streamText = Effect.fnUntraced(function* (options) {
+    const span = yield* Effect.makeSpanScoped("LanguageModel.streamText", {
+      captureStackTrace: false,
+      attributes: {
+        concurrency: options.concurrency,
+        toolChoice: options.toolChoice
+      }
+    });
+    const providerOptions = {
+      prompt: Prompt.make(options.prompt),
+      tools: [],
+      toolChoice: "none",
+      responseFormat: {
+        type: "text"
+      },
+      span
+    };
+    // Resolve the content stream for the request
+    const stream = yield* streamContent(options, providerOptions);
+    // Return the stream immediately if there is no span transformer
+    const spanTransformer = yield* getSpanTransformer;
+    if (Option.isNone(spanTransformer)) {
+      return stream;
+    }
+    // Otherwise aggregate generated content and apply the span transformer
+    // when the stream is finished
+    let content = [];
+    return stream.pipe(Stream.mapChunks(chunk => {
+      content = [...content, ...chunk];
+      return chunk;
+    }), Stream.ensuring(Effect.sync(() => {
+      spanTransformer.value({
+        ...providerOptions,
+        response: content
+      });
+    })));
+  }, Stream.unwrapScoped, Stream.mapError(error => ParseResult.isParseError(error) ? AiError.MalformedOutput.fromParseError({
+    module: "LanguageModel",
+    method: "streamText",
+    error
+  }) : error), Stream.provideService(IdGenerator, idGenerator));
+  const generateContent = Effect.fnUntraced(function* (options, providerOptions) {
+    const toolChoice = options.toolChoice ?? "auto";
+    // If there is no toolkit, the generated content can be returned immediately
+    if (Predicate.isUndefined(options.toolkit)) {
+      const ResponseSchema = Schema.mutable(Schema.Array(Response.Part(Toolkit.empty)));
+      const rawContent = yield* params.generateText(providerOptions);
+      const content = yield* Schema.decodeUnknown(ResponseSchema)(rawContent);
+      return content;
+    }
+    // If there is a toolkit resolve and apply it to the provider options
+    const toolkit = yield* resolveToolkit(options.toolkit);
+    // If the resolved toolkit is empty, return the generated content immediately
+    if (Object.values(toolkit.tools).length === 0) {
+      const ResponseSchema = Schema.mutable(Schema.Array(Response.Part(Toolkit.empty)));
+      const rawContent = yield* params.generateText(providerOptions);
+      const content = yield* Schema.decodeUnknown(ResponseSchema)(rawContent);
+      return content;
+    }
+    const tools = typeof toolChoice === "object" && "oneOf" in toolChoice ? Object.values(toolkit.tools).filter(tool => toolChoice.oneOf.includes(tool.name)) : Object.values(toolkit.tools);
+    providerOptions.tools = tools;
+    providerOptions.toolChoice = toolChoice;
+    // If tool call resolution is disabled, return the response without
+    // resolving the tool calls that were generated
+    if (options.disableToolCallResolution === true) {
+      const ResponseSchema = Schema.mutable(Schema.Array(Response.Part(Toolkit.empty)));
+      const rawContent = yield* params.generateText(providerOptions);
+      const content = yield* Schema.decodeUnknown(ResponseSchema)(rawContent);
+      return content;
+    }
+    const rawContent = yield* params.generateText(providerOptions);
+    // Resolve the generated tool calls
+    const toolResults = yield* resolveToolCalls(rawContent, toolkit, options.concurrency);
+    const ResponseSchema = Schema.mutable(Schema.Array(Response.Part(toolkit)));
+    const content = yield* Schema.decodeUnknown(ResponseSchema)(rawContent);
+    // Return the content merged with the tool call results
+    return [...content, ...toolResults];
+  });
+  const streamContent = Effect.fnUntraced(function* (options, providerOptions) {
+    const toolChoice = options.toolChoice ?? "auto";
+    // If there is no toolkit, return immediately
+    if (Predicate.isUndefined(options.toolkit)) {
+      const schema = Schema.ChunkFromSelf(Response.StreamPart(Toolkit.empty));
+      const decode = Schema.decode(schema);
+      return params.streamText(providerOptions).pipe(Stream.mapChunksEffect(decode));
+    }
+    // If there is a toolkit resolve and apply it to the provider options
+    const toolkit = Effect.isEffect(options.toolkit) ? yield* options.toolkit : options.toolkit;
+    // If the toolkit is empty, return immediately
+    if (Object.values(toolkit.tools).length === 0) {
+      const schema = Schema.ChunkFromSelf(Response.StreamPart(Toolkit.empty));
+      const decode = Schema.decode(schema);
+      return params.streamText(providerOptions).pipe(Stream.mapChunksEffect(decode));
+    }
+    const tools = typeof toolChoice === "object" && "oneOf" in toolChoice ? Object.values(toolkit.tools).filter(tool => toolChoice.oneOf.includes(tool.name)) : Object.values(toolkit.tools);
+    providerOptions.tools = tools;
+    providerOptions.toolChoice = toolChoice;
+    // If tool call resolution is disabled, return the response without
+    // resolving the tool calls that were generated
+    if (options.disableToolCallResolution === true) {
+      const schema = Schema.ChunkFromSelf(Response.StreamPart(toolkit));
+      const decode = Schema.decode(schema);
+      return params.streamText(providerOptions).pipe(Stream.mapChunksEffect(decode));
+    }
+    const ResponseSchema = Schema.Chunk(Response.StreamPart(toolkit));
+    const decode = Schema.decode(ResponseSchema);
+    return params.streamText(providerOptions).pipe(Stream.mapChunksEffect(Effect.fnUntraced(function* (chunk) {
+      const rawContent = Chunk.toArray(chunk);
+      const toolResults = yield* resolveToolCalls(rawContent, toolkit, options.concurrency);
+      const content = yield* decode(rawContent);
+      return Chunk.unsafeFromArray([...content, ...toolResults]);
+    })));
+  });
+  return {
+    generateText,
+    generateObject,
+    streamText
+  };
+});
+// =============================================================================
+// Accessors
+// =============================================================================
+/**
+ * Generate text using a language model.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect } from "effect"
+ *
+ * const program = Effect.gen(function* () {
+ *   const response = yield* LanguageModel.generateText({
+ *     prompt: "Write a haiku about programming",
+ *     toolChoice: "none"
+ *   })
+ *
+ *   console.log(response.text)
+ *   console.log(response.usage.totalTokens)
+ *
+ *   return response
+ * })
+ * ```
+ *
+ * @since 1.0.0
+ * @category Functions
+ */
+export const generateText = /*#__PURE__*/Effect.serviceFunctionEffect(LanguageModel, model => model.generateText);
+/**
+ * Generate a structured object from a schema using a language model.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect, Schema } from "effect"
+ *
+ * const EventSchema = Schema.Struct({
+ *   title: Schema.String,
+ *   date: Schema.String,
+ *   location: Schema.String
+ * })
+ *
+ * const program = Effect.gen(function* () {
+ *   const response = yield* LanguageModel.generateObject({
+ *     prompt: "Extract event info: Tech Conference on March 15th in San Francisco",
+ *     schema: EventSchema,
+ *     objectName: "event"
+ *   })
+ *
+ *   console.log(response.value)
+ *   // { title: "Tech Conference", date: "March 15th", location: "San Francisco" }
+ *
+ *   return response.value
+ * })
+ * ```
+ *
+ * @since 1.0.0
+ * @category Functions
+ */
+export const generateObject = /*#__PURE__*/Effect.serviceFunctionEffect(LanguageModel, model => model.generateObject);
+/**
+ * Generate text using a language model with streaming output.
+ *
+ * Returns a stream of response parts that are emitted as soon as they are
+ * available from the model, enabling real-time text generation experiences.
+ *
+ * @example
+ * ```ts
+ * import { LanguageModel } from "@effect/ai"
+ * import { Effect, Stream, Console } from "effect"
+ *
+ * const program = LanguageModel.streamText({
+ *   prompt: "Write a story about a space explorer"
+ * }).pipe(Stream.runForEach((part) => {
+ *   if (part.type === "text-delta") {
+ *     return Console.log(part.delta)
+ *   }
+ *   return Effect.void
+ * }))
+ * ```
+ *
+ * @since 1.0.0
+ * @category Functions
+ */
+export const streamText = options => Stream.unwrap(LanguageModel.pipe(Effect.map(model => model.streamText(options))));
+// =============================================================================
+// Tool Call Resolution
+// =============================================================================
+const resolveToolCalls = (content, toolkit, concurrency) => {
+  const toolNames = [];
+  const toolCalls = [];
+  for (const part of content) {
+    if (part.type === "tool-call") {
+      toolNames.push(part.name);
+      if (part.providerExecuted === true) {
+        continue;
+      }
+      toolCalls.push(part);
+    }
+  }
+  return Effect.forEach(toolCalls, toolCall => {
+    return toolkit.handle(toolCall.name, toolCall.params).pipe(Effect.map(({
+      encodedResult,
+      result
+    }) => Response.makePart("tool-result", {
+      id: toolCall.id,
+      name: toolCall.name,
+      result,
+      encodedResult,
+      providerName: toolCall.providerName,
+      providerExecuted: false
+    })));
+  }, {
+    concurrency
+  });
+};
+// =============================================================================
+// Utilities
+// =============================================================================
+const resolveToolkit = toolkit => Effect.isEffect(toolkit) ? toolkit : Effect.succeed(toolkit);
+const getObjectName = (objectName, schema) => Predicate.isNotUndefined(objectName) ? objectName : "_tag" in schema ? schema._tag : "identifier" in schema ? schema.identifier : "generateObject";
+const resolveStructuredOutput = /*#__PURE__*/Effect.fnUntraced(function* (response, ResultSchema) {
+  const text = [];
+  for (const part of response) {
+    if (part.type === "text") {
+      text.push(part.text);
+    }
+  }
+  if (text.length === 0) {
+    return yield* new AiError.MalformedOutput({
+      module: "LanguageModel",
+      method: "generateObject",
+      description: "No object was generated by the large language model"
+    });
+  }
+  const decode = Schema.decode(Schema.parseJson(ResultSchema));
+  return yield* Effect.mapError(decode(text.join("")), cause => new AiError.MalformedOutput({
+    module: "LanguageModel",
+    method: "generateObject",
+    description: "Generated object failed to conform to provided schema",
+    cause
+  }));
+});
+const applySpanTransformer = (transformer, response, options) => {
+  if (Option.isSome(transformer)) {
+    transformer.value({
+      ...options,
+      response: response
+    });
+  }
+};
+//# sourceMappingURL=LanguageModel.js.map

package/dist/esm/LanguageModel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LanguageModel.js","names":["Chunk","Context","Effect","Option","ParseResult","Predicate","Schema","Stream","AiError","defaultIdGenerator","IdGenerator","Prompt","Response","CurrentSpanTransformer","Toolkit","LanguageModel","Tag","GenerateTextResponse","content","constructor","text","part","type","push","join","reasoning","filter","reasoningText","length","undefined","toolCalls","toolResults","finishReason","finishPart","find","isUndefined","reason","usage","Usage","inputTokens","outputTokens","totalTokens","reasoningTokens","cachedInputTokens","GenerateObjectResponse","value","make","fnUntraced","params","parentSpanTransformer","serviceOption","getSpanTransformer","pipe","map","orElse","idGenerator","getOrElse","generateText","options","useSpan","captureStackTrace","attributes","concurrency","toolChoice","span","spanTransformer","providerOptions","prompt","tools","responseFormat","generateContent","applySpanTransformer","catchTag","error","MalformedOutput","fromParseError","module","method","effect","withParentSpan","provideService","generateObject","schema","objectName","getObjectName","resolveStructuredOutput","streamText","makeSpanScoped","stream","streamContent","isNone","mapChunks","chunk","ensuring","sync","response","unwrapScoped","mapError","isParseError","toolkit","ResponseSchema","mutable","Array","Part","empty","rawContent","decodeUnknown","resolveToolkit","Object","values","tool","oneOf","includes","name","disableToolCallResolution","resolveToolCalls","ChunkFromSelf","StreamPart","decode","mapChunksEffect","isEffect","toArray","unsafeFromArray","serviceFunctionEffect","model","unwrap","toolNames","providerExecuted","forEach","toolCall","handle","encodedResult","result","makePart","id","providerName","succeed","isNotUndefined","_tag","identifier","ResultSchema","description","parseJson","cause","transformer","isSome"],"sources":["../../src/LanguageModel.ts"],"sourcesContent":[null],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDA,OAAO,KAAKA,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,WAAW,MAAM,oBAAoB;AACjD,OAAO,KAAKC,SAAS,MAAM,kBAAkB;AAC7C,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAGvC,OAAO,KAAKC,OAAO,MAAM,cAAc;AACvC,SAASC,kBAAkB,EAAEC,WAAW,QAAQ,kBAAkB;AAClE,OAAO,KAAKC,MAAM,MAAM,aAAa;AACrC,OAAO,KAAKC,QAAQ,MAAM,eAAe;AAEzC,SAASC,sBAAsB,QAAQ,gBAAgB;AAEvD,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC;AACA;AACA;AAEA;;;;;;;;;;;;;;;;;;;;;;;;AAwBA,OAAM,MAAOC,aAAc,sBAAQd,OAAO,CAACe,GAAG,CAAC,0BAA0B,CAAC,EAGvE;AA0JH;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BA,OAAM,MAAOC,oBAAoB;EACtBC,OAAO;EAEhBC,YAAYD,OAAoC;IAC9C,IAAI,CAACA,OAAO,GAAGA,OAAO;EACxB;EAEA;;;EAGA,IAAIE,IAAIA,CAAA;IACN,MAAMA,IAAI,GAAkB,EAAE;IAC9B,KAAK,MAAMC,IAAI,IAAI,IAAI,CAACH,OAAO,EAAE;MAC/B,IAAIG,IAAI,CAACC,IAAI,KAAK,MAAM,EAAE;QACxBF,IAAI,CAACG,IAAI,CAACF,IAAI,CAACD,IAAI,CAAC;MACtB;IACF;IACA,OAAOA,IAAI,CAACI,IAAI,CAAC,EAAE,CAAC;EACtB;EAEA;;;EAGA,IAAIC,SAASA,CAAA;IACX,OAAO,IAAI,CAACP,OAAO,CAACQ,MAAM,CAAEL,IAAI,IAAKA,IAAI,CAACC,IAAI,KAAK,WAAW,CAAC;EACjE;EAEA;;;EAGA,IAAIK,aAAaA,CAAA;IACf,MAAMP,IAAI,GAAkB,EAAE;IAC9B,KAAK,MAAMC,IAAI,IAAI,IAAI,CAACH,OAAO,EAAE;MAC/B,IAAIG,IAAI,CAACC,IAAI,KAAK,WAAW,EAAE;QAC7BF,IAAI,CAACG,IAAI,CAACF,IAAI,CAACD,IAAI,CAAC;MACtB;IACF;IACA,OAAOA,IAAI,CAACQ,MAAM,KAAK,CAAC,GAAGC,SAAS,GAAGT,IAAI,CAACI,IAAI,CAAC,EAAE,CAAC;EACtD;EAEA;;;EAGA,IAAIM,SAASA,CAAA;IACX,OAAO,IAAI,CAACZ,OAAO,CAACQ,MAAM,CAAEL,IAAI,IAAKA,IAAI,CAACC,IAAI,KAAK,WAAW,CAAC;EACjE;EAEA;;;EAGA,IAAIS,WAAWA,CAAA;IACb,OAAO,IAAI,CAACb,OAAO,CAACQ,MAAM,CAAEL,IAAI,IAAKA,IAAI,CAACC,IAAI,KAAK,aAAa,CAAC;EACnE;EAEA;;;EAGA,IAAIU,YAAYA,CAAA;IACd,MAAMC,UAAU,GAAG,IAAI,CAACf,OAAO,CAACgB,IAAI,CAAEb,IAAI,IAAKA,IAAI,CAACC,IAAI,KAAK,QAAQ,CAAC;IACtE,OAAOjB,SAAS,CAAC8B,WAAW,CAACF,UAAU,CAAC,GAAG,SAAS,GAAGA,UAAU,CAACG,MAAM;EAC1E;EAEA;;;EAGA,IAAIC,KAAKA,CAAA;IACP,MAAMJ,UAAU,GAAG,IAAI,CAACf,OAAO,CAACgB,IAAI,CAAEb,IAAI,IAAKA,IAAI,CAACC,IAAI,KAAK,QAAQ,CAAC;IACtE,IAAIjB,SAAS,CAAC8B,WAAW,CAACF,UAAU,CAAC,EAAE;MACrC,OAAO,IAAIrB,QAAQ,CAAC0B,KAAK,CAAC;QACxBC,WAAW,EAAEV,SAAS;QACtBW,YAAY,EAAEX,SAAS;QACvBY,WAAW,EAAEZ,SAAS;QACtBa,eAAe,EAAEb,SAAS;QAC1Bc,iBAAiB,EAAEd;OACpB,CAAC;IACJ;IACA,OAAOI,UAAU,CAACI,KAAK;EACzB;;AAGF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6BA,OAAM,MAAOO,sBAAkE,SAAQ3B,oBAA2B;EAChH;;;EAGS4B,KAAK;EAEd1B,YAAY0B,KAAQ,EAAE3B,OAAoC;IACxD,KAAK,CAACA,OAAO,CAAC;IACd,IAAI,CAAC2B,KAAK,GAAGA,KAAK;EACpB;;AAsJF;;;;;;;;;AASA,OAAO,MAAMC,IAAI,gBAA0D5C,MAAM,CAAC6C,UAAU,CAC1F,WAAUC,MAAM;EACd,MAAMC,qBAAqB,GAAG,OAAO/C,MAAM,CAACgD,aAAa,CAACrC,sBAAsB,CAAC;EACjF,MAAMsC,kBAAkB,GAAGjD,MAAM,CAACgD,aAAa,CAACrC,sBAAsB,CAAC,CAACuC,IAAI,CAC1ElD,MAAM,CAACmD,GAAG,CAAClD,MAAM,CAACmD,MAAM,CAAC,MAAML,qBAAqB,CAAC,CAAC,CACvD;EAED,MAAMM,WAAW,GAAG,OAAOrD,MAAM,CAACgD,aAAa,CAACxC,WAAW,CAAC,CAAC0C,IAAI,CAC/DlD,MAAM,CAACmD,GAAG,CAAClD,MAAM,CAACqD,SAAS,CAAC,MAAM/C,kBAAkB,CAAC,CAAC,CACvD;EAED,MAAMgD,YAAY,GAGhBC,OAA6C,IAK7CxD,MAAM,CAACyD,OAAO,CACZ,4BAA4B,EAC5B;IACEC,iBAAiB,EAAE,KAAK;IACxBC,UAAU,EAAE;MACVC,WAAW,EAAEJ,OAAO,CAACI,WAAW;MAChCC,UAAU,EAAEL,OAAO,CAACK;;GAEvB,EACD7D,MAAM,CAAC6C,UAAU,CACf,WAAUiB,IAAI;IACZ,MAAMC,eAAe,GAAG,OAAOd,kBAAkB;IAEjD,MAAMe,eAAe,GAA6B;MAChDC,MAAM,EAAExD,MAAM,CAACmC,IAAI,CAACY,OAAO,CAACS,MAAM,CAAC;MACnCC,KAAK,EAAE,EAAE;MACTL,UAAU,EAAE,MAAM;MAClBM,cAAc,EAAE;QAAE/C,IAAI,EAAE;MAAM,CAAE;MAChC0C;KACD;IACD,MAAM9C,OAAO,GAAG,OAAOoD,eAAe,CAACZ,OAAO,EAAEQ,eAAe,CAAC;IAEhEK,oBAAoB,CAACN,eAAe,EAAE/C,OAAc,EAAEgD,eAAe,CAAC;IAEtE,OAAO,IAAIjD,oBAAoB,CAACC,OAAO,CAAC;EAC1C,CAAC,EACDhB,MAAM,CAACsE,QAAQ,CAAC,YAAY,EAAGC,KAAK,IAClCjE,OAAO,CAACkE,eAAe,CAACC,cAAc,CAAC;IACrCC,MAAM,EAAE,eAAe;IACvBC,MAAM,EAAE,cAAc;IACtBJ;GACD,CAAC,CAAC,EACL,CAACK,MAAM,EAAEd,IAAI,KAAK9D,MAAM,CAAC6E,cAAc,CAACD,MAAM,EAAEd,IAAI,CAAC,EACrD9D,MAAM,CAAC8E,cAAc,CAACtE,WAAW,EAAE6C,WAAW,CAAC,CAChD,CACK;EAEV,MAAM0B,cAAc,GAMlBvB,OAAwD,IAItD;IACF,MAAMwB,MAAM,GAA2BxB,OAAO,CAACwB,MAAM;IACrD,MAAMC,UAAU,GAAGC,aAAa,CAAC1B,OAAO,CAACyB,UAAU,EAAED,MAAM,CAAC;IAC5D,OAAOhF,MAAM,CAACyD,OAAO,CACnB,8BAA8B,EAC9B;MACEC,iBAAiB,EAAE,KAAK;MACxBC,UAAU,EAAE;QACVsB,UAAU;QACVrB,WAAW,EAAEJ,OAAO,CAACI,WAAW;QAChCC,UAAU,EAAEL,OAAO,CAACK;;KAEvB,EACD7D,MAAM,CAAC6C,UAAU,CACf,WAAUiB,IAAI;MACZ,MAAMC,eAAe,GAAG,OAAOd,kBAAkB;MAEjD,MAAMe,eAAe,GAA6B;QAChDC,MAAM,EAAExD,MAAM,CAACmC,IAAI,CAACY,OAAO,CAACS,MAAM,CAAC;QACnCC,KAAK,EAAE,EAAE;QACTL,UAAU,EAAE,MAAM;QAClBM,cAAc,EAAE;UAAE/C,IAAI,EAAE,MAAM;UAAE6D,UAAU;UAAED;QAAM,CAAE;QACpDlB;OACD;MAED,MAAM9C,OAAO,GAAG,OAAOoD,eAAe,CAACZ,OAAO,EAAEQ,eAAe,CAAC;MAEhEK,oBAAoB,CAACN,eAAe,EAAE/C,OAAc,EAAEgD,eAAe,CAAC;MAEtE,MAAMrB,KAAK,GAAG,OAAOwC,uBAAuB,CAACnE,OAAc,EAAEgE,MAAM,CAAC;MAEpE,OAAO,IAAItC,sBAAsB,CAACC,KAAK,EAAE3B,OAAO,CAAC;IACnD,CAAC,EACDhB,MAAM,CAACsE,QAAQ,CAAC,YAAY,EAAGC,KAAK,IAClCjE,OAAO,CAACkE,eAAe,CAACC,cAAc,CAAC;MACrCC,MAAM,EAAE,eAAe;MACvBC,MAAM,EAAE,cAAc;MACtBJ;KACD,CAAC,CAAC,EACL,CAACK,MAAM,EAAEd,IAAI,KAAK9D,MAAM,CAAC6E,cAAc,CAACD,MAAM,EAAEd,IAAI,CAAC,EACrD9D,MAAM,CAAC8E,cAAc,CAACtE,WAAW,EAAE6C,WAAW,CAAC,CAChD,CACK;EACV,CAAC;EAED,MAAM+B,UAAU,GAOZpF,MAAM,CAAC6C,UAAU,CACnB,WAGEW,OAA6C;IAC7C,MAAMM,IAAI,GAAG,OAAO9D,MAAM,CAACqF,cAAc,CAAC,0BAA0B,EAAE;MACpE3B,iBAAiB,EAAE,KAAK;MACxBC,UAAU,EAAE;QAAEC,WAAW,EAAEJ,OAAO,CAACI,WAAW;QAAEC,UAAU,EAAEL,OAAO,CAACK;MAAU;KAC/E,CAAC;IAEF,MAAMG,eAAe,GAA6B;MAChDC,MAAM,EAAExD,MAAM,CAACmC,IAAI,CAACY,OAAO,CAACS,MAAM,CAAC;MACnCC,KAAK,EAAE,EAAE;MACTL,UAAU,EAAE,MAAM;MAClBM,cAAc,EAAE;QAAE/C,IAAI,EAAE;MAAM,CAAE;MAChC0C;KACD;IAED;IACA,MAAMwB,MAAM,GAAG,OAAOC,aAAa,CAAC/B,OAAO,EAAEQ,eAAe,CAAC;IAE7D;IACA,MAAMD,eAAe,GAAG,OAAOd,kBAAkB;IACjD,IAAIhD,MAAM,CAACuF,MAAM,CAACzB,eAAe,CAAC,EAAE;MAClC,OAAOuB,MAAM;IACf;IAEA;IACA;IACA,IAAItE,OAAO,GAAsC,EAAE;IACnD,OAAOsE,MAAM,CAACpC,IAAI,CAChB7C,MAAM,CAACoF,SAAS,CAAEC,KAAK,IAAI;MACzB1E,OAAO,GAAG,CAAC,GAAGA,OAAO,EAAE,GAAG0E,KAAK,CAAC;MAChC,OAAOA,KAAK;IACd,CAAC,CAAC,EACFrF,MAAM,CAACsF,QAAQ,CAAC3F,MAAM,CAAC4F,IAAI,CAAC,MAAK;MAC/B7B,eAAe,CAACpB,KAAK,CAAC;QAAE,GAAGqB,eAAe;QAAE6B,QAAQ,EAAE7E;MAAc,CAAE,CAAC;IACzE,CAAC,CAAC,CAAC,CACJ;EACH,CAAC,EACDX,MAAM,CAACyF,YAAY,EACnBzF,MAAM,CAAC0F,QAAQ,CAAExB,KAAK,IACpBrE,WAAW,CAAC8F,YAAY,CAACzB,KAAK,CAAC,GAC3BjE,OAAO,CAACkE,eAAe,CAACC,cAAc,CAAC;IACvCC,MAAM,EAAE,eAAe;IACvBC,MAAM,EAAE,YAAY;IACpBJ;GACD,CAAC,GACAA,KAAK,CACV,EACDlE,MAAM,CAACyE,cAAc,CAACtE,WAAW,EAAE6C,WAAW,CAAC,CACzC;EAER,MAAMe,eAAe,GAOjBpE,MAAM,CAAC6C,UAAU,CACnB,WAGEW,OAA6C,EAAEQ,eAAyC;IACxF,MAAMH,UAAU,GAAGL,OAAO,CAACK,UAAU,IAAI,MAAM;IAE/C;IACA,IAAI1D,SAAS,CAAC8B,WAAW,CAACuB,OAAO,CAACyC,OAAO,CAAC,EAAE;MAC1C,MAAMC,cAAc,GAAG9F,MAAM,CAAC+F,OAAO,CAAC/F,MAAM,CAACgG,KAAK,CAAC1F,QAAQ,CAAC2F,IAAI,CAACzF,OAAO,CAAC0F,KAAK,CAAC,CAAC,CAAC;MACjF,MAAMC,UAAU,GAAG,OAAOzD,MAAM,CAACS,YAAY,CAACS,eAAe,CAAC;MAC9D,MAAMhD,OAAO,GAAG,OAAOZ,MAAM,CAACoG,aAAa,CAACN,cAAc,CAAC,CAACK,UAAU,CAAC;MACvE,OAAOvF,OAAsC;IAC/C;IAEA;IACA,MAAMiF,OAAO,GAAG,OAAOQ,cAAc,CAAkBjD,OAAO,CAACyC,OAAO,CAAC;IAEvE;IACA,IAAIS,MAAM,CAACC,MAAM,CAACV,OAAO,CAAC/B,KAAK,CAAC,CAACxC,MAAM,KAAK,CAAC,EAAE;MAC7C,MAAMwE,cAAc,GAAG9F,MAAM,CAAC+F,OAAO,CAAC/F,MAAM,CAACgG,KAAK,CAAC1F,QAAQ,CAAC2F,IAAI,CAACzF,OAAO,CAAC0F,KAAK,CAAC,CAAC,CAAC;MACjF,MAAMC,UAAU,GAAG,OAAOzD,MAAM,CAACS,YAAY,CAACS,eAAe,CAAC;MAC9D,MAAMhD,OAAO,GAAG,OAAOZ,MAAM,CAACoG,aAAa,CAACN,cAAc,CAAC,CAACK,UAAU,CAAC;MACvE,OAAOvF,OAAsC;IAC/C;IAEA,MAAMkD,KAAK,GAAG,OAAOL,UAAU,KAAK,QAAQ,IAAI,OAAO,IAAIA,UAAU,GACjE6C,MAAM,CAACC,MAAM,CAACV,OAAO,CAAC/B,KAAK,CAAC,CAAC1C,MAAM,CAAEoF,IAAI,IAAK/C,UAAU,CAACgD,KAAK,CAACC,QAAQ,CAACF,IAAI,CAACG,IAAI,CAAC,CAAC,GACnFL,MAAM,CAACC,MAAM,CAACV,OAAO,CAAC/B,KAAK,CAAC;IAChCF,eAAe,CAACE,KAAK,GAAGA,KAAK;IAC7BF,eAAe,CAACH,UAAU,GAAGA,UAAU;IAEvC;IACA;IACA,IAAIL,OAAO,CAACwD,yBAAyB,KAAK,IAAI,EAAE;MAC9C,MAAMd,cAAc,GAAG9F,MAAM,CAAC+F,OAAO,CAAC/F,MAAM,CAACgG,KAAK,CAAC1F,QAAQ,CAAC2F,IAAI,CAACzF,OAAO,CAAC0F,KAAK,CAAC,CAAC,CAAC;MACjF,MAAMC,UAAU,GAAG,OAAOzD,MAAM,CAACS,YAAY,CAACS,eAAe,CAAC;MAC9D,MAAMhD,OAAO,GAAG,OAAOZ,MAAM,CAACoG,aAAa,CAACN,cAAc,CAAC,CAACK,UAAU,CAAC;MACvE,OAAOvF,OAAsC;IAC/C;IAEA,MAAMuF,UAAU,GAAG,OAAOzD,MAAM,CAACS,YAAY,CAACS,eAAe,CAAC;IAE9D;IACA,MAAMnC,WAAW,GAAG,OAAOoF,gBAAgB,CAACV,UAAU,EAAEN,OAAO,EAAEzC,OAAO,CAACI,WAAW,CAAC;IACrF,MAAMsC,cAAc,GAAG9F,MAAM,CAAC+F,OAAO,CAAC/F,MAAM,CAACgG,KAAK,CAAC1F,QAAQ,CAAC2F,IAAI,CAACJ,OAAO,CAAC,CAAC,CAAC;IAC3E,MAAMjF,OAAO,GAAG,OAAOZ,MAAM,CAACoG,aAAa,CAACN,cAAc,CAAC,CAACK,UAAU,CAAC;IAEvE;IACA,OAAO,CAAC,GAAGvF,OAAO,EAAE,GAAGa,WAAW,CAAgC;EACpE,CAAC,CACF;EAED,MAAM0D,aAAa,GAOfvF,MAAM,CAAC6C,UAAU,CACnB,WAGEW,OAA6C,EAAEQ,eAAyC;IACxF,MAAMH,UAAU,GAAGL,OAAO,CAACK,UAAU,IAAI,MAAM;IAE/C;IACA,IAAI1D,SAAS,CAAC8B,WAAW,CAACuB,OAAO,CAACyC,OAAO,CAAC,EAAE;MAC1C,MAAMjB,MAAM,GAAG5E,MAAM,CAAC8G,aAAa,CAACxG,QAAQ,CAACyG,UAAU,CAACvG,OAAO,CAAC0F,KAAK,CAAC,CAAC;MACvE,MAAMc,MAAM,GAAGhH,MAAM,CAACgH,MAAM,CAACpC,MAAM,CAAC;MACpC,OAAOlC,MAAM,CAACsC,UAAU,CAACpB,eAAe,CAAC,CAACd,IAAI,CAC5C7C,MAAM,CAACgH,eAAe,CAACD,MAAM,CAAC,CACqE;IACvG;IAEA;IACA,MAAMnB,OAAO,GAAGjG,MAAM,CAACsH,QAAQ,CAAC9D,OAAO,CAACyC,OAAO,CAAC,GAAG,OAAOzC,OAAO,CAACyC,OAAO,GAAGzC,OAAO,CAACyC,OAAO;IAE3F;IACA,IAAIS,MAAM,CAACC,MAAM,CAACV,OAAO,CAAC/B,KAAK,CAAC,CAACxC,MAAM,KAAK,CAAC,EAAE;MAC7C,MAAMsD,MAAM,GAAG5E,MAAM,CAAC8G,aAAa,CAACxG,QAAQ,CAACyG,UAAU,CAACvG,OAAO,CAAC0F,KAAK,CAAC,CAAC;MACvE,MAAMc,MAAM,GAAGhH,MAAM,CAACgH,MAAM,CAACpC,MAAM,CAAC;MACpC,OAAOlC,MAAM,CAACsC,UAAU,CAACpB,eAAe,CAAC,CAACd,IAAI,CAC5C7C,MAAM,CAACgH,eAAe,CAACD,MAAM,CAAC,CACqE;IACvG;IAEA,MAAMlD,KAAK,GAAG,OAAOL,UAAU,KAAK,QAAQ,IAAI,OAAO,IAAIA,UAAU,GACjE6C,MAAM,CAACC,MAAM,CAACV,OAAO,CAAC/B,KAAK,CAAC,CAAC1C,MAAM,CAAEoF,IAAI,IAAK/C,UAAU,CAACgD,KAAK,CAACC,QAAQ,CAACF,IAAI,CAACG,IAAI,CAAC,CAAC,GACnFL,MAAM,CAACC,MAAM,CAACV,OAAO,CAAC/B,KAAK,CAAC;IAChCF,eAAe,CAACE,KAAK,GAAGA,KAAK;IAC7BF,eAAe,CAACH,UAAU,GAAGA,UAAU;IAEvC;IACA;IACA,IAAIL,OAAO,CAACwD,yBAAyB,KAAK,IAAI,EAAE;MAC9C,MAAMhC,MAAM,GAAG5E,MAAM,CAAC8G,aAAa,CAACxG,QAAQ,CAACyG,UAAU,CAAClB,OAAO,CAAC,CAAC;MACjE,MAAMmB,MAAM,GAAGhH,MAAM,CAACgH,MAAM,CAACpC,MAAM,CAAC;MACpC,OAAOlC,MAAM,CAACsC,UAAU,CAACpB,eAAe,CAAC,CAACd,IAAI,CAC5C7C,MAAM,CAACgH,eAAe,CAACD,MAAM,CAAC,CACqE;IACvG;IAEA,MAAMlB,cAAc,GAAG9F,MAAM,CAACN,KAAK,CAACY,QAAQ,CAACyG,UAAU,CAAClB,OAAO,CAAC,CAAC;IACjE,MAAMmB,MAAM,GAAGhH,MAAM,CAACgH,MAAM,CAAClB,cAAc,CAAC;IAC5C,OAAOpD,MAAM,CAACsC,UAAU,CAACpB,eAAe,CAAC,CAACd,IAAI,CAC5C7C,MAAM,CAACgH,eAAe,CAACrH,MAAM,CAAC6C,UAAU,CAAC,WAAU6C,KAAK;MACtD,MAAMa,UAAU,GAAGzG,KAAK,CAACyH,OAAO,CAAC7B,KAAK,CAAC;MACvC,MAAM7D,WAAW,GAAG,OAAOoF,gBAAgB,CAACV,UAAU,EAAEN,OAAO,EAAEzC,OAAO,CAACI,WAAW,CAAC;MACrF,MAAM5C,OAAO,GAAG,OAAOoG,MAAM,CAACb,UAAU,CAAC;MACzC,OAAOzG,KAAK,CAAC0H,eAAe,CAAC,CAAC,GAAGxG,OAAO,EAAE,GAAGa,WAAW,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC,CACgG;EACvG,CAAC,CACF;EAED,OAAO;IACL0B,YAAY;IACZwB,cAAc;IACdK;GACQ;AACZ,CAAC,CACF;AAED;AACA;AACA;AAEA;;;;;;;;;;;;;;;;;;;;;;;;AAwBA,OAAO,MAAM7B,YAAY,gBAOrBvD,MAAM,CAACyH,qBAAqB,CAAC5G,aAAa,EAAG6G,KAAK,IAAKA,KAAK,CAACnE,YAAY,CAAC;AAE9E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+BA,OAAO,MAAMwB,cAAc,gBAUvB/E,MAAM,CAACyH,qBAAqB,CAAC5G,aAAa,EAAG6G,KAAK,IAAKA,KAAK,CAAC3C,cAAc,CAAC;AAEhF;;;;;;;;;;;;;;;;;;;;;;;;AAwBA,OAAO,MAAMK,UAAU,GAGrB5B,OAA6C,IAI1CnD,MAAM,CAACsH,MAAM,CAAC9G,aAAa,CAACqC,IAAI,CAAClD,MAAM,CAACmD,GAAG,CAAEuE,KAAK,IAAKA,KAAK,CAACtC,UAAU,CAAC5B,OAAO,CAAC,CAAC,CAAC,CAAC;AAExF;AACA;AACA;AAEA,MAAMyD,gBAAgB,GAAGA,CACvBjG,OAAgD,EAChDiF,OAAmC,EACnCrC,WAAoC,KAKlC;EACF,MAAMgE,SAAS,GAAkB,EAAE;EACnC,MAAMhG,SAAS,GAAwC,EAAE;EAEzD,KAAK,MAAMT,IAAI,IAAIH,OAAO,EAAE;IAC1B,IAAIG,IAAI,CAACC,IAAI,KAAK,WAAW,EAAE;MAC7BwG,SAAS,CAACvG,IAAI,CAACF,IAAI,CAAC4F,IAAI,CAAC;MACzB,IAAI5F,IAAI,CAAC0G,gBAAgB,KAAK,IAAI,EAAE;QAClC;MACF;MACAjG,SAAS,CAACP,IAAI,CAACF,IAAI,CAAC;IACtB;EACF;EAEA,OAAOnB,MAAM,CAAC8H,OAAO,CAAClG,SAAS,EAAGmG,QAAQ,IAAI;IAC5C,OAAO9B,OAAO,CAAC+B,MAAM,CAACD,QAAQ,CAAChB,IAAI,EAAEgB,QAAQ,CAACjF,MAAa,CAAC,CAACI,IAAI,CAC/DlD,MAAM,CAACmD,GAAG,CAAC,CAAC;MAAE8E,aAAa;MAAEC;IAAM,CAAE,KACnCxH,QAAQ,CAACyH,QAAQ,CAAC,aAAa,EAAE;MAC/BC,EAAE,EAAEL,QAAQ,CAACK,EAAE;MACfrB,IAAI,EAAEgB,QAAQ,CAAChB,IAAI;MACnBmB,MAAM;MACND,aAAa;MACbI,YAAY,EAAEN,QAAQ,CAACM,YAAY;MACnCR,gBAAgB,EAAE;KACnB,CAAC,CACH,CACF;EACH,CAAC,EAAE;IAAEjE;EAAW,CAAE,CAAC;AACrB,CAAC;AAED;AACA;AACA;AAEA,MAAM6C,cAAc,GAClBR,OAAqF,IACjCjG,MAAM,CAACsH,QAAQ,CAACrB,OAAO,CAAC,GAAGA,OAAO,GAAGjG,MAAM,CAACsI,OAAO,CAACrC,OAAO,CAAC;AAElH,MAAMf,aAAa,GAAGA,CACpBD,UAA8B,EAC9BD,MAA8B,KAE9B7E,SAAS,CAACoI,cAAc,CAACtD,UAAU,CAAC,GAChCA,UAAU,GACV,MAAM,IAAID,MAAM,GAChBA,MAAM,CAACwD,IAAc,GACrB,YAAY,IAAIxD,MAAM,GACtBA,MAAM,CAACyD,UAAoB,GAC3B,gBAAgB;AAEtB,MAAMtD,uBAAuB,gBAAGnF,MAAM,CAAC6C,UAAU,CAC/C,WAAmBgD,QAA+C,EAAE6C,YAAoC;EACtG,MAAMxH,IAAI,GAAkB,EAAE;EAC9B,KAAK,MAAMC,IAAI,IAAI0E,QAAQ,EAAE;IAC3B,IAAI1E,IAAI,CAACC,IAAI,KAAK,MAAM,EAAE;MACxBF,IAAI,CAACG,IAAI,CAACF,IAAI,CAACD,IAAI,CAAC;IACtB;EACF;EAEA,IAAIA,IAAI,CAACQ,MAAM,KAAK,CAAC,EAAE;IACrB,OAAO,OAAO,IAAIpB,OAAO,CAACkE,eAAe,CAAC;MACxCE,MAAM,EAAE,eAAe;MACvBC,MAAM,EAAE,gBAAgB;MACxBgE,WAAW,EAAE;KACd,CAAC;EACJ;EAEA,MAAMvB,MAAM,GAAGhH,MAAM,CAACgH,MAAM,CAAChH,MAAM,CAACwI,SAAS,CAACF,YAAY,CAAC,CAAC;EAC5D,OAAO,OAAO1I,MAAM,CAAC+F,QAAQ,CAACqB,MAAM,CAAClG,IAAI,CAACI,IAAI,CAAC,EAAE,CAAC,CAAC,EAAGuH,KAAK,IACzD,IAAIvI,OAAO,CAACkE,eAAe,CAAC;IAC1BE,MAAM,EAAE,eAAe;IACvBC,MAAM,EAAE,gBAAgB;IACxBgE,WAAW,EAAE,uDAAuD;IACpEE;GACD,CAAC,CAAC;AACP,CAAC,CACF;AAED,MAAMxE,oBAAoB,GAAGA,CAC3ByE,WAA2C,EAC3CjD,QAA+C,EAC/CrC,OAAwB,KAChB;EACR,IAAIvD,MAAM,CAAC8I,MAAM,CAACD,WAAW,CAAC,EAAE;IAC9BA,WAAW,CAACnG,KAAK,CAAC;MAAE,GAAGa,OAAO;MAAEqC,QAAQ,EAAEA;IAAe,CAAE,CAAC;EAC9D;AACF,CAAC","ignoreList":[]}

package/dist/esm/McpServer.js

@@ -19,8 +19,8 @@ import * as RcMap from "effect/RcMap";
 import * as Schema from "effect/Schema";
 import * as AST from "effect/SchemaAST";
 import * as FindMyWay from "find-my-way-ts";
-import * as AiTool from "./AiTool.js";
 import { CallToolResult, ClientNotificationRpcs, ClientRpcs, CompleteResult, Elicit, ElicitationDeclined, GetPromptResult, InternalError, InvalidParams, ListPromptsResult, ListResourcesResult, ListResourceTemplatesResult, ListToolsResult, McpServerClient, McpServerClientMiddleware, ParamAnnotation, Prompt, Resource, ResourceTemplate, ServerNotificationRpcs, ServerRequestRpcs, TextContent, Tool, ToolAnnotations } from "./McpSchema.js";
+import * as AiTool from "./Tool.js";
 /**
  * @since 1.0.0
  * @category McpServer
@@ -68,6 +68,7 @@ export class McpServer extends /*#__PURE__*/Context.Tag("@effect/ai/McpServer")(
     return McpServer.of({
       notifications: notifications.client,
       notificationsMailbox,
+      initializedClients: new Set(),
       get tools() {
         return tools;
       },
@@ -259,7 +260,11 @@ export const run = /*#__PURE__*/Effect.fnUntraced(function* (options) {
       payload: encoded
     };
     const clientIds = yield* patchedProtocol.clientIds;
-    for (const clientId of clientIds) {
+    for (const clientId of server.initializedClients) {
+      if (!clientIds.has(clientId)) {
+        server.initializedClients.delete(clientId);
+        continue;
+      }
       yield* patchedProtocol.send(clientId, message);
     }
   })), Effect.catchAllCause(() => Effect.void), Effect.forever, Effect.forkScoped);
@@ -414,7 +419,7 @@ export const registerToolkit = /*#__PURE__*/Effect.fnUntraced(function* (toolkit
   const registry = yield* McpServer;
   const built = yield* toolkit;
   const context = yield* Effect.context();
-  for (const tool of built.tools) {
+  for (const tool of Object.values(built.tools)) {
     const mcpTool = new Tool({
       name: tool.name,
       description: tool.description,
@@ -700,7 +705,9 @@ const layerHandlers = serverInfo => ClientRpcs.toLayer(Effect.gen(function* () {
   return {
     // Requests
     ping: () => Effect.succeed({}),
-    initialize(params) {
+    initialize(params, {
+      clientId
+    }) {
       const requestedVersion = params.protocolVersion;
       const capabilities = {
         completions: {}
@@ -721,6 +728,7 @@ const layerHandlers = serverInfo => ClientRpcs.toLayer(Effect.gen(function* () {
           listChanged: true
         };
       }
+      server.initializedClients.add(clientId);
       return Effect.succeed({
         capabilities,
         serverInfo,