postgresql

type ColumnInfo struct {
	ColumnName string `json:"column_name"`
	DataType   string `json:"data_type"`
	NotNull    bool   `json:"not_null"`
	Default    string `json:"default,omitempty"`
}

type Config struct {
	ConnectionString string        `json:"connection_string"`
	PoolSize         int           `json:"pool_size"`
	StatementTimeout time.Duration `json:"statement_timeout"`
}

type ExecuteResult struct {
	CommandTag   string `json:"command_tag"`
	RowsAffected int64  `json:"rows_affected"`
}

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

type QueryResult struct {
	Columns  []string                 `json:"columns"`
	Rows     []map[string]interface{} `json:"rows"`
	RowCount int64                    `json:"row_count"`
}

type TableInfo struct {
	Schema      string `json:"schema"`
	Name        string `json:"name"`
	Type        string `json:"type"`
	RowEstimate int64  `json:"row_estimate,omitempty"`
}

type ToolDefinition struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	InputSchema json.RawMessage `json:"inputSchema"`
}

type ToolHandler func(ctx context.Context, args json.RawMessage) (interface{}, error)