// Code generated by protoc-gen-goext. DO NOT EDIT.

package runs

import (
	v1 "github.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1"
	threads "github.com/yandex-cloud/go-genproto/yandex/cloud/ai/assistants/v1/threads"
	common "github.com/yandex-cloud/go-genproto/yandex/cloud/ai/common"
	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
)

func (m *Run) SetId(v string) {
	m.Id = v
}

func (m *Run) SetAssistantId(v string) {
	m.AssistantId = v
}

func (m *Run) SetThreadId(v string) {
	m.ThreadId = v
}

func (m *Run) SetCreatedBy(v string) {
	m.CreatedBy = v
}

func (m *Run) SetCreatedAt(v *timestamppb.Timestamp) {
	m.CreatedAt = v
}

func (m *Run) SetLabels(v map[string]string) {
	m.Labels = v
}

func (m *Run) SetState(v *RunState) {
	m.State = v
}

func (m *Run) SetUsage(v *ContentUsage) {
	m.Usage = v
}

func (m *Run) SetCustomPromptTruncationOptions(v *v1.PromptTruncationOptions) {
	m.CustomPromptTruncationOptions = v
}

func (m *Run) SetCustomCompletionOptions(v *v1.CompletionOptions) {
	m.CustomCompletionOptions = v
}

func (m *Run) SetTools(v []*v1.Tool) {
	m.Tools = v
}

func (m *Run) SetCustomResponseFormat(v *v1.ResponseFormat) {
	m.CustomResponseFormat = v
}

type RunState_StateData = isRunState_StateData

func (m *RunState) SetStateData(v RunState_StateData) {
	m.StateData = v
}

func (m *RunState) SetStatus(v RunState_RunStatus) {
	m.Status = v
}

func (m *RunState) SetError(v *common.Error) {
	m.StateData = &RunState_Error{
		Error: v,
	}
}

func (m *RunState) SetCompletedMessage(v *threads.Message) {
	m.StateData = &RunState_CompletedMessage{
		CompletedMessage: v,
	}
}

func (m *RunState) SetToolCallList(v *v1.ToolCallList) {
	m.StateData = &RunState_ToolCallList{
		ToolCallList: v,
	}
}

func (m *ContentUsage) SetPromptTokens(v int64) {
	m.PromptTokens = v
}

func (m *ContentUsage) SetCompletionTokens(v int64) {
	m.CompletionTokens = v
}

func (m *ContentUsage) SetTotalTokens(v int64) {
	m.TotalTokens = v
}
