Update app.py

app.py

@@ -3,8 +3,9 @@ import asyncio
 import os
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.prompts import SystemMessagePromptTemplate, HumanMessagePromptTemplate
-from langchain_community.vectorstores import Chroma
-from langchain_openai import OpenAIEmbeddings, ChatOpenAI
+from langchain.vectorstores import Chroma
+from langchain.embeddings import OpenAIEmbeddings
+from langchain.chat_models import ChatOpenAI
 from PyPDF2 import PdfReader
 import aiohttp
 from io import BytesIO
@@ -37,8 +38,8 @@ class RetrievalAugmentedQAPipeline:
         formatted_system_prompt = system_role_prompt.format()
         formatted_user_prompt = user_role_prompt.format(question=user_query, context=context_prompt)
 
-        response = await self.llm.ainvoke([formatted_system_prompt, formatted_user_prompt])
-        return {"response": response.content, "context": context_list}
+        response = await self.llm.agenerate([formatted_system_prompt, formatted_user_prompt])
+        return {"response": response.generations[0][0].text, "context": context_list}
 
 # PDF processing functions
 async def fetch_pdf(session, url):