@dxos/functions 0.8.4-main.e8ec1fe → 0.8.4-main.ef1bc66f44

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (57) hide show
  1. package/dist/lib/{browser → neutral}/index.mjs +450 -148
  2. package/dist/lib/neutral/index.mjs.map +7 -0
  3. package/dist/lib/neutral/meta.json +1 -0
  4. package/dist/types/src/errors.d.ts +24 -32
  5. package/dist/types/src/errors.d.ts.map +1 -1
  6. package/dist/types/src/operation-compatibility.test.d.ts +2 -0
  7. package/dist/types/src/operation-compatibility.test.d.ts.map +1 -0
  8. package/dist/types/src/protocol/functions-ai-http-client.d.ts +12 -0
  9. package/dist/types/src/protocol/functions-ai-http-client.d.ts.map +1 -0
  10. package/dist/types/src/protocol/protocol.d.ts.map +1 -1
  11. package/dist/types/src/sdk.d.ts +27 -2
  12. package/dist/types/src/sdk.d.ts.map +1 -1
  13. package/dist/types/src/services/credentials.d.ts +6 -4
  14. package/dist/types/src/services/credentials.d.ts.map +1 -1
  15. package/dist/types/src/services/event-logger.d.ts +25 -31
  16. package/dist/types/src/services/event-logger.d.ts.map +1 -1
  17. package/dist/types/src/services/function-invocation-service.d.ts +5 -0
  18. package/dist/types/src/services/function-invocation-service.d.ts.map +1 -1
  19. package/dist/types/src/services/index.d.ts +0 -1
  20. package/dist/types/src/services/index.d.ts.map +1 -1
  21. package/dist/types/src/services/queues.d.ts +4 -4
  22. package/dist/types/src/services/queues.d.ts.map +1 -1
  23. package/dist/types/src/services/tracing.d.ts +37 -3
  24. package/dist/types/src/services/tracing.d.ts.map +1 -1
  25. package/dist/types/src/types/Function.d.ts +40 -46
  26. package/dist/types/src/types/Function.d.ts.map +1 -1
  27. package/dist/types/src/types/Script.d.ts +9 -16
  28. package/dist/types/src/types/Script.d.ts.map +1 -1
  29. package/dist/types/src/types/Trigger.d.ts +58 -76
  30. package/dist/types/src/types/Trigger.d.ts.map +1 -1
  31. package/dist/types/src/types/TriggerEvent.d.ts +43 -13
  32. package/dist/types/src/types/TriggerEvent.d.ts.map +1 -1
  33. package/dist/types/src/types/url.d.ts +4 -3
  34. package/dist/types/src/types/url.d.ts.map +1 -1
  35. package/dist/types/tsconfig.tsbuildinfo +1 -1
  36. package/package.json +23 -17
  37. package/src/errors.ts +4 -4
  38. package/src/operation-compatibility.test.ts +185 -0
  39. package/src/protocol/functions-ai-http-client.ts +67 -0
  40. package/src/protocol/protocol.ts +184 -67
  41. package/src/sdk.ts +68 -5
  42. package/src/services/credentials.ts +31 -15
  43. package/src/services/event-logger.ts +2 -2
  44. package/src/services/function-invocation-service.ts +14 -0
  45. package/src/services/index.ts +0 -2
  46. package/src/services/queues.ts +5 -7
  47. package/src/services/tracing.ts +63 -4
  48. package/src/types/Function.ts +28 -8
  49. package/src/types/Script.ts +8 -7
  50. package/src/types/Trigger.ts +18 -14
  51. package/src/types/TriggerEvent.ts +29 -29
  52. package/src/types/url.ts +4 -3
  53. package/dist/lib/browser/index.mjs.map +0 -7
  54. package/dist/lib/browser/meta.json +0 -1
  55. package/dist/lib/node-esm/index.mjs +0 -928
  56. package/dist/lib/node-esm/index.mjs.map +0 -7
  57. package/dist/lib/node-esm/meta.json +0 -1
@@ -1,928 +0,0 @@
1
- import { createRequire } from 'node:module';const require = createRequire(import.meta.url);
2
- var __defProp = Object.defineProperty;
3
- var __export = (target, all) => {
4
- for (var name in all)
5
- __defProp(target, name, { get: all[name], enumerable: true });
6
- };
7
-
8
- // src/errors.ts
9
- import { BaseError } from "@dxos/errors";
10
- var ServiceNotAvailableError = class extends BaseError.extend("SERVICE_NOT_AVAILABLE", "Service not available") {
11
- constructor(service, options) {
12
- super({
13
- context: {
14
- service
15
- },
16
- ...options
17
- });
18
- }
19
- };
20
- var FunctionNotFoundError = class extends BaseError.extend("FUNCTION_NOT_FOUND", "Function not found") {
21
- constructor(functionKey, options) {
22
- super({
23
- context: {
24
- function: functionKey
25
- },
26
- ...options
27
- });
28
- }
29
- };
30
- var FunctionError = class extends BaseError.extend("FUNCTION_ERROR", "Function invocation error") {
31
- };
32
- var TriggerStateNotFoundError = class extends BaseError.extend("TRIGGER_STATE_NOT_FOUND", "Trigger state not found") {
33
- };
34
-
35
- // src/example/fib.ts
36
- import * as Effect2 from "effect/Effect";
37
- import * as Schema6 from "effect/Schema";
38
-
39
- // src/sdk.ts
40
- import * as Effect from "effect/Effect";
41
- import * as Schema5 from "effect/Schema";
42
- import { Obj as Obj5, Type as Type5 } from "@dxos/echo";
43
- import { assertArgument, failedInvariant } from "@dxos/invariant";
44
-
45
- // src/types/Function.ts
46
- var Function_exports = {};
47
- __export(Function_exports, {
48
- Function: () => Function,
49
- make: () => make2
50
- });
51
- import * as Schema2 from "effect/Schema";
52
- import { Obj as Obj2, Type as Type2 } from "@dxos/echo";
53
- import { JsonSchemaType, LabelAnnotation as LabelAnnotation2, Ref as Ref2 } from "@dxos/echo/internal";
54
-
55
- // src/types/Script.ts
56
- var Script_exports = {};
57
- __export(Script_exports, {
58
- Script: () => Script,
59
- make: () => make
60
- });
61
- import * as Schema from "effect/Schema";
62
- import { Obj, Ref, Type } from "@dxos/echo";
63
- import { FormAnnotation, LabelAnnotation } from "@dxos/echo/internal";
64
- import { Text } from "@dxos/schema";
65
- var Script = Schema.Struct({
66
- name: Schema.String.pipe(Schema.optional),
67
- description: Schema.String.pipe(Schema.optional),
68
- // TODO(burdon): Change to hash of deployed content.
69
- // Whether source has changed since last deploy.
70
- changed: Schema.Boolean.pipe(FormAnnotation.set(false), Schema.optional),
71
- source: Type.Ref(Text.Text).pipe(FormAnnotation.set(false))
72
- }).pipe(Type.Obj({
73
- typename: "dxos.org/type/Script",
74
- version: "0.1.0"
75
- }), LabelAnnotation.set([
76
- "name"
77
- ]));
78
- var make = ({ source = "", ...props } = {}) => Obj.make(Script, {
79
- ...props,
80
- source: Ref.make(Text.make(source))
81
- });
82
-
83
- // src/types/Function.ts
84
- var Function = Schema2.Struct({
85
- /**
86
- * Global registry ID.
87
- * NOTE: The `key` property refers to the original registry entry.
88
- */
89
- // TODO(burdon): Create Format type for DXN-like ids, such as this and schema type.
90
- // TODO(dmaretskyi): Consider making it part of ECHO meta.
91
- // TODO(dmaretskyi): Make required.
92
- key: Schema2.optional(Schema2.String).annotations({
93
- description: "Unique registration key for the blueprint"
94
- }),
95
- // TODO(burdon): Rename to id/uri?
96
- name: Schema2.NonEmptyString,
97
- version: Schema2.String,
98
- description: Schema2.optional(Schema2.String),
99
- /**
100
- * ISO date string of the last deployment.
101
- */
102
- updated: Schema2.optional(Schema2.String),
103
- // Reference to a source script if it exists within ECHO.
104
- // TODO(burdon): Don't ref ScriptType directly (core).
105
- source: Schema2.optional(Ref2(Script)),
106
- inputSchema: Schema2.optional(JsonSchemaType),
107
- outputSchema: Schema2.optional(JsonSchemaType),
108
- /**
109
- * List of required services.
110
- * Match the Context.Tag keys of the FunctionServices variants.
111
- */
112
- services: Schema2.optional(Schema2.Array(Schema2.String)),
113
- // Local binding to a function name.
114
- binding: Schema2.optional(Schema2.String)
115
- }).pipe(Type2.Obj({
116
- typename: "dxos.org/type/Function",
117
- version: "0.1.0"
118
- }), LabelAnnotation2.set([
119
- "name"
120
- ]));
121
- var make2 = (props) => Obj2.make(Function, props);
122
-
123
- // src/types/Trigger.ts
124
- var Trigger_exports = {};
125
- __export(Trigger_exports, {
126
- EmailSpec: () => EmailSpec,
127
- Kinds: () => Kinds,
128
- QueueSpec: () => QueueSpec,
129
- Spec: () => Spec,
130
- SubscriptionSpec: () => SubscriptionSpec,
131
- TimerSpec: () => TimerSpec,
132
- Trigger: () => Trigger,
133
- WebhookSpec: () => WebhookSpec,
134
- make: () => make3
135
- });
136
- import * as Schema3 from "effect/Schema";
137
- import * as SchemaAST from "effect/SchemaAST";
138
- import { Obj as Obj3, QueryAST, Type as Type3 } from "@dxos/echo";
139
- import { Expando, OptionsAnnotationId, Ref as Ref3 } from "@dxos/echo/internal";
140
- import { DXN } from "@dxos/keys";
141
- var Kinds = [
142
- "email",
143
- "queue",
144
- "subscription",
145
- "timer",
146
- "webhook"
147
- ];
148
- var kindLiteralAnnotations = {
149
- title: "Kind"
150
- };
151
- var EmailSpec = Schema3.Struct({
152
- kind: Schema3.Literal("email").annotations(kindLiteralAnnotations)
153
- }).pipe(Schema3.mutable);
154
- var QueueSpec = Schema3.Struct({
155
- kind: Schema3.Literal("queue").annotations(kindLiteralAnnotations),
156
- // TODO(dmaretskyi): Change to a reference.
157
- queue: DXN.Schema
158
- }).pipe(Schema3.mutable);
159
- var SubscriptionSpec = Schema3.Struct({
160
- kind: Schema3.Literal("subscription").annotations(kindLiteralAnnotations),
161
- query: Schema3.Struct({
162
- raw: Schema3.optional(Schema3.String.annotations({
163
- title: "Query"
164
- })),
165
- ast: QueryAST.Query
166
- }).pipe(Schema3.mutable),
167
- options: Schema3.optional(Schema3.Struct({
168
- // Watch changes to object (not just creation).
169
- deep: Schema3.optional(Schema3.Boolean.annotations({
170
- title: "Nested"
171
- })),
172
- // Debounce changes (delay in ms).
173
- delay: Schema3.optional(Schema3.Number.annotations({
174
- title: "Delay"
175
- }))
176
- }).annotations({
177
- title: "Options"
178
- }))
179
- }).pipe(Schema3.mutable);
180
- var TimerSpec = Schema3.Struct({
181
- kind: Schema3.Literal("timer").annotations(kindLiteralAnnotations),
182
- cron: Schema3.String.annotations({
183
- title: "Cron",
184
- [SchemaAST.ExamplesAnnotationId]: [
185
- "0 0 * * *"
186
- ]
187
- })
188
- }).pipe(Schema3.mutable);
189
- var WebhookSpec = Schema3.Struct({
190
- kind: Schema3.Literal("webhook").annotations(kindLiteralAnnotations),
191
- method: Schema3.optional(Schema3.String.annotations({
192
- title: "Method",
193
- [OptionsAnnotationId]: [
194
- "GET",
195
- "POST"
196
- ]
197
- })),
198
- port: Schema3.optional(Schema3.Number.annotations({
199
- title: "Port"
200
- }))
201
- }).pipe(Schema3.mutable);
202
- var Spec = Schema3.Union(EmailSpec, QueueSpec, SubscriptionSpec, TimerSpec, WebhookSpec).annotations({
203
- title: "Trigger"
204
- });
205
- var Trigger_ = Schema3.Struct({
206
- /**
207
- * Function or workflow to invoke.
208
- */
209
- // TODO(dmaretskyi): Can be a Ref(FunctionType) or Ref(ComputeGraphType).
210
- function: Schema3.optional(Ref3(Expando).annotations({
211
- title: "Function"
212
- })),
213
- /**
214
- * Only used for workflowSchema.
215
- * Specifies the input node in the circuit.
216
- * @deprecated Remove and enforce a single input node in all compute graphSchema.
217
- */
218
- inputNodeId: Schema3.optional(Schema3.String.annotations({
219
- title: "Input Node ID"
220
- })),
221
- enabled: Schema3.optional(Schema3.Boolean.annotations({
222
- title: "Enabled"
223
- })),
224
- spec: Schema3.optional(Spec),
225
- /**
226
- * Passed as the input data to the function.
227
- * Must match the function's input schema.
228
- *
229
- * @example
230
- * {
231
- * item: '{{$.trigger.event}}',
232
- * instructions: 'Summarize and perform entity-extraction'
233
- * mailbox: { '/': 'dxn:echo:AAA:ZZZ' }
234
- * }
235
- */
236
- input: Schema3.optional(Schema3.mutable(Schema3.Record({
237
- key: Schema3.String,
238
- value: Schema3.Any
239
- })))
240
- }).pipe(Type3.Obj({
241
- typename: "dxos.org/type/Trigger",
242
- version: "0.1.0"
243
- }));
244
- var Trigger = Trigger_;
245
- var make3 = (props) => Obj3.make(Trigger, props);
246
-
247
- // src/types/TriggerEvent.ts
248
- var TriggerEvent_exports = {};
249
- __export(TriggerEvent_exports, {
250
- EmailEvent: () => EmailEvent,
251
- QueueEvent: () => QueueEvent,
252
- SubscriptionEvent: () => SubscriptionEvent,
253
- TimerEvent: () => TimerEvent,
254
- WebhookEvent: () => WebhookEvent
255
- });
256
- import * as Schema4 from "effect/Schema";
257
- import { DXN as DXN2, Obj as Obj4, Type as Type4 } from "@dxos/echo";
258
- var EmailEvent = Schema4.mutable(Schema4.Struct({
259
- from: Schema4.String,
260
- to: Schema4.String,
261
- subject: Schema4.String,
262
- created: Schema4.String,
263
- body: Schema4.String
264
- }));
265
- var QueueEvent = Schema4.mutable(Schema4.Struct({
266
- queue: DXN2.Schema,
267
- item: Schema4.Any,
268
- cursor: Schema4.String
269
- }));
270
- var SubscriptionEvent = Schema4.Struct({
271
- /**
272
- * Type of the mutation.
273
- */
274
- // TODO(dmaretskyi): Specify enum.
275
- type: Schema4.String,
276
- /**
277
- * Reference to the object that was changed or created.
278
- */
279
- subject: Type4.Ref(Obj4.Any),
280
- /**
281
- * @deprecated
282
- */
283
- changedObjectId: Schema4.optional(Schema4.String)
284
- }).pipe(Schema4.mutable);
285
- var TimerEvent = Schema4.mutable(Schema4.Struct({
286
- tick: Schema4.Number
287
- }));
288
- var WebhookEvent = Schema4.mutable(Schema4.Struct({
289
- url: Schema4.String,
290
- method: Schema4.Literal("GET", "POST"),
291
- headers: Schema4.Record({
292
- key: Schema4.String,
293
- value: Schema4.String
294
- }),
295
- bodyText: Schema4.String
296
- }));
297
-
298
- // src/types/url.ts
299
- var FUNCTIONS_META_KEY = "dxos.org/service/function";
300
- var FUNCTIONS_PRESET_META_KEY = "dxos.org/service/function-preset";
301
- var getUserFunctionIdInMetadata = (meta) => {
302
- return meta.keys.find((key) => key.source === FUNCTIONS_META_KEY)?.id;
303
- };
304
- var setUserFunctionIdInMetadata = (meta, functionId) => {
305
- const key = meta.keys.find((key2) => key2.source === FUNCTIONS_META_KEY);
306
- if (key) {
307
- if (key.id !== functionId) {
308
- throw new Error("Metadata mismatch");
309
- }
310
- } else {
311
- meta.keys.push({
312
- source: FUNCTIONS_META_KEY,
313
- id: functionId
314
- });
315
- }
316
- };
317
-
318
- // src/sdk.ts
319
- var typeId = Symbol.for("@dxos/functions/FunctionDefinition");
320
- var defineFunction = ({ key, name, description, inputSchema, outputSchema = Schema5.Any, handler, services }) => {
321
- if (!Schema5.isSchema(inputSchema)) {
322
- throw new Error("Input schema must be a valid schema");
323
- }
324
- if (typeof handler !== "function") {
325
- throw new Error("Handler must be a function");
326
- }
327
- const limit = Error.stackTraceLimit;
328
- Error.stackTraceLimit = 2;
329
- const traceError = new Error();
330
- Error.stackTraceLimit = limit;
331
- let cache = false;
332
- const captureStackTrace = () => {
333
- if (cache !== false) {
334
- return cache;
335
- }
336
- if (traceError.stack !== void 0) {
337
- const stack = traceError.stack.split("\n");
338
- if (stack[2] !== void 0) {
339
- cache = stack[2].trim();
340
- return cache;
341
- }
342
- }
343
- };
344
- const handlerWithSpan = (...args) => {
345
- const result = handler(...args);
346
- if (Effect.isEffect(result)) {
347
- return Effect.withSpan(result, `${key ?? name}`, {
348
- captureStackTrace
349
- });
350
- }
351
- return result;
352
- };
353
- return {
354
- [typeId]: true,
355
- key,
356
- name,
357
- description,
358
- inputSchema,
359
- outputSchema,
360
- handler: handlerWithSpan,
361
- services: !services ? [] : getServiceKeys(services)
362
- };
363
- };
364
- var getServiceKeys = (services) => {
365
- return services.map((tag) => {
366
- if (typeof tag.key === "string") {
367
- return tag.key;
368
- }
369
- console.log(tag);
370
- failedInvariant();
371
- });
372
- };
373
- var FunctionDefinition = {
374
- make: defineFunction,
375
- isFunction: (value2) => {
376
- return typeof value2 === "object" && value2 !== null && Symbol.for("@dxos/functions/FunctionDefinition") in value2;
377
- },
378
- serialize: (functionDef) => {
379
- assertArgument(FunctionDefinition.isFunction(functionDef), "functionDef");
380
- return serializeFunction(functionDef);
381
- },
382
- deserialize: (functionObj) => {
383
- assertArgument(Obj5.instanceOf(Function_exports.Function, functionObj), "functionObj");
384
- return deserializeFunction(functionObj);
385
- }
386
- };
387
- var serializeFunction = (functionDef) => {
388
- const fn4 = Function_exports.make({
389
- key: functionDef.key,
390
- name: functionDef.name,
391
- version: "0.1.0",
392
- description: functionDef.description,
393
- inputSchema: Type5.toJsonSchema(functionDef.inputSchema),
394
- outputSchema: !functionDef.outputSchema ? void 0 : Type5.toJsonSchema(functionDef.outputSchema),
395
- services: functionDef.services
396
- });
397
- if (functionDef.meta?.deployedFunctionId) {
398
- setUserFunctionIdInMetadata(Obj5.getMeta(fn4), functionDef.meta.deployedFunctionId);
399
- }
400
- return fn4;
401
- };
402
- var deserializeFunction = (functionObj) => {
403
- return {
404
- [typeId]: true,
405
- // TODO(dmaretskyi): Fix key.
406
- key: functionObj.key ?? functionObj.name,
407
- name: functionObj.name,
408
- description: functionObj.description,
409
- inputSchema: !functionObj.inputSchema ? Schema5.Unknown : Type5.toEffectSchema(functionObj.inputSchema),
410
- outputSchema: !functionObj.outputSchema ? void 0 : Type5.toEffectSchema(functionObj.outputSchema),
411
- // TODO(dmaretskyi): This should throw error.
412
- handler: () => {
413
- },
414
- services: functionObj.services ?? [],
415
- meta: {
416
- deployedFunctionId: getUserFunctionIdInMetadata(Obj5.getMeta(functionObj))
417
- }
418
- };
419
- };
420
-
421
- // src/example/fib.ts
422
- var fib_default = defineFunction({
423
- key: "example.org/function/fib",
424
- name: "Fibonacci",
425
- description: "Function that calculates a Fibonacci number",
426
- inputSchema: Schema6.Struct({
427
- iterations: Schema6.optional(Schema6.Number).annotations({
428
- description: "Number of iterations",
429
- default: 1e5
430
- })
431
- }),
432
- outputSchema: Schema6.Struct({
433
- result: Schema6.String
434
- }),
435
- handler: Effect2.fn(function* ({ data: { iterations = 1e5 } }) {
436
- let a = 0n;
437
- let b = 1n;
438
- for (let i = 0; i < iterations; i++) {
439
- a += b;
440
- b = a - b;
441
- }
442
- return {
443
- result: a.toString()
444
- };
445
- })
446
- });
447
-
448
- // src/example/reply.ts
449
- import * as Console from "effect/Console";
450
- import * as Effect3 from "effect/Effect";
451
- import * as Schema7 from "effect/Schema";
452
- var reply_default = defineFunction({
453
- key: "example.org/function/reply",
454
- name: "Reply",
455
- description: "Function that echoes the input",
456
- inputSchema: Schema7.Any,
457
- outputSchema: Schema7.Any,
458
- handler: Effect3.fn(function* ({ data }) {
459
- yield* Console.log("reply", {
460
- data
461
- });
462
- return data;
463
- })
464
- });
465
-
466
- // src/example/sleep.ts
467
- import * as Effect4 from "effect/Effect";
468
- import * as Schema8 from "effect/Schema";
469
- var sleep_default = defineFunction({
470
- key: "example.org/function/sleep",
471
- name: "Sleep",
472
- description: "Function that sleeps for a given amount of time",
473
- inputSchema: Schema8.Struct({
474
- duration: Schema8.optional(Schema8.Number).annotations({
475
- description: "Milliseconds to sleep",
476
- default: 1e5
477
- })
478
- }),
479
- outputSchema: Schema8.Void,
480
- handler: Effect4.fn(function* ({ data: { duration = 1e5 } }) {
481
- yield* Effect4.sleep(duration);
482
- })
483
- });
484
-
485
- // src/example/index.ts
486
- (function(Example2) {
487
- Example2.fib = fib_default;
488
- Example2.reply = reply_default;
489
- Example2.sleep = sleep_default;
490
- })(Example || (Example = {}));
491
- var Example;
492
-
493
- // src/services/index.ts
494
- import { DatabaseService as DatabaseService2 } from "@dxos/echo-db";
495
-
496
- // src/services/credentials.ts
497
- import * as HttpClient from "@effect/platform/HttpClient";
498
- import * as HttpClientRequest from "@effect/platform/HttpClientRequest";
499
- import * as Context from "effect/Context";
500
- import * as Effect5 from "effect/Effect";
501
- import * as Layer from "effect/Layer";
502
- import * as Redacted from "effect/Redacted";
503
- import { Query } from "@dxos/echo";
504
- import { DatabaseService } from "@dxos/echo-db";
505
- import { AccessToken } from "@dxos/types";
506
- var CredentialsService = class _CredentialsService extends Context.Tag("@dxos/functions/CredentialsService")() {
507
- static getCredential = (query) => Effect5.gen(function* () {
508
- const credentials = yield* _CredentialsService;
509
- return yield* Effect5.promise(() => credentials.getCredential(query));
510
- });
511
- static getApiKey = (query) => Effect5.gen(function* () {
512
- const credential = yield* _CredentialsService.getCredential(query);
513
- if (!credential.apiKey) {
514
- throw new Error(`API key not found for service: ${query.service}`);
515
- }
516
- return Redacted.make(credential.apiKey);
517
- });
518
- static configuredLayer = (credentials) => Layer.succeed(_CredentialsService, new ConfiguredCredentialsService(credentials));
519
- static layerConfig = (credentials) => Layer.effect(_CredentialsService, Effect5.gen(function* () {
520
- const serviceCredentials = yield* Effect5.forEach(credentials, ({ service, apiKey }) => Effect5.gen(function* () {
521
- return {
522
- service,
523
- apiKey: Redacted.value(yield* apiKey)
524
- };
525
- }));
526
- return new ConfiguredCredentialsService(serviceCredentials);
527
- }));
528
- static layerFromDatabase = () => Layer.effect(_CredentialsService, Effect5.gen(function* () {
529
- const dbService = yield* DatabaseService;
530
- const queryCredentials = async (query) => {
531
- const { objects: accessTokens } = await dbService.db.query(Query.type(AccessToken.AccessToken)).run();
532
- return accessTokens.filter((accessToken) => accessToken.source === query.service).map((accessToken) => ({
533
- service: accessToken.source,
534
- apiKey: accessToken.token
535
- }));
536
- };
537
- return {
538
- getCredential: async (query) => {
539
- const credentials = await queryCredentials(query);
540
- if (credentials.length === 0) {
541
- throw new Error(`Credential not found for service: ${query.service}`);
542
- }
543
- return credentials[0];
544
- },
545
- queryCredentials: async (query) => {
546
- return queryCredentials(query);
547
- }
548
- };
549
- }));
550
- };
551
- var ConfiguredCredentialsService = class {
552
- credentials;
553
- constructor(credentials = []) {
554
- this.credentials = credentials;
555
- }
556
- addCredentials(credentials) {
557
- this.credentials.push(...credentials);
558
- return this;
559
- }
560
- async queryCredentials(query) {
561
- return this.credentials.filter((credential) => credential.service === query.service);
562
- }
563
- async getCredential(query) {
564
- const credential = this.credentials.find((credential2) => credential2.service === query.service);
565
- if (!credential) {
566
- throw new Error(`Credential not found for service: ${query.service}`);
567
- }
568
- return credential;
569
- }
570
- };
571
- var withAuthorization = (query, kind) => HttpClient.mapRequestEffect(Effect5.fnUntraced(function* (request) {
572
- const key = yield* CredentialsService.getApiKey(query).pipe(Effect5.map(Redacted.value));
573
- const authorization = kind ? `${kind} ${key}` : key;
574
- return HttpClientRequest.setHeader(request, "Authorization", authorization);
575
- }));
576
-
577
- // src/services/event-logger.ts
578
- import * as Context3 from "effect/Context";
579
- import * as Effect7 from "effect/Effect";
580
- import * as Layer3 from "effect/Layer";
581
- import * as Schema9 from "effect/Schema";
582
- import { Obj as Obj7, Type as Type6 } from "@dxos/echo";
583
- import { invariant } from "@dxos/invariant";
584
- import { LogLevel, log as log2 } from "@dxos/log";
585
-
586
- // src/services/tracing.ts
587
- import * as Context2 from "effect/Context";
588
- import * as Effect6 from "effect/Effect";
589
- import * as Layer2 from "effect/Layer";
590
- import { AgentStatus } from "@dxos/ai";
591
- import { Obj as Obj6 } from "@dxos/echo";
592
- import { Message } from "@dxos/types";
593
- var TracingService = class _TracingService extends Context2.Tag("@dxos/functions/TracingService")() {
594
- static noop = {
595
- getTraceContext: () => ({}),
596
- write: () => {
597
- }
598
- };
599
- static layerNoop = Layer2.succeed(_TracingService, _TracingService.noop);
600
- /**
601
- * Creates a TracingService layer that emits events to the parent tracing service.
602
- */
603
- static layerSubframe = (mapContext) => Layer2.effect(_TracingService, Effect6.gen(function* () {
604
- const tracing = yield* _TracingService;
605
- const context = mapContext(tracing.getTraceContext());
606
- return {
607
- write: (event) => tracing.write(event),
608
- getTraceContext: () => context
609
- };
610
- }));
611
- /**
612
- * Emit the current human-readable execution status.
613
- */
614
- static emitStatus = Effect6.fnUntraced(function* (data) {
615
- const tracing = yield* _TracingService;
616
- tracing.write(Obj6.make(AgentStatus, {
617
- parentMessage: tracing.getTraceContext().parentMessage,
618
- toolCallId: tracing.getTraceContext().toolCallId,
619
- created: (/* @__PURE__ */ new Date()).toISOString(),
620
- ...data
621
- }));
622
- });
623
- static emitConverationMessage = Effect6.fnUntraced(function* (data) {
624
- const tracing = yield* _TracingService;
625
- tracing.write(Obj6.make(Message.Message, {
626
- parentMessage: tracing.getTraceContext().parentMessage,
627
- ...data,
628
- properties: {
629
- [MESSAGE_PROPERTY_TOOL_CALL_ID]: tracing.getTraceContext().toolCallId,
630
- ...data.properties
631
- }
632
- }));
633
- });
634
- };
635
- var MESSAGE_PROPERTY_TOOL_CALL_ID = "toolCallId";
636
-
637
- // src/services/event-logger.ts
638
- var __dxlog_file = "/__w/dxos/dxos/packages/core/functions/src/services/event-logger.ts";
639
- var ComputeEventPayload = Schema9.Union(Schema9.Struct({
640
- type: Schema9.Literal("begin-compute"),
641
- nodeId: Schema9.String,
642
- /**
643
- * Names of the inputs begin computed.
644
- */
645
- inputs: Schema9.Array(Schema9.String)
646
- }), Schema9.Struct({
647
- type: Schema9.Literal("end-compute"),
648
- nodeId: Schema9.String,
649
- /**
650
- * Names of the outputs computed.
651
- */
652
- outputs: Schema9.Array(Schema9.String)
653
- }), Schema9.Struct({
654
- type: Schema9.Literal("compute-input"),
655
- nodeId: Schema9.String,
656
- property: Schema9.String,
657
- value: Schema9.Any
658
- }), Schema9.Struct({
659
- type: Schema9.Literal("compute-output"),
660
- nodeId: Schema9.String,
661
- property: Schema9.String,
662
- value: Schema9.Any
663
- }), Schema9.Struct({
664
- type: Schema9.Literal("custom"),
665
- nodeId: Schema9.String,
666
- event: Schema9.Any
667
- }));
668
- var ComputeEvent = Schema9.Struct({
669
- payload: ComputeEventPayload
670
- }).pipe(Type6.Obj({
671
- typename: "dxos.org/type/ComputeEvent",
672
- version: "0.1.0"
673
- }));
674
- var ComputeEventLogger = class _ComputeEventLogger extends Context3.Tag("@dxos/functions/ComputeEventLogger")() {
675
- static noop = {
676
- log: () => {
677
- },
678
- nodeId: void 0
679
- };
680
- /**
681
- * Implements ComputeEventLogger using TracingService.
682
- */
683
- static layerFromTracing = Layer3.effect(_ComputeEventLogger, Effect7.gen(function* () {
684
- const tracing = yield* TracingService;
685
- return {
686
- log: (event) => {
687
- tracing.write(Obj7.make(ComputeEvent, {
688
- payload: event
689
- }));
690
- },
691
- nodeId: void 0
692
- };
693
- }));
694
- };
695
- var logCustomEvent = (data) => Effect7.gen(function* () {
696
- const logger = yield* ComputeEventLogger;
697
- if (!logger.nodeId) {
698
- throw new Error("logCustomEvent must be called within a node compute function");
699
- }
700
- logger.log({
701
- type: "custom",
702
- nodeId: logger.nodeId,
703
- event: data
704
- });
705
- });
706
- var createDefectLogger = () => Effect7.catchAll((error) => Effect7.gen(function* () {
707
- log2.error("unhandled effect error", {
708
- error
709
- }, {
710
- F: __dxlog_file,
711
- L: 102,
712
- S: this,
713
- C: (f, a) => f(...a)
714
- });
715
- throw error;
716
- }));
717
- var createEventLogger = (level, message = "event") => {
718
- const logFunction = {
719
- [LogLevel.WARN]: log2.warn,
720
- [LogLevel.VERBOSE]: log2.verbose,
721
- [LogLevel.DEBUG]: log2.debug,
722
- [LogLevel.INFO]: log2.info,
723
- [LogLevel.ERROR]: log2.error
724
- }[level];
725
- invariant(logFunction, void 0, {
726
- F: __dxlog_file,
727
- L: 120,
728
- S: void 0,
729
- A: [
730
- "logFunction",
731
- ""
732
- ]
733
- });
734
- return {
735
- log: (event) => {
736
- logFunction(message, event);
737
- },
738
- nodeId: void 0
739
- };
740
- };
741
-
742
- // src/services/function-invocation-service.ts
743
- import * as Context4 from "effect/Context";
744
- import * as Effect8 from "effect/Effect";
745
- var FunctionInvocationService = class _FunctionInvocationService extends Context4.Tag("@dxos/functions/FunctionInvocationService")() {
746
- static invokeFunction = (functionDef, input) => Effect8.serviceFunctionEffect(_FunctionInvocationService, (service) => service.invokeFunction)(functionDef, input);
747
- };
748
-
749
- // src/services/queues.ts
750
- import * as Context5 from "effect/Context";
751
- import * as Effect9 from "effect/Effect";
752
- import * as Layer4 from "effect/Layer";
753
- var QueueService = class _QueueService extends Context5.Tag("@dxos/functions/QueueService")() {
754
- static notAvailable = Layer4.succeed(_QueueService, {
755
- queues: {
756
- get(_dxn) {
757
- throw new Error("Queues not available");
758
- },
759
- create() {
760
- throw new Error("Queues not available");
761
- }
762
- },
763
- queue: void 0
764
- });
765
- static make = (queues, queue) => {
766
- return {
767
- queues,
768
- queue
769
- };
770
- };
771
- static layer = (queues, queue) => Layer4.succeed(_QueueService, _QueueService.make(queues, queue));
772
- /**
773
- * Gets a queue by its DXN.
774
- */
775
- static getQueue = (dxn) => _QueueService.pipe(Effect9.map(({ queues }) => queues.get(dxn)));
776
- /**
777
- * Creates a new queue.
778
- */
779
- static createQueue = (options) => _QueueService.pipe(Effect9.map(({ queues }) => queues.create(options)));
780
- static append = (queue, objects) => Effect9.promise(() => queue.append(objects));
781
- };
782
- var ContextQueueService = class _ContextQueueService extends Context5.Tag("@dxos/functions/ContextQueueService")() {
783
- static layer = (queue) => Layer4.succeed(_ContextQueueService, {
784
- queue
785
- });
786
- };
787
-
788
- // src/protocol/protocol.ts
789
- import * as Effect10 from "effect/Effect";
790
- import * as Layer5 from "effect/Layer";
791
- import * as Schema10 from "effect/Schema";
792
- import * as SchemaAST2 from "effect/SchemaAST";
793
- import { AiService } from "@dxos/ai";
794
- import { Type as Type7 } from "@dxos/echo";
795
- import { EchoClient } from "@dxos/echo-db";
796
- import { acquireReleaseResource } from "@dxos/effect";
797
- import { failedInvariant as failedInvariant2, invariant as invariant2 } from "@dxos/invariant";
798
- import { PublicKey } from "@dxos/keys";
799
- var __dxlog_file2 = "/__w/dxos/dxos/packages/core/functions/src/protocol/protocol.ts";
800
- var wrapFunctionHandler = (func) => {
801
- if (!FunctionDefinition.isFunction(func)) {
802
- throw new TypeError("Invalid function definition");
803
- }
804
- return {
805
- meta: {
806
- key: func.key,
807
- name: func.name,
808
- description: func.description,
809
- inputSchema: Type7.toJsonSchema(func.inputSchema),
810
- outputSchema: func.outputSchema === void 0 ? void 0 : Type7.toJsonSchema(func.outputSchema),
811
- services: func.services
812
- },
813
- handler: async ({ data, context }) => {
814
- if ((func.services.includes(DatabaseService2.key) || func.services.includes(QueueService.key)) && (!context.services.dataService || !context.services.queryService)) {
815
- throw new FunctionError({
816
- message: "Services not provided: dataService, queryService"
817
- });
818
- }
819
- try {
820
- if (!SchemaAST2.isAnyKeyword(func.inputSchema.ast)) {
821
- Schema10.validateSync(func.inputSchema)(data);
822
- }
823
- let result = await func.handler({
824
- // TODO(dmaretskyi): Fix the types.
825
- context,
826
- data
827
- });
828
- if (Effect10.isEffect(result)) {
829
- result = await Effect10.runPromise(result.pipe(Effect10.orDie, Effect10.provide(createServiceLayer(context))));
830
- }
831
- if (func.outputSchema && !SchemaAST2.isAnyKeyword(func.outputSchema.ast)) {
832
- Schema10.validateSync(func.outputSchema)(result);
833
- }
834
- return result;
835
- } catch (error) {
836
- if (FunctionError.is(error)) {
837
- throw error;
838
- } else {
839
- throw new FunctionError({
840
- cause: error,
841
- context: {
842
- func: func.key
843
- }
844
- });
845
- }
846
- }
847
- }
848
- };
849
- };
850
- var createServiceLayer = (context) => {
851
- return Layer5.unwrapScoped(Effect10.gen(function* () {
852
- let client;
853
- if (context.services.dataService && context.services.queryService) {
854
- client = yield* acquireReleaseResource(() => {
855
- invariant2(context.services.dataService && context.services.queryService, void 0, {
856
- F: __dxlog_file2,
857
- L: 99,
858
- S: this,
859
- A: [
860
- "context.services.dataService && context.services.queryService",
861
- ""
862
- ]
863
- });
864
- return new EchoClient().connectToService({
865
- dataService: context.services.dataService,
866
- queryService: context.services.queryService,
867
- queueService: context.services.queueService
868
- });
869
- });
870
- }
871
- const db = client && context.spaceId ? yield* acquireReleaseResource(() => client.constructDatabase({
872
- spaceId: context.spaceId ?? failedInvariant2(),
873
- spaceKey: PublicKey.fromHex(context.spaceKey ?? failedInvariant2("spaceKey missing in context")),
874
- reactiveSchemaQuery: false
875
- })) : void 0;
876
- if (db) {
877
- console.log("Setting space root", context.spaceRootUrl);
878
- yield* Effect10.promise(() => db.setSpaceRoot(context.spaceRootUrl ?? failedInvariant2("spaceRootUrl missing in context")));
879
- }
880
- const queues = client && context.spaceId ? client.constructQueueFactory(context.spaceId) : void 0;
881
- const dbLayer = db ? DatabaseService2.layer(db) : DatabaseService2.notAvailable;
882
- const queuesLayer = queues ? QueueService.layer(queues) : QueueService.notAvailable;
883
- const credentials = dbLayer ? CredentialsService.layerFromDatabase().pipe(Layer5.provide(dbLayer)) : CredentialsService.configuredLayer([]);
884
- const functionInvocationService = MockedFunctionInvocationService;
885
- const aiService = AiService.notAvailable;
886
- const tracing = TracingService.layerNoop;
887
- return Layer5.mergeAll(dbLayer, queuesLayer, credentials, functionInvocationService, aiService, tracing);
888
- }));
889
- };
890
- var MockedFunctionInvocationService = Layer5.succeed(FunctionInvocationService, {
891
- invokeFunction: () => Effect10.die("Calling functions from functions is not implemented yet.")
892
- });
893
- export {
894
- ComputeEvent,
895
- ComputeEventLogger,
896
- ComputeEventPayload,
897
- ConfiguredCredentialsService,
898
- ContextQueueService,
899
- CredentialsService,
900
- DatabaseService2 as DatabaseService,
901
- Example,
902
- FUNCTIONS_META_KEY,
903
- FUNCTIONS_PRESET_META_KEY,
904
- Function_exports as Function,
905
- FunctionDefinition,
906
- FunctionError,
907
- FunctionInvocationService,
908
- FunctionNotFoundError,
909
- MESSAGE_PROPERTY_TOOL_CALL_ID,
910
- QueueService,
911
- Script_exports as Script,
912
- ServiceNotAvailableError,
913
- TracingService,
914
- Trigger_exports as Trigger,
915
- TriggerEvent_exports as TriggerEvent,
916
- TriggerStateNotFoundError,
917
- createDefectLogger,
918
- createEventLogger,
919
- defineFunction,
920
- deserializeFunction,
921
- getUserFunctionIdInMetadata,
922
- logCustomEvent,
923
- serializeFunction,
924
- setUserFunctionIdInMetadata,
925
- withAuthorization,
926
- wrapFunctionHandler
927
- };
928
- //# sourceMappingURL=index.mjs.map