Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

yolo

Browse Source
This commit is contained in:
Josh Long
2024-05-17 17:55:33 +02:00
parent 1b1daa7ee7
commit a2147e8572
157 changed files with 1026 additions and 809 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
vector-stores/spring-ai-hanadb-store/src/test/java/org/springframework/ai/vectorstore/CricketWorldCupHanaController.java
View File

@@ -17,7 +17,7 @@ package org.springframework.ai.vectorstore;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.ai.chat.ChatClient;
import org.springframework.ai.chat.connector.ChatConnector;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.chat.prompt.SystemPromptTemplate;
@@ -51,11 +51,11 @@ public class CricketWorldCupHanaController {
private final VectorStore hanaCloudVectorStore;
private final ChatClient chatClient;
private final ChatConnector chatConnector;
@Autowired
public CricketWorldCupHanaController(ChatClient chatClient, VectorStore hanaCloudVectorStore) {
this.chatClient = chatClient;
public CricketWorldCupHanaController(ChatConnector chatConnector, VectorStore hanaCloudVectorStore) {
this.chatConnector = chatConnector;
this.hanaCloudVectorStore = hanaCloudVectorStore;
}
@@ -88,7 +88,7 @@ public class CricketWorldCupHanaController {
var userMessage = new UserMessage(message);
Prompt prompt = new Prompt(List.of(similarDocsMessage, userMessage));
String generation = chatClient.call(prompt).getResult().getOutput().getContent();
String generation = chatConnector.execute(prompt).getResult().getOutput().getContent();
logger.info("Generation: {}", generation);
return Map.of("generation", generation);
}