Update agent.py
Browse files
agent.py
CHANGED
@@ -10,11 +10,11 @@ from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingF
|
|
10 |
#from langchain_community.tools.tavily_search import TavilySearchResults
|
11 |
from langchain_community.document_loaders import WikipediaLoader
|
12 |
from langchain_community.document_loaders import ArxivLoader
|
13 |
-
from langchain_community.vectorstores import SupabaseVectorStore
|
14 |
from langchain_core.messages import SystemMessage, HumanMessage
|
15 |
from langchain_core.tools import tool
|
16 |
-
from langchain.tools.retriever import create_retriever_tool
|
17 |
-
from supabase.client import Client, create_client
|
18 |
|
19 |
load_dotenv()
|
20 |
|
@@ -179,7 +179,7 @@ def build_graph(provider: str = "huggingface"):
|
|
179 |
llm = ChatHuggingFace(
|
180 |
llm=HuggingFaceEndpoint(
|
181 |
url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
182 |
-
temperature=0,
|
183 |
),
|
184 |
)
|
185 |
else:
|
|
|
10 |
#from langchain_community.tools.tavily_search import TavilySearchResults
|
11 |
from langchain_community.document_loaders import WikipediaLoader
|
12 |
from langchain_community.document_loaders import ArxivLoader
|
13 |
+
#from langchain_community.vectorstores import SupabaseVectorStore
|
14 |
from langchain_core.messages import SystemMessage, HumanMessage
|
15 |
from langchain_core.tools import tool
|
16 |
+
#from langchain.tools.retriever import create_retriever_tool
|
17 |
+
#from supabase.client import Client, create_client
|
18 |
|
19 |
load_dotenv()
|
20 |
|
|
|
179 |
llm = ChatHuggingFace(
|
180 |
llm=HuggingFaceEndpoint(
|
181 |
url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
182 |
+
temperature=0.2,
|
183 |
),
|
184 |
)
|
185 |
else:
|