1
- import dev .langchain4j .model .chat .ChatLanguageModel ;
1
+ import dev .langchain4j .community .rag .content .retriever .neo4j .Neo4jGraph ;
2
+ import dev .langchain4j .community .rag .content .retriever .neo4j .Neo4jText2CypherRetriever ;
2
3
import dev .langchain4j .model .openai .OpenAiChatModel ;
3
4
import dev .langchain4j .rag .content .Content ;
4
- import dev .langchain4j .rag .content .retriever .neo4j .Neo4jGraph ;
5
- import dev .langchain4j .rag .content .retriever .neo4j .Neo4jText2CypherRetriever ;
6
5
import dev .langchain4j .rag .query .Query ;
7
6
import org .neo4j .driver .AuthTokens ;
8
7
import org .neo4j .driver .Driver ;
@@ -19,10 +18,12 @@ public class Neo4jContentRetrieverExample {
19
18
// You can use "demo" api key for demonstration purposes.
20
19
// You can get your own OpenAI API key here: https://platform.openai.com/account/api-keys
21
20
public static final String OPENAI_API_KEY = getOrDefault (System .getenv ("OPENAI_API_KEY" ), "demo" );
21
+ public static final String OPENAI_BASE_URL = getOrDefault (System .getenv ("OPENAI_BASE_URL" ), "http://langchain4j.dev/demo/openai/v1" );
22
22
23
23
public static void main (String [] args ) {
24
24
final OpenAiChatModel chatLanguageModel = OpenAiChatModel .builder ()
25
25
.apiKey (OPENAI_API_KEY )
26
+ .baseUrl (OPENAI_BASE_URL )
26
27
.modelName (GPT_4_O_MINI )
27
28
.build ();
28
29
@@ -39,7 +40,7 @@ public static void main(String[] args) {
39
40
40
41
Neo4jText2CypherRetriever retriever = Neo4jText2CypherRetriever .builder ()
41
42
.graph (graph )
42
- .chatLanguageModel (chatLanguageModel )
43
+ .chatModel (chatLanguageModel )
43
44
.build ();
44
45
45
46
Query query = new Query ("Who is the author of the book 'Dune'?" );
0 commit comments