Update app.py

app.py

@@ -19,6 +19,7 @@ from llama_index.core import SimpleDirectoryReader, KnowledgeGraphIndex, Setting
 from llama_index.core.graph_stores import SimpleGraphStore
 from llama_index.core import StorageContext
 from llama_index.embeddings.openai import OpenAIEmbedding  # Fix import path
+from llama_index.llms.langchain import LangChainLLM
 
 def main():
     st.set_page_config(
@@ -190,10 +191,13 @@ def process_document(file_path, graph, openai_api_key):
     # Configure OpenAI
     os.environ["OPENAI_API_KEY"] = openai_api_key
     Settings.chunk_size = 512
-    Settings.llm = ChatOpenAI(temperature=0, model="gpt-4")
     
-    # Setup embeddings using correct import
-    embed_model = OpenAIEmbedding()  # This should work now
+    # Setup LangChain LLM wrapper
+    llm = ChatOpenAI(temperature=0, model="gpt-4")
+    Settings.llm = LangChainLLM(llm=llm)
+    
+    # Setup embeddings
+    embed_model = OpenAIEmbedding()
     
     # Create graph store
     graph_store = SimpleGraphStore()