@dxos/functions 0.8.4-main.7ace549 → 0.8.4-main.937b3ca

package/dist/lib/node-esm/index.mjs

@@ -7,7 +7,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: {
@@ -17,7 +17,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: {
@@ -27,9 +27,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
@@ -39,8 +39,9 @@ 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 = {};
@@ -70,7 +71,7 @@ var Script = Schema.Struct({
   // Whether source has changed since last deploy.
   changed: Schema.Boolean.pipe(FormInputAnnotation.set(false), Schema.optional),
   source: Type.Ref(Text.Text).pipe(FormInputAnnotation.set(false))
-}).pipe(Type.Obj({
+}).pipe(Type.object({
   typename: "dxos.org/type/Script",
   version: "0.1.0"
 }), Annotation.LabelAnnotation.set([
@@ -112,7 +113,7 @@ 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"
 }), Annotation2.LabelAnnotation.set([
@@ -120,14 +121,16 @@ var Function = Schema2.Struct({
 ]), SystemTypeAnnotation.set(true));
 var make2 = (props) => Obj2.make(Function, props);
 var setFrom = (target, source) => {
-  target.key = source.key ?? target.key;
-  target.name = source.name ?? target.name;
-  target.version = source.version;
-  target.description = source.description;
-  target.updated = source.updated;
-  target.inputSchema = source.inputSchema ? JSON.parse(JSON.stringify(source.inputSchema)) : void 0;
-  target.outputSchema = source.outputSchema ? JSON.parse(JSON.stringify(source.outputSchema)) : void 0;
-  Obj2.getMeta(target).keys = JSON.parse(JSON.stringify(Obj2.getMeta(source).keys));
+  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
@@ -148,6 +151,7 @@ import * as SchemaAST from "effect/SchemaAST";
 import { Obj as Obj3, QueryAST, Type as Type3 } from "@dxos/echo";
 import { OptionsAnnotationId, SystemTypeAnnotation as SystemTypeAnnotation2 } from "@dxos/echo/internal";
 import { DXN } from "@dxos/keys";
+import { Expando } from "@dxos/schema";
 var Kinds = [
   "email",
   "queue",
@@ -212,12 +216,12 @@ var WebhookSpec = Schema3.Struct({
 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(Type3.Ref(Type3.Expando).annotations({
+  function: Schema3.optional(Type3.Ref(Expando.Expando).annotations({
     title: "Function"
   })),
   /**
@@ -228,6 +232,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"
   })),
@@ -247,11 +253,11 @@ var Trigger_ = Schema3.Struct({
     key: Schema3.String,
     value: Schema3.Any
   })))
-}).pipe(Type3.Obj({
+}).pipe(Type3.object({
   typename: "dxos.org/type/Trigger",
   version: "0.1.0"
 }), SystemTypeAnnotation2.set(true));
-var Trigger = Trigger_;
+var Trigger = TriggerSchema;
 var make3 = (props) => Obj3.make(Trigger, props);
 
 // src/types/TriggerEvent.ts
@@ -261,10 +267,11 @@ __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";
+import { DXN as DXN2, Type as Type4 } from "@dxos/echo";
 var EmailEvent = Schema4.mutable(Schema4.Struct({
   from: Schema4.String,
   to: Schema4.String,
@@ -286,7 +293,7 @@ var SubscriptionEvent = Schema4.Struct({
   /**
   * Reference to the object that was changed or created.
   */
-  subject: Type4.Ref(Obj4.Any),
+  subject: Type4.Ref(Type4.Obj),
   /**
   * @deprecated
   */
@@ -304,6 +311,7 @@ var WebhookEvent = Schema4.mutable(Schema4.Struct({
   }),
   bodyText: Schema4.String
 }));
+var TriggerEvent = Schema4.Union(EmailEvent, QueueEvent, SubscriptionEvent, TimerEvent, WebhookEvent);
 
 // src/types/url.ts
 var FUNCTIONS_META_KEY = "dxos.org/service/function";
@@ -377,10 +385,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) => {
@@ -391,9 +428,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({
@@ -406,7 +444,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;
 };
@@ -425,7 +463,7 @@ var deserializeFunction = (functionObj) => {
     services: functionObj.services ?? [],
     types: [],
     meta: {
-      deployedFunctionId: getUserFunctionIdInMetadata(Obj5.getMeta(functionObj))
+      deployedFunctionId: getUserFunctionIdInMetadata(Obj4.getMeta(functionObj))
     }
   };
 };
@@ -502,9 +540,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";
@@ -513,7 +548,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* () {
@@ -537,14 +572,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) => {
@@ -580,18 +624,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";
 
@@ -600,13 +643,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);
   /**
@@ -616,32 +666,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";
@@ -679,7 +740,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"
 }));
@@ -696,9 +757,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
     };
@@ -754,16 +815,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");
@@ -780,7 +847,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.
   */
@@ -792,22 +859,83 @@ 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 { AiModelResolver, AiService } from "@dxos/ai";
+import { AnthropicResolver } from "@dxos/ai/resolvers";
 import { LifecycleState, Resource } from "@dxos/context";
-import { Type as Type7 } from "@dxos/echo";
+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 { runAndForwardErrors } from "@dxos/effect";
 import { assertState, failedInvariant as failedInvariant2, invariant as invariant2 } from "@dxos/invariant";
 import { PublicKey } from "@dxos/keys";
+
+// 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.");
@@ -873,7 +1001,7 @@ function _ts_dispose_resources(env) {
     return next();
   })(env);
 }
-var __dxlog_file2 = "/__w/dxos/dxos/packages/core/functions/src/protocol/protocol.ts";
+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");
@@ -888,7 +1016,7 @@ 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"
         });
@@ -913,23 +1041,24 @@ var wrapFunctionHandler = (func) => {
           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_file2,
-              L: 63,
+              F: __dxlog_file3,
+              L: 68,
               S: void 0,
               A: [
                 "funcContext.db",
                 "'Database is required for functions with types'"
               ]
             });
-            funcContext.db.graph.schemaRegistry.addSchema(func.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
+            data: dataWithDecodedRefs
           });
-          if (Effect10.isEffect(result)) {
-            result = await Effect10.runPromise(result.pipe(Effect10.orDie, Effect10.provide(funcContext.createLayer())));
+          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);
@@ -969,7 +1098,8 @@ var FunctionContext = class extends Resource {
     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
+      reactiveSchemaQuery: false,
+      preloadSchemaOnOpen: false
     }) : void 0;
     await this.db?.setSpaceRoot(this.context.spaceRootUrl ?? failedInvariant2("spaceRootUrl missing in context"));
     await this.db?.open();
@@ -981,18 +1111,87 @@ var FunctionContext = class extends Resource {
   }
   createLayer() {
     assertState(this._lifecycleState === LifecycleState.OPEN, "FunctionContext is not open");
-    const dbLayer = this.db ? DatabaseService2.layer(this.db) : DatabaseService2.notAvailable;
+    const dbLayer = this.db ? Database2.Service.layer(this.db) : Database2.Service.notAvailable;
     const queuesLayer = this.queues ? QueueService.layer(this.queues) : QueueService.notAvailable;
-    const credentials = dbLayer ? CredentialsService.layerFromDatabase().pipe(Layer5.provide(dbLayer)) : CredentialsService.configuredLayer([]);
+    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,
@@ -1000,7 +1199,6 @@ export {
   ConfiguredCredentialsService,
   ContextQueueService,
   CredentialsService,
-  DatabaseService2 as DatabaseService,
   Example,
   FUNCTIONS_META_KEY,
   FUNCTIONS_PRESET_META_KEY,
@@ -1025,6 +1223,7 @@ export {
   logCustomEvent,
   serializeFunction,
   setUserFunctionIdInMetadata,
+  toOperation,
   withAuthorization,
   wrapFunctionHandler
 };