Update ns_updyoga.py

ns_updyoga.py

@@ -12,8 +12,9 @@ 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
+    return retriver, llm
     
 def create_index():
     # load the data
@@ -39,8 +40,8 @@ def create_index():
 def yg_assistant(openai_key,query):
 
     # Setup
-    llm = ChatOpenAI(model="gpt-4")
-    retriever = setup(openai_key)
+    retriever, llm = setup(openai_key)
+    
     # Create the QA chain
     handler = StdOutCallbackHandler()