yao/neo/rag/types.go
Max fdf71f4cfd Update Go version and dependencies; enhance Neo API functionality
- Bump Go version from 1.22.2 to 1.23 for improved performance.
- Update dependencies in go.mod and go.sum, including:
  - github.com/PuerkitoBio/goquery to v1.10.1
  - github.com/dchest/captcha to v1.1.0
  - github.com/evanw/esbuild to v0.24.2
  - github.com/fatih/color to v1.18.0
  - github.com/fsnotify/fsnotify to v1.8.0
  - github.com/spf13/cobra to v1.8.1
  - github.com/stretchr/testify to v1.10.0
  - github.com/xuri/excelize/v2 to v2.9.0
- Introduce RAG (Retrieval-Augmented Generation) initialization in the Neo API, enhancing assistant capabilities.
- Refactor workflows to use Go 1.23 for testing and CI processes, ensuring compatibility with the latest features.
2025-01-02 13:06:36 +08:00

29 lines
995 B
Go

package rag
// Setting RAG settings
type Setting struct {
Engine Engine `json:"engine" yaml:"engine"`
Vectorizer Vectorizer `json:"vectorizer" yaml:"vectorizer"`
Upload Upload `json:"upload" yaml:"upload"`
IndexPrefix string `json:"index_prefix" yaml:"index_prefix"`
}
// Engine the vector database engine settings
type Engine struct {
Driver string `json:"driver" yaml:"driver"`
Options map[string]interface{} `json:"options" yaml:"options"`
}
// Vectorizer the text vectorizer settings
type Vectorizer struct {
Driver string `json:"driver" yaml:"driver"`
Options map[string]interface{} `json:"options" yaml:"options"`
}
// Upload the file upload settings
type Upload struct {
Async bool `json:"async" yaml:"async"`
AllowedTypes []string `json:"allowed_types" yaml:"allowed_types"`
ChunkSize int `json:"chunk_size" yaml:"chunk_size"`
ChunkOverlap int `json:"chunk_overlap" yaml:"chunk_overlap"`
}