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