admin/hermione
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

moved to mongo

Browse Source
This commit is contained in:
andrea.terzani
2024-07-29 12:41:27 +02:00
parent 157dacbe31
commit b1a5602882
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/olympus/hermione/controllers/TestController.java
View File

@@ -15,7 +15,7 @@ public class TestController {
@GetMapping("/test/scenario_execution") @GetMapping("/test/scenario_execution")
public String testScenarioExecution(){ public String testScenarioExecution(){
String result = scenarioExecutionService.executeScenario("66a5ff66fc2ae458a491f161","What's the purpose if an inputset in ATF? How it can be defined?"); String result = scenarioExecutionService.executeScenario("66a5ff66fc2ae458a491f161","How i can modify a query on ATF ?");
return result; return result;
} }
} }

5
src/main/java/com/olympus/hermione/stepSolvers/BasicQueryRagSolver.java
View File

@@ -69,8 +69,7 @@ public class BasicQueryRagSolver extends StepSolver {
@Override @Override
public ScenarioExecution solveStep(){ public ScenarioExecution solveStep(){
System.out.println("Solving step: " + this.step.getName()); logger.info("Solving step: " + this.step.getName());
loadParameters(); loadParameters();
logParameters(); logParameters();
@@ -81,9 +80,11 @@ public class BasicQueryRagSolver extends StepSolver {
if(this.rag_filter != null && !this.rag_filter.isEmpty()){ if(this.rag_filter != null && !this.rag_filter.isEmpty()){
request.withFilterExpression(this.rag_filter); request.withFilterExpression(this.rag_filter);
logger.info("Using Filter expression: " + this.rag_filter);
} }
List<Document> docs = this.vectorStore.similaritySearch(request); List<Document> docs = this.vectorStore.similaritySearch(request);
logger.info("Number of VDB retrieved documents: " + docs.size());
List<String> result = new ArrayList<String>(); List<String> result = new ArrayList<String>();
for (Document doc : docs) { for (Document doc : docs) {

4
src/main/resources/application.properties
View File

@@ -4,10 +4,10 @@ server.port=8081
spring.data.mongodb.uri=mongodb+srv://olympus_adm:26111979@olympus.l6qor4p.mongodb.net/?retryWrites=true&w=majority&appName=Olympus spring.data.mongodb.uri=mongodb+srv://olympus_adm:26111979@olympus.l6qor4p.mongodb.net/?retryWrites=true&w=majority&appName=Olympus
spring.data.mongodb.database=olympus spring.data.mongodb.database=olympus
spring.data.mongodb.username=olympus_adm spring.data.mongodb.username=olympus_adm
spring.data.mongodb.password=XXXXXX spring.data.mongodb.password=26111979
spring.ai.vectorstore.mongodb.indexName=vector_index spring.ai.vectorstore.mongodb.indexName=vector_index
spring.ai.vectorstore.mongodb.collection-name=vector_store spring.ai.vectorstore.mongodb.collection-name=vector_store
spring.ai.vectorstore.mongodb.initialize-schema=false spring.ai.vectorstore.mongodb.initialize-schema=false
spring.ai.openai.api-key=sk-proj-k4jrXXXUYQN8yQG2vNmWT3BlbkFJ0Ge9EfKcrMxduVFQZlyO spring.ai.openai.api-key=sk-proj-j3TFJ0h348DIzMrYYfyUT3BlbkFJjk4HMc8A2ux2Asg8Y7H1