wt002 commited on
Commit
3411224
·
verified ·
1 Parent(s): a222e80

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -14,7 +14,7 @@ from langchain_community.document_loaders import ArxivLoader
14
  from langchain_core.messages import SystemMessage, HumanMessage
15
  from langchain_core.tools import tool
16
  #from langchain_community.tools import create_retriever_tool
17
- from langchain_community.agent_toolkits import create_retriever_tool
18
  from supabase.client import Client, create_client
19
  from sentence_transformers import SentenceTransformer
20
  from langchain.embeddings.base import Embeddings
 
14
  from langchain_core.messages import SystemMessage, HumanMessage
15
  from langchain_core.tools import tool
16
  #from langchain_community.tools import create_retriever_tool
17
+ from langchain.tools.retriever import create_retriever_tool
18
  from supabase.client import Client, create_client
19
  from sentence_transformers import SentenceTransformer
20
  from langchain.embeddings.base import Embeddings