Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -3,6 +3,7 @@ from dotenv import load_dotenv
|
|
| 3 |
from langgraph.graph import START, StateGraph, MessagesState
|
| 4 |
from langgraph.prebuilt import tools_condition, ToolNode
|
| 5 |
from langchain_groq import ChatGroq
|
|
|
|
| 6 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 7 |
from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
|
| 8 |
from langchain_community.vectorstores import SupabaseVectorStore
|
|
@@ -151,7 +152,7 @@ tools = [
|
|
| 151 |
def build_graph():
|
| 152 |
"""Build the graph"""
|
| 153 |
# llm = ChatGroq(model="compound-beta", temperature=0)
|
| 154 |
-
llm =
|
| 155 |
# Bind tools to LLM
|
| 156 |
llm_with_tools = llm.bind_tools(tools)
|
| 157 |
|
|
|
|
| 3 |
from langgraph.graph import START, StateGraph, MessagesState
|
| 4 |
from langgraph.prebuilt import tools_condition, ToolNode
|
| 5 |
from langchain_groq import ChatGroq
|
| 6 |
+
from langchain_openai import ChatOpenAI
|
| 7 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 8 |
from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
|
| 9 |
from langchain_community.vectorstores import SupabaseVectorStore
|
|
|
|
| 152 |
def build_graph():
|
| 153 |
"""Build the graph"""
|
| 154 |
# llm = ChatGroq(model="compound-beta", temperature=0)
|
| 155 |
+
llm = ChatOpenAI(model = "gpt-3.5-turbo", temperature=0)
|
| 156 |
# Bind tools to LLM
|
| 157 |
llm_with_tools = llm.bind_tools(tools)
|
| 158 |
|