Remove prompt version selectors
This commit is contained in:
@@ -24,10 +24,9 @@ import (
|
||||
)
|
||||
|
||||
type fakePromptRepo struct {
|
||||
def *domain.PromptDefinition
|
||||
err error
|
||||
lastID string
|
||||
lastVersion string
|
||||
def *domain.PromptDefinition
|
||||
err error
|
||||
lastID string
|
||||
}
|
||||
|
||||
type fakeExecutionProfileRepo struct {
|
||||
@@ -43,7 +42,6 @@ type staticPromptRepo struct {
|
||||
func (r staticPromptRepo) GetPromptDefinition(
|
||||
context.Context,
|
||||
string,
|
||||
string,
|
||||
) (*domain.PromptDefinition, error) {
|
||||
return r.def, nil
|
||||
}
|
||||
@@ -94,9 +92,8 @@ func (f *fakeExecutionProfileRepo) GetProfile(ctx context.Context, id string) (*
|
||||
return nil, errors.New("profile not found")
|
||||
}
|
||||
|
||||
func (f *fakePromptRepo) GetPromptDefinition(ctx context.Context, id string, version string) (*domain.PromptDefinition, error) {
|
||||
func (f *fakePromptRepo) GetPromptDefinition(ctx context.Context, id string) (*domain.PromptDefinition, error) {
|
||||
f.lastID = id
|
||||
f.lastVersion = version
|
||||
if f.err != nil {
|
||||
return nil, f.err
|
||||
}
|
||||
@@ -358,9 +355,8 @@ func TestRunnerPrepareWithExplicitProfileSelection(t *testing.T) {
|
||||
|
||||
runner := NewRunner(promptRepo, execRepo, reader, renderer, llmClient, nil, nil)
|
||||
prepared, err := runner.Prepare(context.Background(), domain.RunRequest{
|
||||
PromptID: "p",
|
||||
PromptVersion: "1",
|
||||
ProfileID: "exec",
|
||||
PromptID: "p",
|
||||
ProfileID: "exec",
|
||||
Inputs: map[string]domain.ArtifactRef{
|
||||
"transcript": {Type: domain.ArtifactRefFile, URI: "a://t"},
|
||||
"glossary": {Type: domain.ArtifactRefFile, URI: "a://g"},
|
||||
@@ -956,8 +952,8 @@ func TestRunnerPrepareJSONSchemaBuildsStructuredOutputSpec(t *testing.T) {
|
||||
if prepared.StructuredOutput.JSONSchema == nil {
|
||||
t.Fatal("expected structured output json_schema payload")
|
||||
}
|
||||
if prepared.StructuredOutput.JSONSchema.Name != "p_1" {
|
||||
t.Fatalf("expected derived schema name p_1, got %q", prepared.StructuredOutput.JSONSchema.Name)
|
||||
if prepared.StructuredOutput.JSONSchema.Name != "p" {
|
||||
t.Fatalf("expected derived schema name p, got %q", prepared.StructuredOutput.JSONSchema.Name)
|
||||
}
|
||||
if prepared.StructuredOutput.JSONSchema.Strict != true {
|
||||
t.Fatalf("expected strict=true, got %v", prepared.StructuredOutput.JSONSchema.Strict)
|
||||
@@ -1022,28 +1018,25 @@ func TestRunnerRunJSONSchemaSchemaLoadFailureFailsBeforeLLM(t *testing.T) {
|
||||
|
||||
func TestDeriveStructuredSchemaName(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
id string
|
||||
version string
|
||||
want string
|
||||
name string
|
||||
id string
|
||||
want string
|
||||
}{
|
||||
{
|
||||
name: "sanitizes punctuation and keeps dashes",
|
||||
id: "prompt.id/alpha",
|
||||
version: "1.0.0-beta",
|
||||
want: "prompt_id_alpha_1_0_0-beta",
|
||||
name: "sanitizes punctuation and keeps dashes",
|
||||
id: "prompt.id/alpha",
|
||||
want: "prompt_id_alpha",
|
||||
},
|
||||
{
|
||||
name: "fallback when empty",
|
||||
id: "",
|
||||
version: "",
|
||||
want: "promptkit_schema",
|
||||
name: "fallback when empty",
|
||||
id: "",
|
||||
want: "promptkit_schema",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range tests {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
got := deriveStructuredSchemaName(tc.id, tc.version)
|
||||
got := deriveStructuredSchemaName(tc.id)
|
||||
if got != tc.want {
|
||||
t.Fatalf("expected %q, got %q", tc.want, got)
|
||||
}
|
||||
@@ -1253,9 +1246,8 @@ func TestRunnerRunSuccessful(t *testing.T) {
|
||||
|
||||
runner := NewRunner(promptRepo, execRepo, reader, renderer, llmClient, nil, nil)
|
||||
res, err := runner.Run(context.Background(), domain.RunRequest{
|
||||
PromptID: "p",
|
||||
PromptVersion: "1",
|
||||
ProfileID: "exec",
|
||||
PromptID: "p",
|
||||
ProfileID: "exec",
|
||||
Inputs: map[string]domain.ArtifactRef{
|
||||
"transcript": {Type: domain.ArtifactRefFile, URI: "a://t"},
|
||||
"glossary": {Type: domain.ArtifactRefFile, URI: "a://g"},
|
||||
@@ -2325,8 +2317,8 @@ func TestRunnerRepairStateMachine(t *testing.T) {
|
||||
t.Fatalf("initial structured output = %+v, want present %v", initialRequest.StructuredOutput, tc.structured)
|
||||
}
|
||||
if tc.structured && (initialRequest.StructuredOutput.JSONSchema == nil ||
|
||||
initialRequest.StructuredOutput.JSONSchema.Name != "p_1") {
|
||||
t.Fatalf("initial JSON Schema metadata = %+v, want derived schema name p_1", initialRequest.StructuredOutput)
|
||||
initialRequest.StructuredOutput.JSONSchema.Name != "p") {
|
||||
t.Fatalf("initial JSON Schema metadata = %+v, want derived schema name p", initialRequest.StructuredOutput)
|
||||
}
|
||||
|
||||
for index, req := range repairer.reqs {
|
||||
|
||||
Reference in New Issue
Block a user