From 164d86e1b75ce9352d336afa5b336906fa3c45fe Mon Sep 17 00:00:00 2001 From: Darren Shepherd Date: Tue, 5 Nov 2024 11:00:36 -0700 Subject: [PATCH] chore: add simple logging for cred stores --- pkg/gptscript/gptscript.go | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) diff --git a/pkg/gptscript/gptscript.go b/pkg/gptscript/gptscript.go index 63a67647..65aa2209 100644 --- a/pkg/gptscript/gptscript.go +++ b/pkg/gptscript/gptscript.go @@ -288,6 +288,7 @@ type simpleRunner struct { func newSimpleRunner(cache *cache.Client, rm engine.RuntimeManager, env []string) (*simpleRunner, error) { runner, err := runner.New(noopModel{}, credentials.NoopStore{}, runner.Options{ RuntimeManager: rm, + MonitorFactory: simpleMonitorFactory{}, }) if err != nil { return nil, err @@ -319,3 +320,33 @@ func (n noopModel) Call(_ context.Context, _ types.CompletionRequest, _ []string func (n noopModel) ProxyInfo() (string, string, error) { return "", "", errors.New("unsupported") } + +type simpleMonitorFactory struct { +} + +func (s simpleMonitorFactory) Start(_ context.Context, _ *types.Program, _ []string, _ string) (runner.Monitor, error) { + return simpleMonitor{}, nil +} + +func (s simpleMonitorFactory) Pause() func() { + //TODO implement me + panic("implement me") +} + +type simpleMonitor struct { +} + +func (s simpleMonitor) Stop(_ context.Context, _ string, _ error) { +} + +func (s simpleMonitor) Event(event runner.Event) { + if event.Type == runner.EventTypeCallProgress { + if !strings.HasPrefix(event.Content, "{") { + fmt.Println(event.Content) + } + } +} + +func (s simpleMonitor) Pause() func() { + return func() {} +}