tracespec

type ModelAudioURL struct {
	Name string `json:"name,omitempty"`
	// Required. You can enter a valid audio URL of audio.
	URL string `json:"url,omitempty"`
}

type ModelCallOption struct {
	Temperature      float32  `json:"temperature"`
	MaxTokens        int64    `json:"max_tokens,omitempty"`
	Stop             []string `json:"stop,omitempty"`
	TopP             float32  `json:"top_p,omitempty"`
	N                int64    `json:"n,omitempty"`
	TopK             *int64   `json:"top_k,omitempty"`
	PresencePenalty  *float32 `json:"presence_penalty,omitempty"`
	FrequencyPenalty *float32 `json:"frequency_penalty,omitempty"`
	ReasoningEffort  string   `json:"reasoning_effort,omitempty"`
}

type ModelChoice struct {
	FinishReason string        `json:"finish_reason"`
	Index        int64         `json:"index"`
	Message      *ModelMessage `json:"message"`
}

type ModelFileURL struct {
	Name string `json:"name,omitempty"`
	// Required. You can enter a valid file URL or MDN Base64 data of file.
	// MDN: https://developer.mozilla.org/en-US/docs/Web/URI/Reference/Schemes/data#syntax
	URL    string `json:"url,omitempty"`
	Detail string `json:"detail,omitempty"`
	Suffix string `json:"suffix,omitempty"`
}

type ModelImageURL struct {
	Name string `json:"name,omitempty"`
	// Required. You can enter a valid image URL or MDN Base64 data of image.
	// MDN: https://developer.mozilla.org/en-US/docs/Web/URI/Reference/Schemes/data#syntax
	URL    string `json:"url,omitempty"`
	Detail string `json:"detail,omitempty"`
}

type ModelInput struct {
	Messages           []*ModelMessage  `json:"messages,omitempty"`
	Tools              []*ModelTool     `json:"tools,omitempty"`
	ModelToolChoice    *ModelToolChoice `json:"tool_choice,omitempty"`
	PreviousResponseID string           `json:"previous_response_id,omitempty"`
}

type ModelMessage struct {
	Role             string              `json:"role"`                        // from enum VRole in span_value
	Content          string              `json:"content,omitempty"`           // single content
	ReasoningContent string              `json:"reasoning_content,omitempty"` // only for output
	Parts            []*ModelMessagePart `json:"parts,omitempty"`             // multi-modality content
	Name             string              `json:"name,omitempty"`
	ToolCalls        []*ModelToolCall    `json:"tool_calls,omitempty"`
	ToolCallID       string              `json:"tool_call_id,omitempty"`
	Signature        string              `json:"signature,omitempty"`
	Metadata         map[string]string   `json:"metadata,omitempty"`
}

type ModelMessagePart struct {
	Type      ModelMessagePartType `json:"type"` // Required. The type of the content.
	Text      string               `json:"text,omitempty"`
	ImageURL  *ModelImageURL       `json:"image_url,omitempty"`
	FileURL   *ModelFileURL        `json:"file_url,omitempty"`
	AudioURL  *ModelAudioURL       `json:"audio_url,omitempty"`
	VideoURL  *ModelVideoURL       `json:"video_url,omitempty"`
	Signature string               `json:"signature,omitempty"`
}

type ModelMessagePartType string

type ModelOutput struct {
	ID      string         `json:"id,omitempty"`
	Choices []*ModelChoice `json:"choices"`
}

type ModelTool struct {
	Type     string             `json:"type"` // Always be: "function"
	Function *ModelToolFunction `json:"function"`
}

type ModelToolCall struct {
	ID        string                 `json:"id,omitempty"`
	Type      string                 `json:"type,omitempty"` // Always be: "function"
	Function  *ModelToolCallFunction `json:"function"`
	Signature string                 `json:"signature,omitempty"`
}

type ModelToolCallFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments,omitempty"`
}

type ModelToolChoice struct {
	Type     string                 `json:"type"`               // from enum VToolChoice in span_value
	Function *ModelToolCallFunction `json:"function,omitempty"` // field name only.
}

type ModelToolFunction struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	Parameters  json.RawMessage `json:"parameters"`
}

type ModelVideoURL struct {
	Name string `json:"name,omitempty"`
	// Required. You can enter a valid video URL of video.
	URL string `json:"url,omitempty"`
}

type PromptArgument struct {
	Key       string                  `json:"key"`
	Value     any                     `json:"value"`
	Source    string                  `json:"source"` // from enum VPromptArgSource in span_value.go
	ValueType PromptArgumentValueType `json:"value_type"`
}

type PromptArgumentValueType string

type PromptInput struct {
	Templates []*ModelMessage   `json:"templates"`
	Arguments []*PromptArgument `json:"arguments"`
}

type PromptOutput struct {
	Prompts []*ModelMessage `json:"prompts"`
}

type RetrieverCallOption struct {
	TopK     int64    `json:"top_k,omitempty"`
	MinScore *float64 `json:"min_score,omitempty"`
	Filter   string   `json:"filter,omitempty"`
}

type RetrieverDocument struct {
	ID      string    `json:"id,omitempty"`
	Index   string    `json:"index,omitempty"`
	Content string    `json:"content"`
	Vector  []float64 `json:"vector,omitempty"`
	Score   float64   `json:"score"`
}

type RetrieverInput struct {
	Query string `json:"query,omitempty"`
}

type RetrieverOutput struct {
	Documents []*RetrieverDocument `json:"documents,omitempty"`
}

type Runtime struct {
	Language     string `json:"language"`          // from enum VLang in span_value.go
	Library      string `json:"library,omitempty"` // integration library, from enum VLib in span_value.go
	Scene        string `json:"scene,omitempty"`   // usage scene, from enum VScene in span_value.go
	SceneVersion string `json:"scene_version,omitempty"`

	// Dependency Versions.
	LibraryVersion string `json:"library_version,omitempty"`
	LoopSDKVersion string `json:"loop_sdk_version,omitempty"`

	// Extra info.
	Extra map[string]interface{} `json:"extra,omitempty"`
}