Merge pull request #79 from thedadams/add-helpers

feat: add a few helper functions for using code outside of this package

Author: thedadams

pkg/mvl/log.go

@@ -46,7 +46,7 @@ func SetError() {
 
 func Package() Logger {
 	_, p, _, _ := runtime.Caller(1)
-	_, suffix, _ := strings.Cut(p, "gptscript/")
+	_, suffix, _ := strings.Cut(p, "gptscript")
 	i := strings.LastIndex(suffix, "/")
 	if i > 0 {
 		return New(suffix[:i])

pkg/server/server.go

@@ -45,9 +45,7 @@ func New(model engine.Model, opts ...Options) (*Server, error) {
 
 	opt := complete(opts)
 	r, err := runner.New(model, runner.Options{
-		MonitorFactory: &SessionFactory{
-			events: events,
-		},
+		MonitorFactory: NewSessionFactory(events),
 	})
 	if err != nil {
 		return nil, err
@@ -84,6 +82,14 @@ var (
 
 type execKey struct{}
 
+func ContextWithNewID(ctx context.Context) context.Context {
+	return context.WithValue(ctx, execKey{}, fmt.Sprint(atomic.AddInt64(&execID, 1)))
+}
+
+func IDFromContext(ctx context.Context) string {
+	return ctx.Value(execKey{}).(string)
+}
+
 func (s *Server) list(rw http.ResponseWriter, req *http.Request) {
 	rw.Header().Set("Content-Type", "application/json")
 	enc := json.NewEncoder(rw)
@@ -276,6 +282,12 @@ type SessionFactory struct {
 	events *broadcaster.Broadcaster[Event]
 }
 
+func NewSessionFactory(events *broadcaster.Broadcaster[Event]) *SessionFactory {
+	return &SessionFactory{
+		events: events,
+	}
+}
+
 func (s SessionFactory) Start(ctx context.Context, prg *types.Program, env []string, input string) (runner.Monitor, error) {
 	id, _ := ctx.Value(execKey{}).(string)