protocol

type Checkpoint struct {
	CheckpointID       string                 `json:"checkpoint_id" bson:"_id"`
	ThreadID           string                 `json:"thread_id" bson:"thread_id"`
	ParentCheckpointID string                 `json:"parent_checkpoint_id,omitempty" bson:"parent_checkpoint_id,omitempty"`
	Values             map[string]interface{} `json:"values" bson:"values"`
	Messages           []Message              `json:"messages,omitempty" bson:"messages,omitempty"`
	Metadata           map[string]interface{} `json:"metadata,omitempty" bson:"metadata,omitempty"`
	PendingWrites      map[string]interface{} `json:"pending_writes,omitempty" bson:"pending_writes,omitempty"`
	CreatedAt          time.Time              `json:"created_at" bson:"created_at"`
}

type ErrorResponse struct {
	Code    string `json:"code,omitempty"`
	Message string `json:"message"`
}

type Message struct {
	Role     string                 `json:"role" bson:"role"`
	Content  string                 `json:"content" bson:"content"`
	ID       string                 `json:"id,omitempty" bson:"id,omitempty"`
	Metadata map[string]interface{} `json:"metadata,omitempty" bson:"metadata,omitempty"`
}

type Run struct {
	RunID     string                 `json:"run_id" bson:"_id"`
	ThreadID  string                 `json:"thread_id,omitempty" bson:"thread_id,omitempty"`
	AgentID   string                 `json:"agent_id,omitempty" bson:"agent_id,omitempty"`
	Status    RunStatus              `json:"status" bson:"status"`
	Input     map[string]interface{} `json:"input,omitempty" bson:"input,omitempty"`
	Output    map[string]interface{} `json:"output,omitempty" bson:"output,omitempty"`
	Metadata  map[string]interface{} `json:"metadata" bson:"metadata"`
	CreatedAt time.Time              `json:"created_at" bson:"created_at"`
	UpdatedAt time.Time              `json:"updated_at" bson:"updated_at"`
}

type RunCreate struct {
	AgentID    string                 `json:"agent_id,omitempty"`
	ThreadID   string                 `json:"thread_id,omitempty"`
	Input      map[string]interface{} `json:"input"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
	Config     map[string]interface{} `json:"config,omitempty"`
	StreamMode []string               `json:"stream_mode,omitempty"`
}

type RunSearchRequest struct {
	ThreadID string                 `json:"thread_id,omitempty"`
	AgentID  string                 `json:"agent_id,omitempty"`
	Status   RunStatus              `json:"status,omitempty"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
	Limit    int                    `json:"limit,omitempty"`
	Offset   int                    `json:"offset,omitempty"`
}

type RunStatus string

type StoreDeleteRequest struct {
	Namespace []string `json:"namespace,omitempty"`
	Key       string   `json:"key"`
}

type StoreGetRequest struct {
	Namespace []string `json:"namespace,omitempty"`
	Key       string   `json:"key"`
}

type StoreItem struct {
	Namespace []string               `json:"namespace" bson:"namespace"`
	Key       string                 `json:"key" bson:"key"`
	Value     map[string]interface{} `json:"value" bson:"value"`
	CreatedAt time.Time              `json:"created_at" bson:"created_at"`
	UpdatedAt time.Time              `json:"updated_at" bson:"updated_at"`
}

type StorePutRequest struct {
	Namespace []string               `json:"namespace"`
	Key       string                 `json:"key"`
	Value     map[string]interface{} `json:"value"`
}

type StoreSearchRequest struct {
	NamespacePrefix []string               `json:"namespace_prefix,omitempty"`
	Filter          map[string]interface{} `json:"filter,omitempty"`
	Limit           int                    `json:"limit,omitempty"`
	Offset          int                    `json:"offset,omitempty"`
}

type Thread struct {
	ThreadID  string                 `json:"thread_id" bson:"_id"`
	CreatedAt time.Time              `json:"created_at" bson:"created_at"`
	UpdatedAt time.Time              `json:"updated_at" bson:"updated_at"`
	Metadata  map[string]interface{} `json:"metadata" bson:"metadata"`
	Status    ThreadStatus           `json:"status" bson:"status"`
	Values    map[string]interface{} `json:"values,omitempty" bson:"values,omitempty"`
	Messages  []Message              `json:"messages,omitempty" bson:"messages,omitempty"`
}

type ThreadCreate struct {
	ThreadID string                 `json:"thread_id,omitempty"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
	IfExists string                 `json:"if_exists,omitempty"` // "raise" or "do_nothing"
}

type ThreadPatch struct {
	Metadata map[string]interface{} `json:"metadata,omitempty"`
	Values   map[string]interface{} `json:"values,omitempty"`
	Messages []Message              `json:"messages,omitempty"`
}

type ThreadSearchRequest struct {
	Metadata map[string]interface{} `json:"metadata,omitempty"`
	Status   ThreadStatus           `json:"status,omitempty"`
	Limit    int                    `json:"limit,omitempty"`
	Offset   int                    `json:"offset,omitempty"`
}

type ThreadStatus string