schema

type ChatMessageType string

type CodeChunk struct {
	Content         string            `json:"content"`
	LineStart       int               `json:"lineStart"`
	LineEnd         int               `json:"lineEnd"`
	Type            string            `json:"type"`
	Identifier      string            `json:"identifier"`
	Annotations     map[string]string `json:"annotations"`
	TokenCount      int               `json:"tokenCount"`
	EnrichedContent string            `json:"enrichedContent"`
	ParentContext   string            `json:"parentContext"`
	ContextLevel    int               `json:"contextLevel"`
}

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

type CodeEntityDefinition struct {
	Type          string `json:"type"`
	Name          string `json:"name"`
	LineStart     int    `json:"line_start"`
	LineEnd       int    `json:"line_end"`
	Visibility    string `json:"visibility"`
	Signature     string `json:"signature"`
	Documentation string `json:"documentation"`
}

type CodeSymbol struct {
	Name      string `json:"name"`
	Type      string `json:"type"`
	LineStart int    `json:"line_start"`
	LineEnd   int    `json:"line_end"`
	IsExport  bool   `json:"is_export"`
}

type ContentChoice struct {
	Content          string
	StopReason       string
	GenerationInfo   map[string]any
	ReasoningContent string
}

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

type ContentResponse struct {
	Choices []*ContentChoice
}

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

type FileMetadata struct {
	FilePath    string                 `json:"file_path"`
	Language    string                 `json:"language"`
	PackageName string                 `json:"package_name"`
	Imports     []string               `json:"imports"`
	Definitions []CodeEntityDefinition `json:"definitions"`
	Symbols     []CodeSymbol           `json:"symbols"`
	Properties  map[string]string      `json:"properties"`
}

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 TextContent struct {
	Text string
}