Update rag.py

rag.py

@@ -101,4 +101,9 @@ def rag_chain(llm, prompt, db):
                                             return_source_documents = True,
                                             verbose = False)
     completion = rag_chain({"query": prompt})
-    return completion, rag_chain
+    return completion, rag_chain
+
+def get_llm():
+    return ChatOpenAI(model_name = config["model_name"], 
+                      openai_api_key = openai_api_key, 
+                      temperature = config["temperature"])