Package com.bytedesk.ai.springai.rag

Class SpringAIRagController

java.lang.Object

com.bytedesk.ai.springai.rag.SpringAIRagController

@RestController
@RequestMapping("/spring/ai/rag")
public class SpringAIRagController
extends Object

RAG知识库问答 https://docs.spring.io/spring-ai/reference/api/retrieval-augmented-generation.html https://mp.weixin.qq.com/s/ETmbEAE7lNligcM_A_GF8A https://www.xiaohongshu.com/explore/67b422600000000029008c9f?note_flow_source=wechat&xsec_token=CBKBzlIFWDRmY2lKlLXYLivyqzKeFCE-9sd-_DNTLH-rM=

Field Summary

Fields

Modifier and Type	Field	Description
private final io.micrometer.observation.ObservationRegistry	observationRegistry
private final Optional<org.springframework.ai.ollama.OllamaChatModel>	ollamaChatModel
private final org.springframework.ai.vectorstore.VectorStore	ollamaRedisVectorStore
private final SpringAIVectorStoreService	springAIVectorService

Constructor Summary

Constructors

Constructor	Description
SpringAIRagController()

Method Summary

Modifier and Type	Method	Description
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	advancedRag(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	compressionQueryTransformer(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	contextualQueryAugmenter(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	filter(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	multiQueryExpander(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	naiveRag(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	observedChat(String message)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	rag(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	rewriteQueryTransformer(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	search(String query, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	translationQueryTransformer(String message, String kbUid)
(package private) org.springframework.http.ResponseEntity<JsonResult<?>>	vectorStoreDocumentRetriever(String message, String kbUid)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

ollamaRedisVectorStore

@Qualifier("ollamaRedisVectorStore")
private final org.springframework.ai.vectorstore.VectorStore ollamaRedisVectorStore

ollamaChatModel

@Qualifier("ollamaChatModel")
private final Optional<org.springframework.ai.ollama.OllamaChatModel> ollamaChatModel

springAIVectorService

private final SpringAIVectorStoreService springAIVectorService

observationRegistry

private final io.micrometer.observation.ObservationRegistry observationRegistry

Constructor Details

SpringAIRagController

public SpringAIRagController()

Method Details

rag

@GetMapping("/rag")
org.springframework.http.ResponseEntity<JsonResult<?>> rag(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

filter

@GetMapping("/filter")
org.springframework.http.ResponseEntity<JsonResult<?>> filter(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

naiveRag

@GetMapping("/naive-rag")
org.springframework.http.ResponseEntity<JsonResult<?>> naiveRag(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

advancedRag

@GetMapping("/advanced-rag")
org.springframework.http.ResponseEntity<JsonResult<?>> advancedRag(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

compressionQueryTransformer

@GetMapping("/compression-query-transformer")
org.springframework.http.ResponseEntity<JsonResult<?>> compressionQueryTransformer(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

rewriteQueryTransformer

@GetMapping("/rewrite-query-transformer")
org.springframework.http.ResponseEntity<JsonResult<?>> rewriteQueryTransformer(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

translationQueryTransformer

@GetMapping("/translation-query-transformer")
org.springframework.http.ResponseEntity<JsonResult<?>> translationQueryTransformer(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

multiQueryExpander

@GetMapping("/multi-query-expander")
org.springframework.http.ResponseEntity<JsonResult<?>> multiQueryExpander(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

vectorStoreDocumentRetriever

@GetMapping("/vector-store-document-retriever")
org.springframework.http.ResponseEntity<JsonResult<?>> vectorStoreDocumentRetriever(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

contextualQueryAugmenter

@GetMapping("/contextual-query-augmenter")
org.springframework.http.ResponseEntity<JsonResult<?>> contextualQueryAugmenter(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message,
 @RequestParam(value="kbUid",defaultValue="")
 String kbUid)

search

@GetMapping("/search")
org.springframework.http.ResponseEntity<JsonResult<?>> search(@RequestParam(value="query",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String query,
 @RequestParam(value="kbUid",required=true)
 String kbUid)

observedChat

@GetMapping("/observed")
org.springframework.http.ResponseEntity<JsonResult<?>> observedChat(@RequestParam(value="message",defaultValue="\u4ec0\u4e48\u65f6\u95f4\u8003\u8bd5\uff1f")
 String message)