@dxos/functions 0.8.4-main.72ec0f3 → 0.8.4-main.8342792842

package/dist/lib/{browser → neutral}/index.mjs

@@ -6,7 +6,7 @@ var __export = (target, all) => {
 
 // src/errors.ts
 import { BaseError } from "@dxos/errors";
-var ServiceNotAvailableError = class extends BaseError.extend("SERVICE_NOT_AVAILABLE", "Service not available") {
+var ServiceNotAvailableError = class extends BaseError.extend("ServiceNotAvailable", "Service not available") {
   constructor(service, options) {
     super({
       context: {
@@ -16,7 +16,7 @@ var ServiceNotAvailableError = class extends BaseError.extend("SERVICE_NOT_AVAIL
     });
   }
 };
-var FunctionNotFoundError = class extends BaseError.extend("FUNCTION_NOT_FOUND", "Function not found") {
+var FunctionNotFoundError = class extends BaseError.extend("FunctionNotFound", "Function not found") {
   constructor(functionKey, options) {
     super({
       context: {
@@ -26,9 +26,9 @@ var FunctionNotFoundError = class extends BaseError.extend("FUNCTION_NOT_FOUND",
     });
   }
 };
-var FunctionError = class extends BaseError.extend("FUNCTION_ERROR", "Function invocation error") {
+var FunctionError = class extends BaseError.extend("FunctionError", "Function invocation error") {
 };
-var TriggerStateNotFoundError = class extends BaseError.extend("TRIGGER_STATE_NOT_FOUND", "Trigger state not found") {
+var TriggerStateNotFoundError = class extends BaseError.extend("TriggerStateNotFound", "Trigger state not found") {
 };
 
 // src/example/fib.ts
@@ -38,18 +38,20 @@ import * as Schema6 from "effect/Schema";
 // src/sdk.ts
 import * as Effect from "effect/Effect";
 import * as Schema5 from "effect/Schema";
-import { Obj as Obj5, Type as Type5 } from "@dxos/echo";
+import { Obj as Obj4, Type as Type5 } from "@dxos/echo";
 import { assertArgument, failedInvariant } from "@dxos/invariant";
+import { Operation } from "@dxos/operation";
 
 // src/types/Function.ts
 var Function_exports = {};
 __export(Function_exports, {
   Function: () => Function,
-  make: () => make2
+  make: () => make2,
+  setFrom: () => setFrom
 });
 import * as Schema2 from "effect/Schema";
-import { Obj as Obj2, Type as Type2 } from "@dxos/echo";
-import { JsonSchemaType, LabelAnnotation as LabelAnnotation2, Ref as Ref2 } from "@dxos/echo/internal";
+import { Annotation as Annotation2, JsonSchema, Obj as Obj2, Type as Type2 } from "@dxos/echo";
+import { SystemTypeAnnotation } from "@dxos/echo/internal";
 
 // src/types/Script.ts
 var Script_exports = {};
@@ -58,20 +60,20 @@ __export(Script_exports, {
   make: () => make
 });
 import * as Schema from "effect/Schema";
-import { Obj, Ref, Type } from "@dxos/echo";
-import { FormAnnotation, LabelAnnotation } from "@dxos/echo/internal";
+import { Annotation, Obj, Ref, Type } from "@dxos/echo";
+import { FormInputAnnotation } from "@dxos/echo/internal";
 import { Text } from "@dxos/schema";
 var Script = Schema.Struct({
   name: Schema.String.pipe(Schema.optional),
   description: Schema.String.pipe(Schema.optional),
   // TODO(burdon): Change to hash of deployed content.
   // Whether source has changed since last deploy.
-  changed: Schema.Boolean.pipe(FormAnnotation.set(false), Schema.optional),
-  source: Type.Ref(Text.Text).pipe(FormAnnotation.set(false))
-}).pipe(Type.Obj({
+  changed: Schema.Boolean.pipe(FormInputAnnotation.set(false), Schema.optional),
+  source: Type.Ref(Text.Text).pipe(FormInputAnnotation.set(false))
+}).pipe(Type.object({
   typename: "dxos.org/type/Script",
   version: "0.1.0"
-}), LabelAnnotation.set([
+}), Annotation.LabelAnnotation.set([
   "name"
 ]));
 var make = ({ source = "", ...props } = {}) => Obj.make(Script, {
@@ -91,7 +93,6 @@ var Function = Schema2.Struct({
   key: Schema2.optional(Schema2.String).annotations({
     description: "Unique registration key for the blueprint"
   }),
-  // TODO(burdon): Rename to id/uri?
   name: Schema2.NonEmptyString,
   version: Schema2.String,
   description: Schema2.optional(Schema2.String),
@@ -101,9 +102,9 @@ var Function = Schema2.Struct({
   updated: Schema2.optional(Schema2.String),
   // Reference to a source script if it exists within ECHO.
   // TODO(burdon): Don't ref ScriptType directly (core).
-  source: Schema2.optional(Ref2(Script)),
-  inputSchema: Schema2.optional(JsonSchemaType),
-  outputSchema: Schema2.optional(JsonSchemaType),
+  source: Schema2.optional(Type2.Ref(Script)),
+  inputSchema: Schema2.optional(JsonSchema.JsonSchema),
+  outputSchema: Schema2.optional(JsonSchema.JsonSchema),
   /**
   * List of required services.
   * Match the Context.Tag keys of the FunctionServices variants.
@@ -111,13 +112,25 @@ var Function = Schema2.Struct({
   services: Schema2.optional(Schema2.Array(Schema2.String)),
   // Local binding to a function name.
   binding: Schema2.optional(Schema2.String)
-}).pipe(Type2.Obj({
+}).pipe(Type2.object({
   typename: "dxos.org/type/Function",
   version: "0.1.0"
-}), LabelAnnotation2.set([
+}), Annotation2.LabelAnnotation.set([
   "name"
-]));
+]), SystemTypeAnnotation.set(true));
 var make2 = (props) => Obj2.make(Function, props);
+var setFrom = (target, source) => {
+  Obj2.change(target, (t) => {
+    t.key = source.key ?? target.key;
+    t.name = source.name ?? target.name;
+    t.version = source.version;
+    t.description = source.description;
+    t.updated = source.updated;
+    t.inputSchema = source.inputSchema ? JSON.parse(JSON.stringify(source.inputSchema)) : void 0;
+    t.outputSchema = source.outputSchema ? JSON.parse(JSON.stringify(source.outputSchema)) : void 0;
+    Obj2.getMeta(t).keys = JSON.parse(JSON.stringify(Obj2.getMeta(source).keys));
+  });
+};
 
 // src/types/Trigger.ts
 var Trigger_exports = {};
@@ -135,8 +148,9 @@ __export(Trigger_exports, {
 import * as Schema3 from "effect/Schema";
 import * as SchemaAST from "effect/SchemaAST";
 import { Obj as Obj3, QueryAST, Type as Type3 } from "@dxos/echo";
-import { Expando, OptionsAnnotationId, Ref as Ref3 } from "@dxos/echo/internal";
+import { OptionsAnnotationId, SystemTypeAnnotation as SystemTypeAnnotation2 } from "@dxos/echo/internal";
 import { DXN } from "@dxos/keys";
+import { Expando } from "@dxos/schema";
 var Kinds = [
   "email",
   "queue",
@@ -149,12 +163,12 @@ var kindLiteralAnnotations = {
 };
 var EmailSpec = Schema3.Struct({
   kind: Schema3.Literal("email").annotations(kindLiteralAnnotations)
-}).pipe(Schema3.mutable);
+});
 var QueueSpec = Schema3.Struct({
   kind: Schema3.Literal("queue").annotations(kindLiteralAnnotations),
   // TODO(dmaretskyi): Change to a reference.
   queue: DXN.Schema
-}).pipe(Schema3.mutable);
+});
 var SubscriptionSpec = Schema3.Struct({
   kind: Schema3.Literal("subscription").annotations(kindLiteralAnnotations),
   query: Schema3.Struct({
@@ -162,7 +176,7 @@ var SubscriptionSpec = Schema3.Struct({
       title: "Query"
     })),
     ast: QueryAST.Query
-  }).pipe(Schema3.mutable),
+  }),
   options: Schema3.optional(Schema3.Struct({
     // Watch changes to object (not just creation).
     deep: Schema3.optional(Schema3.Boolean.annotations({
@@ -175,7 +189,7 @@ var SubscriptionSpec = Schema3.Struct({
   }).annotations({
     title: "Options"
   }))
-}).pipe(Schema3.mutable);
+});
 var TimerSpec = Schema3.Struct({
   kind: Schema3.Literal("timer").annotations(kindLiteralAnnotations),
   cron: Schema3.String.annotations({
@@ -184,7 +198,7 @@ var TimerSpec = Schema3.Struct({
       "0 0 * * *"
     ]
   })
-}).pipe(Schema3.mutable);
+});
 var WebhookSpec = Schema3.Struct({
   kind: Schema3.Literal("webhook").annotations(kindLiteralAnnotations),
   method: Schema3.optional(Schema3.String.annotations({
@@ -197,16 +211,16 @@ var WebhookSpec = Schema3.Struct({
   port: Schema3.optional(Schema3.Number.annotations({
     title: "Port"
   }))
-}).pipe(Schema3.mutable);
+});
 var Spec = Schema3.Union(EmailSpec, QueueSpec, SubscriptionSpec, TimerSpec, WebhookSpec).annotations({
   title: "Trigger"
 });
-var Trigger_ = Schema3.Struct({
+var TriggerSchema = Schema3.Struct({
   /**
   * Function or workflow to invoke.
   */
   // TODO(dmaretskyi): Can be a Ref(FunctionType) or Ref(ComputeGraphType).
-  function: Schema3.optional(Ref3(Expando).annotations({
+  function: Schema3.optional(Type3.Ref(Expando.Expando).annotations({
     title: "Function"
   })),
   /**
@@ -217,6 +231,8 @@ var Trigger_ = Schema3.Struct({
   inputNodeId: Schema3.optional(Schema3.String.annotations({
     title: "Input Node ID"
   })),
+  // TODO(burdon): NO BOOLEAN PROPERTIES (enabld/disabled/paused, etc.)
+  //  Need lint rule; or agent rule to require PR review for "boolean" key word.
   enabled: Schema3.optional(Schema3.Boolean.annotations({
     title: "Enabled"
   })),
@@ -232,15 +248,15 @@ var Trigger_ = Schema3.Struct({
   *   mailbox: { '/': 'dxn:echo:AAA:ZZZ' }
   * }
   */
-  input: Schema3.optional(Schema3.mutable(Schema3.Record({
+  input: Schema3.optional(Schema3.Record({
     key: Schema3.String,
     value: Schema3.Any
-  })))
-}).pipe(Type3.Obj({
+  }))
+}).pipe(Type3.object({
   typename: "dxos.org/type/Trigger",
   version: "0.1.0"
-}));
-var Trigger = Trigger_;
+}), SystemTypeAnnotation2.set(true));
+var Trigger = TriggerSchema;
 var make3 = (props) => Obj3.make(Trigger, props);
 
 // src/types/TriggerEvent.ts
@@ -250,22 +266,23 @@ __export(TriggerEvent_exports, {
   QueueEvent: () => QueueEvent,
   SubscriptionEvent: () => SubscriptionEvent,
   TimerEvent: () => TimerEvent,
+  TriggerEvent: () => TriggerEvent,
   WebhookEvent: () => WebhookEvent
 });
 import * as Schema4 from "effect/Schema";
-import { DXN as DXN2, Obj as Obj4, Type as Type4 } from "@dxos/echo";
-var EmailEvent = Schema4.mutable(Schema4.Struct({
+import { DXN as DXN2, Type as Type4 } from "@dxos/echo";
+var EmailEvent = Schema4.Struct({
   from: Schema4.String,
   to: Schema4.String,
   subject: Schema4.String,
   created: Schema4.String,
   body: Schema4.String
-}));
-var QueueEvent = Schema4.mutable(Schema4.Struct({
+});
+var QueueEvent = Schema4.Struct({
   queue: DXN2.Schema,
   item: Schema4.Any,
   cursor: Schema4.String
-}));
+});
 var SubscriptionEvent = Schema4.Struct({
   /**
   * Type of the mutation.
@@ -275,16 +292,16 @@ var SubscriptionEvent = Schema4.Struct({
   /**
   * Reference to the object that was changed or created.
   */
-  subject: Type4.Ref(Obj4.Any),
+  subject: Type4.Ref(Type4.Obj),
   /**
   * @deprecated
   */
   changedObjectId: Schema4.optional(Schema4.String)
-}).pipe(Schema4.mutable);
-var TimerEvent = Schema4.mutable(Schema4.Struct({
+});
+var TimerEvent = Schema4.Struct({
   tick: Schema4.Number
-}));
-var WebhookEvent = Schema4.mutable(Schema4.Struct({
+});
+var WebhookEvent = Schema4.Struct({
   url: Schema4.String,
   method: Schema4.Literal("GET", "POST"),
   headers: Schema4.Record({
@@ -292,7 +309,8 @@ var WebhookEvent = Schema4.mutable(Schema4.Struct({
     value: Schema4.String
   }),
   bodyText: Schema4.String
-}));
+});
+var TriggerEvent = Schema4.Union(EmailEvent, QueueEvent, SubscriptionEvent, TimerEvent, WebhookEvent);
 
 // src/types/url.ts
 var FUNCTIONS_META_KEY = "dxos.org/service/function";
@@ -316,7 +334,7 @@ var setUserFunctionIdInMetadata = (meta, functionId) => {
 
 // src/sdk.ts
 var typeId = Symbol.for("@dxos/functions/FunctionDefinition");
-var defineFunction = ({ key, name, description, inputSchema, outputSchema = Schema5.Any, handler, services }) => {
+var defineFunction = ({ key, name, description, inputSchema, outputSchema = Schema5.Any, handler, types, services }) => {
   if (!Schema5.isSchema(inputSchema)) {
     throw new Error("Input schema must be a valid schema");
   }
@@ -357,6 +375,7 @@ var defineFunction = ({ key, name, description, inputSchema, outputSchema = Sche
     inputSchema,
     outputSchema,
     handler: handlerWithSpan,
+    types: types ?? [],
     services: !services ? [] : getServiceKeys(services)
   };
 };
@@ -365,10 +384,39 @@ var getServiceKeys = (services) => {
     if (typeof tag.key === "string") {
       return tag.key;
     }
-    console.log(tag);
     failedInvariant();
   });
 };
+var toOperation = (functionDef) => {
+  const op = Operation.make({
+    schema: {
+      input: functionDef.inputSchema,
+      output: functionDef.outputSchema ?? Schema5.Any
+    },
+    meta: {
+      key: functionDef.key,
+      name: functionDef.name,
+      description: functionDef.description
+    }
+  });
+  const operationHandler = (input) => {
+    const result = functionDef.handler({
+      context: {},
+      data: input
+    });
+    if (Effect.isEffect(result)) {
+      return result;
+    }
+    if (result instanceof Promise) {
+      return Effect.tryPromise(() => result);
+    }
+    return Effect.succeed(result);
+  };
+  return {
+    ...op,
+    handler: operationHandler
+  };
+};
 var FunctionDefinition = {
   make: defineFunction,
   isFunction: (value2) => {
@@ -379,9 +427,10 @@ var FunctionDefinition = {
     return serializeFunction(functionDef);
   },
   deserialize: (functionObj) => {
-    assertArgument(Obj5.instanceOf(Function_exports.Function, functionObj), "functionObj");
+    assertArgument(Obj4.instanceOf(Function_exports.Function, functionObj), "functionObj");
     return deserializeFunction(functionObj);
-  }
+  },
+  toOperation
 };
 var serializeFunction = (functionDef) => {
   const fn4 = Function_exports.make({
@@ -394,7 +443,7 @@ var serializeFunction = (functionDef) => {
     services: functionDef.services
   });
   if (functionDef.meta?.deployedFunctionId) {
-    setUserFunctionIdInMetadata(Obj5.getMeta(fn4), functionDef.meta.deployedFunctionId);
+    Obj4.change(fn4, (fn5) => setUserFunctionIdInMetadata(Obj4.getMeta(fn5), functionDef.meta.deployedFunctionId));
   }
   return fn4;
 };
@@ -411,8 +460,9 @@ var deserializeFunction = (functionObj) => {
     handler: () => {
     },
     services: functionObj.services ?? [],
+    types: [],
     meta: {
-      deployedFunctionId: getUserFunctionIdInMetadata(Obj5.getMeta(functionObj))
+      deployedFunctionId: getUserFunctionIdInMetadata(Obj4.getMeta(functionObj))
     }
   };
 };
@@ -489,9 +539,6 @@ var sleep_default = defineFunction({
 })(Example || (Example = {}));
 var Example;
 
-// src/services/index.ts
-import { DatabaseService as DatabaseService2 } from "@dxos/echo-db";
-
 // src/services/credentials.ts
 import * as HttpClient from "@effect/platform/HttpClient";
 import * as HttpClientRequest from "@effect/platform/HttpClientRequest";
@@ -500,7 +547,7 @@ import * as Effect5 from "effect/Effect";
 import * as Layer from "effect/Layer";
 import * as Redacted from "effect/Redacted";
 import { Query } from "@dxos/echo";
-import { DatabaseService } from "@dxos/echo-db";
+import { Database } from "@dxos/echo";
 import { AccessToken } from "@dxos/types";
 var CredentialsService = class _CredentialsService extends Context.Tag("@dxos/functions/CredentialsService")() {
   static getCredential = (query) => Effect5.gen(function* () {
@@ -524,14 +571,23 @@ var CredentialsService = class _CredentialsService extends Context.Tag("@dxos/fu
     }));
     return new ConfiguredCredentialsService(serviceCredentials);
   }));
-  static layerFromDatabase = () => Layer.effect(_CredentialsService, Effect5.gen(function* () {
-    const dbService = yield* DatabaseService;
+  static layerFromDatabase = ({ caching = false } = {}) => Layer.effect(_CredentialsService, Effect5.gen(function* () {
+    const dbService = yield* Database.Service;
+    const cache = /* @__PURE__ */ new Map();
     const queryCredentials = async (query) => {
-      const { objects: accessTokens } = await dbService.db.query(Query.type(AccessToken.AccessToken)).run();
-      return accessTokens.filter((accessToken) => accessToken.source === query.service).map((accessToken) => ({
+      const cacheKey = JSON.stringify(query);
+      if (caching && cache.has(cacheKey)) {
+        return cache.get(cacheKey);
+      }
+      const accessTokens = await dbService.db.query(Query.type(AccessToken.AccessToken)).run();
+      const credentials = accessTokens.filter((accessToken) => accessToken.source === query.service).map((accessToken) => ({
         service: accessToken.source,
         apiKey: accessToken.token
       }));
+      if (caching) {
+        cache.set(cacheKey, credentials);
+      }
+      return credentials;
     };
     return {
       getCredential: async (query) => {
@@ -567,18 +623,17 @@ var ConfiguredCredentialsService = class {
     return credential;
   }
 };
-var withAuthorization = (query, kind) => HttpClient.mapRequestEffect(Effect5.fnUntraced(function* (request) {
-  const key = yield* CredentialsService.getApiKey(query).pipe(Effect5.map(Redacted.value));
-  const authorization = kind ? `${kind} ${key}` : key;
+var withAuthorization = (token, kind) => HttpClient.mapRequest((request) => {
+  const authorization = kind ? `${kind} ${token}` : token;
   return HttpClientRequest.setHeader(request, "Authorization", authorization);
-}));
+});
 
 // src/services/event-logger.ts
 import * as Context3 from "effect/Context";
 import * as Effect7 from "effect/Effect";
 import * as Layer3 from "effect/Layer";
 import * as Schema9 from "effect/Schema";
-import { Obj as Obj7, Type as Type6 } from "@dxos/echo";
+import { Obj as Obj6, Type as Type6 } from "@dxos/echo";
 import { invariant } from "@dxos/invariant";
 import { LogLevel, log as log2 } from "@dxos/log";
 
@@ -587,13 +642,20 @@ import * as Context2 from "effect/Context";
 import * as Effect6 from "effect/Effect";
 import * as Layer2 from "effect/Layer";
 import { AgentStatus } from "@dxos/ai";
-import { Obj as Obj6 } from "@dxos/echo";
+import { Obj as Obj5 } from "@dxos/echo";
+import { ObjectId } from "@dxos/keys";
 import { Message } from "@dxos/types";
 var TracingService = class _TracingService extends Context2.Tag("@dxos/functions/TracingService")() {
   static noop = {
     getTraceContext: () => ({}),
     write: () => {
-    }
+    },
+    traceInvocationStart: () => Effect6.sync(() => ({
+      invocationId: ObjectId.random(),
+      invocationTraceQueue: void 0
+    })),
+    traceInvocationEnd: () => Effect6.sync(() => {
+    })
   };
   static layerNoop = Layer2.succeed(_TracingService, _TracingService.noop);
   /**
@@ -603,32 +665,43 @@ var TracingService = class _TracingService extends Context2.Tag("@dxos/functions
     const tracing = yield* _TracingService;
     const context = mapContext(tracing.getTraceContext());
     return {
-      write: (event) => tracing.write(event),
-      getTraceContext: () => context
+      write: (event, context2) => tracing.write(event, context2),
+      getTraceContext: () => context,
+      traceInvocationStart: () => Effect6.die("Tracing invocation inside another invocation is not supported."),
+      traceInvocationEnd: () => Effect6.die("Tracing invocation inside another invocation is not supported.")
     };
   }));
   /**
+  * Create sublayer to trace an invocation.
+  * @param data
+  * @returns
+  */
+  static layerInvocation = (data) => _TracingService.layerSubframe((context) => ({
+    ...context,
+    currentInvocation: data
+  }));
+  /**
   * Emit the current human-readable execution status.
   */
   static emitStatus = Effect6.fnUntraced(function* (data) {
     const tracing = yield* _TracingService;
-    tracing.write(Obj6.make(AgentStatus, {
+    tracing.write(Obj5.make(AgentStatus, {
       parentMessage: tracing.getTraceContext().parentMessage,
       toolCallId: tracing.getTraceContext().toolCallId,
       created: (/* @__PURE__ */ new Date()).toISOString(),
       ...data
-    }));
+    }), tracing.getTraceContext());
   });
   static emitConverationMessage = Effect6.fnUntraced(function* (data) {
     const tracing = yield* _TracingService;
-    tracing.write(Obj6.make(Message.Message, {
+    tracing.write(Obj5.make(Message.Message, {
       parentMessage: tracing.getTraceContext().parentMessage,
       ...data,
       properties: {
         [MESSAGE_PROPERTY_TOOL_CALL_ID]: tracing.getTraceContext().toolCallId,
         ...data.properties
       }
-    }));
+    }), tracing.getTraceContext());
   });
 };
 var MESSAGE_PROPERTY_TOOL_CALL_ID = "toolCallId";
@@ -666,7 +739,7 @@ var ComputeEventPayload = Schema9.Union(Schema9.Struct({
 }));
 var ComputeEvent = Schema9.Struct({
   payload: ComputeEventPayload
-}).pipe(Type6.Obj({
+}).pipe(Type6.object({
   typename: "dxos.org/type/ComputeEvent",
   version: "0.1.0"
 }));
@@ -683,9 +756,9 @@ var ComputeEventLogger = class _ComputeEventLogger extends Context3.Tag("@dxos/f
     const tracing = yield* TracingService;
     return {
       log: (event) => {
-        tracing.write(Obj7.make(ComputeEvent, {
+        tracing.write(Obj6.make(ComputeEvent, {
           payload: event
-        }));
+        }), tracing.getTraceContext());
       },
       nodeId: void 0
     };
@@ -741,16 +814,22 @@ var createEventLogger = (level, message = "event") => {
 // src/services/function-invocation-service.ts
 import * as Context4 from "effect/Context";
 import * as Effect8 from "effect/Effect";
+import * as Layer4 from "effect/Layer";
 var FunctionInvocationService = class _FunctionInvocationService extends Context4.Tag("@dxos/functions/FunctionInvocationService")() {
+  static layerNotAvailable = Layer4.succeed(_FunctionInvocationService, {
+    invokeFunction: () => Effect8.die("FunctionInvocationService is not avaialble."),
+    resolveFunction: () => Effect8.die("FunctionInvocationService is not available.")
+  });
   static invokeFunction = (functionDef, input) => Effect8.serviceFunctionEffect(_FunctionInvocationService, (service) => service.invokeFunction)(functionDef, input);
+  static resolveFunction = (key) => Effect8.serviceFunctionEffect(_FunctionInvocationService, (service) => service.resolveFunction)(key);
 };
 
 // src/services/queues.ts
 import * as Context5 from "effect/Context";
 import * as Effect9 from "effect/Effect";
-import * as Layer4 from "effect/Layer";
+import * as Layer5 from "effect/Layer";
 var QueueService = class _QueueService extends Context5.Tag("@dxos/functions/QueueService")() {
-  static notAvailable = Layer4.succeed(_QueueService, {
+  static notAvailable = Layer5.succeed(_QueueService, {
     queues: {
       get(_dxn) {
         throw new Error("Queues not available");
@@ -767,7 +846,7 @@ var QueueService = class _QueueService extends Context5.Tag("@dxos/functions/Que
       queue
     };
   };
-  static layer = (queues, queue) => Layer4.succeed(_QueueService, _QueueService.make(queues, queue));
+  static layer = (queues, queue) => Layer5.succeed(_QueueService, _QueueService.make(queues, queue));
   /**
   * Gets a queue by its DXN.
   */
@@ -779,23 +858,149 @@ var QueueService = class _QueueService extends Context5.Tag("@dxos/functions/Que
   static append = (queue, objects) => Effect9.promise(() => queue.append(objects));
 };
 var ContextQueueService = class _ContextQueueService extends Context5.Tag("@dxos/functions/ContextQueueService")() {
-  static layer = (queue) => Layer4.succeed(_ContextQueueService, {
+  static layer = (queue) => Layer5.succeed(_ContextQueueService, {
     queue
   });
 };
 
 // src/protocol/protocol.ts
-import * as Effect10 from "effect/Effect";
-import * as Layer5 from "effect/Layer";
+import * as AnthropicClient from "@effect/ai-anthropic/AnthropicClient";
+import * as Effect11 from "effect/Effect";
+import * as Layer7 from "effect/Layer";
 import * as Schema10 from "effect/Schema";
 import * as SchemaAST2 from "effect/SchemaAST";
-import { AiService } from "@dxos/ai";
-import { Type as Type7 } from "@dxos/echo";
+import { AiModelResolver, AiService } from "@dxos/ai";
+import { AnthropicResolver } from "@dxos/ai/resolvers";
+import { LifecycleState, Resource } from "@dxos/context";
+import { Database as Database2, Ref as Ref2, Type as Type7 } from "@dxos/echo";
+import { refFromEncodedReference } from "@dxos/echo/internal";
 import { EchoClient } from "@dxos/echo-db";
-import { acquireReleaseResource } from "@dxos/effect";
-import { failedInvariant as failedInvariant2, invariant as invariant2 } from "@dxos/invariant";
+import { runAndForwardErrors } from "@dxos/effect";
+import { assertState, failedInvariant as failedInvariant2, invariant as invariant2 } from "@dxos/invariant";
 import { PublicKey } from "@dxos/keys";
-var __dxlog_file2 = "/__w/dxos/dxos/packages/core/functions/src/protocol/protocol.ts";
+
+// src/protocol/functions-ai-http-client.ts
+import * as Headers from "@effect/platform/Headers";
+import * as HttpClient2 from "@effect/platform/HttpClient";
+import * as HttpClientError from "@effect/platform/HttpClientError";
+import * as HttpClientResponse from "@effect/platform/HttpClientResponse";
+import * as Effect10 from "effect/Effect";
+import * as FiberRef from "effect/FiberRef";
+import * as Layer6 from "effect/Layer";
+import * as Stream from "effect/Stream";
+import { log as log3 } from "@dxos/log";
+import { ErrorCodec } from "@dxos/protocols";
+var __dxlog_file2 = "/__w/dxos/dxos/packages/core/functions/src/protocol/functions-ai-http-client.ts";
+var requestInitTagKey = "@effect/platform/FetchHttpClient/FetchOptions";
+var FunctionsAiHttpClient = class _FunctionsAiHttpClient {
+  static make = (service) => HttpClient2.make((request, url, signal, fiber) => {
+    const context = fiber.getFiberRef(FiberRef.currentContext);
+    const options = context.unsafeMap.get(requestInitTagKey) ?? {};
+    const headers = options.headers ? Headers.merge(Headers.fromInput(options.headers), request.headers) : request.headers;
+    const send = (body) => Effect10.tryPromise({
+      try: () => service.fetch(new Request(url, {
+        ...options,
+        method: request.method,
+        headers,
+        body
+      })),
+      catch: (cause) => {
+        log3.error("Failed to fetch", {
+          errorSerialized: ErrorCodec.encode(cause)
+        }, {
+          F: __dxlog_file2,
+          L: 43,
+          S: this,
+          C: (f, a) => f(...a)
+        });
+        return new HttpClientError.RequestError({
+          request,
+          reason: "Transport",
+          cause
+        });
+      }
+    }).pipe(Effect10.map((response) => HttpClientResponse.fromWeb(request, response)));
+    switch (request.body._tag) {
+      case "Raw":
+      case "Uint8Array":
+        return send(request.body.body);
+      case "FormData":
+        return send(request.body.formData);
+      case "Stream":
+        return Stream.toReadableStreamEffect(request.body.stream).pipe(Effect10.flatMap(send));
+    }
+    return send(void 0);
+  });
+  static layer = (service) => Layer6.succeed(HttpClient2.HttpClient, _FunctionsAiHttpClient.make(service));
+};
+
+// src/protocol/protocol.ts
+function _ts_add_disposable_resource(env, value2, async) {
+  if (value2 !== null && value2 !== void 0) {
+    if (typeof value2 !== "object" && typeof value2 !== "function") throw new TypeError("Object expected.");
+    var dispose, inner;
+    if (async) {
+      if (!Symbol.asyncDispose) throw new TypeError("Symbol.asyncDispose is not defined.");
+      dispose = value2[Symbol.asyncDispose];
+    }
+    if (dispose === void 0) {
+      if (!Symbol.dispose) throw new TypeError("Symbol.dispose is not defined.");
+      dispose = value2[Symbol.dispose];
+      if (async) inner = dispose;
+    }
+    if (typeof dispose !== "function") throw new TypeError("Object not disposable.");
+    if (inner) dispose = function() {
+      try {
+        inner.call(this);
+      } catch (e) {
+        return Promise.reject(e);
+      }
+    };
+    env.stack.push({
+      value: value2,
+      dispose,
+      async
+    });
+  } else if (async) {
+    env.stack.push({
+      async: true
+    });
+  }
+  return value2;
+}
+function _ts_dispose_resources(env) {
+  var _SuppressedError = typeof SuppressedError === "function" ? SuppressedError : function(error, suppressed, message) {
+    var e = new Error(message);
+    return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
+  };
+  return (_ts_dispose_resources = function _ts_dispose_resources2(env2) {
+    function fail(e) {
+      env2.error = env2.hasError ? new _SuppressedError(e, env2.error, "An error was suppressed during disposal.") : e;
+      env2.hasError = true;
+    }
+    var r, s = 0;
+    function next() {
+      while (r = env2.stack.pop()) {
+        try {
+          if (!r.async && s === 1) return s = 0, env2.stack.push(r), Promise.resolve().then(next);
+          if (r.dispose) {
+            var result = r.dispose.call(r.value);
+            if (r.async) return s |= 2, Promise.resolve(result).then(next, function(e) {
+              fail(e);
+              return next();
+            });
+          } else s |= 1;
+        } catch (e) {
+          fail(e);
+        }
+      }
+      if (s === 1) return env2.hasError ? Promise.reject(env2.error) : Promise.resolve();
+      if (env2.hasError) throw env2.error;
+    }
+    return next();
+  })(env);
+}
+var __dxlog_file3 = "/__w/dxos/dxos/packages/core/functions/src/protocol/protocol.ts";
 var wrapFunctionHandler = (func) => {
   if (!FunctionDefinition.isFunction(func)) {
     throw new TypeError("Invalid function definition");
@@ -810,85 +1015,182 @@ var wrapFunctionHandler = (func) => {
       services: func.services
     },
     handler: async ({ data, context }) => {
-      if ((func.services.includes(DatabaseService2.key) || func.services.includes(QueueService.key)) && (!context.services.dataService || !context.services.queryService)) {
+      if ((func.services.includes(Database2.Service.key) || func.services.includes(QueueService.key)) && (!context.services.dataService || !context.services.queryService)) {
         throw new FunctionError({
           message: "Services not provided: dataService, queryService"
         });
       }
       try {
-        if (!SchemaAST2.isAnyKeyword(func.inputSchema.ast)) {
-          Schema10.validateSync(func.inputSchema)(data);
-        }
-        let result = await func.handler({
-          // TODO(dmaretskyi): Fix the types.
-          context,
-          data
-        });
-        if (Effect10.isEffect(result)) {
-          result = await Effect10.runPromise(result.pipe(Effect10.orDie, Effect10.provide(createServiceLayer(context))));
-        }
-        if (func.outputSchema && !SchemaAST2.isAnyKeyword(func.outputSchema.ast)) {
-          Schema10.validateSync(func.outputSchema)(result);
-        }
-        return result;
-      } catch (error) {
-        if (FunctionError.is(error)) {
-          throw error;
-        } else {
-          throw new FunctionError({
-            cause: error,
-            context: {
-              func: func.key
+        const env = {
+          stack: [],
+          error: void 0,
+          hasError: false
+        };
+        try {
+          if (!SchemaAST2.isAnyKeyword(func.inputSchema.ast)) {
+            try {
+              Schema10.validateSync(func.inputSchema)(data);
+            } catch (error) {
+              throw new FunctionError({
+                message: "Invalid input schema",
+                cause: error
+              });
             }
+          }
+          const funcContext = _ts_add_disposable_resource(env, await new FunctionContext(context).open(), true);
+          if (func.types.length > 0) {
+            invariant2(funcContext.db, "Database is required for functions with types", {
+              F: __dxlog_file3,
+              L: 68,
+              S: void 0,
+              A: [
+                "funcContext.db",
+                "'Database is required for functions with types'"
+              ]
+            });
+            await funcContext.db.graph.schemaRegistry.register(func.types);
+          }
+          const dataWithDecodedRefs = funcContext.db && !SchemaAST2.isAnyKeyword(func.inputSchema.ast) ? decodeRefsFromSchema(func.inputSchema.ast, data, funcContext.db) : data;
+          let result = await func.handler({
+            // TODO(dmaretskyi): Fix the types.
+            context,
+            data: dataWithDecodedRefs
           });
+          if (Effect11.isEffect(result)) {
+            result = await runAndForwardErrors(result.pipe(Effect11.orDie, Effect11.provide(funcContext.createLayer())));
+          }
+          if (func.outputSchema && !SchemaAST2.isAnyKeyword(func.outputSchema.ast)) {
+            Schema10.validateSync(func.outputSchema)(result);
+          }
+          return result;
+        } catch (e) {
+          env.error = e;
+          env.hasError = true;
+        } finally {
+          const result = _ts_dispose_resources(env);
+          if (result) await result;
         }
+      } catch (error) {
+        throw error;
       }
     }
   };
 };
-var createServiceLayer = (context) => {
-  return Layer5.unwrapScoped(Effect10.gen(function* () {
-    let client;
+var FunctionContext = class extends Resource {
+  context;
+  client;
+  db;
+  queues;
+  constructor(context) {
+    super();
+    this.context = context;
     if (context.services.dataService && context.services.queryService) {
-      client = yield* acquireReleaseResource(() => {
-        invariant2(context.services.dataService && context.services.queryService, void 0, {
-          F: __dxlog_file2,
-          L: 99,
-          S: this,
-          A: [
-            "context.services.dataService && context.services.queryService",
-            ""
-          ]
-        });
-        return new EchoClient().connectToService({
-          dataService: context.services.dataService,
-          queryService: context.services.queryService,
-          queueService: context.services.queueService
-        });
+      this.client = new EchoClient().connectToService({
+        dataService: context.services.dataService,
+        queryService: context.services.queryService,
+        queueService: context.services.queueService
       });
     }
-    const db = client && context.spaceId ? yield* acquireReleaseResource(() => client.constructDatabase({
-      spaceId: context.spaceId ?? failedInvariant2(),
-      spaceKey: PublicKey.fromHex(context.spaceKey ?? failedInvariant2("spaceKey missing in context")),
-      reactiveSchemaQuery: false
-    })) : void 0;
-    if (db) {
-      console.log("Setting space root", context.spaceRootUrl);
-      yield* Effect10.promise(() => db.setSpaceRoot(context.spaceRootUrl ?? failedInvariant2("spaceRootUrl missing in context")));
-    }
-    const queues = client && context.spaceId ? client.constructQueueFactory(context.spaceId) : void 0;
-    const dbLayer = db ? DatabaseService2.layer(db) : DatabaseService2.notAvailable;
-    const queuesLayer = queues ? QueueService.layer(queues) : QueueService.notAvailable;
-    const credentials = dbLayer ? CredentialsService.layerFromDatabase().pipe(Layer5.provide(dbLayer)) : CredentialsService.configuredLayer([]);
+  }
+  async _open() {
+    await this.client?.open();
+    this.db = this.client && this.context.spaceId ? this.client.constructDatabase({
+      spaceId: this.context.spaceId ?? failedInvariant2(),
+      spaceKey: PublicKey.fromHex(this.context.spaceKey ?? failedInvariant2("spaceKey missing in context")),
+      reactiveSchemaQuery: false,
+      preloadSchemaOnOpen: false
+    }) : void 0;
+    await this.db?.setSpaceRoot(this.context.spaceRootUrl ?? failedInvariant2("spaceRootUrl missing in context"));
+    await this.db?.open();
+    this.queues = this.client && this.context.spaceId ? this.client.constructQueueFactory(this.context.spaceId) : void 0;
+  }
+  async _close() {
+    await this.db?.close();
+    await this.client?.close();
+  }
+  createLayer() {
+    assertState(this._lifecycleState === LifecycleState.OPEN, "FunctionContext is not open");
+    const dbLayer = this.db ? Database2.layer(this.db) : Database2.notAvailable;
+    const queuesLayer = this.queues ? QueueService.layer(this.queues) : QueueService.notAvailable;
+    const credentials = dbLayer ? CredentialsService.layerFromDatabase({
+      caching: true
+    }).pipe(Layer7.provide(dbLayer)) : CredentialsService.configuredLayer([]);
     const functionInvocationService = MockedFunctionInvocationService;
-    const aiService = AiService.notAvailable;
     const tracing = TracingService.layerNoop;
-    return Layer5.mergeAll(dbLayer, queuesLayer, credentials, functionInvocationService, aiService, tracing);
-  }));
+    const aiLayer = this.context.services.functionsAiService ? AiModelResolver.AiModelResolver.buildAiService.pipe(Layer7.provide(AnthropicResolver.make().pipe(Layer7.provide(AnthropicClient.layer({
+      // Note: It doesn't matter what is base url here, it will be proxied to ai gateway in edge.
+      apiUrl: "http://internal/provider/anthropic"
+    }).pipe(Layer7.provide(FunctionsAiHttpClient.layer(this.context.services.functionsAiService))))))) : AiService.notAvailable;
+    return Layer7.mergeAll(dbLayer, queuesLayer, credentials, functionInvocationService, aiLayer, tracing);
+  }
 };
-var MockedFunctionInvocationService = Layer5.succeed(FunctionInvocationService, {
-  invokeFunction: () => Effect10.die("Calling functions from functions is not implemented yet.")
+var MockedFunctionInvocationService = Layer7.succeed(FunctionInvocationService, {
+  invokeFunction: () => Effect11.die("Calling functions from functions is not implemented yet."),
+  resolveFunction: () => Effect11.die("Not implemented.")
 });
+var decodeRefsFromSchema = (ast, value2, db) => {
+  if (value2 == null) {
+    return value2;
+  }
+  const encoded = SchemaAST2.encodedBoundAST(ast);
+  if (Ref2.isRefType(encoded)) {
+    if (Ref2.isRef(value2)) {
+      return value2;
+    }
+    if (typeof value2 === "object" && value2 !== null && typeof value2["/"] === "string") {
+      const resolver = db.graph.createRefResolver({
+        context: {
+          space: db.spaceId
+        }
+      });
+      return refFromEncodedReference(value2, resolver);
+    }
+    return value2;
+  }
+  switch (encoded._tag) {
+    case "TypeLiteral": {
+      if (typeof value2 !== "object" || value2 === null || Array.isArray(value2)) {
+        return value2;
+      }
+      const result = {
+        ...value2
+      };
+      for (const prop of SchemaAST2.getPropertySignatures(encoded)) {
+        const key = prop.name.toString();
+        if (key in result) {
+          result[key] = decodeRefsFromSchema(prop.type, result[key], db);
+        }
+      }
+      return result;
+    }
+    case "TupleType": {
+      if (!Array.isArray(value2)) {
+        return value2;
+      }
+      if (encoded.elements.length === 0 && encoded.rest.length === 1) {
+        const elementType = encoded.rest[0].type;
+        return value2.map((item) => decodeRefsFromSchema(elementType, item, db));
+      }
+      return value2;
+    }
+    case "Union": {
+      const nonUndefined = encoded.types.filter((t) => !SchemaAST2.isUndefinedKeyword(t));
+      if (nonUndefined.length === 1) {
+        return decodeRefsFromSchema(nonUndefined[0], value2, db);
+      }
+      return value2;
+    }
+    case "Suspend": {
+      return decodeRefsFromSchema(encoded.f(), value2, db);
+    }
+    case "Refinement": {
+      return decodeRefsFromSchema(encoded.from, value2, db);
+    }
+    default: {
+      return value2;
+    }
+  }
+};
 export {
   ComputeEvent,
   ComputeEventLogger,
@@ -896,7 +1198,6 @@ export {
   ConfiguredCredentialsService,
   ContextQueueService,
   CredentialsService,
-  DatabaseService2 as DatabaseService,
   Example,
   FUNCTIONS_META_KEY,
   FUNCTIONS_PRESET_META_KEY,
@@ -921,6 +1222,7 @@ export {
   logCustomEvent,
   serializeFunction,
   setUserFunctionIdInMetadata,
+  toOperation,
   withAuthorization,
   wrapFunctionHandler
 };