107 lines
4.1 KiB
Go
107 lines
4.1 KiB
Go
package artifacts
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/source"
|
|
)
|
|
|
|
type ArtifactCandidate struct {
|
|
Index int `json:"index"`
|
|
ExtractorKey string `json:"extractor_key"`
|
|
ArtifactType string `json:"artifact_type"`
|
|
SchemaVersion string `json:"schema_version"`
|
|
Payload json.RawMessage `json:"payload"`
|
|
SourceRefs []source.SourceRef `json:"source_refs,omitempty"`
|
|
Metadata map[string]any `json:"metadata,omitempty"`
|
|
}
|
|
|
|
type Artifact struct {
|
|
ExtractorKey string `json:"extractor_key"`
|
|
ArtifactType string `json:"artifact_type"`
|
|
SchemaVersion string `json:"schema_version"`
|
|
Payload json.RawMessage `json:"payload"`
|
|
SourceRefs []source.SourceRef `json:"source_refs,omitempty"`
|
|
Metadata map[string]any `json:"metadata,omitempty"`
|
|
}
|
|
|
|
type RejectedArtifact struct {
|
|
Candidate ArtifactCandidate `json:"candidate"`
|
|
ValidatorName string `json:"validator_name"`
|
|
ReasonCode string `json:"reason_code"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type ArtifactLaneManifest struct {
|
|
ID string `json:"id"`
|
|
Extractor string `json:"extractor"`
|
|
Merger string `json:"merger"`
|
|
Normalizer string `json:"normalizer"`
|
|
Validators []string `json:"validators,omitempty"`
|
|
Metadata map[string]any `json:"metadata,omitempty"`
|
|
}
|
|
|
|
type LLMProfileManifest struct {
|
|
ID string `json:"id"`
|
|
Provider string `json:"provider,omitempty"`
|
|
Model string `json:"model,omitempty"`
|
|
}
|
|
|
|
type ReferenceProvenance struct {
|
|
Stage string `json:"stage,omitempty"`
|
|
LaneID string `json:"lane_id,omitempty"`
|
|
SlotName string `json:"slot_name"`
|
|
OriginType string `json:"origin_type"`
|
|
OriginURI string `json:"origin_uri,omitempty"`
|
|
Digest string `json:"digest,omitempty"`
|
|
MediaType string `json:"media_type,omitempty"`
|
|
SizeBytes int64 `json:"size_bytes,omitempty"`
|
|
BindingSource string `json:"binding_source,omitempty"`
|
|
}
|
|
|
|
type RunManifest struct {
|
|
RunID string `json:"run_id,omitempty"`
|
|
PipelineID string `json:"pipeline_id,omitempty"`
|
|
PipelineDigest string `json:"pipeline_digest,omitempty"`
|
|
InputModule string `json:"input_module,omitempty"`
|
|
Chunker string `json:"chunker,omitempty"`
|
|
SourceDigests []string `json:"source_digests,omitempty"`
|
|
Extractors []string `json:"extractors,omitempty"`
|
|
Merger string `json:"merger,omitempty"`
|
|
Normalizer string `json:"normalizer,omitempty"`
|
|
OutputEncoder string `json:"output_encoder,omitempty"`
|
|
ModuleMetadata map[string]map[string]any `json:"module_metadata,omitempty"`
|
|
ArtifactLanes []ArtifactLaneManifest `json:"artifact_lanes,omitempty"`
|
|
References []ReferenceProvenance `json:"references,omitempty"`
|
|
LLMProfiles []LLMProfileManifest `json:"llm_profiles,omitempty"`
|
|
Metadata map[string]any `json:"metadata,omitempty"`
|
|
SchemaVersion string `json:"schema_version,omitempty"`
|
|
ValidationStatus string `json:"validation_status,omitempty"`
|
|
StartedAt *time.Time `json:"started_at,omitempty"`
|
|
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
|
}
|
|
|
|
func ArtifactFromCandidate(candidate ArtifactCandidate) Artifact {
|
|
return Artifact{
|
|
ExtractorKey: candidate.ExtractorKey,
|
|
ArtifactType: candidate.ArtifactType,
|
|
SchemaVersion: candidate.SchemaVersion,
|
|
Payload: append(json.RawMessage(nil), candidate.Payload...),
|
|
SourceRefs: append([]source.SourceRef(nil), candidate.SourceRefs...),
|
|
Metadata: copyMetadata(candidate.Metadata),
|
|
}
|
|
}
|
|
|
|
func copyMetadata(metadata map[string]any) map[string]any {
|
|
if len(metadata) == 0 {
|
|
return nil
|
|
}
|
|
|
|
copied := make(map[string]any, len(metadata))
|
|
for key, value := range metadata {
|
|
copied[key] = value
|
|
}
|
|
return copied
|
|
}
|