From d01f9f5f95371a0361bbdeebcba03070fb7dda69 Mon Sep 17 00:00:00 2001 From: Darren Shepherd Date: Mon, 6 May 2024 13:50:58 -0700 Subject: [PATCH] bug: sys.chat.finish did not return proper message --- pkg/builtin/builtin.go | 2 +- pkg/tests/runner_test.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pkg/builtin/builtin.go b/pkg/builtin/builtin.go index f5799baf..184bda30 100644 --- a/pkg/builtin/builtin.go +++ b/pkg/builtin/builtin.go @@ -630,7 +630,7 @@ func (e *ErrChatFinish) Error() string { func SysChatFinish(ctx context.Context, env []string, input string) (string, error) { var params struct { - Message string `json:"message,omitempty"` + Message string `json:"summary,omitempty"` } if err := json.Unmarshal([]byte(input), ¶ms); err != nil { return "", &ErrChatFinish{ diff --git a/pkg/tests/runner_test.go b/pkg/tests/runner_test.go index cb55118e..ea450b53 100644 --- a/pkg/tests/runner_test.go +++ b/pkg/tests/runner_test.go @@ -64,7 +64,7 @@ func TestDualSubChat(t *testing.T) { r.RespondWith(tester.Result{ Func: types.CompletionFunctionCall{ Name: types.ToolNormalizer("sys.chat.finish"), - Arguments: `{"message":"Chat done"}`, + Arguments: `{"summary":"Chat done"}`, }, }) @@ -89,7 +89,7 @@ func TestDualSubChat(t *testing.T) { r.RespondWith(tester.Result{ Func: types.CompletionFunctionCall{ Name: types.ToolNormalizer("sys.chat.finish"), - Arguments: `{"message":"Chat done2"}`, + Arguments: `{"summary":"Chat done2"}`, }, }, tester.Result{ Text: "And we're done",