Optimize MilvusExample HamaWhiteGG#38

RazorChen · Aug 5, 2023 · d4ec521 · d4ec521
1 parent 609737e
commit d4ec521
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/langchain-examples/src/main/java/com/hw/langchain/examples/chains/MilvusExample.java b/langchain-examples/src/main/java/com/hw/langchain/examples/chains/MilvusExample.java
@@ -18,13 +18,16 @@
 
 package com.hw.langchain.examples.chains;
 
+import com.hw.langchain.chains.retrieval.qa.base.RetrievalQa;
 import com.hw.langchain.document.loaders.text.TextLoader;
 import com.hw.langchain.embeddings.openai.OpenAIEmbeddings;
+import com.hw.langchain.llms.openai.OpenAI;
 import com.hw.langchain.text.splitter.CharacterTextSplitter;
 import com.hw.langchain.vectorstores.milvus.Milvus;
 
 import io.milvus.param.ConnectParam;
 
+import static com.hw.langchain.chains.question.answering.ChainType.STUFF;
 import static com.hw.langchain.examples.utils.PrintUtils.println;
 
 /**
@@ -58,7 +61,13 @@ public static void main(String[] args) {
         var query = "What did the president say about Ketanji Brown Jackson";
         docs = milvus.similaritySearch(query);
 
-        var result = docs.get(0).getPageContent();
+        var pageContent = docs.get(0).getPageContent();
+        println(pageContent);
+
+        var llm = OpenAI.builder().temperature(0).requestTimeout(30).build().init();
+        var qa = RetrievalQa.fromChainType(llm, STUFF, milvus.asRetriever());
+
+        var result = qa.run(query);
         println(result);
     }
 }