move import

langchain-streamlit-demo/llm_resources.py

@@ -16,6 +16,7 @@ from langchain.chat_models import (
 from langchain.document_loaders import PyPDFLoader
 from langchain.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
 from langchain.llms.base import BaseLLM
+from langchain.memory import ConversationBufferMemory
 from langchain.prompts import MessagesPlaceholder, ChatPromptTemplate
 from langchain.retrievers import EnsembleRetriever
 from langchain.retrievers.multi_query import MultiQueryRetriever
@@ -59,7 +60,6 @@ def get_agent(
  # memory_key=memory_key,
  # llm=llm,
  # )
- from langchain.memory import ConversationBufferMemory
 
  agent_memory = ConversationBufferMemory(
  chat_memory=chat_history,