feat: tab completion

internal/api/chat/get_citation_keys.go

@@ -0,0 +1,42 @@
+package chat
+
+import (
+	"context"
+
+	"paperdebugger/internal/libs/contextutil"
+	"paperdebugger/internal/models"
+	chatv2 "paperdebugger/pkg/gen/api/chat/v2"
+)
+
+func (s *ChatServerV2) GetCitationKeys(
+	ctx context.Context,
+	req *chatv2.GetCitationKeysRequest,
+) (*chatv2.GetCitationKeysResponse, error) {
+	actor, err := contextutil.GetActor(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	settings, err := s.userService.GetUserSettings(ctx, actor.ID)
+	if err != nil {
+		return nil, err
+	}
+
+	llmProvider := &models.LLMProviderConfig{
+		APIKey: settings.OpenAIAPIKey,
+	}
+
+	citationKeys, err := s.aiClientV2.GetCitationKeys(
+		ctx,
+		req.GetSentence(),
+		req.GetBibliography(),
+		llmProvider,
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	return &chatv2.GetCitationKeysResponse{
+		CitationKeys: citationKeys,
+	}, nil
+}

internal/services/toolkit/client/get_citation_keys.go

@@ -0,0 +1,37 @@
+package client
+
+// TODO: This file should not place in the client package.
+import (
+	"context"
+	"fmt"
+	"paperdebugger/internal/models"
+	"strings"
+
+	"github.com/openai/openai-go/v3"
+)
+
+func (a *AIClientV2) GetCitationKeys(ctx context.Context, sentence string, bibliography string, llmProvider *models.LLMProviderConfig) (string, error) {
+	emptyCitation := "none"
+	message := fmt.Sprintf("Sentence: %s\nBibliography: %s\nBased on the sentence and bibliography, suggest relevant citation keys separated by commas. If no relevant citations are found, return '%s'.", sentence, bibliography, emptyCitation)
+
+	_, resp, err := a.ChatCompletionV2(ctx, "gpt-5-nano", OpenAIChatHistory{
+		openai.SystemMessage("You are a helpful assistant that suggests relevant citation keys."),
+		openai.UserMessage(message),
+	}, llmProvider)
+
+	if err != nil {
+		return "", err
+	}
+
+	if len(resp) == 0 {
+		return "", nil
+	}
+
+	citationKeys := strings.TrimSpace(resp[0].Payload.GetAssistant().GetContent())
+
+	if citationKeys == emptyCitation {
+		return "", nil
+	}
+
+	return citationKeys, nil
+}