IamRulo commited on
Commit
9d5e618
·
verified ·
1 Parent(s): 6c7a180

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -4
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: