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":
|