Go knowledge compiler with scheduler-driven dataset compilation (#17913)

Ports dataset knowledge compilation (wiki/graph/tree/mindmap) to the Go
scheduler with a status contract, aligns wiki storage/retrieval with
Python, sizes prompts by content_length, and resolves embedding batch
size from provider capability.
This commit is contained in:
Zhichang Yu
2026-08-06 15:54:00 +08:00
committed by GitHub
parent addc5acdc0
commit 2e37997ab9
116 changed files with 6678 additions and 1942 deletions

View File

@@ -30,8 +30,8 @@ type Config struct {
admin AdminConfig
apiServer APIServerConfig
ingestor IngestorConfig
syncer SyncerConfig
ingestor IngestorConfig
log LogConfig
smtp common.SMTPConfig

View File

@@ -12,19 +12,26 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package config
import "github.com/spf13/viper"
type IngestorConfig struct {
// MaxConcurrentWorkers bounds how many ingestion tasks the ingestor runs in
// parallel (the task channel width and dataset-level compile worker count
// default to this value). 0/negative falls back to runtime.NumCPU().
MaxConcurrentWorkers int `mapstructure:"max_concurrent_workers"`
// CompilerPoolSize bounds the process-wide knowledge-compilation worker
// pool that drives the cross-doc KNN / LLM-merge / write stages. 0/negative
// falls back to runtime.NumCPU() (or KC_COMPILE_CONCURRENCY if set).
CompilerPoolSize int `mapstructure:"compiler_pool_size"`
}
func (c *Config) ParseIngestorConfig(v *viper.Viper) error {
// Default Ingestor config
c.ingestor.MaxConcurrentWorkers = 1
c.ingestor.CompilerPoolSize = 0
if !v.IsSet("ingestor") {
return nil
@@ -38,5 +45,13 @@ func (c *Config) ParseIngestorConfig(v *viper.Viper) error {
c.ingestor.MaxConcurrentWorkers = sub.GetInt("max_concurrent_workers")
}
if sub.IsSet("compiler_pool_size") {
c.ingestor.CompilerPoolSize = sub.GetInt("compiler_pool_size")
}
return nil
}
func (c *Config) GetIngestorConfig() *IngestorConfig {
return &c.ingestor
}