Merge pull request #319 from rumpl/test-runtime

Add some tests for the runtime

Author: dgageot

pkg/runtime/runtime.go

@@ -26,6 +26,10 @@ import (
 
 type ResumeType string
 
+type modelStore interface {
+	GetModel(ctx context.Context, modelID string) (*modelsdev.Model, error)
+}
+
 const (
 	ResumeTypeApprove        ResumeType = "approve"
 	ResumeTypeApproveSession ResumeType = "approve-session"
@@ -61,7 +65,7 @@ type runtime struct {
 	resumeChan        chan ResumeType
 	oauthManager      oauth.Manager
 	tracer            trace.Tracer
-	modelsStore       *modelsdev.Store
+	modelsStore       modelStore
 	sessionCompaction bool
 }
 
@@ -86,6 +90,12 @@ func WithSessionCompaction(sessionCompaction bool) Opt {
 	}
 }
 
+func WithModelStore(store modelStore) Opt {
+	return func(r *runtime) {
+		r.modelsStore = store
+	}
+}
+
 // New creates a new runtime for an agent and its team
 func New(agents *team.Team, opts ...Opt) (Runtime, error) {
 	modelsStore, err := modelsdev.NewStore()