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

Signed-off-by: Donnie Adams <[email protected]>

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, "/")
 	return New(suffix[:i])
 }

pkg/server/server.go

@@ -49,9 +49,7 @@ func New(opts ...Options) (*Server, error) {
 
 	runnerOpts, opt := complete(opts)
 	r, err := runner.New(append(runnerOpts, runner.Options{
-		MonitorFactory: &SessionFactory{
-			events: events,
-		},
+		MonitorFactory: NewSessionFactory(events),
 	})...)
 	if err != nil {
 		return nil, err
@@ -61,9 +59,7 @@ func New(opts ...Options) (*Server, error) {
 		CacheOptions: runner.CacheOptions{
 			Cache: new(bool),
 		},
-		MonitorFactory: &SessionFactory{
-			events: events,
-		},
+		MonitorFactory: NewSessionFactory(events),
 	})...)
 	if err != nil {
 		return nil, err
@@ -102,6 +98,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)
@@ -283,6 +287,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)