aljaz/opencode
1
0
Fork 0
mirror of https://github.com/aljazceru/opencode.git synced 2025-12-24 03:04:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

better discriminator

Browse Source
This commit is contained in:
Dax Raad
2025-05-28 13:35:05 -04:00
parent 3b7b7f4bea
commit f3b224090c
3 changed files with 96 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
pkg/client/generated-client.go
View File

@@ -7,6 +7,7 @@ import (
"bytes"
"context"
"encoding/json"
"errors"
"fmt"
"io"
"net/http"
@@ -75,13 +76,6 @@ type SessionMessagePart struct {
union json.RawMessage
}
// SessionMessagePartData defines model for Session.Message.Part.Data.
type SessionMessagePartData struct {
Data *interface{} `json:"data,omitempty"`
Id *string `json:"id,omitempty"`
Type *interface{} `json:"type,omitempty"`
}
// SessionMessagePartFile defines model for Session.Message.Part.File.
type SessionMessagePartFile struct {
Filename *string `json:"filename,omitempty"`
@@ -192,6 +186,7 @@ func (t SessionMessagePart) AsSessionMessagePartText() (SessionMessagePartText,
// FromSessionMessagePartText overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartText
func (t *SessionMessagePart) FromSessionMessagePartText(v SessionMessagePartText) error {
v.Type = "text"
b, err := json.Marshal(v)
t.union = b
return err
@@ -199,6 +194,7 @@ func (t *SessionMessagePart) FromSessionMessagePartText(v SessionMessagePartText
// MergeSessionMessagePartText performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartText
func (t *SessionMessagePart) MergeSessionMessagePartText(v SessionMessagePartText) error {
v.Type = "text"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -218,6 +214,7 @@ func (t SessionMessagePart) AsSessionMessagePartReasoning() (SessionMessagePartR
// FromSessionMessagePartReasoning overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartReasoning
func (t *SessionMessagePart) FromSessionMessagePartReasoning(v SessionMessagePartReasoning) error {
v.Type = "reasoning"
b, err := json.Marshal(v)
t.union = b
return err
@@ -225,6 +222,7 @@ func (t *SessionMessagePart) FromSessionMessagePartReasoning(v SessionMessagePar
// MergeSessionMessagePartReasoning performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartReasoning
func (t *SessionMessagePart) MergeSessionMessagePartReasoning(v SessionMessagePartReasoning) error {
v.Type = "reasoning"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -244,6 +242,7 @@ func (t SessionMessagePart) AsSessionMessagePartToolInvocation() (SessionMessage
// FromSessionMessagePartToolInvocation overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartToolInvocation
func (t *SessionMessagePart) FromSessionMessagePartToolInvocation(v SessionMessagePartToolInvocation) error {
v.Type = "tool-invocation"
b, err := json.Marshal(v)
t.union = b
return err
@@ -251,6 +250,7 @@ func (t *SessionMessagePart) FromSessionMessagePartToolInvocation(v SessionMessa
// MergeSessionMessagePartToolInvocation performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartToolInvocation
func (t *SessionMessagePart) MergeSessionMessagePartToolInvocation(v SessionMessagePartToolInvocation) error {
v.Type = "tool-invocation"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -270,6 +270,7 @@ func (t SessionMessagePart) AsSessionMessagePartSourceUrl() (SessionMessagePartS
// FromSessionMessagePartSourceUrl overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartSourceUrl
func (t *SessionMessagePart) FromSessionMessagePartSourceUrl(v SessionMessagePartSourceUrl) error {
v.Type = "source-url"
b, err := json.Marshal(v)
t.union = b
return err
@@ -277,6 +278,7 @@ func (t *SessionMessagePart) FromSessionMessagePartSourceUrl(v SessionMessagePar
// MergeSessionMessagePartSourceUrl performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartSourceUrl
func (t *SessionMessagePart) MergeSessionMessagePartSourceUrl(v SessionMessagePartSourceUrl) error {
v.Type = "source-url"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -296,6 +298,7 @@ func (t SessionMessagePart) AsSessionMessagePartFile() (SessionMessagePartFile,
// FromSessionMessagePartFile overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartFile
func (t *SessionMessagePart) FromSessionMessagePartFile(v SessionMessagePartFile) error {
v.Type = "file"
b, err := json.Marshal(v)
t.union = b
return err
@@ -303,6 +306,7 @@ func (t *SessionMessagePart) FromSessionMessagePartFile(v SessionMessagePartFile
// MergeSessionMessagePartFile performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartFile
func (t *SessionMessagePart) MergeSessionMessagePartFile(v SessionMessagePartFile) error {
v.Type = "file"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -322,6 +326,7 @@ func (t SessionMessagePart) AsSessionMessagePartStepStart() (SessionMessagePartS
// FromSessionMessagePartStepStart overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartStepStart
func (t *SessionMessagePart) FromSessionMessagePartStepStart(v SessionMessagePartStepStart) error {
v.Type = "step-start"
b, err := json.Marshal(v)
t.union = b
return err
@@ -329,6 +334,7 @@ func (t *SessionMessagePart) FromSessionMessagePartStepStart(v SessionMessagePar
// MergeSessionMessagePartStepStart performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartStepStart
func (t *SessionMessagePart) MergeSessionMessagePartStepStart(v SessionMessagePartStepStart) error {
v.Type = "step-start"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -339,30 +345,35 @@ func (t *SessionMessagePart) MergeSessionMessagePartStepStart(v SessionMessagePa
return err
}
// AsSessionMessagePartData returns the union data inside the SessionMessagePart as a SessionMessagePartData
func (t SessionMessagePart) AsSessionMessagePartData() (SessionMessagePartData, error) {
var body SessionMessagePartData
err := json.Unmarshal(t.union, &body)
return body, err
}
// FromSessionMessagePartData overwrites any union data inside the SessionMessagePart as the provided SessionMessagePartData
func (t *SessionMessagePart) FromSessionMessagePartData(v SessionMessagePartData) error {
b, err := json.Marshal(v)
t.union = b
return err
}
// MergeSessionMessagePartData performs a merge with any union data inside the SessionMessagePart, using the provided SessionMessagePartData
func (t *SessionMessagePart) MergeSessionMessagePartData(v SessionMessagePartData) error {
b, err := json.Marshal(v)
if err != nil {
return err
func (t SessionMessagePart) Discriminator() (string, error) {
var discriminator struct {
Discriminator string `json:"type"`
}
err := json.Unmarshal(t.union, &discriminator)
return discriminator.Discriminator, err
}
merged, err := runtime.JSONMerge(t.union, b)
t.union = merged
return err
func (t SessionMessagePart) ValueByDiscriminator() (interface{}, error) {
discriminator, err := t.Discriminator()
if err != nil {
return nil, err
}
switch discriminator {
case "file":
return t.AsSessionMessagePartFile()
case "reasoning":
return t.AsSessionMessagePartReasoning()
case "source-url":
return t.AsSessionMessagePartSourceUrl()
case "step-start":
return t.AsSessionMessagePartStepStart()
case "text":
return t.AsSessionMessagePartText()
case "tool-invocation":
return t.AsSessionMessagePartToolInvocation()
default:
return nil, errors.New("unknown discriminator value: " + discriminator)
}
}
func (t SessionMessagePart) MarshalJSON() ([]byte, error) {
@@ -384,6 +395,7 @@ func (t SessionMessageToolInvocation) AsSessionMessageToolInvocationToolCall() (
// FromSessionMessageToolInvocationToolCall overwrites any union data inside the SessionMessageToolInvocation as the provided SessionMessageToolInvocationToolCall
func (t *SessionMessageToolInvocation) FromSessionMessageToolInvocationToolCall(v SessionMessageToolInvocationToolCall) error {
v.State = "call"
b, err := json.Marshal(v)
t.union = b
return err
@@ -391,6 +403,7 @@ func (t *SessionMessageToolInvocation) FromSessionMessageToolInvocationToolCall(
// MergeSessionMessageToolInvocationToolCall performs a merge with any union data inside the SessionMessageToolInvocation, using the provided SessionMessageToolInvocationToolCall
func (t *SessionMessageToolInvocation) MergeSessionMessageToolInvocationToolCall(v SessionMessageToolInvocationToolCall) error {
v.State = "call"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -410,6 +423,7 @@ func (t SessionMessageToolInvocation) AsSessionMessageToolInvocationToolPartialC
// FromSessionMessageToolInvocationToolPartialCall overwrites any union data inside the SessionMessageToolInvocation as the provided SessionMessageToolInvocationToolPartialCall
func (t *SessionMessageToolInvocation) FromSessionMessageToolInvocationToolPartialCall(v SessionMessageToolInvocationToolPartialCall) error {
v.State = "partial-call"
b, err := json.Marshal(v)
t.union = b
return err
@@ -417,6 +431,7 @@ func (t *SessionMessageToolInvocation) FromSessionMessageToolInvocationToolParti
// MergeSessionMessageToolInvocationToolPartialCall performs a merge with any union data inside the SessionMessageToolInvocation, using the provided SessionMessageToolInvocationToolPartialCall
func (t *SessionMessageToolInvocation) MergeSessionMessageToolInvocationToolPartialCall(v SessionMessageToolInvocationToolPartialCall) error {
v.State = "partial-call"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -436,6 +451,7 @@ func (t SessionMessageToolInvocation) AsSessionMessageToolInvocationToolResult()
// FromSessionMessageToolInvocationToolResult overwrites any union data inside the SessionMessageToolInvocation as the provided SessionMessageToolInvocationToolResult
func (t *SessionMessageToolInvocation) FromSessionMessageToolInvocationToolResult(v SessionMessageToolInvocationToolResult) error {
v.State = "result"
b, err := json.Marshal(v)
t.union = b
return err
@@ -443,6 +459,7 @@ func (t *SessionMessageToolInvocation) FromSessionMessageToolInvocationToolResul
// MergeSessionMessageToolInvocationToolResult performs a merge with any union data inside the SessionMessageToolInvocation, using the provided SessionMessageToolInvocationToolResult
func (t *SessionMessageToolInvocation) MergeSessionMessageToolInvocationToolResult(v SessionMessageToolInvocationToolResult) error {
v.State = "result"
b, err := json.Marshal(v)
if err != nil {
return err
@@ -453,6 +470,31 @@ func (t *SessionMessageToolInvocation) MergeSessionMessageToolInvocationToolResu
return err
}
func (t SessionMessageToolInvocation) Discriminator() (string, error) {
var discriminator struct {
Discriminator string `json:"state"`
}
err := json.Unmarshal(t.union, &discriminator)
return discriminator.Discriminator, err
}
func (t SessionMessageToolInvocation) ValueByDiscriminator() (interface{}, error) {
discriminator, err := t.Discriminator()
if err != nil {
return nil, err
}
switch discriminator {
case "call":
return t.AsSessionMessageToolInvocationToolCall()
case "partial-call":
return t.AsSessionMessageToolInvocationToolPartialCall()
case "result":
return t.AsSessionMessageToolInvocationToolResult()
default:
return nil, errors.New("unknown discriminator value: " + discriminator)
}
}
func (t SessionMessageToolInvocation) MarshalJSON() ([]byte, error) {
b, err := t.union.MarshalJSON()
return b, err