assistant

type A2AArtifact struct {
	Kind        string          `json:"kind"`
	ArtifactID  string          `json:"artifactId"`
	Name        string          `json:"name,omitempty"`
	Description string          `json:"description,omitempty"`
	Parts       []A2APart       `json:"parts"`
	Index       int             `json:"index,omitempty"`
	LastChunk   bool            `json:"lastChunk,omitempty"`
	Metadata    json.RawMessage `json:"metadata,omitempty"`
}

type A2AArtifactUpdate struct {
	Kind      string      `json:"kind"`
	TaskID    string      `json:"taskId"`
	ContextID string      `json:"contextId"`
	Artifact  A2AArtifact `json:"artifact"`
	Append    bool        `json:"append,omitempty"`
	LastChunk bool        `json:"lastChunk,omitempty"`
}

type A2AEndpoints struct {
	// contains filtered or unexported fields
}

type A2AFileContent struct {
	Name     string `json:"name,omitempty"`
	MimeType string `json:"mimeType,omitempty"`
	Bytes    string `json:"bytes,omitempty"`
	URI      string `json:"uri,omitempty"`
}

type A2AMessage struct {
	Kind      string         `json:"kind"`
	Role      string         `json:"role"`
	Parts     []A2APart      `json:"parts"`
	MessageID string         `json:"messageId"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type A2APart struct {
	Kind     string          `json:"kind"`
	Text     string          `json:"text,omitempty"`
	File     *A2AFileContent `json:"file,omitempty"`
	Data     json.RawMessage `json:"data,omitempty"`
	Metadata json.RawMessage `json:"metadata,omitempty"`
}

type A2AStatus struct {
	State   string      `json:"state"`
	Message *A2AMessage `json:"message,omitempty"`
}

type A2AStatusUpdate struct {
	Kind      string    `json:"kind"`
	TaskID    string    `json:"taskId"`
	ContextID string    `json:"contextId"`
	Status    A2AStatus `json:"status"`
	Final     bool      `json:"final,omitempty"`
}

type A2ATask struct {
	Kind      string           `json:"kind"`
	ID        string           `json:"id"`
	ContextID string           `json:"contextId"`
	Status    A2AStatus        `json:"status"`
	Artifacts []A2AArtifact    `json:"artifacts,omitempty"`
	History   []A2AMessage     `json:"history,omitempty"`
	Metadata  *A2ATaskMetadata `json:"metadata,omitempty"`
}

type A2ATaskMetadata struct {
	Error string `json:"error,omitempty"`
}

type ApprovalHandler interface {
	// HandleApproval is called when an approval request is received.
	// Returns true if approved, false if denied.
	HandleApproval(req ApprovalRequest) bool
}

type ApprovalRequest struct {
	ID          string          `json:"id"`
	ChatID      string          `json:"chatId"`
	TenantID    string          `json:"tenantId"`
	UserID      string          `json:"userId"`
	ToolName    string          `json:"toolName"`
	ToolInput   json.RawMessage `json:"toolInput,omitempty"`
	Description string          `json:"description,omitempty"`
}

type ApprovalResponse struct {
	ID       string `json:"id"`
	ChatID   string `json:"chatId"`
	TenantID string `json:"tenantId"`
	UserID   string `json:"userId"`
	Approved bool   `json:"approved"`
}

type Chat struct {
	ID       string `json:"id"`
	Name     string `json:"name"`
	Category string `json:"category,omitempty"`
	Source   string `json:"source"`
}

type ChatEndpoints struct {
	// contains filtered or unexported fields
}

type ChatMessage struct {
	ID        string      `json:"id"`
	Role      string      `json:"role"`
	Content   ContentJSON `json:"content"`
	CreatedAt string      `json:"created"`
	Type      string      `json:"type,omitempty"`
	Hidden    bool        `json:"hidden,omitempty"`
}

type Client struct {
	// contains filtered or unexported fields
}

type ClientOptions struct {
	GrafanaURL     string
	Token          string
	APIEndpoint    string
	TokenRefresher TokenRefresher
}

type ContentBlock struct {
	Type string `json:"type"`
	Text string `json:"text,omitempty"`
}

type ContentJSON []ContentBlock

type InteractiveApprovalHandler struct {
	Logger Logger
}

type JSONRPCError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Data    string `json:"data,omitempty"`
}

type JSONRPCRequest struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      string          `json:"id"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params"`
}

type JSONRPCResponse struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      string          `json:"id,omitempty"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *JSONRPCError   `json:"error,omitempty"`
}

type Logger interface {
	Info(message string)
	Debug(message string)
	Warning(message string)
}

type MessageSendParams struct {
	Message   A2AMessage `json:"message"`
	ContextID string     `json:"contextId,omitempty"`
}

type NopLogger struct{}

type State struct {
	// LastContextID is the context ID of the last chat session.
	LastContextID string `yaml:"last-context-id,omitempty"`
}

type StreamEvent struct {
	Type      string `json:"type"`
	TaskID    string `json:"taskId,omitempty"`
	ContextID string `json:"contextId,omitempty"`
	State     string `json:"state,omitempty"`
	Final     bool   `json:"final,omitempty"`
	ToolName  string `json:"toolName,omitempty"`
	Text      string `json:"text,omitempty"`
	Error     string `json:"error,omitempty"`
	Timeout   int    `json:"timeout,omitempty"`
}

type StreamOptions struct {
	Timeout   int
	ContextID string
	OnEvent   func(StreamEvent)
}

type StreamResult struct {
	TaskID            string
	ContextID         string
	Completed         bool
	TimedOut          bool
	Failed            bool
	Canceled          bool
	ErrorMessage      string
	Response          string
	ErrorEventEmitted bool
}

type TokenRefresher func() (string, error)