PradeepBodhi commited on
Commit
0eed05c
·
verified ·
1 Parent(s): 5da66a8

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -11,7 +11,6 @@ from langchain_core.tools import tool
11
  from langchain.tools.retriever import create_retriever_tool
12
  from langchain_huggingface import HuggingFaceEmbeddings
13
  from supabase.client import Client, create_client
14
- from langchain.llms import HuggingFaceHub
15
 
16
  load_dotenv()
17
 
@@ -151,7 +150,8 @@ tools = [
151
  # Build graph function
152
  def build_graph():
153
  """Build the graph"""
154
- llm = ChatGroq(model="compound-beta", temperature=0)
 
155
  # Bind tools to LLM
156
  llm_with_tools = llm.bind_tools(tools)
157
 
 
11
  from langchain.tools.retriever import create_retriever_tool
12
  from langchain_huggingface import HuggingFaceEmbeddings
13
  from supabase.client import Client, create_client
 
14
 
15
  load_dotenv()
16
 
 
150
  # Build graph function
151
  def build_graph():
152
  """Build the graph"""
153
+ # llm = ChatGroq(model="compound-beta", temperature=0)
154
+ llm = ChatGroq(model="compound-beta", temperature=0, api_key=os.getenv("GROQ_API_KEY"))
155
  # Bind tools to LLM
156
  llm_with_tools = llm.bind_tools(tools)
157