supratipb commited on
Commit
5db08b2
Β·
verified Β·
1 Parent(s): e2dc69f

Upload 3 files

Browse files
Files changed (2) hide show
  1. agent.py +3 -2
  2. requirements.txt +3 -3
agent.py CHANGED
@@ -15,6 +15,9 @@ 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
 
19
  load_dotenv()
20
 
@@ -171,10 +174,8 @@ def build_graph(provider: str = "openai"):
171
  """Build the graph"""
172
  # Load environment variables from .env file
173
  if provider == "openai":
174
- from langchain.chat_models import ChatOpenAI
175
  llm = ChatOpenAI(model_name="gpt-4", temperature=0)
176
  elif provider == "anthropic":
177
- from langchain.chat_models import ChatAnthropic
178
  llm = ChatAnthropic(model="claude-v1", temperature=0)
179
  elif provider == "google":
180
  # Google Gemini
 
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
 
 
174
  """Build the graph"""
175
  # Load environment variables from .env file
176
  if provider == "openai":
 
177
  llm = ChatOpenAI(model_name="gpt-4", temperature=0)
178
  elif provider == "anthropic":
 
179
  llm = ChatAnthropic(model="claude-v1", temperature=0)
180
  elif provider == "google":
181
  # Google Gemini
requirements.txt CHANGED
@@ -1,7 +1,6 @@
1
  gradio
2
  requests
3
  langchain
4
- langchain-community
5
  langchain-core
6
  langchain-google-genai
7
  langchain-huggingface
@@ -17,6 +16,7 @@ wikipedia
17
  pgvector
18
  python-dotenv
19
  sentence-transformers
20
- langchain
21
- openai
22
  anthropic
 
 
 
1
  gradio
2
  requests
3
  langchain
 
4
  langchain-core
5
  langchain-google-genai
6
  langchain-huggingface
 
16
  pgvector
17
  python-dotenv
18
  sentence-transformers
19
+ openai
 
20
  anthropic
21
+ langchain-openai
22
+ langchain-anthropic