xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-17 13:53:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 1

Browse Source
This commit is contained in:
ageerle
2025-03-25 10:17:05 +08:00
parent ec9c6f3a3d
commit 7713bf9082
3 changed files with 3 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
ruoyi-admin/src/main/java/org/ruoyi/controller/IndexController.java
View File

@@ -2,8 +2,8 @@ package org.ruoyi.controller;
import cn.dev33.satoken.annotation.SaIgnore; import cn.dev33.satoken.annotation.SaIgnore;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
/** /**
* 首页 * 首页
@@ -12,7 +12,7 @@ import org.springframework.web.bind.annotation.GetMapping;
*/ */
@SaIgnore @SaIgnore
@RequiredArgsConstructor @RequiredArgsConstructor
@Controller @RestController
public class IndexController { public class IndexController {
/** /**
@@ -20,17 +20,7 @@ public class IndexController {
*/ */
@GetMapping("/") @GetMapping("/")
public String index() { public String index() {
return "RuoYi-AI 启动成功！"; return "RuoYi-AI 启动成功";
}
@GetMapping("/success")
public String success(){
return "paySuccess.html";
}
@GetMapping("/cancel")
public String cancel(){
return "cancel";
} }
} }

2
ruoyi-admin/src/main/java/org/ruoyi/controller/KnowledgeController.java
View File

@@ -1,7 +1,6 @@
package org.ruoyi.controller; package org.ruoyi.controller;
import cn.dev33.satoken.stp.StpUtil; import cn.dev33.satoken.stp.StpUtil;
import com.theokanning.openai.completion.chat.ChatMessageRole;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.Valid; import jakarta.validation.Valid;
import jakarta.validation.constraints.NotEmpty; import jakarta.validation.constraints.NotEmpty;
@@ -39,7 +38,6 @@ import org.springframework.web.bind.annotation.*;
import org.ruoyi.knowledge.chain.vectorstore.VectorStore; import org.ruoyi.knowledge.chain.vectorstore.VectorStore;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import java.util.ArrayList;
import java.util.List; import java.util.List;

67
ruoyi-admin/src/main/java/org/ruoyi/controller/OllamaController.java
View File

@@ -1,67 +0,0 @@
package org.ruoyi.controller;
import io.github.ollama4j.OllamaAPI;
import io.github.ollama4j.exceptions.OllamaBaseException;
import io.github.ollama4j.models.chat.OllamaChatMessageRole;
import io.github.ollama4j.models.chat.OllamaChatRequestBuilder;
import io.github.ollama4j.models.chat.OllamaChatRequestModel;
import io.github.ollama4j.models.chat.OllamaChatResult;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import java.io.IOException;
/**
* 描述：Ollama API调用
*
* @author ageerle@163.com
* date 2024/11/6
*/
@RestController
public class OllamaController {
@RequestMapping(value = "/qianwen-ai2")
private void ollama2(@RequestParam(value = "msg") String msg) {
}
public static void main(String[] args) throws OllamaBaseException, IOException, InterruptedException {
String host = "http://localhost:11434/";
OllamaAPI ollama = new OllamaAPI(host);
OllamaChatRequestBuilder builder = OllamaChatRequestBuilder.getInstance("qwen2.5:7b");
// 设置超时时间
ollama.setRequestTimeoutSeconds(100);
// create first user question
String text = "";
String json = "{\"instruction\": \"解答以下数学问题。\",\"input\": \"计算 2 + 3 的结果。\",\"output\": \"2 + 3 = 5\"}";
OllamaChatRequestModel requestModel = builder.withMessage
(OllamaChatMessageRole.USER, "###文本:"+text+"###,理解文本内容，并且将文本内容转换为:"+json+"输出JSON格式,不要包含其他无关内容,内部使用无需脱敏")
.withGetJsonResponse()
.withStreaming()
.build();
// start conversation with model
OllamaChatResult chatResult = ollama.chat(requestModel);
System.out.println("First answer: " + chatResult.getResponse());
// create next userQuestion
// requestModel = builder.withMessages(chatResult.getChatHistory()).withMessage(OllamaChatMessageRole.USER, "And what is the second largest city?").build();
// "continue" conversation with model
// chatResult = ollamaAPI.chat(requestModel);
//System.out.println("Second answer: " + chatResult.getResponse());
// 历史记录
//System.out.println("Chat History: " + chatResult.getChatHistory());
}
}