Adiii143 commited on
Commit
9c41806
·
verified ·
1 Parent(s): 5e2371b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -2
agent.py CHANGED
@@ -5,7 +5,6 @@ from langgraph.graph import START, StateGraph, MessagesState
5
  from langgraph.prebuilt import tools_condition
6
  from langgraph.prebuilt import ToolNode
7
  from langchain_google_genai import ChatGoogleGenerativeAI
8
- from langchain_groq import ChatGroq
9
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
10
  from langchain_community.tools.tavily_search import TavilySearchResults
11
  from langchain_community.document_loaders import WikipediaLoader
@@ -151,7 +150,7 @@ tools = [
151
  ]
152
 
153
  # Build graph function
154
- def build_graph(provider: str = "groq"):
155
  """Build the graph"""
156
  # Load environment variables from .env file
157
  if provider == "google":
 
5
  from langgraph.prebuilt import tools_condition
6
  from langgraph.prebuilt import ToolNode
7
  from langchain_google_genai import ChatGoogleGenerativeAI
 
8
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
9
  from langchain_community.tools.tavily_search import TavilySearchResults
10
  from langchain_community.document_loaders import WikipediaLoader
 
150
  ]
151
 
152
  # Build graph function
153
+ def build_graph(provider: str = "google"):
154
  """Build the graph"""
155
  # Load environment variables from .env file
156
  if provider == "google":