session.go 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644
  1. // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
  2. package opencode
  3. import (
  4. "context"
  5. "errors"
  6. "fmt"
  7. "net/http"
  8. "reflect"
  9. "github.com/sst/opencode-sdk-go/internal/apijson"
  10. "github.com/sst/opencode-sdk-go/internal/param"
  11. "github.com/sst/opencode-sdk-go/internal/requestconfig"
  12. "github.com/sst/opencode-sdk-go/option"
  13. "github.com/sst/opencode-sdk-go/shared"
  14. "github.com/tidwall/gjson"
  15. )
  16. // SessionService contains methods and other services that help with interacting
  17. // with the opencode API.
  18. //
  19. // Note, unlike clients, this service does not read variables from the environment
  20. // automatically. You should not instantiate this service directly, and instead use
  21. // the [NewSessionService] method instead.
  22. type SessionService struct {
  23. Options []option.RequestOption
  24. }
  25. // NewSessionService generates a new service that applies the given options to each
  26. // request. These options are applied after the parent client's options (if there
  27. // is one), and before any request-specific options.
  28. func NewSessionService(opts ...option.RequestOption) (r *SessionService) {
  29. r = &SessionService{}
  30. r.Options = opts
  31. return
  32. }
  33. // Create a new session
  34. func (r *SessionService) New(ctx context.Context, opts ...option.RequestOption) (res *Session, err error) {
  35. opts = append(r.Options[:], opts...)
  36. path := "session"
  37. err = requestconfig.ExecuteNewRequest(ctx, http.MethodPost, path, nil, &res, opts...)
  38. return
  39. }
  40. // List all sessions
  41. func (r *SessionService) List(ctx context.Context, opts ...option.RequestOption) (res *[]Session, err error) {
  42. opts = append(r.Options[:], opts...)
  43. path := "session"
  44. err = requestconfig.ExecuteNewRequest(ctx, http.MethodGet, path, nil, &res, opts...)
  45. return
  46. }
  47. // Delete a session and all its data
  48. func (r *SessionService) Delete(ctx context.Context, id string, opts ...option.RequestOption) (res *bool, err error) {
  49. opts = append(r.Options[:], opts...)
  50. if id == "" {
  51. err = errors.New("missing required id parameter")
  52. return
  53. }
  54. path := fmt.Sprintf("session/%s", id)
  55. err = requestconfig.ExecuteNewRequest(ctx, http.MethodDelete, path, nil, &res, opts...)
  56. return
  57. }
  58. // Abort a session
  59. func (r *SessionService) Abort(ctx context.Context, id string, opts ...option.RequestOption) (res *bool, err error) {
  60. opts = append(r.Options[:], opts...)
  61. if id == "" {
  62. err = errors.New("missing required id parameter")
  63. return
  64. }
  65. path := fmt.Sprintf("session/%s/abort", id)
  66. err = requestconfig.ExecuteNewRequest(ctx, http.MethodPost, path, nil, &res, opts...)
  67. return
  68. }
  69. // Create and send a new message to a session
  70. func (r *SessionService) Chat(ctx context.Context, id string, body SessionChatParams, opts ...option.RequestOption) (res *AssistantMessage, err error) {
  71. opts = append(r.Options[:], opts...)
  72. if id == "" {
  73. err = errors.New("missing required id parameter")
  74. return
  75. }
  76. path := fmt.Sprintf("session/%s/message", id)
  77. err = requestconfig.ExecuteNewRequest(ctx, http.MethodPost, path, body, &res, opts...)
  78. return
  79. }
  80. // Analyze the app and create an AGENTS.md file
  81. func (r *SessionService) Init(ctx context.Context, id string, body SessionInitParams, opts ...option.RequestOption) (res *bool, err error) {
  82. opts = append(r.Options[:], opts...)
  83. if id == "" {
  84. err = errors.New("missing required id parameter")
  85. return
  86. }
  87. path := fmt.Sprintf("session/%s/init", id)
  88. err = requestconfig.ExecuteNewRequest(ctx, http.MethodPost, path, body, &res, opts...)
  89. return
  90. }
  91. // List messages for a session
  92. func (r *SessionService) Messages(ctx context.Context, id string, opts ...option.RequestOption) (res *[]SessionMessagesResponse, err error) {
  93. opts = append(r.Options[:], opts...)
  94. if id == "" {
  95. err = errors.New("missing required id parameter")
  96. return
  97. }
  98. path := fmt.Sprintf("session/%s/message", id)
  99. err = requestconfig.ExecuteNewRequest(ctx, http.MethodGet, path, nil, &res, opts...)
  100. return
  101. }
  102. // Share a session
  103. func (r *SessionService) Share(ctx context.Context, id string, opts ...option.RequestOption) (res *Session, err error) {
  104. opts = append(r.Options[:], opts...)
  105. if id == "" {
  106. err = errors.New("missing required id parameter")
  107. return
  108. }
  109. path := fmt.Sprintf("session/%s/share", id)
  110. err = requestconfig.ExecuteNewRequest(ctx, http.MethodPost, path, nil, &res, opts...)
  111. return
  112. }
  113. // Summarize the session
  114. func (r *SessionService) Summarize(ctx context.Context, id string, body SessionSummarizeParams, opts ...option.RequestOption) (res *bool, err error) {
  115. opts = append(r.Options[:], opts...)
  116. if id == "" {
  117. err = errors.New("missing required id parameter")
  118. return
  119. }
  120. path := fmt.Sprintf("session/%s/summarize", id)
  121. err = requestconfig.ExecuteNewRequest(ctx, http.MethodPost, path, body, &res, opts...)
  122. return
  123. }
  124. // Unshare the session
  125. func (r *SessionService) Unshare(ctx context.Context, id string, opts ...option.RequestOption) (res *Session, err error) {
  126. opts = append(r.Options[:], opts...)
  127. if id == "" {
  128. err = errors.New("missing required id parameter")
  129. return
  130. }
  131. path := fmt.Sprintf("session/%s/share", id)
  132. err = requestconfig.ExecuteNewRequest(ctx, http.MethodDelete, path, nil, &res, opts...)
  133. return
  134. }
  135. type AssistantMessage struct {
  136. ID string `json:"id,required"`
  137. Cost float64 `json:"cost,required"`
  138. ModelID string `json:"modelID,required"`
  139. Path AssistantMessagePath `json:"path,required"`
  140. ProviderID string `json:"providerID,required"`
  141. Role AssistantMessageRole `json:"role,required"`
  142. SessionID string `json:"sessionID,required"`
  143. System []string `json:"system,required"`
  144. Time AssistantMessageTime `json:"time,required"`
  145. Tokens AssistantMessageTokens `json:"tokens,required"`
  146. Error AssistantMessageError `json:"error"`
  147. Summary bool `json:"summary"`
  148. JSON assistantMessageJSON `json:"-"`
  149. }
  150. // assistantMessageJSON contains the JSON metadata for the struct
  151. // [AssistantMessage]
  152. type assistantMessageJSON struct {
  153. ID apijson.Field
  154. Cost apijson.Field
  155. ModelID apijson.Field
  156. Path apijson.Field
  157. ProviderID apijson.Field
  158. Role apijson.Field
  159. SessionID apijson.Field
  160. System apijson.Field
  161. Time apijson.Field
  162. Tokens apijson.Field
  163. Error apijson.Field
  164. Summary apijson.Field
  165. raw string
  166. ExtraFields map[string]apijson.Field
  167. }
  168. func (r *AssistantMessage) UnmarshalJSON(data []byte) (err error) {
  169. return apijson.UnmarshalRoot(data, r)
  170. }
  171. func (r assistantMessageJSON) RawJSON() string {
  172. return r.raw
  173. }
  174. func (r AssistantMessage) implementsMessage() {}
  175. type AssistantMessagePath struct {
  176. Cwd string `json:"cwd,required"`
  177. Root string `json:"root,required"`
  178. JSON assistantMessagePathJSON `json:"-"`
  179. }
  180. // assistantMessagePathJSON contains the JSON metadata for the struct
  181. // [AssistantMessagePath]
  182. type assistantMessagePathJSON struct {
  183. Cwd apijson.Field
  184. Root apijson.Field
  185. raw string
  186. ExtraFields map[string]apijson.Field
  187. }
  188. func (r *AssistantMessagePath) UnmarshalJSON(data []byte) (err error) {
  189. return apijson.UnmarshalRoot(data, r)
  190. }
  191. func (r assistantMessagePathJSON) RawJSON() string {
  192. return r.raw
  193. }
  194. type AssistantMessageRole string
  195. const (
  196. AssistantMessageRoleAssistant AssistantMessageRole = "assistant"
  197. )
  198. func (r AssistantMessageRole) IsKnown() bool {
  199. switch r {
  200. case AssistantMessageRoleAssistant:
  201. return true
  202. }
  203. return false
  204. }
  205. type AssistantMessageTime struct {
  206. Created float64 `json:"created,required"`
  207. Completed float64 `json:"completed"`
  208. JSON assistantMessageTimeJSON `json:"-"`
  209. }
  210. // assistantMessageTimeJSON contains the JSON metadata for the struct
  211. // [AssistantMessageTime]
  212. type assistantMessageTimeJSON struct {
  213. Created apijson.Field
  214. Completed apijson.Field
  215. raw string
  216. ExtraFields map[string]apijson.Field
  217. }
  218. func (r *AssistantMessageTime) UnmarshalJSON(data []byte) (err error) {
  219. return apijson.UnmarshalRoot(data, r)
  220. }
  221. func (r assistantMessageTimeJSON) RawJSON() string {
  222. return r.raw
  223. }
  224. type AssistantMessageTokens struct {
  225. Cache AssistantMessageTokensCache `json:"cache,required"`
  226. Input float64 `json:"input,required"`
  227. Output float64 `json:"output,required"`
  228. Reasoning float64 `json:"reasoning,required"`
  229. JSON assistantMessageTokensJSON `json:"-"`
  230. }
  231. // assistantMessageTokensJSON contains the JSON metadata for the struct
  232. // [AssistantMessageTokens]
  233. type assistantMessageTokensJSON struct {
  234. Cache apijson.Field
  235. Input apijson.Field
  236. Output apijson.Field
  237. Reasoning apijson.Field
  238. raw string
  239. ExtraFields map[string]apijson.Field
  240. }
  241. func (r *AssistantMessageTokens) UnmarshalJSON(data []byte) (err error) {
  242. return apijson.UnmarshalRoot(data, r)
  243. }
  244. func (r assistantMessageTokensJSON) RawJSON() string {
  245. return r.raw
  246. }
  247. type AssistantMessageTokensCache struct {
  248. Read float64 `json:"read,required"`
  249. Write float64 `json:"write,required"`
  250. JSON assistantMessageTokensCacheJSON `json:"-"`
  251. }
  252. // assistantMessageTokensCacheJSON contains the JSON metadata for the struct
  253. // [AssistantMessageTokensCache]
  254. type assistantMessageTokensCacheJSON struct {
  255. Read apijson.Field
  256. Write apijson.Field
  257. raw string
  258. ExtraFields map[string]apijson.Field
  259. }
  260. func (r *AssistantMessageTokensCache) UnmarshalJSON(data []byte) (err error) {
  261. return apijson.UnmarshalRoot(data, r)
  262. }
  263. func (r assistantMessageTokensCacheJSON) RawJSON() string {
  264. return r.raw
  265. }
  266. type AssistantMessageError struct {
  267. // This field can have the runtime type of [shared.ProviderAuthErrorData],
  268. // [shared.UnknownErrorData], [interface{}].
  269. Data interface{} `json:"data,required"`
  270. Name AssistantMessageErrorName `json:"name,required"`
  271. JSON assistantMessageErrorJSON `json:"-"`
  272. union AssistantMessageErrorUnion
  273. }
  274. // assistantMessageErrorJSON contains the JSON metadata for the struct
  275. // [AssistantMessageError]
  276. type assistantMessageErrorJSON struct {
  277. Data apijson.Field
  278. Name apijson.Field
  279. raw string
  280. ExtraFields map[string]apijson.Field
  281. }
  282. func (r assistantMessageErrorJSON) RawJSON() string {
  283. return r.raw
  284. }
  285. func (r *AssistantMessageError) UnmarshalJSON(data []byte) (err error) {
  286. *r = AssistantMessageError{}
  287. err = apijson.UnmarshalRoot(data, &r.union)
  288. if err != nil {
  289. return err
  290. }
  291. return apijson.Port(r.union, &r)
  292. }
  293. // AsUnion returns a [AssistantMessageErrorUnion] interface which you can cast to
  294. // the specific types for more type safety.
  295. //
  296. // Possible runtime types of the union are [shared.ProviderAuthError],
  297. // [shared.UnknownError], [AssistantMessageErrorMessageOutputLengthError],
  298. // [shared.MessageAbortedError].
  299. func (r AssistantMessageError) AsUnion() AssistantMessageErrorUnion {
  300. return r.union
  301. }
  302. // Union satisfied by [shared.ProviderAuthError], [shared.UnknownError],
  303. // [AssistantMessageErrorMessageOutputLengthError] or [shared.MessageAbortedError].
  304. type AssistantMessageErrorUnion interface {
  305. ImplementsAssistantMessageError()
  306. }
  307. func init() {
  308. apijson.RegisterUnion(
  309. reflect.TypeOf((*AssistantMessageErrorUnion)(nil)).Elem(),
  310. "name",
  311. apijson.UnionVariant{
  312. TypeFilter: gjson.JSON,
  313. Type: reflect.TypeOf(shared.ProviderAuthError{}),
  314. DiscriminatorValue: "ProviderAuthError",
  315. },
  316. apijson.UnionVariant{
  317. TypeFilter: gjson.JSON,
  318. Type: reflect.TypeOf(shared.UnknownError{}),
  319. DiscriminatorValue: "UnknownError",
  320. },
  321. apijson.UnionVariant{
  322. TypeFilter: gjson.JSON,
  323. Type: reflect.TypeOf(AssistantMessageErrorMessageOutputLengthError{}),
  324. DiscriminatorValue: "MessageOutputLengthError",
  325. },
  326. apijson.UnionVariant{
  327. TypeFilter: gjson.JSON,
  328. Type: reflect.TypeOf(shared.MessageAbortedError{}),
  329. DiscriminatorValue: "MessageAbortedError",
  330. },
  331. )
  332. }
  333. type AssistantMessageErrorMessageOutputLengthError struct {
  334. Data interface{} `json:"data,required"`
  335. Name AssistantMessageErrorMessageOutputLengthErrorName `json:"name,required"`
  336. JSON assistantMessageErrorMessageOutputLengthErrorJSON `json:"-"`
  337. }
  338. // assistantMessageErrorMessageOutputLengthErrorJSON contains the JSON metadata for
  339. // the struct [AssistantMessageErrorMessageOutputLengthError]
  340. type assistantMessageErrorMessageOutputLengthErrorJSON struct {
  341. Data apijson.Field
  342. Name apijson.Field
  343. raw string
  344. ExtraFields map[string]apijson.Field
  345. }
  346. func (r *AssistantMessageErrorMessageOutputLengthError) UnmarshalJSON(data []byte) (err error) {
  347. return apijson.UnmarshalRoot(data, r)
  348. }
  349. func (r assistantMessageErrorMessageOutputLengthErrorJSON) RawJSON() string {
  350. return r.raw
  351. }
  352. func (r AssistantMessageErrorMessageOutputLengthError) ImplementsAssistantMessageError() {}
  353. type AssistantMessageErrorMessageOutputLengthErrorName string
  354. const (
  355. AssistantMessageErrorMessageOutputLengthErrorNameMessageOutputLengthError AssistantMessageErrorMessageOutputLengthErrorName = "MessageOutputLengthError"
  356. )
  357. func (r AssistantMessageErrorMessageOutputLengthErrorName) IsKnown() bool {
  358. switch r {
  359. case AssistantMessageErrorMessageOutputLengthErrorNameMessageOutputLengthError:
  360. return true
  361. }
  362. return false
  363. }
  364. type AssistantMessageErrorName string
  365. const (
  366. AssistantMessageErrorNameProviderAuthError AssistantMessageErrorName = "ProviderAuthError"
  367. AssistantMessageErrorNameUnknownError AssistantMessageErrorName = "UnknownError"
  368. AssistantMessageErrorNameMessageOutputLengthError AssistantMessageErrorName = "MessageOutputLengthError"
  369. AssistantMessageErrorNameMessageAbortedError AssistantMessageErrorName = "MessageAbortedError"
  370. )
  371. func (r AssistantMessageErrorName) IsKnown() bool {
  372. switch r {
  373. case AssistantMessageErrorNameProviderAuthError, AssistantMessageErrorNameUnknownError, AssistantMessageErrorNameMessageOutputLengthError, AssistantMessageErrorNameMessageAbortedError:
  374. return true
  375. }
  376. return false
  377. }
  378. type FilePart struct {
  379. ID string `json:"id,required"`
  380. MessageID string `json:"messageID,required"`
  381. Mime string `json:"mime,required"`
  382. SessionID string `json:"sessionID,required"`
  383. Type FilePartType `json:"type,required"`
  384. URL string `json:"url,required"`
  385. Filename string `json:"filename"`
  386. JSON filePartJSON `json:"-"`
  387. }
  388. // filePartJSON contains the JSON metadata for the struct [FilePart]
  389. type filePartJSON struct {
  390. ID apijson.Field
  391. MessageID apijson.Field
  392. Mime apijson.Field
  393. SessionID apijson.Field
  394. Type apijson.Field
  395. URL apijson.Field
  396. Filename apijson.Field
  397. raw string
  398. ExtraFields map[string]apijson.Field
  399. }
  400. func (r *FilePart) UnmarshalJSON(data []byte) (err error) {
  401. return apijson.UnmarshalRoot(data, r)
  402. }
  403. func (r filePartJSON) RawJSON() string {
  404. return r.raw
  405. }
  406. func (r FilePart) implementsPart() {}
  407. type FilePartType string
  408. const (
  409. FilePartTypeFile FilePartType = "file"
  410. )
  411. func (r FilePartType) IsKnown() bool {
  412. switch r {
  413. case FilePartTypeFile:
  414. return true
  415. }
  416. return false
  417. }
  418. type FilePartParam struct {
  419. ID param.Field[string] `json:"id,required"`
  420. MessageID param.Field[string] `json:"messageID,required"`
  421. Mime param.Field[string] `json:"mime,required"`
  422. SessionID param.Field[string] `json:"sessionID,required"`
  423. Type param.Field[FilePartType] `json:"type,required"`
  424. URL param.Field[string] `json:"url,required"`
  425. Filename param.Field[string] `json:"filename"`
  426. }
  427. func (r FilePartParam) MarshalJSON() (data []byte, err error) {
  428. return apijson.MarshalRoot(r)
  429. }
  430. func (r FilePartParam) implementsSessionChatParamsPartUnion() {}
  431. type Message struct {
  432. ID string `json:"id,required"`
  433. Role MessageRole `json:"role,required"`
  434. SessionID string `json:"sessionID,required"`
  435. // This field can have the runtime type of [UserMessageTime],
  436. // [AssistantMessageTime].
  437. Time interface{} `json:"time,required"`
  438. Cost float64 `json:"cost"`
  439. // This field can have the runtime type of [AssistantMessageError].
  440. Error interface{} `json:"error"`
  441. ModelID string `json:"modelID"`
  442. // This field can have the runtime type of [AssistantMessagePath].
  443. Path interface{} `json:"path"`
  444. ProviderID string `json:"providerID"`
  445. Summary bool `json:"summary"`
  446. // This field can have the runtime type of [[]string].
  447. System interface{} `json:"system"`
  448. // This field can have the runtime type of [AssistantMessageTokens].
  449. Tokens interface{} `json:"tokens"`
  450. JSON messageJSON `json:"-"`
  451. union MessageUnion
  452. }
  453. // messageJSON contains the JSON metadata for the struct [Message]
  454. type messageJSON struct {
  455. ID apijson.Field
  456. Role apijson.Field
  457. SessionID apijson.Field
  458. Time apijson.Field
  459. Cost apijson.Field
  460. Error apijson.Field
  461. ModelID apijson.Field
  462. Path apijson.Field
  463. ProviderID apijson.Field
  464. Summary apijson.Field
  465. System apijson.Field
  466. Tokens apijson.Field
  467. raw string
  468. ExtraFields map[string]apijson.Field
  469. }
  470. func (r messageJSON) RawJSON() string {
  471. return r.raw
  472. }
  473. func (r *Message) UnmarshalJSON(data []byte) (err error) {
  474. *r = Message{}
  475. err = apijson.UnmarshalRoot(data, &r.union)
  476. if err != nil {
  477. return err
  478. }
  479. return apijson.Port(r.union, &r)
  480. }
  481. // AsUnion returns a [MessageUnion] interface which you can cast to the specific
  482. // types for more type safety.
  483. //
  484. // Possible runtime types of the union are [UserMessage], [AssistantMessage].
  485. func (r Message) AsUnion() MessageUnion {
  486. return r.union
  487. }
  488. // Union satisfied by [UserMessage] or [AssistantMessage].
  489. type MessageUnion interface {
  490. implementsMessage()
  491. }
  492. func init() {
  493. apijson.RegisterUnion(
  494. reflect.TypeOf((*MessageUnion)(nil)).Elem(),
  495. "role",
  496. apijson.UnionVariant{
  497. TypeFilter: gjson.JSON,
  498. Type: reflect.TypeOf(UserMessage{}),
  499. DiscriminatorValue: "user",
  500. },
  501. apijson.UnionVariant{
  502. TypeFilter: gjson.JSON,
  503. Type: reflect.TypeOf(AssistantMessage{}),
  504. DiscriminatorValue: "assistant",
  505. },
  506. )
  507. }
  508. type MessageRole string
  509. const (
  510. MessageRoleUser MessageRole = "user"
  511. MessageRoleAssistant MessageRole = "assistant"
  512. )
  513. func (r MessageRole) IsKnown() bool {
  514. switch r {
  515. case MessageRoleUser, MessageRoleAssistant:
  516. return true
  517. }
  518. return false
  519. }
  520. type Part struct {
  521. ID string `json:"id,required"`
  522. MessageID string `json:"messageID,required"`
  523. SessionID string `json:"sessionID,required"`
  524. Type PartType `json:"type,required"`
  525. CallID string `json:"callID"`
  526. Cost float64 `json:"cost"`
  527. Filename string `json:"filename"`
  528. Mime string `json:"mime"`
  529. Snapshot string `json:"snapshot"`
  530. // This field can have the runtime type of [ToolPartState].
  531. State interface{} `json:"state"`
  532. Synthetic bool `json:"synthetic"`
  533. Text string `json:"text"`
  534. // This field can have the runtime type of [TextPartTime].
  535. Time interface{} `json:"time"`
  536. // This field can have the runtime type of [StepFinishPartTokens].
  537. Tokens interface{} `json:"tokens"`
  538. Tool string `json:"tool"`
  539. URL string `json:"url"`
  540. JSON partJSON `json:"-"`
  541. union PartUnion
  542. }
  543. // partJSON contains the JSON metadata for the struct [Part]
  544. type partJSON struct {
  545. ID apijson.Field
  546. MessageID apijson.Field
  547. SessionID apijson.Field
  548. Type apijson.Field
  549. CallID apijson.Field
  550. Cost apijson.Field
  551. Filename apijson.Field
  552. Mime apijson.Field
  553. Snapshot apijson.Field
  554. State apijson.Field
  555. Synthetic apijson.Field
  556. Text apijson.Field
  557. Time apijson.Field
  558. Tokens apijson.Field
  559. Tool apijson.Field
  560. URL apijson.Field
  561. raw string
  562. ExtraFields map[string]apijson.Field
  563. }
  564. func (r partJSON) RawJSON() string {
  565. return r.raw
  566. }
  567. func (r *Part) UnmarshalJSON(data []byte) (err error) {
  568. *r = Part{}
  569. err = apijson.UnmarshalRoot(data, &r.union)
  570. if err != nil {
  571. return err
  572. }
  573. return apijson.Port(r.union, &r)
  574. }
  575. // AsUnion returns a [PartUnion] interface which you can cast to the specific types
  576. // for more type safety.
  577. //
  578. // Possible runtime types of the union are [TextPart], [FilePart], [ToolPart],
  579. // [StepStartPart], [StepFinishPart], [SnapshotPart].
  580. func (r Part) AsUnion() PartUnion {
  581. return r.union
  582. }
  583. // Union satisfied by [TextPart], [FilePart], [ToolPart], [StepStartPart],
  584. // [StepFinishPart] or [SnapshotPart].
  585. type PartUnion interface {
  586. implementsPart()
  587. }
  588. func init() {
  589. apijson.RegisterUnion(
  590. reflect.TypeOf((*PartUnion)(nil)).Elem(),
  591. "type",
  592. apijson.UnionVariant{
  593. TypeFilter: gjson.JSON,
  594. Type: reflect.TypeOf(TextPart{}),
  595. DiscriminatorValue: "text",
  596. },
  597. apijson.UnionVariant{
  598. TypeFilter: gjson.JSON,
  599. Type: reflect.TypeOf(FilePart{}),
  600. DiscriminatorValue: "file",
  601. },
  602. apijson.UnionVariant{
  603. TypeFilter: gjson.JSON,
  604. Type: reflect.TypeOf(ToolPart{}),
  605. DiscriminatorValue: "tool",
  606. },
  607. apijson.UnionVariant{
  608. TypeFilter: gjson.JSON,
  609. Type: reflect.TypeOf(StepStartPart{}),
  610. DiscriminatorValue: "step-start",
  611. },
  612. apijson.UnionVariant{
  613. TypeFilter: gjson.JSON,
  614. Type: reflect.TypeOf(StepFinishPart{}),
  615. DiscriminatorValue: "step-finish",
  616. },
  617. apijson.UnionVariant{
  618. TypeFilter: gjson.JSON,
  619. Type: reflect.TypeOf(SnapshotPart{}),
  620. DiscriminatorValue: "snapshot",
  621. },
  622. )
  623. }
  624. type PartType string
  625. const (
  626. PartTypeText PartType = "text"
  627. PartTypeFile PartType = "file"
  628. PartTypeTool PartType = "tool"
  629. PartTypeStepStart PartType = "step-start"
  630. PartTypeStepFinish PartType = "step-finish"
  631. PartTypeSnapshot PartType = "snapshot"
  632. )
  633. func (r PartType) IsKnown() bool {
  634. switch r {
  635. case PartTypeText, PartTypeFile, PartTypeTool, PartTypeStepStart, PartTypeStepFinish, PartTypeSnapshot:
  636. return true
  637. }
  638. return false
  639. }
  640. type Session struct {
  641. ID string `json:"id,required"`
  642. Time SessionTime `json:"time,required"`
  643. Title string `json:"title,required"`
  644. Version string `json:"version,required"`
  645. ParentID string `json:"parentID"`
  646. Revert SessionRevert `json:"revert"`
  647. Share SessionShare `json:"share"`
  648. JSON sessionJSON `json:"-"`
  649. }
  650. // sessionJSON contains the JSON metadata for the struct [Session]
  651. type sessionJSON struct {
  652. ID apijson.Field
  653. Time apijson.Field
  654. Title apijson.Field
  655. Version apijson.Field
  656. ParentID apijson.Field
  657. Revert apijson.Field
  658. Share apijson.Field
  659. raw string
  660. ExtraFields map[string]apijson.Field
  661. }
  662. func (r *Session) UnmarshalJSON(data []byte) (err error) {
  663. return apijson.UnmarshalRoot(data, r)
  664. }
  665. func (r sessionJSON) RawJSON() string {
  666. return r.raw
  667. }
  668. type SessionTime struct {
  669. Created float64 `json:"created,required"`
  670. Updated float64 `json:"updated,required"`
  671. JSON sessionTimeJSON `json:"-"`
  672. }
  673. // sessionTimeJSON contains the JSON metadata for the struct [SessionTime]
  674. type sessionTimeJSON struct {
  675. Created apijson.Field
  676. Updated apijson.Field
  677. raw string
  678. ExtraFields map[string]apijson.Field
  679. }
  680. func (r *SessionTime) UnmarshalJSON(data []byte) (err error) {
  681. return apijson.UnmarshalRoot(data, r)
  682. }
  683. func (r sessionTimeJSON) RawJSON() string {
  684. return r.raw
  685. }
  686. type SessionRevert struct {
  687. MessageID string `json:"messageID,required"`
  688. Part float64 `json:"part,required"`
  689. Snapshot string `json:"snapshot"`
  690. JSON sessionRevertJSON `json:"-"`
  691. }
  692. // sessionRevertJSON contains the JSON metadata for the struct [SessionRevert]
  693. type sessionRevertJSON struct {
  694. MessageID apijson.Field
  695. Part apijson.Field
  696. Snapshot apijson.Field
  697. raw string
  698. ExtraFields map[string]apijson.Field
  699. }
  700. func (r *SessionRevert) UnmarshalJSON(data []byte) (err error) {
  701. return apijson.UnmarshalRoot(data, r)
  702. }
  703. func (r sessionRevertJSON) RawJSON() string {
  704. return r.raw
  705. }
  706. type SessionShare struct {
  707. URL string `json:"url,required"`
  708. JSON sessionShareJSON `json:"-"`
  709. }
  710. // sessionShareJSON contains the JSON metadata for the struct [SessionShare]
  711. type sessionShareJSON struct {
  712. URL apijson.Field
  713. raw string
  714. ExtraFields map[string]apijson.Field
  715. }
  716. func (r *SessionShare) UnmarshalJSON(data []byte) (err error) {
  717. return apijson.UnmarshalRoot(data, r)
  718. }
  719. func (r sessionShareJSON) RawJSON() string {
  720. return r.raw
  721. }
  722. type SnapshotPart struct {
  723. ID string `json:"id,required"`
  724. MessageID string `json:"messageID,required"`
  725. SessionID string `json:"sessionID,required"`
  726. Snapshot string `json:"snapshot,required"`
  727. Type SnapshotPartType `json:"type,required"`
  728. JSON snapshotPartJSON `json:"-"`
  729. }
  730. // snapshotPartJSON contains the JSON metadata for the struct [SnapshotPart]
  731. type snapshotPartJSON struct {
  732. ID apijson.Field
  733. MessageID apijson.Field
  734. SessionID apijson.Field
  735. Snapshot apijson.Field
  736. Type apijson.Field
  737. raw string
  738. ExtraFields map[string]apijson.Field
  739. }
  740. func (r *SnapshotPart) UnmarshalJSON(data []byte) (err error) {
  741. return apijson.UnmarshalRoot(data, r)
  742. }
  743. func (r snapshotPartJSON) RawJSON() string {
  744. return r.raw
  745. }
  746. func (r SnapshotPart) implementsPart() {}
  747. type SnapshotPartType string
  748. const (
  749. SnapshotPartTypeSnapshot SnapshotPartType = "snapshot"
  750. )
  751. func (r SnapshotPartType) IsKnown() bool {
  752. switch r {
  753. case SnapshotPartTypeSnapshot:
  754. return true
  755. }
  756. return false
  757. }
  758. type StepFinishPart struct {
  759. ID string `json:"id,required"`
  760. Cost float64 `json:"cost,required"`
  761. MessageID string `json:"messageID,required"`
  762. SessionID string `json:"sessionID,required"`
  763. Tokens StepFinishPartTokens `json:"tokens,required"`
  764. Type StepFinishPartType `json:"type,required"`
  765. JSON stepFinishPartJSON `json:"-"`
  766. }
  767. // stepFinishPartJSON contains the JSON metadata for the struct [StepFinishPart]
  768. type stepFinishPartJSON struct {
  769. ID apijson.Field
  770. Cost apijson.Field
  771. MessageID apijson.Field
  772. SessionID apijson.Field
  773. Tokens apijson.Field
  774. Type apijson.Field
  775. raw string
  776. ExtraFields map[string]apijson.Field
  777. }
  778. func (r *StepFinishPart) UnmarshalJSON(data []byte) (err error) {
  779. return apijson.UnmarshalRoot(data, r)
  780. }
  781. func (r stepFinishPartJSON) RawJSON() string {
  782. return r.raw
  783. }
  784. func (r StepFinishPart) implementsPart() {}
  785. type StepFinishPartTokens struct {
  786. Cache StepFinishPartTokensCache `json:"cache,required"`
  787. Input float64 `json:"input,required"`
  788. Output float64 `json:"output,required"`
  789. Reasoning float64 `json:"reasoning,required"`
  790. JSON stepFinishPartTokensJSON `json:"-"`
  791. }
  792. // stepFinishPartTokensJSON contains the JSON metadata for the struct
  793. // [StepFinishPartTokens]
  794. type stepFinishPartTokensJSON struct {
  795. Cache apijson.Field
  796. Input apijson.Field
  797. Output apijson.Field
  798. Reasoning apijson.Field
  799. raw string
  800. ExtraFields map[string]apijson.Field
  801. }
  802. func (r *StepFinishPartTokens) UnmarshalJSON(data []byte) (err error) {
  803. return apijson.UnmarshalRoot(data, r)
  804. }
  805. func (r stepFinishPartTokensJSON) RawJSON() string {
  806. return r.raw
  807. }
  808. type StepFinishPartTokensCache struct {
  809. Read float64 `json:"read,required"`
  810. Write float64 `json:"write,required"`
  811. JSON stepFinishPartTokensCacheJSON `json:"-"`
  812. }
  813. // stepFinishPartTokensCacheJSON contains the JSON metadata for the struct
  814. // [StepFinishPartTokensCache]
  815. type stepFinishPartTokensCacheJSON struct {
  816. Read apijson.Field
  817. Write apijson.Field
  818. raw string
  819. ExtraFields map[string]apijson.Field
  820. }
  821. func (r *StepFinishPartTokensCache) UnmarshalJSON(data []byte) (err error) {
  822. return apijson.UnmarshalRoot(data, r)
  823. }
  824. func (r stepFinishPartTokensCacheJSON) RawJSON() string {
  825. return r.raw
  826. }
  827. type StepFinishPartType string
  828. const (
  829. StepFinishPartTypeStepFinish StepFinishPartType = "step-finish"
  830. )
  831. func (r StepFinishPartType) IsKnown() bool {
  832. switch r {
  833. case StepFinishPartTypeStepFinish:
  834. return true
  835. }
  836. return false
  837. }
  838. type StepStartPart struct {
  839. ID string `json:"id,required"`
  840. MessageID string `json:"messageID,required"`
  841. SessionID string `json:"sessionID,required"`
  842. Type StepStartPartType `json:"type,required"`
  843. JSON stepStartPartJSON `json:"-"`
  844. }
  845. // stepStartPartJSON contains the JSON metadata for the struct [StepStartPart]
  846. type stepStartPartJSON struct {
  847. ID apijson.Field
  848. MessageID apijson.Field
  849. SessionID apijson.Field
  850. Type apijson.Field
  851. raw string
  852. ExtraFields map[string]apijson.Field
  853. }
  854. func (r *StepStartPart) UnmarshalJSON(data []byte) (err error) {
  855. return apijson.UnmarshalRoot(data, r)
  856. }
  857. func (r stepStartPartJSON) RawJSON() string {
  858. return r.raw
  859. }
  860. func (r StepStartPart) implementsPart() {}
  861. type StepStartPartType string
  862. const (
  863. StepStartPartTypeStepStart StepStartPartType = "step-start"
  864. )
  865. func (r StepStartPartType) IsKnown() bool {
  866. switch r {
  867. case StepStartPartTypeStepStart:
  868. return true
  869. }
  870. return false
  871. }
  872. type TextPart struct {
  873. ID string `json:"id,required"`
  874. MessageID string `json:"messageID,required"`
  875. SessionID string `json:"sessionID,required"`
  876. Text string `json:"text,required"`
  877. Type TextPartType `json:"type,required"`
  878. Synthetic bool `json:"synthetic"`
  879. Time TextPartTime `json:"time"`
  880. JSON textPartJSON `json:"-"`
  881. }
  882. // textPartJSON contains the JSON metadata for the struct [TextPart]
  883. type textPartJSON struct {
  884. ID apijson.Field
  885. MessageID apijson.Field
  886. SessionID apijson.Field
  887. Text apijson.Field
  888. Type apijson.Field
  889. Synthetic apijson.Field
  890. Time apijson.Field
  891. raw string
  892. ExtraFields map[string]apijson.Field
  893. }
  894. func (r *TextPart) UnmarshalJSON(data []byte) (err error) {
  895. return apijson.UnmarshalRoot(data, r)
  896. }
  897. func (r textPartJSON) RawJSON() string {
  898. return r.raw
  899. }
  900. func (r TextPart) implementsPart() {}
  901. type TextPartType string
  902. const (
  903. TextPartTypeText TextPartType = "text"
  904. )
  905. func (r TextPartType) IsKnown() bool {
  906. switch r {
  907. case TextPartTypeText:
  908. return true
  909. }
  910. return false
  911. }
  912. type TextPartTime struct {
  913. Start float64 `json:"start,required"`
  914. End float64 `json:"end"`
  915. JSON textPartTimeJSON `json:"-"`
  916. }
  917. // textPartTimeJSON contains the JSON metadata for the struct [TextPartTime]
  918. type textPartTimeJSON struct {
  919. Start apijson.Field
  920. End apijson.Field
  921. raw string
  922. ExtraFields map[string]apijson.Field
  923. }
  924. func (r *TextPartTime) UnmarshalJSON(data []byte) (err error) {
  925. return apijson.UnmarshalRoot(data, r)
  926. }
  927. func (r textPartTimeJSON) RawJSON() string {
  928. return r.raw
  929. }
  930. type TextPartParam struct {
  931. ID param.Field[string] `json:"id,required"`
  932. MessageID param.Field[string] `json:"messageID,required"`
  933. SessionID param.Field[string] `json:"sessionID,required"`
  934. Text param.Field[string] `json:"text,required"`
  935. Type param.Field[TextPartType] `json:"type,required"`
  936. Synthetic param.Field[bool] `json:"synthetic"`
  937. Time param.Field[TextPartTimeParam] `json:"time"`
  938. }
  939. func (r TextPartParam) MarshalJSON() (data []byte, err error) {
  940. return apijson.MarshalRoot(r)
  941. }
  942. func (r TextPartParam) implementsSessionChatParamsPartUnion() {}
  943. type TextPartTimeParam struct {
  944. Start param.Field[float64] `json:"start,required"`
  945. End param.Field[float64] `json:"end"`
  946. }
  947. func (r TextPartTimeParam) MarshalJSON() (data []byte, err error) {
  948. return apijson.MarshalRoot(r)
  949. }
  950. type ToolPart struct {
  951. ID string `json:"id,required"`
  952. CallID string `json:"callID,required"`
  953. MessageID string `json:"messageID,required"`
  954. SessionID string `json:"sessionID,required"`
  955. State ToolPartState `json:"state,required"`
  956. Tool string `json:"tool,required"`
  957. Type ToolPartType `json:"type,required"`
  958. JSON toolPartJSON `json:"-"`
  959. }
  960. // toolPartJSON contains the JSON metadata for the struct [ToolPart]
  961. type toolPartJSON struct {
  962. ID apijson.Field
  963. CallID apijson.Field
  964. MessageID apijson.Field
  965. SessionID apijson.Field
  966. State apijson.Field
  967. Tool apijson.Field
  968. Type apijson.Field
  969. raw string
  970. ExtraFields map[string]apijson.Field
  971. }
  972. func (r *ToolPart) UnmarshalJSON(data []byte) (err error) {
  973. return apijson.UnmarshalRoot(data, r)
  974. }
  975. func (r toolPartJSON) RawJSON() string {
  976. return r.raw
  977. }
  978. func (r ToolPart) implementsPart() {}
  979. type ToolPartState struct {
  980. Status ToolPartStateStatus `json:"status,required"`
  981. Error string `json:"error"`
  982. // This field can have the runtime type of [interface{}], [map[string]interface{}].
  983. Input interface{} `json:"input"`
  984. // This field can have the runtime type of [map[string]interface{}].
  985. Metadata interface{} `json:"metadata"`
  986. Output string `json:"output"`
  987. // This field can have the runtime type of [ToolStateRunningTime],
  988. // [ToolStateCompletedTime], [ToolStateErrorTime].
  989. Time interface{} `json:"time"`
  990. Title string `json:"title"`
  991. JSON toolPartStateJSON `json:"-"`
  992. union ToolPartStateUnion
  993. }
  994. // toolPartStateJSON contains the JSON metadata for the struct [ToolPartState]
  995. type toolPartStateJSON struct {
  996. Status apijson.Field
  997. Error apijson.Field
  998. Input apijson.Field
  999. Metadata apijson.Field
  1000. Output apijson.Field
  1001. Time apijson.Field
  1002. Title apijson.Field
  1003. raw string
  1004. ExtraFields map[string]apijson.Field
  1005. }
  1006. func (r toolPartStateJSON) RawJSON() string {
  1007. return r.raw
  1008. }
  1009. func (r *ToolPartState) UnmarshalJSON(data []byte) (err error) {
  1010. *r = ToolPartState{}
  1011. err = apijson.UnmarshalRoot(data, &r.union)
  1012. if err != nil {
  1013. return err
  1014. }
  1015. return apijson.Port(r.union, &r)
  1016. }
  1017. // AsUnion returns a [ToolPartStateUnion] interface which you can cast to the
  1018. // specific types for more type safety.
  1019. //
  1020. // Possible runtime types of the union are [ToolStatePending], [ToolStateRunning],
  1021. // [ToolStateCompleted], [ToolStateError].
  1022. func (r ToolPartState) AsUnion() ToolPartStateUnion {
  1023. return r.union
  1024. }
  1025. // Union satisfied by [ToolStatePending], [ToolStateRunning], [ToolStateCompleted]
  1026. // or [ToolStateError].
  1027. type ToolPartStateUnion interface {
  1028. implementsToolPartState()
  1029. }
  1030. func init() {
  1031. apijson.RegisterUnion(
  1032. reflect.TypeOf((*ToolPartStateUnion)(nil)).Elem(),
  1033. "status",
  1034. apijson.UnionVariant{
  1035. TypeFilter: gjson.JSON,
  1036. Type: reflect.TypeOf(ToolStatePending{}),
  1037. DiscriminatorValue: "pending",
  1038. },
  1039. apijson.UnionVariant{
  1040. TypeFilter: gjson.JSON,
  1041. Type: reflect.TypeOf(ToolStateRunning{}),
  1042. DiscriminatorValue: "running",
  1043. },
  1044. apijson.UnionVariant{
  1045. TypeFilter: gjson.JSON,
  1046. Type: reflect.TypeOf(ToolStateCompleted{}),
  1047. DiscriminatorValue: "completed",
  1048. },
  1049. apijson.UnionVariant{
  1050. TypeFilter: gjson.JSON,
  1051. Type: reflect.TypeOf(ToolStateError{}),
  1052. DiscriminatorValue: "error",
  1053. },
  1054. )
  1055. }
  1056. type ToolPartStateStatus string
  1057. const (
  1058. ToolPartStateStatusPending ToolPartStateStatus = "pending"
  1059. ToolPartStateStatusRunning ToolPartStateStatus = "running"
  1060. ToolPartStateStatusCompleted ToolPartStateStatus = "completed"
  1061. ToolPartStateStatusError ToolPartStateStatus = "error"
  1062. )
  1063. func (r ToolPartStateStatus) IsKnown() bool {
  1064. switch r {
  1065. case ToolPartStateStatusPending, ToolPartStateStatusRunning, ToolPartStateStatusCompleted, ToolPartStateStatusError:
  1066. return true
  1067. }
  1068. return false
  1069. }
  1070. type ToolPartType string
  1071. const (
  1072. ToolPartTypeTool ToolPartType = "tool"
  1073. )
  1074. func (r ToolPartType) IsKnown() bool {
  1075. switch r {
  1076. case ToolPartTypeTool:
  1077. return true
  1078. }
  1079. return false
  1080. }
  1081. type ToolStateCompleted struct {
  1082. Input map[string]interface{} `json:"input,required"`
  1083. Metadata map[string]interface{} `json:"metadata,required"`
  1084. Output string `json:"output,required"`
  1085. Status ToolStateCompletedStatus `json:"status,required"`
  1086. Time ToolStateCompletedTime `json:"time,required"`
  1087. Title string `json:"title,required"`
  1088. JSON toolStateCompletedJSON `json:"-"`
  1089. }
  1090. // toolStateCompletedJSON contains the JSON metadata for the struct
  1091. // [ToolStateCompleted]
  1092. type toolStateCompletedJSON struct {
  1093. Input apijson.Field
  1094. Metadata apijson.Field
  1095. Output apijson.Field
  1096. Status apijson.Field
  1097. Time apijson.Field
  1098. Title apijson.Field
  1099. raw string
  1100. ExtraFields map[string]apijson.Field
  1101. }
  1102. func (r *ToolStateCompleted) UnmarshalJSON(data []byte) (err error) {
  1103. return apijson.UnmarshalRoot(data, r)
  1104. }
  1105. func (r toolStateCompletedJSON) RawJSON() string {
  1106. return r.raw
  1107. }
  1108. func (r ToolStateCompleted) implementsToolPartState() {}
  1109. type ToolStateCompletedStatus string
  1110. const (
  1111. ToolStateCompletedStatusCompleted ToolStateCompletedStatus = "completed"
  1112. )
  1113. func (r ToolStateCompletedStatus) IsKnown() bool {
  1114. switch r {
  1115. case ToolStateCompletedStatusCompleted:
  1116. return true
  1117. }
  1118. return false
  1119. }
  1120. type ToolStateCompletedTime struct {
  1121. End float64 `json:"end,required"`
  1122. Start float64 `json:"start,required"`
  1123. JSON toolStateCompletedTimeJSON `json:"-"`
  1124. }
  1125. // toolStateCompletedTimeJSON contains the JSON metadata for the struct
  1126. // [ToolStateCompletedTime]
  1127. type toolStateCompletedTimeJSON struct {
  1128. End apijson.Field
  1129. Start apijson.Field
  1130. raw string
  1131. ExtraFields map[string]apijson.Field
  1132. }
  1133. func (r *ToolStateCompletedTime) UnmarshalJSON(data []byte) (err error) {
  1134. return apijson.UnmarshalRoot(data, r)
  1135. }
  1136. func (r toolStateCompletedTimeJSON) RawJSON() string {
  1137. return r.raw
  1138. }
  1139. type ToolStateError struct {
  1140. Error string `json:"error,required"`
  1141. Input map[string]interface{} `json:"input,required"`
  1142. Status ToolStateErrorStatus `json:"status,required"`
  1143. Time ToolStateErrorTime `json:"time,required"`
  1144. JSON toolStateErrorJSON `json:"-"`
  1145. }
  1146. // toolStateErrorJSON contains the JSON metadata for the struct [ToolStateError]
  1147. type toolStateErrorJSON struct {
  1148. Error apijson.Field
  1149. Input apijson.Field
  1150. Status apijson.Field
  1151. Time apijson.Field
  1152. raw string
  1153. ExtraFields map[string]apijson.Field
  1154. }
  1155. func (r *ToolStateError) UnmarshalJSON(data []byte) (err error) {
  1156. return apijson.UnmarshalRoot(data, r)
  1157. }
  1158. func (r toolStateErrorJSON) RawJSON() string {
  1159. return r.raw
  1160. }
  1161. func (r ToolStateError) implementsToolPartState() {}
  1162. type ToolStateErrorStatus string
  1163. const (
  1164. ToolStateErrorStatusError ToolStateErrorStatus = "error"
  1165. )
  1166. func (r ToolStateErrorStatus) IsKnown() bool {
  1167. switch r {
  1168. case ToolStateErrorStatusError:
  1169. return true
  1170. }
  1171. return false
  1172. }
  1173. type ToolStateErrorTime struct {
  1174. End float64 `json:"end,required"`
  1175. Start float64 `json:"start,required"`
  1176. JSON toolStateErrorTimeJSON `json:"-"`
  1177. }
  1178. // toolStateErrorTimeJSON contains the JSON metadata for the struct
  1179. // [ToolStateErrorTime]
  1180. type toolStateErrorTimeJSON struct {
  1181. End apijson.Field
  1182. Start apijson.Field
  1183. raw string
  1184. ExtraFields map[string]apijson.Field
  1185. }
  1186. func (r *ToolStateErrorTime) UnmarshalJSON(data []byte) (err error) {
  1187. return apijson.UnmarshalRoot(data, r)
  1188. }
  1189. func (r toolStateErrorTimeJSON) RawJSON() string {
  1190. return r.raw
  1191. }
  1192. type ToolStatePending struct {
  1193. Status ToolStatePendingStatus `json:"status,required"`
  1194. JSON toolStatePendingJSON `json:"-"`
  1195. }
  1196. // toolStatePendingJSON contains the JSON metadata for the struct
  1197. // [ToolStatePending]
  1198. type toolStatePendingJSON struct {
  1199. Status apijson.Field
  1200. raw string
  1201. ExtraFields map[string]apijson.Field
  1202. }
  1203. func (r *ToolStatePending) UnmarshalJSON(data []byte) (err error) {
  1204. return apijson.UnmarshalRoot(data, r)
  1205. }
  1206. func (r toolStatePendingJSON) RawJSON() string {
  1207. return r.raw
  1208. }
  1209. func (r ToolStatePending) implementsToolPartState() {}
  1210. type ToolStatePendingStatus string
  1211. const (
  1212. ToolStatePendingStatusPending ToolStatePendingStatus = "pending"
  1213. )
  1214. func (r ToolStatePendingStatus) IsKnown() bool {
  1215. switch r {
  1216. case ToolStatePendingStatusPending:
  1217. return true
  1218. }
  1219. return false
  1220. }
  1221. type ToolStateRunning struct {
  1222. Status ToolStateRunningStatus `json:"status,required"`
  1223. Time ToolStateRunningTime `json:"time,required"`
  1224. Input interface{} `json:"input"`
  1225. Metadata map[string]interface{} `json:"metadata"`
  1226. Title string `json:"title"`
  1227. JSON toolStateRunningJSON `json:"-"`
  1228. }
  1229. // toolStateRunningJSON contains the JSON metadata for the struct
  1230. // [ToolStateRunning]
  1231. type toolStateRunningJSON struct {
  1232. Status apijson.Field
  1233. Time apijson.Field
  1234. Input apijson.Field
  1235. Metadata apijson.Field
  1236. Title apijson.Field
  1237. raw string
  1238. ExtraFields map[string]apijson.Field
  1239. }
  1240. func (r *ToolStateRunning) UnmarshalJSON(data []byte) (err error) {
  1241. return apijson.UnmarshalRoot(data, r)
  1242. }
  1243. func (r toolStateRunningJSON) RawJSON() string {
  1244. return r.raw
  1245. }
  1246. func (r ToolStateRunning) implementsToolPartState() {}
  1247. type ToolStateRunningStatus string
  1248. const (
  1249. ToolStateRunningStatusRunning ToolStateRunningStatus = "running"
  1250. )
  1251. func (r ToolStateRunningStatus) IsKnown() bool {
  1252. switch r {
  1253. case ToolStateRunningStatusRunning:
  1254. return true
  1255. }
  1256. return false
  1257. }
  1258. type ToolStateRunningTime struct {
  1259. Start float64 `json:"start,required"`
  1260. JSON toolStateRunningTimeJSON `json:"-"`
  1261. }
  1262. // toolStateRunningTimeJSON contains the JSON metadata for the struct
  1263. // [ToolStateRunningTime]
  1264. type toolStateRunningTimeJSON struct {
  1265. Start apijson.Field
  1266. raw string
  1267. ExtraFields map[string]apijson.Field
  1268. }
  1269. func (r *ToolStateRunningTime) UnmarshalJSON(data []byte) (err error) {
  1270. return apijson.UnmarshalRoot(data, r)
  1271. }
  1272. func (r toolStateRunningTimeJSON) RawJSON() string {
  1273. return r.raw
  1274. }
  1275. type UserMessage struct {
  1276. ID string `json:"id,required"`
  1277. Role UserMessageRole `json:"role,required"`
  1278. SessionID string `json:"sessionID,required"`
  1279. Time UserMessageTime `json:"time,required"`
  1280. JSON userMessageJSON `json:"-"`
  1281. }
  1282. // userMessageJSON contains the JSON metadata for the struct [UserMessage]
  1283. type userMessageJSON struct {
  1284. ID apijson.Field
  1285. Role apijson.Field
  1286. SessionID apijson.Field
  1287. Time apijson.Field
  1288. raw string
  1289. ExtraFields map[string]apijson.Field
  1290. }
  1291. func (r *UserMessage) UnmarshalJSON(data []byte) (err error) {
  1292. return apijson.UnmarshalRoot(data, r)
  1293. }
  1294. func (r userMessageJSON) RawJSON() string {
  1295. return r.raw
  1296. }
  1297. func (r UserMessage) implementsMessage() {}
  1298. type UserMessageRole string
  1299. const (
  1300. UserMessageRoleUser UserMessageRole = "user"
  1301. )
  1302. func (r UserMessageRole) IsKnown() bool {
  1303. switch r {
  1304. case UserMessageRoleUser:
  1305. return true
  1306. }
  1307. return false
  1308. }
  1309. type UserMessageTime struct {
  1310. Created float64 `json:"created,required"`
  1311. JSON userMessageTimeJSON `json:"-"`
  1312. }
  1313. // userMessageTimeJSON contains the JSON metadata for the struct [UserMessageTime]
  1314. type userMessageTimeJSON struct {
  1315. Created apijson.Field
  1316. raw string
  1317. ExtraFields map[string]apijson.Field
  1318. }
  1319. func (r *UserMessageTime) UnmarshalJSON(data []byte) (err error) {
  1320. return apijson.UnmarshalRoot(data, r)
  1321. }
  1322. func (r userMessageTimeJSON) RawJSON() string {
  1323. return r.raw
  1324. }
  1325. type SessionMessagesResponse struct {
  1326. Info Message `json:"info,required"`
  1327. Parts []Part `json:"parts,required"`
  1328. JSON sessionMessagesResponseJSON `json:"-"`
  1329. }
  1330. // sessionMessagesResponseJSON contains the JSON metadata for the struct
  1331. // [SessionMessagesResponse]
  1332. type sessionMessagesResponseJSON struct {
  1333. Info apijson.Field
  1334. Parts apijson.Field
  1335. raw string
  1336. ExtraFields map[string]apijson.Field
  1337. }
  1338. func (r *SessionMessagesResponse) UnmarshalJSON(data []byte) (err error) {
  1339. return apijson.UnmarshalRoot(data, r)
  1340. }
  1341. func (r sessionMessagesResponseJSON) RawJSON() string {
  1342. return r.raw
  1343. }
  1344. type SessionChatParams struct {
  1345. MessageID param.Field[string] `json:"messageID,required"`
  1346. Mode param.Field[string] `json:"mode,required"`
  1347. ModelID param.Field[string] `json:"modelID,required"`
  1348. Parts param.Field[[]SessionChatParamsPartUnion] `json:"parts,required"`
  1349. ProviderID param.Field[string] `json:"providerID,required"`
  1350. }
  1351. func (r SessionChatParams) MarshalJSON() (data []byte, err error) {
  1352. return apijson.MarshalRoot(r)
  1353. }
  1354. type SessionChatParamsPart struct {
  1355. ID param.Field[string] `json:"id,required"`
  1356. MessageID param.Field[string] `json:"messageID,required"`
  1357. SessionID param.Field[string] `json:"sessionID,required"`
  1358. Type param.Field[SessionChatParamsPartsType] `json:"type,required"`
  1359. Filename param.Field[string] `json:"filename"`
  1360. Mime param.Field[string] `json:"mime"`
  1361. Synthetic param.Field[bool] `json:"synthetic"`
  1362. Text param.Field[string] `json:"text"`
  1363. Time param.Field[interface{}] `json:"time"`
  1364. URL param.Field[string] `json:"url"`
  1365. }
  1366. func (r SessionChatParamsPart) MarshalJSON() (data []byte, err error) {
  1367. return apijson.MarshalRoot(r)
  1368. }
  1369. func (r SessionChatParamsPart) implementsSessionChatParamsPartUnion() {}
  1370. // Satisfied by [FilePartParam], [TextPartParam], [SessionChatParamsPart].
  1371. type SessionChatParamsPartUnion interface {
  1372. implementsSessionChatParamsPartUnion()
  1373. }
  1374. type SessionChatParamsPartsType string
  1375. const (
  1376. SessionChatParamsPartsTypeFile SessionChatParamsPartsType = "file"
  1377. SessionChatParamsPartsTypeText SessionChatParamsPartsType = "text"
  1378. )
  1379. func (r SessionChatParamsPartsType) IsKnown() bool {
  1380. switch r {
  1381. case SessionChatParamsPartsTypeFile, SessionChatParamsPartsTypeText:
  1382. return true
  1383. }
  1384. return false
  1385. }
  1386. type SessionInitParams struct {
  1387. MessageID param.Field[string] `json:"messageID,required"`
  1388. ModelID param.Field[string] `json:"modelID,required"`
  1389. ProviderID param.Field[string] `json:"providerID,required"`
  1390. }
  1391. func (r SessionInitParams) MarshalJSON() (data []byte, err error) {
  1392. return apijson.MarshalRoot(r)
  1393. }
  1394. type SessionSummarizeParams struct {
  1395. ModelID param.Field[string] `json:"modelID,required"`
  1396. ProviderID param.Field[string] `json:"providerID,required"`
  1397. }
  1398. func (r SessionSummarizeParams) MarshalJSON() (data []byte, err error) {
  1399. return apijson.MarshalRoot(r)
  1400. }