fix some bug

examples/interrupt_agent/main.go

@@ -22,16 +22,30 @@ import (
 	"github.com/vearne/agentscope-go/pkg/memory"
 	"github.com/vearne/agentscope-go/pkg/message"
 	"github.com/vearne/agentscope-go/pkg/model"
+	"github.com/vearne/agentscope-go/pkg/studio"
 	"github.com/vearne/agentscope-go/pkg/tool"
 )
 
 func main() {
+	ctx := context.Background()
+
+	if err := studio.Init(
+		studio.WithURL("http://localhost:3000"),
+		studio.WithProject("interrupt-studio"),
+	); err != nil {
+		log.Printf("Warning: studio init failed (studio may not be running): %v", err)
+		log.Println("Continuing without studio integration...")
+	}
+	defer func() { _ = studio.Shutdown(ctx) }()
+
 	apiKey := os.Getenv("OPENAI_API_KEY")
 	if apiKey == "" {
 		log.Fatal("Please set OPENAI_API_KEY environment variable")
 	}
 
-	m := model.NewOpenAIChatModel("gpt-4o", apiKey, "", false)
+	baseURL := os.Getenv("OPENAI_BASE_URL")
+	modelName := os.Getenv("OPENAI_MODEL")
+	m := model.NewOpenAIChatModel(modelName, apiKey, baseURL, false)
 	f := formatter.NewOpenAIChatFormatter()
 
 	tk := tool.NewToolkit()

pkg/agent/deep.go

@@ -211,6 +211,11 @@ func (a *DeepAgent) Interrupt() {
 }
 
 func (a *DeepAgent) HandleInterrupt(ctx context.Context, msg *message.Msg) (*message.Msg, error) {
+	// Use a context that is not cancelled so that Studio forwarding and
+	// post-reply hooks can complete. The caller's ctx is already cancelled
+	// because Interrupt() invoked cancelFunc().
+	studioCtx := context.WithoutCancel(ctx)
+
 	resp := &message.Msg{
 		ID:        utils.ShortUUID(),
 		Name:      a.name,
@@ -219,16 +224,16 @@ func (a *DeepAgent) HandleInterrupt(ctx context.Context, msg *message.Msg) (*mes
 		Metadata:  map[string]interface{}{"_is_interrupted": true},
 		Timestamp: time.Now().Format("2006-01-02 15:04:05.000"),
 	}
-	if err := a.mem.Add(ctx, resp); err != nil {
+	if err := a.mem.Add(studioCtx, resp); err != nil {
 		return nil, fmt.Errorf("add interrupted message to memory: %w", err)
 	}
 
 	for _, h := range a.hooks.postReply {
-		h(ctx, a, msg, resp)
+		h(studioCtx, a, msg, resp)
 	}
 
 	if sc := studio.GetClient(); sc != nil {
-		studio.ForwardMessage(ctx, a.name, "assistant", resp)
+		studio.ForwardMessage(studioCtx, a.name, "assistant", resp)
 	}
 	return resp, nil
 }

pkg/agent/react.go

@@ -494,6 +494,11 @@ func (a *ReActAgent) Interrupt() {
 }
 
 func (a *ReActAgent) HandleInterrupt(ctx context.Context, msg *message.Msg) (*message.Msg, error) {
+	// Use a context that is not cancelled so that Studio forwarding and
+	// post-reply hooks can complete. The caller's ctx is already cancelled
+	// because Interrupt() invoked cancelFunc().
+	studioCtx := context.WithoutCancel(ctx)
+
 	resp := &message.Msg{
 		ID:        utils.ShortUUID(),
 		Name:      a.name,
@@ -502,16 +507,16 @@ func (a *ReActAgent) HandleInterrupt(ctx context.Context, msg *message.Msg) (*me
 		Metadata:  map[string]interface{}{"_is_interrupted": true},
 		Timestamp: time.Now().Format("2006-01-02 15:04:05.000"),
 	}
-	if err := a.mem.Add(ctx, resp); err != nil {
+	if err := a.mem.Add(studioCtx, resp); err != nil {
 		return nil, fmt.Errorf("add interrupted message to memory: %w", err)
 	}
 
 	for _, h := range a.hooks.postReply {
-		h(ctx, a, msg, resp)
+		h(studioCtx, a, msg, resp)
 	}
 
 	if sc := studio.GetClient(); sc != nil {
-		studio.ForwardMessage(ctx, a.name, "assistant", resp)
+		studio.ForwardMessage(studioCtx, a.name, "assistant", resp)
 	}
 	return resp, nil
 }

pkg/model/anthropic.go

@@ -38,6 +38,7 @@ func NewAnthropicChatModel(modelName, apiKey, baseURL string, stream bool, opts
 		httpClient: &http.Client{
 			Timeout: 300 * time.Second,
 		},
+		config: &ModelConfig{},
 	}
 
 	for _, opt := range opts {

pkg/model/gemini.go

@@ -38,6 +38,7 @@ func NewGeminiChatModel(modelName, apiKey, baseURL string, stream bool, opts ...
 		httpClient: &http.Client{
 			Timeout: 300 * time.Second,
 		},
+		config: &ModelConfig{},
 	}
 
 	for _, opt := range opts {

pkg/model/openai.go

@@ -38,6 +38,7 @@ func NewOpenAIChatModel(modelName, apiKey, baseURL string, stream bool, opts ...
 		httpClient: &http.Client{
 			Timeout: 300 * time.Second,
 		},
+		config: &ModelConfig{},
 	}
 
 	for _, opt := range opts {