diff --git a/callback/prompt_layer.go b/callback/prompt_layer.go index 0016f60..141cb88 100644 --- a/callback/prompt_layer.go +++ b/callback/prompt_layer.go @@ -95,6 +95,7 @@ func (cb PromptLayerHandler) OnModelEnd(ctx context.Context, input *schema.Model invocationParams, _ := runInfo["invocationParams"].(map[string]any) var kwargs map[string]any + prompt, ok := runInfo["prompt"].(string) if ok { kwargs = map[string]any{