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.
616616 driver = GraphDatabase.driver(URI , auth = AUTH )
617617
618618 # Create LLM object
619- llm = OpenAILLM(model_name = " gpt-3.5-turbo-instruct " )
619+ llm = OpenAILLM(model_name = " gpt-3.5-turbo" )
620620
621621 # (Optional) Specify your own Neo4j schema
622622 neo4j_schema = """
Original file line number Diff line number Diff line change 1- from langchain_openai import OpenAI
21from neo4j import GraphDatabase
2+ from neo4j_genai .llm import OpenAILLM
33from neo4j_genai .retrievers .text2cypher import Text2CypherRetriever
44
55URI = "neo4j://localhost:7687"
99driver = GraphDatabase .driver (URI , auth = AUTH )
1010
1111# 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 } )
1313
1414# (Optional) Specify your own Neo4j schema
1515neo4j_schema = """
3636# Initialize the retriever
3737retriever = Text2CypherRetriever (
3838 driver = driver ,
39- llm = llm , # type: ignore
39+ llm = llm ,
4040 neo4j_schema = neo4j_schema ,
4141 examples = examples ,
4242)
You can’t perform that action at this time.
0 commit comments