chat

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

type AppliedJoinEdge struct {
	JoinPath     RelationInstanceID `json:"joinPath"`
	SourcePath   RelationInstanceID `json:"sourcePath"`
	ConstraintID string             `json:"constraintId"`
	Direction    string             `json:"direction"`
	CandidateID  JoinCandidateID    `json:"candidateId"`
	Fields       []JoinFieldPair    `json:"fields"`
}

type Bookmark struct {
	ID         string
	ProjectID  string
	SourceID   string
	TargetKind string
	Title      string
	Tags       []string
	CreatedAt  time.Time
	UpdatedAt  time.Time
	Snapshot   BookmarkSnapshot
}

type BookmarkSnapshot struct {
	SourceID  string
	RecordSet RecordSet
	View      *RecordSetView
	Selection *Selection
}

type CellRange struct {
	FirstRow int `json:"firstRow"`
	LastRow  int `json:"lastRow"`
	FirstCol int `json:"firstCol"`
	LastCol  int `json:"lastCol"`
}

type ChartCandidate struct {
	Spec   ChartSpec
	Score  int
	Reason string
}

type ChartKind string

type ChartPoint struct {
	Label string
	Value float64
}

type ChartSpec struct {
	Kind          ChartKind
	Title         string
	Dimension     string
	Measure       string
	Aggregation   string
	Ordering      string
	Limit         int
	SourceColumns []string
	Labels        []string
	Bucket        string
	Points        []ChartPoint
}

type ChatMessage struct {
	ID          string
	Role        string
	Kind        string
	Text        string
	QueryID     string
	RecordSetID string
	CreatedAt   time.Time
}

type ChatScope struct {
	// ProjectID partitions retained project artefacts. It deliberately does not
	// participate in the existing session scope hash: Phase 2 sessions retain
	// their exact historic scope identity during the Phase 4 migration.
	ProjectID         string
	Environment       string
	Database          string
	AccessFingerprint string
	Sources           map[string]string
}

type ChatSession struct {
	ID         string
	Title      string
	CreatedAt  time.Time
	UpdatedAt  time.Time
	Messages   []ChatMessage
	Queries    []ExecutedQuery
	RecordSets map[string]RecordSet
	Bookmarks  map[string]Bookmark
	Workspace  WorkspaceState
}

type ContextReference struct {
	Kind      string `json:"kind"`
	ProjectID string `json:"projectId,omitempty"`
	SourceID  string `json:"sourceId,omitempty"`
	ObjectID  string `json:"objectId"`
	Title     string `json:"title"`
}

type ContextualConversation interface {
	AskWithContext(context.Context, string, string) (Turn, error)
}

type Conversation interface {
	Ask(context.Context, string) (Turn, error)
}

type DTQLExecutor interface {
	RunDTQL(context.Context, string, []byte, map[string]any) (secureread.Result, error)
}

type Dock struct {
	ID        string           `json:"id"`
	Reference ContextReference `json:"reference"`
	Title     string           `json:"title"`
}

type ExecutedQuery struct {
	ID              string
	OriginMessageID string
	Title           string
	DTQL            string
	Source          string
	Parameters      map[string]any
	ExecutedAt      time.Time
	Error           string
}

type ForeignKey struct {
	ConstraintID string
	Schema       string
	FromRelation string
	FromFields   []string
	ToSchema     string
	ToRelation   string
	ToFields     []string
}

type ForeignKeyJoinApplication struct {
	Source   string
	Snapshot ForeignKeySnapshot
	// Refresh is called for candidate exposure and again immediately before
	// apply. It is normally LoadSQLiteForeignKeySnapshot bound to the selected
	// source; a stale edge never falls back to the startup snapshot.
	Refresh       func(context.Context) (ForeignKeySnapshot, error)
	CanReadTarget func(context.Context, RelationInstance) error
	Executor      DTQLExecutor
	Secure        bool
}

type ForeignKeySnapshot struct {
	Source string
	Keys   []ForeignKey
	// Columns contains the source's ordered physical columns, keyed by
	// lowercase schema.relation. It is used to expand wildcards before a JOIN.
	Columns map[string][]string
}

type GridColumn struct {
	Name    string
	Numeric bool
}

type GridModel struct {
	Columns []GridColumn
	Rows    [][]string
	// RawRows preserves the structured source values alongside formatted cells
	// for future typed interactions without leaking database types into the UI
	// component adapter.
	RawRows [][]any
	// SourceRows maps a displayed (possibly sorted) row back to its immutable
	// RecordSet row index for durable selections.
	SourceRows []int
	// contains filtered or unexported fields
}

type InterpretProvider struct {
	Protocol string `json:"protocol"`
	BaseURL  string `json:"baseUrl"`
	Model    string `json:"model"`
	APIKey   string `json:"apiKey"`
}

type InterpretRequest struct {
	Question string            `json:"question"`
	Schema   string            `json:"schema"`
	Provider InterpretProvider `json:"provider"`
}

type InterpretResult struct {
	DTQL  string      `json:"dtql"`
	Usage *TokenUsage `json:"usage,omitempty"`
}

type JoinApplication interface {
	Candidates(context.Context, RecordSet) ([]JoinCandidate, error)
	Apply(context.Context, RecordSet, JoinCandidateID) (QueryResult, error)
}

type JoinCandidate struct {
	ID           JoinCandidateID
	Source       RelationInstance
	Target       RelationInstance
	ConstraintID string
	Direction    string // outgoing or incoming, relative to Source
	Cardinality  string // many-to-one or one-to-many, relative to Source
	Fields       []JoinFieldPair
	Evidence     string // foreign-key
}

type JoinCandidateID string

type JoinFieldPair struct{ SourceField, TargetField string }

type JoinLineage struct {
	ParentRecordSetID string            `json:"parentRecordSetId"`
	CandidateID       JoinCandidateID   `json:"candidateId"`
	AppliedEdges      []AppliedJoinEdge `json:"appliedEdges,omitempty"`
}

type Option func(*conversationConfig) error

type ProjectCatalog struct {
	ID      string
	Title   string
	Objects []ProjectObject
}

type ProjectChoice struct {
	Key    string
	Title  string
	Detail string
}

type ProjectObject struct {
	Reference   ContextReference
	Columns     []string
	ColumnTypes map[string]string
}

type QueryResult struct {
	// Title is presentation metadata supplied by the same structured tool
	// action as DTQL. It is never included in, or interpreted as, executable
	// query text.
	Title       string
	DTQL        string
	QueryID     string
	RecordSetID string
	Result      secureread.Result
	Parameters  map[string]any
	Source      string
	SourceID    string
	// Lineage is DataTug-owned execution provenance, never model supplied.
	Lineage *JoinLineage
	Err     error
}

type RecordSet struct {
	ID              string
	SessionID       string
	QueryID         string
	OriginMessageID string
	Title           string
	DTQL            string
	Source          string
	Environment     string
	Database        string
	Parameters      map[string]any
	CreatedAt       time.Time
	Result          secureread.Result
	Lineage         *JoinLineage
}

type RecordSetView struct {
	ID          string    `json:"id"`
	RecordSetID string    `json:"recordSetId"`
	Title       string    `json:"title"`
	RowIndices  []int     `json:"rowIndices"`
	Columns     []string  `json:"columns,omitempty"`
	OrderBy     string    `json:"orderBy,omitempty"`
	Descending  bool      `json:"descending,omitempty"`
	CreatedAt   time.Time `json:"createdAt"`
}

type RelationInstance struct {
	ID       RelationInstanceID
	Schema   string
	Relation string
	Alias    string
}

type RelationInstanceID string

type Selection struct {
	ID        string      `json:"id"`
	ViewID    string      `json:"viewId"`
	Title     string      `json:"title"`
	Rows      []int       `json:"rows,omitempty"`
	Columns   []string    `json:"columns,omitempty"`
	Ranges    []CellRange `json:"ranges,omitempty"`
	CreatedAt time.Time   `json:"createdAt"`
}

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

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

type TokenUsage struct {
	InputTokens  int64 `json:"inputTokens"`
	OutputTokens int64 `json:"outputTokens"`
	TotalTokens  int64 `json:"totalTokens"`
}

type Turn struct {
	Text    string
	Queries []QueryResult
	Actions []WorkspaceActionResult
	Usage   *TokenUsage
}

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

type WorkspaceAction struct {
	Kind        string           `json:"kind"`
	Reference   ContextReference `json:"reference,omitempty" jsonschema:"Exact existing object reference; omit for dock to dock the current selection"`
	RecordSetID string           `json:"recordSetId,omitempty"`
	ViewID      string           `json:"viewId,omitempty"`
	Title       string           `json:"title,omitempty"`
	Column      string           `json:"column,omitempty"`
	Equals      string           `json:"equals,omitempty"`
	Contains    string           `json:"contains,omitempty"`
	OrderBy     string           `json:"orderBy,omitempty"`
	Descending  bool             `json:"descending,omitempty"`
	Limit       int              `json:"limit,omitempty"`
	RowStart    int              `json:"rowStart,omitempty"`
	RowEnd      int              `json:"rowEnd,omitempty"`
	Columns     []string         `json:"columns,omitempty"`
	Rows        []int            `json:"rows,omitempty"`
	Ranges      []CellRange      `json:"ranges,omitempty"`
	DockID      string           `json:"dockId,omitempty"`
	BookmarkID  string           `json:"bookmarkId,omitempty"`
	Tag         string           `json:"tag,omitempty"`
	Search      string           `json:"search,omitempty"`
	Tags        []string         `json:"tags,omitempty"`
}

type WorkspaceActionResult struct {
	Reference ContextReference
	Summary   string
	Error     string
	Err       error
}

type WorkspaceState struct {
	Views              map[string]RecordSetView `json:"views,omitempty"`
	Selections         map[string]Selection     `json:"selections,omitempty"`
	Attachments        []ContextReference       `json:"attachments,omitempty"`
	Docks              []Dock                   `json:"docks,omitempty"`
	CurrentSelectionID string                   `json:"currentSelectionId,omitempty"`
	ActiveTab          string                   `json:"activeTab,omitempty"`
}