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

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -13,7 +13,8 @@ from langchain_community.utilities import WikipediaAPIWrapper
13
  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 supabase.client import Client, create_client
18
  from sentence_transformers import SentenceTransformer
19
  from langchain.embeddings.base import Embeddings
 
13
  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