fix: improve error handling and session management in GitHubCopilotProvider

Author: Lixeer

pkg/providers/github_copilot_provider.go

@@ -2,60 +2,78 @@ package providers
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
-
-	json "encoding/json"
+	"sync"
 
 	copilot "github.com/github/copilot-sdk/go"
 )
 
 type GitHubCopilotProvider struct {
 	uri         string
-	connectMode string // `stdio` or `grpc``
+	connectMode string // "stdio" or "grpc"
 
+	client  *copilot.Client
 	session *copilot.Session
+
+	
+	mu sync.Mutex
 }
 
 func NewGitHubCopilotProvider(uri string, connectMode string, model string) (*GitHubCopilotProvider, error) {
-
-	var session *copilot.Session
 	if connectMode == "" {
 		connectMode = "grpc"
 	}
-	switch connectMode {
 
+	switch connectMode {
 	case "stdio":
-		//todo
+		// TODO: 
+		return nil, fmt.Errorf("stdio mode not implemented")
 	case "grpc":
 		client := copilot.NewClient(&copilot.ClientOptions{
 			CLIUrl: uri,
 		})
 		if err := client.Start(context.Background()); err != nil {
-			return nil, fmt.Errorf("Can't connect to Github Copilot, https://github.com/github/copilot-sdk/blob/main/docs/getting-started.md#connecting-to-an-external-cli-server for details")
+			return nil, fmt.Errorf("can't connect to Github Copilot: %w; see docs for details", err)
 		}
-		defer client.Stop()
-		session, _ = client.CreateSession(context.Background(), &copilot.SessionConfig{
+
+
+		session, err := client.CreateSession(context.Background(), &copilot.SessionConfig{
 			Model: model,
 			Hooks: &copilot.SessionHooks{},
 		})
+		if err != nil {
+
+			client.Stop()
+			return nil, fmt.Errorf("create session failed: %w", err)
+		}
 
+		return &GitHubCopilotProvider{
+			uri:         uri,
+			connectMode: connectMode,
+			client:      client,
+			session:     session,
+		}, nil
+	default:
+		return nil, fmt.Errorf("unknown connect mode: %s", connectMode)
 	}
+}
 
-	return &GitHubCopilotProvider{
-		uri:         uri,
-		connectMode: connectMode,
-		session:     session,
-	}, nil
+func (p *GitHubCopilotProvider) Close() {
+	if p.client != nil {
+		p.client.Stop()
+		p.client = nil
+		p.session = nil
+	}
 }
 
-// Chat sends a chat request to GitHub Copilot
+
 func (p *GitHubCopilotProvider) Chat(ctx context.Context, messages []Message, tools []ToolDefinition, model string, options map[string]interface{}) (*LLMResponse, error) {
 	type tempMessage struct {
 		Role    string `json:"role"`
 		Content string `json:"content"`
 	}
 	out := make([]tempMessage, 0, len(messages))
-
 	for _, msg := range messages {
 		out = append(out, tempMessage{
 			Role:    msg.Role,
@@ -64,18 +82,28 @@ func (p *GitHubCopilotProvider) Chat(ctx context.Context, messages []Message, to
 	}
 
 	fullcontent, _ := json.Marshal(out)
+	p.mu.Lock()
+	defer p.mu.Unlock()
 
-	content, _ := p.session.Send(ctx, copilot.MessageOptions{
+
+	resp, err := p.session.SendAndWait(ctx, copilot.MessageOptions{
 		Prompt: string(fullcontent),
 	})
+	if err != nil {
+		return nil, err
+	}
+
+
+	var content string
+	if resp != nil && resp.Data.Content != nil {
+		content = *resp.Data.Content
+	}
 
 	return &LLMResponse{
 		FinishReason: "stop",
 		Content:      content,
 	}, nil
-
 }
-
 func (p *GitHubCopilotProvider) GetDefaultModel() string {
 
 	return "gpt-4.1"