Modelin Cevabını Uygulamada Kullanmak — Doğrulamadan Kaydetme

1
enum Category { MARKET, FATURA, KIRA, ULASIM, DIGER }
2

3
// Jackson rejects a value outside the enum ("TRANSPORT") with InvalidFormatException.
4
record CategoryReply(Category category) {
5
    CategoryReply {
6
        if (category == null) throw new IllegalArgumentException("category is missing");
7
    }
8
}

1
interface CategoryModel {
2
    String ask(String prompt);
3
}
4

5
@Component
6
class SpringAiCategoryModel implements CategoryModel {
7

8
    private final ChatClient chat;
9

10
    SpringAiCategoryModel(ChatClient.Builder builder) {
11
        this.chat = builder.defaultSystem("""
12
                Classify the card transaction. Answer with JSON only: {"category": "..."}.
13
                Allowed values: MARKET, FATURA, KIRA, ULASIM, DIGER.
14
                """).build();
15
    }
16

17
    @Override
18
    public String ask(String prompt) {
19
        return chat.prompt().user(prompt).call().content();
20
    }
21
}

1
@Service
2
class TransactionCategorizer {
3

4
    private final CategoryModel model;
5
    private final ObjectMapper json;
6
    private final ReviewQueue review;
7

8
    TransactionCategorizer(CategoryModel model, ObjectMapper json, ReviewQueue review) {
9
        this.model = model;
10
        this.json = json;
11
        this.review = review;
12
    }
13

14
    Optional<Category> categorize(Transaction tx) {
15
        String prompt = "Transaction: " + tx.description();
16
        try {
17
            return Optional.of(read(model.ask(prompt)));
18
        } catch (JsonProcessingException first) {
19
            String again = prompt + "\nYour previous answer was invalid: " + first.getOriginalMessage();
20
            try {
21
                return Optional.of(read(model.ask(again)));   // one retry, never a loop
22
            } catch (JsonProcessingException second) {
23
                review.add(tx.id(), second.getOriginalMessage());   // a person decides; the batch goes on
24
                return Optional.empty();
25
            }
26
        }
27
    }
28

29
    private Category read(String reply) throws JsonProcessingException {
30
        return json.readValue(reply, CategoryReply.class).category();
31
    }
32
}

1
class TransactionCategorizerTest {
2

3
    private final ReviewQueue review = new InMemoryReviewQueue();
4

5
    private TransactionCategorizer withReplies(String... replies) {
6
        Deque<String> script = new ArrayDeque<>(List.of(replies));
7
        return new TransactionCategorizer(prompt -> script.pop(), new ObjectMapper(), review);
8
    }
9

10
    @Test
11
    void anInventedCategoryIsFixedOnTheSecondTry() {
12
        var categorizer = withReplies("{\"category\":\"TRANSPORT\"}", "{\"category\":\"ULASIM\"}");
13
        assertThat(categorizer.categorize(new Transaction(4, "İstanbulkart dolum"))).contains(Category.ULASIM);
14
        assertThat(review.size()).isZero();
15
    }
16

17
    @Test
18
    void twoBadRepliesGoToReviewInsteadOfThrowing() {
19
        var categorizer = withReplies("Tabii! {\"category\":\"ULASIM\"}", "{\"category\":\"EGLENCE\"}");
20
        assertThat(categorizer.categorize(new Transaction(5, "Shell akaryakıt"))).isEmpty();
21
        assertThat(review.size()).isEqualTo(1);
22
    }
23
}