Function ReLocateComment

internal/diff/relocation.go:20–75

ReLocateComment calls the LLM to regenerate a precise existing_code snippet when text-based matching fails, then retries ResolveComment with the new snippet. Returns (success, response, requestMessages) so the caller can record session history and track token usage. Response and messages are nil on

(
	ctx context.Context,
	cm *model.LlmComment,
	d *model.Diff,
	client llm.LLMClient,
	task *template.LlmConversation,
	modelName string,
	maxTokens int,
)

// Returns (success, response, requestMessages) so the caller can record session
// history and track token usage. Response and messages are nil on early exits.
func ReLocateComment(
	ctx context.Context,
	cm *model.LlmComment,
	d *model.Diff,
	client llm.LLMClient,
	task *template.LlmConversation,
	modelName string,
	maxTokens int,
) (bool, *llm.ChatResponse, []llm.Message) {
	if task == nil || len(task.Messages) == 0 {
		return false, nil, nil
	}

	messages := make([]llm.Message, 0, len(task.Messages))
	for _, m := range task.Messages {
		content := m.Content
		content = strings.ReplaceAll(content, "{diff}", d.Diff)
		content = strings.ReplaceAll(content, "{existing_code}", cm.ExistingCode)
		content = strings.ReplaceAll(content, "{suggestion_content}", cm.Content)
		messages = append(messages, llm.NewTextMessage(m.Role, content))
	}

	startTime := time.Now()
	_, llmSpan := telemetry.StartLLMSpan(ctx, modelName)
	resp, err := client.CompletionsWithCtx(ctx, llm.ChatRequest{
		Model:     modelName,
		Messages:  messages,
		MaxTokens: maxTokens,
	})
	duration := time.Since(startTime)
	if err != nil {
		telemetry.RecordLLMResult(llmSpan, duration, 0, err)
		llmSpan.End()
		fmt.Fprintf(stdout.Writer(), "[ocr] Re-location LLM call failed for %s: %v\n", cm.Path, err)
		return false, nil, messages
	}
	var totalTokens int64
	if resp.Usage != nil {
		totalTokens = resp.Usage.TotalTokens
	}
	telemetry.RecordLLMResult(llmSpan, duration, totalTokens, nil)
	llmSpan.End()

	code := extractCodeBlock(resp.Content())
	if code == "" {
		return false, resp, messages
	}

	original := cm.ExistingCode
	cm.ExistingCode = code
	if ResolveComment(cm, d) {
		return true, resp, messages
	}
	cm.ExistingCode = original
	return false, resp, messages
}

// extractCodeBlock extracts the content of the first fenced code block from text.