Expose structured generation errors to consumers
This commit is contained in:
@@ -1164,8 +1164,9 @@ func TestArtifactReaderFailuresPreserveArtifactLoadErrors(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestRunAddsLLMGenerateToCollaboratorPublicError(t *testing.T) {
|
||||
injectedErr := errors.New("injected model client failure")
|
||||
engine := newContractEngineWithOptions(t, frameworkSchemaDir,
|
||||
promptkit.WithLLMClient(&fakeLLMClient{err: promptkit.ErrArtifactLoad}),
|
||||
promptkit.WithLLMClient(&fakeLLMClient{err: injectedErr}),
|
||||
)
|
||||
|
||||
_, err := engine.Run(context.Background(), promptkit.RunRequest{
|
||||
@@ -1178,8 +1179,12 @@ func TestRunAddsLLMGenerateToCollaboratorPublicError(t *testing.T) {
|
||||
if !errors.Is(err, promptkit.ErrLLMGenerate) {
|
||||
t.Fatalf("expected ErrLLMGenerate, got %v", err)
|
||||
}
|
||||
if !errors.Is(err, promptkit.ErrArtifactLoad) {
|
||||
t.Fatalf("expected preserved ErrArtifactLoad, got %v", err)
|
||||
if !errors.Is(err, injectedErr) {
|
||||
t.Fatalf("expected preserved injected error, got %v", err)
|
||||
}
|
||||
var generationErr *promptkit.GenerationError
|
||||
if errors.As(err, &generationErr) {
|
||||
t.Fatalf("injected error became GenerationError: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user