@effect/ai 0.10.2 → 0.10.4

package/dist/esm/AiModels.js

@@ -0,0 +1,44 @@
+/**
+ * @since 1.0.0
+ */
+import * as Context from "effect/Context";
+import * as Effect from "effect/Effect";
+import * as Equal from "effect/Equal";
+import * as Hash from "effect/Hash";
+import * as Layer from "effect/Layer";
+import * as RcMap from "effect/RcMap";
+/**
+ * @since 1.0.0
+ * @category tags
+ */
+export class AiModels extends /*#__PURE__*/Context.Tag("@effect/ai/AiModels")() {}
+class AiModelsKey {
+  model;
+  service;
+  constructor(model, service) {
+    this.model = model;
+    this.service = service;
+  }
+  [Equal.symbol](that) {
+    return this.service === that.service && this.model.cacheKey === that.model.cacheKey;
+  }
+  [Hash.symbol]() {
+    return Hash.combine(Hash.hash(this.service))(Hash.hash(this.model.cacheKey));
+  }
+}
+const make = /*#__PURE__*/Effect.gen(function* () {
+  const services = yield* RcMap.make({
+    idleTimeToLive: "1 minute",
+    lookup: key => Effect.provideService(key.model.provides, key.model.requires, key.service)
+  });
+  const build = (model, context) => Effect.map(RcMap.get(services, new AiModelsKey(model, Context.get(context, model.requires))), context => model.updateContext(context));
+  return {
+    build
+  };
+});
+/**
+ * @since 1.0.0
+ * @category layers
+ */
+export const layer = /*#__PURE__*/Layer.scoped(AiModels, make);
+//# sourceMappingURL=AiModels.js.map

package/dist/esm/AiModels.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AiModels.js","names":["Context","Effect","Equal","Hash","Layer","RcMap","AiModels","Tag","AiModelsKey","model","service","constructor","symbol","that","cacheKey","combine","hash","make","gen","services","idleTimeToLive","lookup","key","provideService","provides","requires","build","context","map","get","updateContext","layer","scoped"],"sources":["../../src/AiModels.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,IAAI,MAAM,aAAa;AACnC,OAAO,KAAKC,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,KAAK,MAAM,cAAc;AAIrC;;;;AAIA,OAAM,MAAOC,QAAS,sBAAQN,OAAO,CAACO,GAAG,CAAC,qBAAqB,CAAC,EAG7D;AAkBH,MAAMC,WAAW;EAEJC,KAAA;EACAC,OAAA;EAFXC,YACWF,KAAwB,EACxBC,OAAgB;IADhB,KAAAD,KAAK,GAALA,KAAK;IACL,KAAAC,OAAO,GAAPA,OAAO;EACf;EAEH,CAACR,KAAK,CAACU,MAAM,EAAEC,IAAiB;IAC9B,OAAO,IAAI,CAACH,OAAO,KAAKG,IAAI,CAACH,OAAO,IAAI,IAAI,CAACD,KAAK,CAACK,QAAQ,KAAKD,IAAI,CAACJ,KAAK,CAACK,QAAQ;EACrF;EACA,CAACX,IAAI,CAACS,MAAM,IAAC;IACX,OAAOT,IAAI,CAACY,OAAO,CAACZ,IAAI,CAACa,IAAI,CAAC,IAAI,CAACN,OAAO,CAAC,CAAC,CAACP,IAAI,CAACa,IAAI,CAAC,IAAI,CAACP,KAAK,CAACK,QAAQ,CAAC,CAAC;EAC9E;;AAGF,MAAMG,IAAI,gBAAGhB,MAAM,CAACiB,GAAG,CAAC,aAAS;EAC/B,MAAMC,QAAQ,GAAG,OAAOd,KAAK,CAACY,IAAI,CAAC;IACjCG,cAAc,EAAE,UAAU;IAC1BC,MAAM,EAAGC,GAAgB,IAAKrB,MAAM,CAACsB,cAAc,CAACD,GAAG,CAACb,KAAK,CAACe,QAAQ,EAAEF,GAAG,CAACb,KAAK,CAACgB,QAAQ,EAAEH,GAAG,CAACZ,OAAO;GACxG,CAAC;EAEF,MAAMgB,KAAK,GAAGA,CACZjB,KAAkC,EAClCkB,OAAkC,KAElC1B,MAAM,CAAC2B,GAAG,CACRvB,KAAK,CAACwB,GAAG,CACPV,QAAQ,EACR,IAAIX,WAAW,CAACC,KAAK,EAAET,OAAO,CAAC6B,GAAG,CAACF,OAAO,EAAElB,KAAK,CAACgB,QAAe,CAAC,CAAC,CACpE,EACAE,OAAO,IAAKlB,KAAK,CAACqB,aAAa,CAACH,OAAO,CAAC,CAC1C;EAEH,OAAO;IAAED;EAAK,CAAW;AAC3B,CAAC,CAAC;AAEF;;;;AAIA,OAAO,MAAMK,KAAK,gBAA0B3B,KAAK,CAAC4B,MAAM,CAAC1B,QAAQ,EAAEW,IAAI,CAAC","ignoreList":[]}

package/dist/esm/AiPlan.js

@@ -0,0 +1,22 @@
+import * as Internal from "./internal/aiPlan.js";
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export const TypeId = Internal.TypeId;
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const fromModel = Internal.fromModel;
+/**
+ * @since 1.0.0
+ * @category error handling
+ */
+export const withFallback = Internal.withFallback;
+/**
+ * @since 1.0.0
+ * @category combination
+ */
+export const concatSteps = Internal.concatSteps;
+//# sourceMappingURL=AiPlan.js.map

package/dist/esm/AiPlan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AiPlan.js","names":["Internal","TypeId","fromModel","withFallback","concatSteps"],"sources":["../../src/AiPlan.ts"],"sourcesContent":[null],"mappings":"AAWA,OAAO,KAAKA,QAAQ,MAAM,sBAAsB;AAEhD;;;;AAIA,OAAO,MAAMC,MAAM,GAAkBD,QAAQ,CAACC,MAAM;AAmEpD;;;;AAIA,OAAO,MAAMC,SAAS,GAO+BF,QAAQ,CAACE,SAAS;AAEvE;;;;AAIA,OAAO,MAAMC,YAAY,GA4BrBH,QAAQ,CAACG,YAAY;AAEzB;;;;AAIA,OAAO,MAAMC,WAAW,GAgBpBJ,QAAQ,CAACI,WAAW","ignoreList":[]}

package/dist/esm/index.js

@@ -10,6 +10,18 @@ export * as AiError from "./AiError.js";
  * @since 1.0.0
  */
 export * as AiInput from "./AiInput.js";
+/**
+ * @since 1.0.0
+ */
+export * as AiModel from "./AiModel.js";
+/**
+ * @since 1.0.0
+ */
+export * as AiModels from "./AiModels.js";
+/**
+ * @since 1.0.0
+ */
+export * as AiPlan from "./AiPlan.js";
 /**
  * @since 1.0.0
  */

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":["AiChat","AiError","AiInput","AiResponse","AiRole","AiTelemetry","AiToolkit","Completions","Embeddings","Tokenizer"],"sources":["../../src/index.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,MAAM,MAAM,aAAa;AAErC;;;AAGA,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC;;;AAGA,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC;;;AAGA,OAAO,KAAKC,UAAU,MAAM,iBAAiB;AAE7C;;;AAGA,OAAO,KAAKC,MAAM,MAAM,aAAa;AAErC;;;AAGA,OAAO,KAAKC,WAAW,MAAM,kBAAkB;AAE/C;;;AAGA,OAAO,KAAKC,SAAS,MAAM,gBAAgB;AAE3C;;;AAGA,OAAO,KAAKC,WAAW,MAAM,kBAAkB;AAE/C;;;AAGA,OAAO,KAAKC,UAAU,MAAM,iBAAiB;AAE7C;;;AAGA,OAAO,KAAKC,SAAS,MAAM,gBAAgB","ignoreList":[]}
+{"version":3,"file":"index.js","names":["AiChat","AiError","AiInput","AiModel","AiModels","AiPlan","AiResponse","AiRole","AiTelemetry","AiToolkit","Completions","Embeddings","Tokenizer"],"sources":["../../src/index.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,MAAM,MAAM,aAAa;AAErC;;;AAGA,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC;;;AAGA,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC;;;AAGA,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC;;;AAGA,OAAO,KAAKC,QAAQ,MAAM,eAAe;AAEzC;;;AAGA,OAAO,KAAKC,MAAM,MAAM,aAAa;AAErC;;;AAGA,OAAO,KAAKC,UAAU,MAAM,iBAAiB;AAE7C;;;AAGA,OAAO,KAAKC,MAAM,MAAM,aAAa;AAErC;;;AAGA,OAAO,KAAKC,WAAW,MAAM,kBAAkB;AAE/C;;;AAGA,OAAO,KAAKC,SAAS,MAAM,gBAAgB;AAE3C;;;AAGA,OAAO,KAAKC,WAAW,MAAM,kBAAkB;AAE/C;;;AAGA,OAAO,KAAKC,UAAU,MAAM,iBAAiB;AAE7C;;;AAGA,OAAO,KAAKC,SAAS,MAAM,gBAAgB","ignoreList":[]}

package/dist/esm/internal/aiPlan.js

@@ -0,0 +1,82 @@
+import * as Cause from "effect/Cause";
+import * as Context from "effect/Context";
+import * as Effect from "effect/Effect";
+import { CommitPrototype } from "effect/Effectable";
+import * as Exit from "effect/Exit";
+import { dual, identity } from "effect/Function";
+import * as Option from "effect/Option";
+import { pipeArguments } from "effect/Pipeable";
+import * as Predicate from "effect/Predicate";
+import * as Schedule from "effect/Schedule";
+import * as Scope from "effect/Scope";
+import { AiModels } from "../AiModels.js";
+/** @internal */
+export const TypeId = /*#__PURE__*/Symbol.for("@effect/ai/AiPlan");
+/** @internal */
+export const PlanPrototype = {
+  ...CommitPrototype,
+  [TypeId]: TypeId,
+  commit() {
+    return buildPlan(this);
+  },
+  pipe() {
+    return pipeArguments(this, arguments);
+  }
+};
+const makePlan = steps => {
+  const self = Object.create(PlanPrototype);
+  self.steps = steps;
+  return self;
+};
+const buildPlan = plan => Effect.map(Effect.context(), context => {
+  const models = Context.get(context, AiModels);
+  return Effect.fnUntraced(function* (effect) {
+    let exit = undefined;
+    for (const step of plan.steps) {
+      if (exit !== undefined && Exit.isFailure(exit) && Option.isSome(step.check)) {
+        const check = step.check.value(Cause.squash(exit.cause));
+        const isFatalError = !(Effect.isEffect(check) ? yield* check : check);
+        if (isFatalError) break;
+      }
+      const retryOptions = getRetryOptions(step);
+      exit = yield* Effect.scopedWith(scope => models.build(step.model, context).pipe(Scope.extend(scope), Effect.flatMap(context => effect.pipe(Option.isSome(retryOptions) ? Effect.retry(retryOptions.value) : identity, Effect.provide(context))), Effect.exit));
+      if (Exit.isSuccess(exit)) break;
+    }
+    return yield* exit;
+  });
+});
+const getRetryOptions = step => {
+  if (Option.isNone(step.schedule) && Option.isNone(step.check)) {
+    return Option.none();
+  }
+  return Option.some({
+    schedule: Option.getOrUndefined(step.schedule),
+    while: Option.getOrUndefined(step.check)
+  });
+};
+/** @internal */
+export const fromModel = (model, options) => makePlan([{
+  model,
+  check: Option.fromNullable(options?.while),
+  schedule: resolveSchedule(options ?? {})
+}]);
+/** @internal */
+export const withFallback = /*#__PURE__*/dual(2, (self, options) => makePlan([...self.steps, {
+  model: options.model,
+  check: Option.fromNullable(options.while),
+  schedule: resolveSchedule(options)
+}]));
+const resolveSchedule = options => {
+  if (Predicate.isUndefined(options.attempts) && Predicate.isUndefined(options.schedule)) return Option.none();
+  let schedule = options.schedule ?? Schedule.forever;
+  if (Predicate.isNotUndefined(options.attempts)) {
+    // In an `AiPlan`, the `attempts` represents the total number of times to
+    // attempt the call, not the number of retries, thus we subtract one from
+    // the total number of recurrences
+    schedule = Schedule.intersect(schedule, Schedule.recurs(options.attempts - 1));
+  }
+  return Option.some(schedule);
+};
+/** @internal */
+export const concatSteps = /*#__PURE__*/dual(2, (self, other) => makePlan([...self.steps, ...other.steps]));
+//# sourceMappingURL=aiPlan.js.map

package/dist/esm/internal/aiPlan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aiPlan.js","names":["Cause","Context","Effect","CommitPrototype","Exit","dual","identity","Option","pipeArguments","Predicate","Schedule","Scope","AiModels","TypeId","Symbol","for","PlanPrototype","commit","buildPlan","pipe","arguments","makePlan","steps","self","Object","create","plan","map","context","models","get","fnUntraced","effect","exit","undefined","step","isFailure","isSome","check","value","squash","cause","isFatalError","isEffect","retryOptions","getRetryOptions","scopedWith","scope","build","model","extend","flatMap","retry","provide","isSuccess","isNone","schedule","none","some","getOrUndefined","while","fromModel","options","fromNullable","resolveSchedule","withFallback","isUndefined","attempts","forever","isNotUndefined","intersect","recurs","concatSteps","other"],"sources":["../../../src/internal/aiPlan.ts"],"sourcesContent":[null],"mappings":"AAAA,OAAO,KAAKA,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,SAASC,eAAe,QAAQ,mBAAmB;AACnD,OAAO,KAAKC,IAAI,MAAM,aAAa;AACnC,SAASC,IAAI,EAAEC,QAAQ,QAAQ,iBAAiB;AAChD,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,SAASC,aAAa,QAAQ,iBAAiB;AAC/C,OAAO,KAAKC,SAAS,MAAM,kBAAkB;AAC7C,OAAO,KAAKC,QAAQ,MAAM,iBAAiB;AAC3C,OAAO,KAAKC,KAAK,MAAM,cAAc;AAErC,SAASC,QAAQ,QAAQ,gBAAgB;AAGzC;AACA,OAAO,MAAMC,MAAM,gBAAkBC,MAAM,CAACC,GAAG,CAAC,mBAAmB,CAAkB;AAErF;AACA,OAAO,MAAMC,aAAa,GAAG;EAC3B,GAAGb,eAAe;EAClB,CAACU,MAAM,GAAGA,MAAM;EAChBI,MAAMA,CAAA;IACJ,OAAOC,SAAS,CAAC,IAAI,CAAC;EACxB,CAAC;EACDC,IAAIA,CAAA;IACF,OAAOX,aAAa,CAAC,IAAI,EAAEY,SAAS,CAAC;EACvC;CACM;AAER,MAAMC,QAAQ,GAEZC,KAAY,IAAkC;EAC9C,MAAMC,IAAI,GAAGC,MAAM,CAACC,MAAM,CAACT,aAAa,CAAC;EACzCO,IAAI,CAACD,KAAK,GAAGA,KAAK;EAClB,OAAOC,IAAI;AACb,CAAC;AAED,MAAML,SAAS,GACbQ,IAA8C,IAM9CxB,MAAM,CAACyB,GAAG,CAACzB,MAAM,CAAC0B,OAAO,EAAuB,EAAGA,OAAO,IAAI;EAC5D,MAAMC,MAAM,GAAG5B,OAAO,CAAC6B,GAAG,CAACF,OAAO,EAAEhB,QAAQ,CAAC;EAC7C,OAAOV,MAAM,CAAC6B,UAAU,CAAC,WAAmBC,MAA8B;IACxE,IAAIC,IAAI,GAAgCC,SAAS;IACjD,KAAK,MAAMC,IAAI,IAAIT,IAAI,CAACJ,KAAK,EAAE;MAC7B,IAAIW,IAAI,KAAKC,SAAS,IAAI9B,IAAI,CAACgC,SAAS,CAACH,IAAI,CAAC,IAAI1B,MAAM,CAAC8B,MAAM,CAACF,IAAI,CAACG,KAAK,CAAC,EAAE;QAC3E,MAAMA,KAAK,GAAGH,IAAI,CAACG,KAAK,CAACC,KAAK,CAACvC,KAAK,CAACwC,MAAM,CAACP,IAAI,CAACQ,KAAK,CAAU,CAAC;QACjE,MAAMC,YAAY,GAAG,EAAExC,MAAM,CAACyC,QAAQ,CAACL,KAAK,CAAC,GAAG,OAAOA,KAAK,GAAGA,KAAK,CAAC;QACrE,IAAII,YAAY,EAAE;MACpB;MACA,MAAME,YAAY,GAAGC,eAAe,CAACV,IAAI,CAAC;MAC1CF,IAAI,GAAG,OAAO/B,MAAM,CAAC4C,UAAU,CAAEC,KAAK,IACpClB,MAAM,CAACmB,KAAK,CAACb,IAAI,CAACc,KAAK,EAAErB,OAAO,CAAC,CAACT,IAAI,CACpCR,KAAK,CAACuC,MAAM,CAACH,KAAK,CAAC,EACnB7C,MAAM,CAACiD,OAAO,CAAEvB,OAAO,IACrBI,MAAM,CAACb,IAAI,CACTZ,MAAM,CAAC8B,MAAM,CAACO,YAAY,CAAC,GACvB1C,MAAM,CAACkD,KAAK,CAACR,YAAY,CAACL,KAAK,CAAC,GAChCjC,QAAQ,EACZJ,MAAM,CAACmD,OAAO,CAACzB,OAAO,CAAC,CACxB,CACF,EACD1B,MAAM,CAAC+B,IAAI,CACZ,CACF;MACD,IAAI7B,IAAI,CAACkD,SAAS,CAACrB,IAAI,CAAC,EAAE;IAC5B;IACA,OAAO,OAAOA,IAAK;EACrB,CAAC,CAAC;AACJ,CAAC,CAAC;AAEJ,MAAMY,eAAe,GACnBV,IAAmD,IACP;EAC5C,IAAI5B,MAAM,CAACgD,MAAM,CAACpB,IAAI,CAACqB,QAAQ,CAAC,IAAIjD,MAAM,CAACgD,MAAM,CAACpB,IAAI,CAACG,KAAK,CAAC,EAAE;IAC7D,OAAO/B,MAAM,CAACkD,IAAI,EAAE;EACtB;EACA,OAAOlD,MAAM,CAACmD,IAAI,CAAC;IACjBF,QAAQ,EAAEjD,MAAM,CAACoD,cAAc,CAACxB,IAAI,CAACqB,QAAQ,CAAC;IAC9CI,KAAK,EAAErD,MAAM,CAACoD,cAAc,CAACxB,IAAI,CAACG,KAAK;GACxC,CAAC;AACJ,CAAC;AAED;AACA,OAAO,MAAMuB,SAAS,GAAGA,CACvBZ,KAA0C,EAC1Ca,OAIC,KAEDzC,QAAQ,CAAC,CAAC;EACR4B,KAAK;EACLX,KAAK,EAAE/B,MAAM,CAACwD,YAAY,CAACD,OAAO,EAAEF,KAAK,CAAQ;EACjDJ,QAAQ,EAAEQ,eAAe,CAACF,OAAO,IAAI,EAAE;CACxC,CAAC,CAAC;AAEL;AACA,OAAO,MAAMG,YAAY,gBAAG5D,IAAI,CAAC,CAAC,EAAE,CAYlCkB,IAA0C,EAC1CuC,OAKC,KAEDzC,QAAQ,CAAC,CACP,GAAGE,IAAI,CAACD,KAAK,EACb;EACE2B,KAAK,EAAEa,OAAO,CAACb,KAAK;EACpBX,KAAK,EAAE/B,MAAM,CAACwD,YAAY,CAACD,OAAO,CAACF,KAAK,CAAQ;EAChDJ,QAAQ,EAAEQ,eAAe,CAACF,OAAO;CAClC,CACF,CAAC,CAAC;AAEL,MAAME,eAAe,GAAmBF,OAGvC,IAAsD;EACrD,IACErD,SAAS,CAACyD,WAAW,CAACJ,OAAO,CAACK,QAAQ,CAAC,IACvC1D,SAAS,CAACyD,WAAW,CAACJ,OAAO,CAACN,QAAQ,CAAC,EACvC,OAAOjD,MAAM,CAACkD,IAAI,EAAE;EACtB,IAAID,QAAQ,GAAIM,OAAO,CAACN,QAAQ,IAAI9C,QAAQ,CAAC0D,OAA6C;EAC1F,IAAI3D,SAAS,CAAC4D,cAAc,CAACP,OAAO,CAACK,QAAQ,CAAC,EAAE;IAC9C;IACA;IACA;IACAX,QAAQ,GAAG9C,QAAQ,CAAC4D,SAAS,CAACd,QAAQ,EAAE9C,QAAQ,CAAC6D,MAAM,CAACT,OAAO,CAACK,QAAQ,GAAG,CAAC,CAAC,CAAC;EAChF;EACA,OAAO5D,MAAM,CAACmD,IAAI,CAACF,QAAQ,CAAC;AAC9B,CAAC;AAED;AACA,OAAO,MAAMgB,WAAW,gBAAGnE,IAAI,CAU7B,CAAC,EAAE,CAACkB,IAAI,EAAEkD,KAAK,KAAKpD,QAAQ,CAAC,CAAC,GAAGE,IAAI,CAACD,KAAK,EAAE,GAAGmD,KAAK,CAACnD,KAAK,CAAC,CAAC,CAAC","ignoreList":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@effect/ai",
-  "version": "0.10.2",
+  "version": "0.10.4",
   "description": "Effect modules for working with AI apis",
   "license": "MIT",
   "repository": {
@@ -11,9 +11,9 @@
   "sideEffects": [],
   "dependencies": {},
   "peerDependencies": {
-    "@effect/experimental": "^0.41.2",
-    "@effect/platform": "^0.77.2",
-    "effect": "^3.13.2"
+    "@effect/experimental": "^0.41.4",
+    "effect": "^3.13.4",
+    "@effect/platform": "^0.77.4"
   },
   "publishConfig": {
     "provenance": true
@@ -43,6 +43,21 @@
       "import": "./dist/esm/AiInput.js",
       "default": "./dist/cjs/AiInput.js"
     },
+    "./AiModel": {
+      "types": "./dist/dts/AiModel.d.ts",
+      "import": "./dist/esm/AiModel.js",
+      "default": "./dist/cjs/AiModel.js"
+    },
+    "./AiModels": {
+      "types": "./dist/dts/AiModels.d.ts",
+      "import": "./dist/esm/AiModels.js",
+      "default": "./dist/cjs/AiModels.js"
+    },
+    "./AiPlan": {
+      "types": "./dist/dts/AiPlan.d.ts",
+      "import": "./dist/esm/AiPlan.js",
+      "default": "./dist/cjs/AiPlan.js"
+    },
     "./AiResponse": {
       "types": "./dist/dts/AiResponse.d.ts",
       "import": "./dist/esm/AiResponse.js",
@@ -90,6 +105,15 @@
       "AiInput": [
         "./dist/dts/AiInput.d.ts"
       ],
+      "AiModel": [
+        "./dist/dts/AiModel.d.ts"
+      ],
+      "AiModels": [
+        "./dist/dts/AiModels.d.ts"
+      ],
+      "AiPlan": [
+        "./dist/dts/AiPlan.d.ts"
+      ],
       "AiResponse": [
         "./dist/dts/AiResponse.d.ts"
       ],

package/src/AiModel.ts

@@ -0,0 +1,91 @@
+/**
+ * @since 1.0.0
+ */
+import type * as Context from "effect/Context"
+import type * as Effect from "effect/Effect"
+import * as Option from "effect/Option"
+import type * as Scope from "effect/Scope"
+import type * as AiPlan from "./AiPlan.js"
+import * as InternalAiPlan from "./internal/aiPlan.js"
+
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export const TypeId: unique symbol = Symbol.for("@effect/ai/AiModel")
+
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export type TypeId = typeof TypeId
+
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export const PlanTypeId: unique symbol = Symbol.for("@effect/ai/Plan")
+
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export type PlanTypeId = typeof TypeId
+
+/**
+ * @since 1.0.0
+ * @category models
+ */
+export interface AiModel<in out Provides, in out Requires> extends AiPlan.AiPlan<unknown, Provides, Requires> {
+  readonly [TypeId]: TypeId
+  readonly model: string
+  readonly cacheKey: symbol
+  readonly requires: Context.Tag<Requires, any>
+  readonly provides: AiModel.ContextBuilder<Provides, Requires>
+  readonly updateContext: (context: Context.Context<Provides>) => Context.Context<Provides>
+}
+
+/**
+ * @since 1.0.0
+ */
+export declare namespace AiModel {
+  /**
+   * @since 1.0.0
+   * @category AiModel
+   */
+  export type ContextBuilder<Provides, Requires> = Effect.Effect<
+    Context.Context<Provides>,
+    never,
+    Requires | Scope.Scope
+  >
+}
+
+const AiModelProto = {
+  ...InternalAiPlan.PlanPrototype,
+  [TypeId]: TypeId
+}
+
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const make = <Provides, Requires>(options: {
+  readonly model: string
+  readonly cacheKey: symbol
+  readonly requires: Context.Tag<Requires, any>
+  readonly provides: AiModel.ContextBuilder<Provides, Requires>
+  readonly updateContext: (context: Context.Context<Provides>) => Context.Context<Provides>
+}): AiModel<Provides, Requires> => {
+  const self = Object.create(AiModelProto)
+  self.cacheKey = options.cacheKey
+  self.model = options.model
+  self.provides = options.provides
+  self.requires = options.requires
+  self.updateContext = options.updateContext
+  self.steps = [{
+    model: self,
+    check: Option.none(),
+    schedule: Option.none()
+  }]
+  return self
+}

package/src/AiModels.ts

@@ -0,0 +1,77 @@
+/**
+ * @since 1.0.0
+ */
+import * as Context from "effect/Context"
+import * as Effect from "effect/Effect"
+import * as Equal from "effect/Equal"
+import * as Hash from "effect/Hash"
+import * as Layer from "effect/Layer"
+import * as RcMap from "effect/RcMap"
+import type * as Scope from "effect/Scope"
+import type { AiModel } from "./AiModel.js"
+
+/**
+ * @since 1.0.0
+ * @category tags
+ */
+export class AiModels extends Context.Tag("@effect/ai/AiModels")<
+  AiModels,
+  AiModels.Service
+>() {}
+
+/**
+ * @since 1.0.0
+ */
+export declare namespace AiModels {
+  /**
+   * @since 1.0.0
+   * @category models
+   */
+  export interface Service {
+    readonly build: <Provides, Requires>(
+      model: AiModel<Provides, Requires>,
+      context: Context.Context<Requires>
+    ) => Effect.Effect<Context.Context<Provides>, never, Scope.Scope>
+  }
+}
+
+class AiModelsKey {
+  constructor(
+    readonly model: AiModel<any, any>,
+    readonly service: unknown
+  ) {}
+
+  [Equal.symbol](that: AiModelsKey): boolean {
+    return this.service === that.service && this.model.cacheKey === that.model.cacheKey
+  }
+  [Hash.symbol](): number {
+    return Hash.combine(Hash.hash(this.service))(Hash.hash(this.model.cacheKey))
+  }
+}
+
+const make = Effect.gen(function*() {
+  const services = yield* RcMap.make({
+    idleTimeToLive: "1 minute",
+    lookup: (key: AiModelsKey) => Effect.provideService(key.model.provides, key.model.requires, key.service)
+  })
+
+  const build = <Provides, Requires>(
+    model: AiModel<Provides, Requires>,
+    context: Context.Context<Requires>
+  ): Effect.Effect<Context.Context<Provides>, never, Scope.Scope> =>
+    Effect.map(
+      RcMap.get(
+        services,
+        new AiModelsKey(model, Context.get(context, model.requires as any))
+      ),
+      (context) => model.updateContext(context)
+    )
+
+  return { build } as const
+})
+
+/**
+ * @since 1.0.0
+ * @category layers
+ */
+export const layer: Layer.Layer<AiModels> = Layer.scoped(AiModels, make)

package/src/AiPlan.ts

@@ -0,0 +1,152 @@
+/**
+ * @since 1.0.0
+ */
+import type { NonEmptyReadonlyArray } from "effect/Array"
+import type * as Effect from "effect/Effect"
+import type * as Option from "effect/Option"
+import type { Pipeable } from "effect/Pipeable"
+import type * as Schedule from "effect/Schedule"
+import type * as Unify from "effect/Unify"
+import type * as AiModel from "./AiModel.js"
+import type { AiModels } from "./AiModels.js"
+import * as Internal from "./internal/aiPlan.js"
+
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export const TypeId: unique symbol = Internal.TypeId
+
+/**
+ * @since 1.0.0
+ * @category type ids
+ */
+export type TypeId = typeof TypeId
+
+/**
+ * @since 1.0.0
+ * @category AiPlan
+ */
+export interface AiPlan<in Error, in out Provides, in out Requires>
+  extends Pipeable, AiPlan.Builder<Provides, Requires>
+{
+  readonly [TypeId]: TypeId
+  readonly [Unify.typeSymbol]?: unknown
+  readonly [Unify.unifySymbol]?: AiPlanUnify<this>
+  readonly [Unify.ignoreSymbol]?: AiPlanUnifyIgnore
+  readonly steps: NonEmptyReadonlyArray<AiPlan.Step<Error, Provides, Requires>>
+}
+
+/**
+ * @since 1.0.0
+ * @category Plan
+ */
+export interface AiPlanUnify<A extends { [Unify.typeSymbol]?: any }> extends Effect.EffectUnify<A> {
+  AiPlan?: () => Extract<A[Unify.typeSymbol], AiPlan<any, any, any>>
+}
+
+/**
+ * @since 1.0.0
+ * @category Plan
+ */
+export interface AiPlanUnifyIgnore extends Effect.EffectUnifyIgnore {
+  Effect?: true
+}
+
+/**
+ * @since 1.0.0
+ */
+export declare namespace AiPlan {
+  /**
+   * @since 1.0.0
+   * @category AiPlan
+   */
+  export type Builder<Provides, Requires> = Effect.Effect<Provider<Provides>, never, AiModels | Requires>
+
+  /**
+   * @since 1.0.0
+   * @category AiPlan
+   */
+  export interface Provider<Provides> {
+    readonly provide: <A, E, R>(effect: Effect.Effect<A, E, R>) => Effect.Effect<A, E, Exclude<R, Provides>>
+  }
+
+  /**
+   * @since 1.0.0
+   * @category AiPlan
+   */
+  export interface Step<Error, Provides, Requires> {
+    readonly model: AiModel.AiModel<Provides, Requires>
+    readonly check: Option.Option<(error: Error) => boolean | Effect.Effect<boolean>>
+    readonly schedule: Option.Option<Schedule.Schedule<any, Error, Requires>>
+  }
+}
+
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const fromModel: <Provides, Requires, EW, Out, ES, RW = never, RS = never>(
+  model: AiModel.AiModel<Provides, Requires>,
+  options?: {
+    readonly attempts?: number | undefined
+    readonly while?: ((error: EW) => boolean | Effect.Effect<boolean, never, RW>) | undefined
+    readonly schedule?: Schedule.Schedule<Out, ES, RS> | undefined
+  }
+) => AiPlan<EW & ES, Provides, RW | RS | Requires> = Internal.fromModel
+
+/**
+ * @since 1.0.0
+ * @category error handling
+ */
+export const withFallback: {
+  /**
+   * @since 1.0.0
+   * @category error handling
+   */
+  <Provides, Provides2 extends Provides, Requires2, Out, EW, ES, RW = never, RS = never>(
+    options: {
+      readonly model: AiModel.AiModel<Provides2, Requires2>
+      readonly attempts?: number | undefined
+      readonly while?: ((error: EW) => boolean | Effect.Effect<boolean, never, RW>) | undefined
+      readonly schedule?: Schedule.Schedule<Out, ES, RS> | undefined
+    }
+  ): <E, Requires>(
+    self: AiPlan<E, Provides, Requires>
+  ) => AiPlan<E & EW & ES, Provides & Provides2, Requires | Requires2 | RW | RS>
+  /**
+   * @since 1.0.0
+   * @category error handling
+   */
+  <E, Provides, Requires, Provides2 extends Provides, Requires2, Out, EW, ES, RW = never, RS = never>(
+    self: AiPlan<E, Provides, Requires>,
+    options: {
+      readonly model: AiModel.AiModel<Provides2, Requires2>
+      readonly attempts?: number | undefined
+      readonly while?: ((error: EW) => boolean | Effect.Effect<boolean, never, RW>) | undefined
+      readonly schedule?: Schedule.Schedule<Out, ES, RS> | undefined
+    }
+  ): AiPlan<E & EW & ES, Provides & Provides2, Requires | Requires2 | RW | RS>
+} = Internal.withFallback
+
+/**
+ * @since 1.0.0
+ * @category combination
+ */
+export const concatSteps: {
+  /**
+   * @since 1.0.0
+   * @category combination
+   */
+  <Error2, Provides2, Requires2>(other: AiPlan<Error2, Provides2, Requires2>): <Error, Provides, Requires>(
+    self: AiPlan<Error, Provides, Requires>
+  ) => AiPlan<Error & Error2, Provides & Provides2, Requires | Requires2>
+  /**
+   * @since 1.0.0
+   * @category combination
+   */
+  <Error, Provides, Requires, Error2, Provides2, Requires2>(
+    self: AiPlan<Error, Provides, Requires>,
+    other: AiPlan<Error2, Provides2, Requires2>
+  ): AiPlan<Error & Error2, Provides & Provides2, Requires | Requires2>
+} = Internal.concatSteps

package/src/index.ts

@@ -13,6 +13,21 @@ export * as AiError from "./AiError.js"
  */
 export * as AiInput from "./AiInput.js"
 
+/**
+ * @since 1.0.0
+ */
+export * as AiModel from "./AiModel.js"
+
+/**
+ * @since 1.0.0
+ */
+export * as AiModels from "./AiModels.js"
+
+/**
+ * @since 1.0.0
+ */
+export * as AiPlan from "./AiPlan.js"
+
 /**
  * @since 1.0.0
  */