Tidy-up, close cypher-qa tag

neo4j-graphacademy · Nov 9, 2023 · 21bfe3a · 21bfe3a
1 parent 1e8c515
commit 21bfe3a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/solutions/llm.py b/solutions/llm.py
@@ -2,7 +2,10 @@
 import streamlit as st
 from langchain.chat_models import ChatOpenAI
 
-llm = ChatOpenAI(openai_api_key=st.secrets["OPENAI_API_KEY"], model='gpt-4')
+llm = ChatOpenAI(
+    openai_api_key=st.secrets["OPENAI_API_KEY"],
+    model=st.secrets["OPENAI_MODEL"],
+)
 # end::llm[]
 
 # tag::embedding[]

diff --git a/solutions/tools/fewshot.py b/solutions/tools/fewshot.py
@@ -62,4 +62,4 @@
     verbose=True,
     cypher_prompt=cypher_prompt
 )
-# tag::cypher-qa[]
+# end::cypher-qa[]