mongodb

type BranchDocument struct {
	ID          string    `bson:"id"          json:"id"`
	Name        string    `bson:"name"        json:"name"`
	Description string    `bson:"description" json:"description,omitempty"`
	ParentIndex int       `bson:"parent_index" json:"parent_index"`
	IsActive    bool      `bson:"is_active"   json:"is_active"`
	CreatedAt   time.Time `bson:"created_at"  json:"created_at"`
}

type ConversationDocument struct {
	ID        string            `bson:"_id"        json:"id"`
	AgentID   string            `bson:"agent_id"   json:"agent_id"`
	TenantID  string            `bson:"tenant_id"  json:"tenant_id"`
	UserID    string            `bson:"user_id"    json:"user_id"`
	Title     string            `bson:"title"      json:"title,omitempty"`
	Messages  []MessageDocument `bson:"messages"   json:"messages"`
	Branches  []BranchDocument  `bson:"branches"   json:"branches,omitempty"`
	Metadata  map[string]any    `bson:"metadata"   json:"metadata,omitempty"`
	CreatedAt time.Time         `bson:"created_at" json:"created_at"`
	UpdatedAt time.Time         `bson:"updated_at" json:"updated_at"`
}

type ConversationFilter struct {
	AgentID   string     `json:"agent_id,omitempty"`
	TenantID  string     `json:"tenant_id,omitempty"`
	UserID    string     `json:"user_id,omitempty"`
	StartTime *time.Time `json:"start_time,omitempty"`
	EndTime   *time.Time `json:"end_time,omitempty"`
	Limit     int        `json:"limit,omitempty"`
	Offset    int        `json:"offset,omitempty"`
}

type ConversationStore interface {
	// Create inserts a new conversation.
	Create(ctx context.Context, doc *ConversationDocument) error
	// GetByID retrieves a conversation by ID.
	GetByID(ctx context.Context, id string) (*ConversationDocument, error)
	// AppendMessage adds a message to an existing conversation.
	AppendMessage(ctx context.Context, conversationID string, msg MessageDocument) error
	// GetMessages returns a paginated slice of messages from a conversation.
	GetMessages(ctx context.Context, conversationID string, limit, offset int) ([]MessageDocument, error)
	// List returns conversations matching the filter.
	List(ctx context.Context, filter ConversationFilter) ([]*ConversationDocument, error)
	// Delete removes a conversation by ID.
	Delete(ctx context.Context, id string) error
	// UpdateBranches replaces the branches array for a conversation.
	UpdateBranches(ctx context.Context, conversationID string, branches []BranchDocument) error
}

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

type MessageDocument struct {
	Role       string               `bson:"role"         json:"role"`
	Content    string               `bson:"content"      json:"content,omitempty"`
	Name       string               `bson:"name"         json:"name,omitempty"`
	ToolCalls  []types.ToolCall     `bson:"tool_calls"   json:"tool_calls,omitempty"`
	ToolCallID string               `bson:"tool_call_id" json:"tool_call_id,omitempty"`
	Images     []types.ImageContent `bson:"images"      json:"images,omitempty"`
	Metadata   any                  `bson:"metadata"     json:"metadata,omitempty"`
	Timestamp  time.Time            `bson:"timestamp"    json:"timestamp,omitempty"`
}

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

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

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

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

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

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

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

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

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

type PromptDocument struct {
	ID        string    `bson:"_id"        json:"id"`
	AgentType string    `bson:"agent_type" json:"agent_type"`
	Name      string    `bson:"name"       json:"name"`
	Version   string    `bson:"version"    json:"version"`
	TenantID  string    `bson:"tenant_id"  json:"tenant_id"`
	IsActive  bool      `bson:"is_active"  json:"is_active"`
	CreatedAt time.Time `bson:"created_at" json:"created_at"`
	UpdatedAt time.Time `bson:"updated_at" json:"updated_at"`

	// Embedded PromptBundle fields.
	System      agent.SystemPrompt `bson:"system"      json:"system"`
	Tools       []llm.ToolSchema   `bson:"tools"       json:"tools,omitempty"`
	Examples    []agent.Example    `bson:"examples"    json:"examples,omitempty"`
	Constraints []string           `bson:"constraints" json:"constraints,omitempty"`
}

type PromptStore interface {
	// Create inserts a new prompt document.
	Create(ctx context.Context, doc *PromptDocument) error
	// GetByID retrieves a prompt by its ID.
	GetByID(ctx context.Context, id string) (*PromptDocument, error)
	// Update replaces an existing prompt document.
	Update(ctx context.Context, doc *PromptDocument) error
	// Delete removes a prompt by ID.
	Delete(ctx context.Context, id string) error
	// ListByAgentType returns prompts matching the given agent type and tenant.
	ListByAgentType(ctx context.Context, agentType, tenantID string, limit, offset int) ([]*PromptDocument, error)
	// GetActive returns the active prompt for a given agent type, name, and tenant.
	GetActive(ctx context.Context, agentType, name, tenantID string) (*PromptDocument, error)
	// SetActive marks a specific version as active and deactivates others.
	SetActive(ctx context.Context, id string) error
}

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

type RunDocument struct {
	ID        string         `bson:"_id"         json:"id"`
	AgentID   string         `bson:"agent_id"    json:"agent_id"`
	TenantID  string         `bson:"tenant_id"   json:"tenant_id"`
	TraceID   string         `bson:"trace_id"    json:"trace_id,omitempty"`
	Status    string         `bson:"status"      json:"status"`
	Input     RunInput       `bson:"input"       json:"input"`
	Output    *RunOutput     `bson:"output"      json:"output,omitempty"`
	Error     string         `bson:"error"       json:"error,omitempty"`
	StartTime time.Time      `bson:"start_time"  json:"start_time"`
	EndTime   *time.Time     `bson:"end_time"    json:"end_time,omitempty"`
	Duration  time.Duration  `bson:"duration"    json:"duration,omitempty"`
	Metadata  map[string]any `bson:"metadata"    json:"metadata,omitempty"`
}

type RunFilter struct {
	AgentID   string     `json:"agent_id,omitempty"`
	TenantID  string     `json:"tenant_id,omitempty"`
	Status    string     `json:"status,omitempty"`
	StartTime *time.Time `json:"start_time,omitempty"`
	EndTime   *time.Time `json:"end_time,omitempty"`
	Limit     int        `json:"limit,omitempty"`
	Offset    int        `json:"offset,omitempty"`
}

type RunInput struct {
	Content   string            `bson:"content"    json:"content"`
	Variables map[string]string `bson:"variables"  json:"variables,omitempty"`
	Context   map[string]any    `bson:"context"    json:"context,omitempty"`
}

type RunOutput struct {
	Content      string         `bson:"content"       json:"content"`
	TokensUsed   int            `bson:"tokens_used"   json:"tokens_used,omitempty"`
	Cost         float64        `bson:"cost"          json:"cost,omitempty"`
	FinishReason string         `bson:"finish_reason" json:"finish_reason,omitempty"`
	Metadata     map[string]any `bson:"metadata"      json:"metadata,omitempty"`
}

type RunStats struct {
	TotalRuns   int64   `bson:"total_runs"   json:"total_runs"`
	Completed   int64   `bson:"completed"    json:"completed"`
	Failed      int64   `bson:"failed"       json:"failed"`
	TotalTokens int64   `bson:"total_tokens" json:"total_tokens"`
	TotalCost   float64 `bson:"total_cost"   json:"total_cost"`
}

type RunStore interface {
	// RecordRun inserts a new run document.
	RecordRun(ctx context.Context, doc *RunDocument) error
	// GetByID retrieves a run by ID.
	GetByID(ctx context.Context, id string) (*RunDocument, error)
	// UpdateStatus updates the status, output, error, and timing of a run.
	UpdateStatus(ctx context.Context, id, status string, output *RunOutput, errMsg string) error
	// List returns runs matching the filter.
	List(ctx context.Context, filter RunFilter) ([]*RunDocument, error)
	// Stats returns aggregated statistics for runs matching the filter.
	Stats(ctx context.Context, agentID, tenantID string) (*RunStats, error)
}

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