Update ns_updyoga.py

ns_updyoga.py

@@ -9,12 +9,7 @@ from langchain_openai import ChatOpenAI
 from langchain.chains import RetrievalQA
 from langchain.callbacks import StdOutCallbackHandler
 
-def setup(openai_key):
-     # Set the API key for OpenAI
-    os.environ["OPENAI_API_KEY"] = openai_key
-    llm = ChatOpenAI(model="gpt-4")
-    retriver = create_index()
-    return retriver, llm
+
     
 def create_index():
     # load the data
@@ -36,6 +31,13 @@ def create_index():
     return vector_store.as_retriever()
 
 
+def setup(openai_key):
+     # Set the API key for OpenAI
+    os.environ["OPENAI_API_KEY"] = openai_key
+    llm = ChatOpenAI(model="gpt-4")
+    retriver = create_index()
+    return retriver, llm
+
 
 def yg_assistant(openai_key,query):