supratipb commited on
Commit
d5e9128
Β·
verified Β·
1 Parent(s): 10131a9

Upload agent.py

Browse files
Files changed (1) hide show
  1. agent.py +9 -2
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 = "openai"):
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":