Function ReLocateComment

internal/diff/relocation.go:49–95

ReLocateComment calls the LLM to regenerate a precise existing_code snippet when text-based matching fails, then retries ResolveComment with the new snippet. messages comes from BuildReLocationMessages; the caller has already recorded it in the session, so only (success, response) are returned here.

(
	ctx context.Context,
	cm *model.LlmComment,
	d *model.Diff,
	client llm.LLMClient,
	messages []llm.Message,
	modelName string,
	maxTokens int,
)

// recorded it in the session, so only (success, response) are returned here.
// Response is nil when the request failed.
func ReLocateComment(
	ctx context.Context,
	cm *model.LlmComment,
	d *model.Diff,
	client llm.LLMClient,
	messages []llm.Message,
	modelName string,
	maxTokens int,
) (bool, *llm.ChatResponse) {
	if len(messages) == 0 {
		return false, nil
	}

	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
	}
	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
	}

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

// extractCodeBlock extracts the content of the first fenced code block from text.
// Returns empty string if no code block is found.