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