72 lines
2.8 KiB
Go
72 lines
2.8 KiB
Go
package format
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"gitea.maximumdirect.net/eric/promptkit"
|
|
)
|
|
|
|
type PromptInspection struct {
|
|
PromptID string `json:"prompt_id"`
|
|
PromptVersion string `json:"prompt_version"`
|
|
PromptHash string `json:"prompt_hash"`
|
|
DefaultProfileID string `json:"default_profile_id"`
|
|
Inputs []PromptInspectionInput `json:"inputs"`
|
|
OutputContract OutputContract `json:"output_contract"`
|
|
}
|
|
|
|
type PromptInspectionInput struct {
|
|
Name string `json:"name"`
|
|
Required bool `json:"required"`
|
|
ContentType string `json:"content_type"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
type OutputContract struct {
|
|
Format string `json:"format"`
|
|
ValidationMode string `json:"validation_mode"`
|
|
SchemaPath string `json:"schema_path"`
|
|
RepairAttempts int `json:"repair_attempts"`
|
|
}
|
|
|
|
func FormatPromptInspection(value *promptkit.PromptInspection, outputFormat OutputFormat) ([]byte, error) {
|
|
if value == nil {
|
|
return nil, errors.New("prompt inspection is nil")
|
|
}
|
|
dto := PromptInspection{
|
|
PromptID: value.PromptID, PromptVersion: value.PromptVersion, PromptHash: value.PromptHash, DefaultProfileID: value.DefaultProfileID,
|
|
Inputs: make([]PromptInspectionInput, len(value.Inputs)),
|
|
OutputContract: OutputContract{Format: string(value.OutputContract.Format), ValidationMode: string(value.OutputContract.ValidationMode), SchemaPath: value.OutputContract.SchemaPath, RepairAttempts: value.OutputContract.RepairAttempts},
|
|
}
|
|
for i, input := range value.Inputs {
|
|
dto.Inputs[i] = PromptInspectionInput{Name: input.Name, Required: input.Required, ContentType: input.ContentType, Description: input.Description}
|
|
}
|
|
switch outputFormat {
|
|
case OutputFormatJSON:
|
|
data, err := json.MarshalIndent(dto, "", " ")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return append(data, '\n'), nil
|
|
case OutputFormatText:
|
|
var b bytes.Buffer
|
|
fmt.Fprintf(&b, "prompt_id: %s\nprompt_version: %s\nprompt_hash: %s\ndefault_profile_id: %s\ninputs:", dto.PromptID, dto.PromptVersion, dto.PromptHash, dto.DefaultProfileID)
|
|
if len(dto.Inputs) == 0 {
|
|
fmt.Fprintln(&b, " []")
|
|
} else {
|
|
fmt.Fprintln(&b)
|
|
for _, input := range dto.Inputs {
|
|
fmt.Fprintf(&b, " - name: %s\n required: %t\n content_type: %s\n description: %s\n", input.Name, input.Required, input.ContentType, strconv.Quote(input.Description))
|
|
}
|
|
}
|
|
fmt.Fprintf(&b, "output_contract:\n format: %s\n validation_mode: %s\n schema_path: %s\n repair_attempts: %d\n", dto.OutputContract.Format, dto.OutputContract.ValidationMode, dto.OutputContract.SchemaPath, dto.OutputContract.RepairAttempts)
|
|
return b.Bytes(), nil
|
|
default:
|
|
return nil, fmt.Errorf("%w: %q", ErrUnknownPreparedRunFormat, outputFormat)
|
|
}
|
|
}
|