Skip to content
8 changes: 4 additions & 4 deletions examples/RAG_with_graph_db.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -620,7 +620,7 @@
},
{
"cell_type": "code",
"execution_count": 15,
"execution_count": 84,
"id": "83100e64",
"metadata": {},
"outputs": [],
Expand All @@ -629,7 +629,7 @@
"client = OpenAI(api_key=os.environ.get(\"OPENAI_API_KEY\", \"<your OpenAI API key if not set as env var>\"))\n",
"\n",
"# Define the entities to look for\n",
"def define_query(prompt, model=\"gpt-4-1106-preview\"):\n",
"def define_query(prompt, model=\"gpt-4o\"):\n",
" completion = client.chat.completions.create(\n",
" model=model,\n",
" temperature=0,\n",
Expand Down Expand Up @@ -1220,7 +1220,7 @@
},
{
"cell_type": "code",
"execution_count": 39,
"execution_count": null,
"id": "14f76f9d",
"metadata": {},
"outputs": [],
Expand All @@ -1230,7 +1230,7 @@
"from langchain.agents.output_parsers.openai_tools import OpenAIToolsAgentOutputParser\n",
"\n",
"\n",
"llm = ChatOpenAI(temperature=0, model=\"gpt-4\")\n",
"llm = ChatOpenAI(temperature=0, model=\"gpt-4o\")\n",
"\n",
"# LLM chain consisting of the LLM and a prompt\n",
"llm_chain = LLMChain(llm=llm, prompt=prompt)\n",
Expand Down
Loading