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

modelcall and chatclient rework

Browse Source
This commit is contained in:
Josh Long
2024-05-18 21:30:51 +02:00
committed by Christian Tzolov
parent 2abf10dbf9
commit c4bfb5709b
158 changed files with 1144 additions and 662 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.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 ChatClient chatClient;
private final ModelCall modelCall;
@Autowired
public CricketWorldCupHanaController(ChatClient chatClient, VectorStore hanaCloudVectorStore) {
this.chatClient = chatClient;
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 = chatClient.call(prompt).getResult().getOutput().getContent();
String generation = modelCall.call(prompt).getResult().getOutput().getContent();
logger.info("Generation: {}", generation);
return Map.of("generation", generation);
}