Update main.py

main.py

@@ -12,7 +12,7 @@ from operator import itemgetter
 from pinecone import Pinecone
 
 from langchain.memory import ConversationBufferMemory
-from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableConfig
+from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableConfig, RunnableLambda
 from langchain.callbacks.base import BaseCallbackHandler
 
 import chainlit as cl