refactor: refactor completion methods to use context and consistent naming

- Added `context.Context` as a parameter to `CreateCompletion` calls across multiple files.
- Renamed `EngineExecOutput` to `CompletionOutput` and updated related methods and fields.
- Updated field names in `StreamCompletionOutput` to use PascalCase for consistency.
- Removed unused `defaultChatID` constant and `SummaryContentOutput` type.
- Added a new `setupChatContext` method call in the `CreateCompletion` function.

Signed-off-by: codiing-hui <wecoding@yeah.net>

Author: coding-hui

internal/cli/commit/commit.go

@@ -1,6 +1,7 @@
 package commit
 
 import (
+	"context"
 	"fmt"
 	"html"
 	"os"
@@ -246,7 +247,7 @@ func (o *Options) codeReview(engine *llm.Engine, vars map[string]any) error {
 		return err
 	}
 
-	resp, err := engine.CreateCompletion(p.Messages())
+	resp, err := engine.CreateCompletion(context.Background(), p.Messages())
 	if err != nil {
 		return err
 	}
@@ -265,7 +266,7 @@ func (o *Options) summarizeTitle(engine *llm.Engine, vars map[string]any) error
 		return err
 	}
 
-	resp, err := engine.CreateCompletion(p.Messages())
+	resp, err := engine.CreateCompletion(context.Background(), p.Messages())
 	if err != nil {
 		return err
 	}
@@ -285,7 +286,7 @@ func (o *Options) summarizePrefix(engine *llm.Engine, vars map[string]any) error
 		return err
 	}
 
-	resp, err := engine.CreateCompletion(p.Messages())
+	resp, err := engine.CreateCompletion(context.Background(), p.Messages())
 	if err != nil {
 		return err
 	}
@@ -330,7 +331,7 @@ func (o *Options) generateCommitMsg(engine *llm.Engine, vars map[string]any) (st
 			return "", err
 		}
 
-		resp, err := engine.CreateCompletion(translationPrompt.Messages())
+		resp, err := engine.CreateCompletion(context.Background(), translationPrompt.Messages())
 		if err != nil {
 			return "", err
 		}

internal/cli/review/review.go

@@ -1,6 +1,7 @@
 package review
 
 import (
+	"context"
 	"errors"
 	"strings"
 
@@ -74,7 +75,7 @@ func (o *Options) reviewCode(cmd *cobra.Command, args []string) error {
 
 	// Get summarize comment from diff datas
 	color.Cyan("We are trying to review code changes")
-	reviewResp, err := llmEngine.CreateCompletion(reviewPrompt.Messages())
+	reviewResp, err := llmEngine.CreateCompletion(context.Background(), reviewPrompt.Messages())
 	if err != nil {
 		return err
 	}
@@ -92,7 +93,7 @@ func (o *Options) reviewCode(cmd *cobra.Command, args []string) error {
 		}
 
 		color.Cyan("we are trying to translate code review to " + o.commitLang + " language")
-		translationResp, err := llmEngine.CreateCompletion(translationPrompt.Messages())
+		translationResp, err := llmEngine.CreateCompletion(context.Background(), translationPrompt.Messages())
 		if err != nil {
 			return err
 		}

internal/llm/engine.go

@@ -20,8 +20,7 @@ import (
 )
 
 const (
-	noExec        = "[noexec]"
-	defaultChatID = "temp_session"
+	noExec = "[noexec]"
 )
 
 type Engine struct {
@@ -137,32 +136,35 @@ func (e *Engine) GetConvoStore() convo.Store {
 
 func (e *Engine) Interrupt() {
 	e.channel <- StreamCompletionOutput{
-		content:    "[Interrupt]",
-		last:       true,
-		interrupt:  true,
-		executable: false,
+		Content:    "[Interrupt]",
+		Last:       true,
+		Interrupt:  true,
+		Executable: false,
 	}
 
 	e.running = false
 }
 
-func (e *Engine) CreateCompletion(messages []llms.ChatMessage) (*EngineExecOutput, error) {
-	ctx := context.Background()
-
+func (e *Engine) CreateCompletion(ctx context.Context, messages []llms.ChatMessage) (*CompletionOutput, error) {
 	e.running = true
 
+	if err := e.setupChatContext(ctx, &messages); err != nil {
+		return nil, err
+	}
+
 	rsp, err := e.Model.GenerateContent(ctx, slices.Map(messages, convert), e.callOptions()...)
 	if err != nil {
 		return nil, errbook.Wrap("Failed to create completion.", err)
 	}
 
 	content := rsp.Choices[0].Content
+
 	e.appendAssistantMessage(content)
 
-	var output EngineExecOutput
+	var output CompletionOutput
 	err = json.Unmarshal([]byte(content), &output)
 	if err != nil {
-		output = EngineExecOutput{
+		output = CompletionOutput{
 			Command:     "",
 			Explanation: content,
 			Executable:  false,
@@ -178,8 +180,8 @@ func (e *Engine) CreateStreamCompletion(ctx context.Context, messages []llms.Cha
 	streamingFunc := func(ctx context.Context, chunk []byte) error {
 		if !e.config.Quiet {
 			e.channel <- StreamCompletionOutput{
-				content: string(chunk),
-				last:    false,
+				Content: string(chunk),
+				Last:    false,
 			}
 		}
 		return nil
@@ -214,18 +216,19 @@ func (e *Engine) CreateStreamCompletion(ctx context.Context, messages []llms.Cha
 
 	if !e.config.Quiet {
 		e.channel <- StreamCompletionOutput{
-			content:    "",
-			last:       true,
-			executable: executable,
+			Content:    "",
+			Last:       true,
+			Executable: executable,
 		}
 	}
 	e.running = false
+
 	e.appendAssistantMessage(output)
 
 	return &StreamCompletionOutput{
-		content:    output,
-		last:       true,
-		executable: executable,
+		Content:    output,
+		Last:       true,
+		Executable: executable,
 	}, nil
 }
 

internal/llm/types.go

@@ -17,26 +17,22 @@ func (m EngineMode) String() string {
 	}
 }
 
-type SummaryContentOutput struct {
-	Content string `json:"content"`
-}
-
-type EngineExecOutput struct {
+type CompletionOutput struct {
 	Command     string `json:"cmd"`
 	Explanation string `json:"exp"`
 	Executable  bool   `json:"exec"`
 }
 
-func (eo EngineExecOutput) GetCommand() string {
-	return eo.Command
+func (c CompletionOutput) GetCommand() string {
+	return c.Command
 }
 
-func (eo EngineExecOutput) GetExplanation() string {
-	return eo.Explanation
+func (c CompletionOutput) GetExplanation() string {
+	return c.Explanation
 }
 
-func (eo EngineExecOutput) IsExecutable() bool {
-	return eo.Executable
+func (c CompletionOutput) IsExecutable() bool {
+	return c.Executable
 }
 
 // CompletionInput is a tea.Msg that wraps the content read from stdin.
@@ -46,32 +42,24 @@ type CompletionInput struct {
 
 // StreamCompletionOutput a tea.Msg that wraps the content returned from llm.
 type StreamCompletionOutput struct {
-	content    string
-	last       bool
-	interrupt  bool
-	executable bool
-}
-
-func (co StreamCompletionOutput) GetContent() string {
-	return co.content
-}
-
-func (co StreamCompletionOutput) IsLast() bool {
-	return co.last
+	Content    string
+	Last       bool
+	Interrupt  bool
+	Executable bool
 }
 
-func (co StreamCompletionOutput) IsInterrupt() bool {
-	return co.interrupt
+func (c StreamCompletionOutput) GetContent() string {
+	return c.Content
 }
 
-func (co StreamCompletionOutput) IsExecutable() bool {
-	return co.executable
+func (c StreamCompletionOutput) IsLast() bool {
+	return c.Last
 }
 
-type EngineLoggingCallbackOutput struct {
-	content string
+func (c StreamCompletionOutput) IsInterrupt() bool {
+	return c.Interrupt
 }
 
-func (ec EngineLoggingCallbackOutput) GetContent() string {
-	return ec.content
+func (c StreamCompletionOutput) IsExecutable() bool {
+	return c.Executable
 }