Changed model in t2c examples to gpt-3.5-turbo-0125 (#125)

* Changed model in t2c examples to gpt-3.5-turbo-0125 * Use more generic gpt-3.5-turbo in example and doc for Text2CypherRetriever
neo4j · Sep 10, 2024 · 19fbace · 19fbace
1 parent c284b08
commit 19fbace
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docs/source/user_guide_rag.rst b/docs/source/user_guide_rag.rst
@@ -616,7 +616,7 @@ LLMs can be different.
     driver = GraphDatabase.driver(URI, auth=AUTH)
 
     # Create LLM object
-    llm = OpenAILLM(model_name="gpt-3.5-turbo-instruct")
+    llm = OpenAILLM(model_name="gpt-3.5-turbo")
 
     # (Optional) Specify your own Neo4j schema
     neo4j_schema = """

diff --git a/examples/text2cypher_search.py b/examples/text2cypher_search.py
@@ -1,5 +1,5 @@
-from langchain_openai import OpenAI
 from neo4j import GraphDatabase
+from neo4j_genai.llm import OpenAILLM
 from neo4j_genai.retrievers.text2cypher import Text2CypherRetriever
 
 URI = "neo4j://localhost:7687"
@@ -9,7 +9,7 @@
 driver = GraphDatabase.driver(URI, auth=AUTH)
 
 # Create LLM object
-llm = OpenAI(model="gpt-3.5-turbo-instruct")
+llm = OpenAILLM(model_name="gpt-3.5-turbo", model_params={"temperature": 0})
 
 # (Optional) Specify your own Neo4j schema
 neo4j_schema = """
@@ -36,7 +36,7 @@
 # Initialize the retriever
 retriever = Text2CypherRetriever(
     driver=driver,
-    llm=llm,  # type: ignore
+    llm=llm,
     neo4j_schema=neo4j_schema,
     examples=examples,
 )