mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-31 04:59:24 +08:00
90f46b0b4d
Ports doc-level auto-metadata extraction to Go and adds the knowledge_compiler component with scheduler/routing. Fixes Extractor metadata injection type assertion and enable_metadata default-on.
48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
//
|
|
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// 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 common
|
|
|
|
const (
|
|
TaskTypeIngestionTask = "ingestion_task"
|
|
TaskTypeIngestionTest = "ingestion_test"
|
|
)
|
|
|
|
type TaskMessage struct {
|
|
TaskID string `json:"task_id" binding:"required"`
|
|
TaskType string `json:"task_type" binding:"required"`
|
|
}
|
|
|
|
type TaskHandle interface {
|
|
GetMessage() TaskMessage
|
|
Ack() error
|
|
Nack() error
|
|
|
|
// InProgress resets the AckWait timer without acknowledging the message,
|
|
// signalling the broker that the worker is still processing. Call
|
|
// periodically during long tasks to avoid in-flight redelivery.
|
|
InProgress() error
|
|
}
|
|
|
|
// RawMessage is a broker message carrying opaque bytes (used by the dataset-level
|
|
// compile consumer, which publishes arbitrary JSON payloads rather than the
|
|
// TaskMessage shape). The NATS engine returns RawMessage from FetchKnowledgeCompileMessages.
|
|
type RawMessage interface {
|
|
Data() []byte
|
|
Ack() error
|
|
Nak() error
|
|
}
|