diff --git a/src/main/java/com/wok/supportbot/controller/AiController.java b/src/main/java/com/wok/supportbot/controller/AiController.java index cb44a4f..ef6c15b 100644 --- a/src/main/java/com/wok/supportbot/controller/AiController.java +++ b/src/main/java/com/wok/supportbot/controller/AiController.java @@ -1,6 +1,9 @@ package com.wok.supportbot.controller; +import cn.hutool.json.JSONUtil; import com.wok.supportbot.app.AssistantApp; +import com.wok.supportbot.app.ProductInfoApp; +import com.wok.supportbot.entity.ProductInfo; import jakarta.annotation.Resource; import org.springframework.ai.chat.model.ChatModel; import org.springframework.ai.tool.ToolCallback; @@ -20,6 +23,21 @@ public class AiController { @Resource private AssistantApp assistantApp; + @Resource + private ProductInfoApp productInfoApp; + + + /** + * 同步调用 AI 提取结构化商品信息 + * + * @param message + * @return + */ + @GetMapping("/product_info_app/chat/sync") + public String doChatWithProductInfoAppSync(String message) { + ProductInfo productInfo = productInfoApp.extractProductInfo(message); + return JSONUtil.toJsonStr(productInfo); + } /** * 同步调用 AI 智能客服应用 diff --git a/src/main/java/com/wok/supportbot/extract/MarkdownDocumentLoader.java b/src/main/java/com/wok/supportbot/document/extract/MarkdownDocumentLoader.java similarity index 97% rename from src/main/java/com/wok/supportbot/extract/MarkdownDocumentLoader.java rename to src/main/java/com/wok/supportbot/document/extract/MarkdownDocumentLoader.java index 42bf090..494943d 100644 --- a/src/main/java/com/wok/supportbot/extract/MarkdownDocumentLoader.java +++ b/src/main/java/com/wok/supportbot/document/extract/MarkdownDocumentLoader.java @@ -1,4 +1,4 @@ -package com.wok.supportbot.extract; +package com.wok.supportbot.document.extract; import lombok.extern.slf4j.Slf4j; import org.springframework.ai.document.Document; diff --git a/src/main/java/com/wok/supportbot/extract/MyJsonReader.java b/src/main/java/com/wok/supportbot/document/extract/MyJsonReader.java similarity index 96% rename from src/main/java/com/wok/supportbot/extract/MyJsonReader.java rename to src/main/java/com/wok/supportbot/document/extract/MyJsonReader.java index d3a808e..c7be227 100644 --- a/src/main/java/com/wok/supportbot/extract/MyJsonReader.java +++ b/src/main/java/com/wok/supportbot/document/extract/MyJsonReader.java @@ -1,4 +1,4 @@ -package com.wok.supportbot.extract; +package com.wok.supportbot.document.extract; import org.springframework.ai.document.Document; import org.springframework.ai.reader.JsonReader; diff --git a/src/main/java/com/wok/supportbot/transform/MyKeywordEnricher.java b/src/main/java/com/wok/supportbot/document/transform/MyKeywordEnricher.java similarity index 94% rename from src/main/java/com/wok/supportbot/transform/MyKeywordEnricher.java rename to src/main/java/com/wok/supportbot/document/transform/MyKeywordEnricher.java index e513acc..0c14c13 100644 --- a/src/main/java/com/wok/supportbot/transform/MyKeywordEnricher.java +++ b/src/main/java/com/wok/supportbot/document/transform/MyKeywordEnricher.java @@ -1,4 +1,4 @@ -package com.wok.supportbot.transform; +package com.wok.supportbot.document.transform; import jakarta.annotation.Resource; import org.springframework.ai.chat.model.ChatModel; diff --git a/src/main/java/com/wok/supportbot/transform/MyTokenTextSplitter.java b/src/main/java/com/wok/supportbot/document/transform/MyTokenTextSplitter.java similarity index 95% rename from src/main/java/com/wok/supportbot/transform/MyTokenTextSplitter.java rename to src/main/java/com/wok/supportbot/document/transform/MyTokenTextSplitter.java index 38c228a..0584b96 100644 --- a/src/main/java/com/wok/supportbot/transform/MyTokenTextSplitter.java +++ b/src/main/java/com/wok/supportbot/document/transform/MyTokenTextSplitter.java @@ -1,4 +1,4 @@ -package com.wok.supportbot.transform; +package com.wok.supportbot.document.transform; import org.springframework.ai.document.Document; import org.springframework.ai.transformer.splitter.TokenTextSplitter;