wt002 commited on
Commit
2d14e5a
·
verified ·
1 Parent(s): e0e02ab

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -2
agent.py CHANGED
@@ -13,7 +13,7 @@ from langchain_community.document_loaders import WikipediaLoader
13
  from langchain_community.utilities import WikipediaAPIWrapper
14
  from langchain_community.document_loaders import ArxivLoader
15
  from langchain_core.messages import SystemMessage, HumanMessage
16
- #from langchain_core.tools import tool
17
  from langchain.tools.retriever import create_retriever_tool
18
  from supabase.client import Client, create_client
19
  from sentence_transformers import SentenceTransformer
@@ -55,7 +55,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
55
  from huggingface_hub import login
56
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
57
  from langchain_huggingface import HuggingFaceEndpoint
58
- from langchain.agents.tools import Tool
59
 
60
 
61
  load_dotenv()
 
13
  from langchain_community.utilities import WikipediaAPIWrapper
14
  from langchain_community.document_loaders import ArxivLoader
15
  from langchain_core.messages import SystemMessage, HumanMessage
16
+ from langchain_core.tools import tool
17
  from langchain.tools.retriever import create_retriever_tool
18
  from supabase.client import Client, create_client
19
  from sentence_transformers import SentenceTransformer
 
55
  from huggingface_hub import login
56
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
57
  from langchain_huggingface import HuggingFaceEndpoint
 
58
 
59
 
60
  load_dotenv()