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

polish the building of messages

Browse Source
This commit is contained in:
Josh Long
2024-05-18 18:08:36 +02:00
parent 2c5ae0837f
commit 703e3d3a8f
157 changed files with 708 additions and 682 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.connector.ChatConnector;
import org.springframework.ai.chat.connector.ModelCall;
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 ChatConnector chatConnector;
private final ModelCall modelCall;
@Autowired
public CricketWorldCupHanaController(ChatConnector chatConnector, VectorStore hanaCloudVectorStore) {
this.chatConnector = chatConnector;
public CricketWorldCupHanaController(ModelCall modelCall, VectorStore hanaCloudVectorStore) {
this.modelCall = modelCall;
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 = chatConnector.call(prompt).getResult().getOutput().getContent();
String generation = modelCall.call(prompt).getResult().getOutput().getContent();
logger.info("Generation: {}", generation);
return Map.of("generation", generation);
}