File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -616,7 +616,7 @@ LLMs can be different.
616
616
driver = GraphDatabase.driver(URI , auth = AUTH )
617
617
618
618
# Create LLM object
619
- llm = OpenAILLM(model_name = " gpt-3.5-turbo-instruct " )
619
+ llm = OpenAILLM(model_name = " gpt-3.5-turbo" )
620
620
621
621
# (Optional) Specify your own Neo4j schema
622
622
neo4j_schema = """
Original file line number Diff line number Diff line change 1
- from langchain_openai import OpenAI
2
1
from neo4j import GraphDatabase
2
+ from neo4j_genai .llm import OpenAILLM
3
3
from neo4j_genai .retrievers .text2cypher import Text2CypherRetriever
4
4
5
5
URI = "neo4j://localhost:7687"
9
9
driver = GraphDatabase .driver (URI , auth = AUTH )
10
10
11
11
# Create LLM object
12
- llm = OpenAI ( model = "gpt-3.5-turbo-instruct" )
12
+ llm = OpenAILLM ( model_name = "gpt-3.5-turbo" , model_params = { "temperature" : 0 } )
13
13
14
14
# (Optional) Specify your own Neo4j schema
15
15
neo4j_schema = """
36
36
# Initialize the retriever
37
37
retriever = Text2CypherRetriever (
38
38
driver = driver ,
39
- llm = llm , # type: ignore
39
+ llm = llm ,
40
40
neo4j_schema = neo4j_schema ,
41
41
examples = examples ,
42
42
)
You can’t perform that action at this time.
0 commit comments