Update ChatCompletionClient to accept context.Context

pkg/cmd/util/util.go

@@ -1,6 +1,7 @@
 package util
 
 import (
+	"context"
 	"fmt"
 	"os"
 	"strings"
@@ -35,7 +36,7 @@ func FetchAndShowCompletion(ctx *lmcli.Context, messages []model.Message) ([]mod
 
 	var apiReplies []model.Message
 	response, err := completionProvider.CreateChatCompletionStream(
-		requestParams, messages, &apiReplies, content,
+		context.Background(), requestParams, messages, &apiReplies, content,
 	)
 	if response != "" {
 		// there was some content, so break to a new line after it
@@ -153,7 +154,7 @@ func GenerateTitle(ctx *lmcli.Context, c *model.Conversation) (string, error) {
 		MaxTokens: 25,
 	}
 
-	response, err := completionProvider.CreateChatCompletion(requestParams, generateRequest, nil)
+	response, err := completionProvider.CreateChatCompletion(context.Background(), requestParams, generateRequest, nil)
 	if err != nil {
 		return "", err
 	}