openAgent

1import { Schema } from "effect"
2import { optional } from "./schema"
3import { statics } from "./schema"
4 
5export interface Source extends Schema.Schema.Type<typeof Source> {}
6export const Source = Schema.Struct({
7  start: Schema.Finite,
8  end: Schema.Finite,
9  text: Schema.String,
10}).annotate({ identifier: "Prompt.Source" })
11 
12export interface FileAttachment extends Schema.Schema.Type<typeof FileAttachment> {}
13export const FileAttachment = Schema.Struct({
14  uri: Schema.String,
15  mime: Schema.String,
16  name: Schema.String.pipe(optional),
17  description: Schema.String.pipe(optional),
18  source: Source.pipe(optional),
19})
20  .annotate({ identifier: "Prompt.FileAttachment" })
21  .pipe(
22    statics((schema) => ({
23      create: (input: FileAttachment) =>
24        schema.make({
25          uri: input.uri,
26          mime: input.mime,
27          name: input.name,
28          description: input.description,
29          source: input.source,
30        }),
31    })),
32  )
33 
34export interface AgentAttachment extends Schema.Schema.Type<typeof AgentAttachment> {}
35export const AgentAttachment = Schema.Struct({
36  name: Schema.String,
37  source: Source.pipe(optional),
38}).annotate({ identifier: "Prompt.AgentAttachment" })
39 
40export interface Prompt extends Schema.Schema.Type<typeof Prompt> {}
41export const Prompt = Schema.Struct({
42  text: Schema.String,
43  files: Schema.Array(FileAttachment).pipe(optional),
44  agents: Schema.Array(AgentAttachment).pipe(optional),
45})
46  .annotate({ identifier: "Prompt" })
47  .pipe(
48    statics((schema) => ({
49      equivalence: Schema.toEquivalence(schema),
50      fromUserMessage: (input: Pick<Prompt, "text" | "files" | "agents">) =>
51        schema.make({
52          text: input.text,
53          ...(input.files === undefined ? {} : { files: input.files }),
54          ...(input.agents === undefined ? {} : { agents: input.agents }),
55        }),
56    })),
57  )
58