Spaces:
Running
Running
Update agent.py
Browse files
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 |
-
|
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()
|