openAgent

1export * as Model from "./model"
2 
3import { Schema } from "effect"
4import { optional } from "./schema"
5import { Provider } from "./provider"
6import { statics } from "./schema"
7 
8export const ID = Schema.String.pipe(Schema.brand("ModelV2.ID"))
9export type ID = typeof ID.Type
10 
11export const VariantID = Schema.String.pipe(Schema.brand("VariantID"))
12export type VariantID = typeof VariantID.Type
13 
14export const Ref = Schema.Struct({
15  id: ID,
16  providerID: Provider.ID,
17  variant: VariantID.pipe(optional),
18}).annotate({ identifier: "Model.Ref" })
19export interface Ref extends Schema.Schema.Type<typeof Ref> {}
20 
21export const Family = Schema.String.pipe(Schema.brand("Family"))
22export type Family = typeof Family.Type
23 
24export interface Capabilities extends Schema.Schema.Type<typeof Capabilities> {}
25export const Capabilities = Schema.Struct({
26  tools: Schema.Boolean,
27  input: Schema.Array(Schema.String),
28  output: Schema.Array(Schema.String),
29}).annotate({ identifier: "Model.Capabilities" })
30 
31export interface Cost extends Schema.Schema.Type<typeof Cost> {}
32export const Cost = Schema.Struct({
33  tier: Schema.Struct({
34    type: Schema.Literal("context"),
35    size: Schema.Int,
36  }).pipe(optional),
37  input: Schema.Finite,
38  output: Schema.Finite,
39  cache: Schema.Struct({
40    read: Schema.Finite,
41    write: Schema.Finite,
42  }),
43}).annotate({ identifier: "Model.Cost" })
44 
45export const Api = Schema.Union([
46  Schema.Struct({
47    id: ID,
48    ...Provider.AISDK.fields,
49  }),
50  Schema.Struct({
51    id: ID,
52    ...Provider.Native.fields,
53  }),
54])
55  .pipe(Schema.toTaggedUnion("type"))
56  .annotate({ identifier: "Model.Api" })
57export type Api = typeof Api.Type
58 
59export interface Info extends Schema.Schema.Type<typeof Info> {}
60export const Info = Schema.Struct({
61  id: ID,
62  providerID: Provider.ID,
63  family: Family.pipe(optional),
64  name: Schema.String,
65  api: Api,
66  capabilities: Capabilities,
67  request: Schema.Struct({
68    ...Provider.Request.fields,
69    variant: Schema.String.pipe(optional),
70  }),
71  variants: Schema.Struct({
72    id: VariantID,
73    ...Provider.Request.fields,
74  }).pipe(Schema.Array),
75  time: Schema.Struct({
76    released: Schema.Finite,
77  }),
78  cost: Schema.Array(Cost),
79  status: Schema.Literals(["alpha", "beta", "deprecated", "active"]),
80  enabled: Schema.Boolean,
81  limit: Schema.Struct({
82    context: Schema.Int,
83    input: Schema.Int.pipe(optional),
84    output: Schema.Int,
85  }),
86})
87  .annotate({ identifier: "ModelV2.Info" })
88  .pipe(
89    statics((schema) => ({
90      empty: (providerID: Provider.ID, modelID: ID) =>
91        schema.make({
92          id: modelID,
93          providerID,
94          name: modelID,
95          api: { id: modelID, type: "native", settings: {} },
96          capabilities: { tools: false, input: [], output: [] },
97          request: { headers: {}, body: {} },
98          variants: [],
99          time: { released: 0 },
100          cost: [],
101          status: "active",
102          enabled: true,
103          limit: { context: 0, output: 0 },
104        }),
105    })),
106  )
107