first implementation

2026-03-31 20:58:47 +02:00
parent dc0bcab36e
commit 618e28b354
1878 changed files with 1381732 additions and 5 deletions
@@ -0,0 +1,108 @@
+package com.aiteacher.topic;
+
+import com.aiteacher.book.BookRepository;
+import com.aiteacher.book.BookStatus;
+import com.aiteacher.book.NoKnowledgeSourceException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.vectorstore.QuestionAnswerAdvisor;
+import org.springframework.ai.chat.model.ChatResponse;
+import org.springframework.ai.document.Document;
+import org.springframework.ai.vectorstore.VectorStore;
+import org.springframework.stereotype.Service;
+
+import java.time.Instant;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+@Service
+public class TopicSummaryService {
+
+    private static final Logger log = LoggerFactory.getLogger(TopicSummaryService.class);
+
+    private static final String SYSTEM_PROMPT = """
+        You are an expert neurosurgery educator. Your role is to provide accurate,
+        clinically relevant summaries based ONLY on the content retrieved from the
+        uploaded medical textbooks. Do not use any knowledge outside the provided context.
+
+        When answering:
+        - Structure your response clearly with key points
+        - If the context mentions specific book titles and page numbers, reference them
+        - If the retrieved context does not contain sufficient information on the topic,
+          explicitly state: "The uploaded books do not contain sufficient information on this topic."
+        - Never hallucinate or fabricate clinical information
+        """;
+
+    private final ChatClient chatClient;
+    private final VectorStore vectorStore;
+    private final BookRepository bookRepository;
+
+    public TopicSummaryService(ChatClient chatClient, VectorStore vectorStore,
+                                BookRepository bookRepository) {
+        this.chatClient = chatClient;
+        this.vectorStore = vectorStore;
+        this.bookRepository = bookRepository;
+    }
+
+    public TopicSummaryResponse generateSummary(Topic topic) {
+        if (!bookRepository.existsByStatus(BookStatus.READY)) {
+            throw new NoKnowledgeSourceException(
+                "No books are available as knowledge sources. Please upload and process at least one book.");
+        }
+
+        String question = buildQuestion(topic);
+
+        ChatResponse response = chatClient.prompt()
+            .system(SYSTEM_PROMPT)
+            .advisors(QuestionAnswerAdvisor.builder(vectorStore).build())
+            .user(question)
+            .call()
+            .chatResponse();
+
+        String summary = response.getResult().getOutput().getText();
+        List<TopicSummaryResponse.SourceReference> sources = extractSources(response);
+
+        return new TopicSummaryResponse(
+            topic.id(),
+            topic.name(),
+            summary,
+            sources,
+            Instant.now()
+        );
+    }
+
+    private String buildQuestion(Topic topic) {
+        return String.format(
+            "Please provide a comprehensive educational summary of the following neurosurgery topic: " +
+            "%s. Topic description: %s. " +
+            "Include key concepts, clinical considerations, and important details that a neurosurgeon should know.",
+            topic.name(), topic.description()
+        );
+    }
+
+    private List<TopicSummaryResponse.SourceReference> extractSources(ChatResponse response) {
+        List<TopicSummaryResponse.SourceReference> sources = new ArrayList<>();
+
+        if (response.getMetadata() != null) {
+            Object retrieved = response.getMetadata().get(QuestionAnswerAdvisor.RETRIEVED_DOCUMENTS);
+            if (retrieved instanceof List<?> docs) {
+                for (Object docObj : docs) {
+                    if (docObj instanceof Document doc) {
+                        Map<String, Object> metadata = doc.getMetadata();
+                        String bookTitle = (String) metadata.get("book_title");
+                        Object pageObj = metadata.get("page_number");
+                        Integer page = pageObj instanceof Number n ? n.intValue() : null;
+                        if (bookTitle != null) {
+                            sources.add(new TopicSummaryResponse.SourceReference(bookTitle, page));
+                        }
+                    }
+                }
+            }
+        }
+
+        // Deduplicate by bookTitle + page
+        return sources.stream().distinct().toList();
+    }
+}