diff --git a/templates/sql-research-assistant/sql_research_assistant/search/sql.py b/templates/sql-research-assistant/sql_research_assistant/search/sql.py
index c67afa884e846..cb976e2d2114e 100644
--- a/templates/sql-research-assistant/sql_research_assistant/search/sql.py
+++ b/templates/sql-research-assistant/sql_research_assistant/search/sql.py
@@ -2,10 +2,10 @@
 
 from langchain.chat_models import ChatOllama, ChatOpenAI
 from langchain.memory import ConversationBufferMemory
-from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langchain.prompts import ChatPromptTemplate
 from langchain.pydantic_v1 import BaseModel
 from langchain.schema.output_parser import StrOutputParser
-from langchain.schema.runnable import RunnableLambda, RunnablePassthrough
+from langchain.schema.runnable import RunnablePassthrough
 from langchain.utilities import SQLDatabase
 
 # Add the LLM downloaded from Ollama
diff --git a/templates/sql-research-assistant/sql_research_assistant/search/web.py b/templates/sql-research-assistant/sql_research_assistant/search/web.py
index afd69c26fba99..eb84202febd38 100644
--- a/templates/sql-research-assistant/sql_research_assistant/search/web.py
+++ b/templates/sql-research-assistant/sql_research_assistant/search/web.py
@@ -5,11 +5,9 @@
 from bs4 import BeautifulSoup
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts import ChatPromptTemplate
-from langchain.retrievers.tavily_search_api import TavilySearchAPIRetriever
 from langchain.schema.messages import SystemMessage
 from langchain.schema.output_parser import StrOutputParser
 from langchain.schema.runnable import (
-    ConfigurableField,
     Runnable,
     RunnableLambda,
     RunnableParallel,