Spaces:
Runtime error
Runtime error
Upload agent.py
Browse files
agent.py
CHANGED
@@ -1,9 +1,12 @@
|
|
1 |
"""LangGraph Agent"""
|
|
|
2 |
import os
|
3 |
from dotenv import load_dotenv
|
|
|
4 |
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
|
@@ -14,10 +17,12 @@ from langchain_community.vectorstores import SupabaseVectorStore
|
|
14 |
from langchain_core.messages import SystemMessage, HumanMessage
|
15 |
from langchain_core.tools import tool
|
16 |
from langchain.tools.retriever import create_retriever_tool
|
17 |
-
from supabase.client import Client, create_client
|
18 |
from langchain_openai import ChatOpenAI
|
19 |
from langchain_anthropic import ChatAnthropic
|
20 |
|
|
|
|
|
|
|
21 |
|
22 |
load_dotenv()
|
23 |
|
@@ -86,6 +91,8 @@ def wiki_search(query: str) -> str:
|
|
86 |
for doc in search_docs
|
87 |
])
|
88 |
return {"wiki_results": formatted_search_docs}
|
|
|
|
|
89 |
|
90 |
@tool
|
91 |
def web_search(query: str) -> str:
|
@@ -170,7 +177,7 @@ tools = [
|
|
170 |
]
|
171 |
|
172 |
# Build graph function
|
173 |
-
def build_graph(provider: str = "
|
174 |
"""Build the graph"""
|
175 |
# Load environment variables from .env file
|
176 |
if provider == "openai":
|
|
|
1 |
"""LangGraph Agent"""
|
2 |
+
|
3 |
import os
|
4 |
from dotenv import load_dotenv
|
5 |
+
|
6 |
from langgraph.graph import START, StateGraph, MessagesState
|
7 |
from langgraph.prebuilt import tools_condition
|
8 |
from langgraph.prebuilt import ToolNode
|
9 |
+
|
10 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
11 |
from langchain_groq import ChatGroq
|
12 |
from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
|
|
|
17 |
from langchain_core.messages import SystemMessage, HumanMessage
|
18 |
from langchain_core.tools import tool
|
19 |
from langchain.tools.retriever import create_retriever_tool
|
|
|
20 |
from langchain_openai import ChatOpenAI
|
21 |
from langchain_anthropic import ChatAnthropic
|
22 |
|
23 |
+
from supabase.client import Client, create_client
|
24 |
+
|
25 |
+
|
26 |
|
27 |
load_dotenv()
|
28 |
|
|
|
91 |
for doc in search_docs
|
92 |
])
|
93 |
return {"wiki_results": formatted_search_docs}
|
94 |
+
|
95 |
+
|
96 |
|
97 |
@tool
|
98 |
def web_search(query: str) -> str:
|
|
|
177 |
]
|
178 |
|
179 |
# Build graph function
|
180 |
+
def build_graph(provider: str = "groq"):
|
181 |
"""Build the graph"""
|
182 |
# Load environment variables from .env file
|
183 |
if provider == "openai":
|