schema

type AIChatMessage struct {
	Content string
}

type ChatMessage interface {
	GetType() ChatMessageType
	GetContent() string
}

type ChatMessageType string

type CodeChunk struct {
	Content         string
	LineStart       int
	LineEnd         int
	Type            string
	Identifier      string
	Annotations     map[string]string
	TokenCount      int
	EnrichedContent string
	ParentContext   string
	ContextLevel    int
}

type CodeChunkingOptions struct {
	ChunkSize         int
	OverlapTokens     int
	PreserveStructure bool
	LanguageHints     []string
	MaxLinesPerChunk  int
	MinCharsPerChunk  int
}

type CodeEntityDefinition struct {
	Type          string
	Name          string
	LineStart     int
	LineEnd       int
	Visibility    string
	Signature     string
	Documentation string
}

type CodeSymbol struct {
	Name      string
	Type      string
	LineStart int
	LineEnd   int
	IsExport  bool
}

type CollectionInfo struct {
	Name           string `json:"name"`            // Name of the collection.
	PointsCount    uint64 `json:"points_count"`    // Number of points (vectors) in the collection.
	VectorSize     uint64 `json:"vector_size"`     // Dimensionality of the vectors in this collection.
	VectorDistance string `json:"vector_distance"` // Distance metric used by the collection (e.g., "Cosine").
}

type ContentPart interface {
	String() string
	// contains filtered or unexported methods
}

type Document struct {
	PageContent string
	Metadata    map[string]any
}

type FileMetadata struct {
	FilePath    string
	Language    string
	Imports     []string
	Definitions []CodeEntityDefinition
	Symbols     []CodeSymbol
	Properties  map[string]string
}

type HumanChatMessage struct {
	Content string
}

type MessageContent struct {
	Role  ChatMessageType
	Parts []ContentPart
}

type ModelDetails struct {
	Family        string
	ParameterSize string
	Quantization  string
	Dimension     int64
}

type ParserPlugin interface {
	Name() string
	Extensions() []string
	CanHandle(path string, info fs.FileInfo) bool
	Chunk(content string, path string, opts *CodeChunkingOptions) ([]CodeChunk, error)
	ExtractMetadata(content string, path string) (FileMetadata, error)
}

type Retriever interface {
	GetRelevantDocuments(ctx context.Context, query string) ([]Document, error)
}

type SystemChatMessage struct {
	Content string
}

type TextContent struct {
	Text string
}