yoda1976 commited on
Commit
0c3e951
·
1 Parent(s): 77bac94

Add huggingface end point instead of openai

Browse files
Files changed (1) hide show
  1. tools.py +4 -2
tools.py CHANGED
@@ -4,7 +4,7 @@ from langchain_core.messages import SystemMessage, HumanMessage, AnyMessage
4
  from langgraph.graph.message import add_messages
5
  from langchain_community.vectorstores import SupabaseVectorStore
6
  from supabase.client import create_client
7
- from langchain_huggingface import HuggingFaceEmbeddings
8
  from serpapi import GoogleSearch
9
  from dotenv import load_dotenv
10
  import os
@@ -46,7 +46,9 @@ def web_search(query: str) -> str:
46
  )
47
  return context if context else "No results found."
48
 
49
- llm = ChatOpenAI(model = "gpt-3.5-turbo")
 
 
50
  tools = [add, subtract, divide, web_search]
51
  llm_with_tools = llm.bind_tools(tools, parallel_tool_calls = False)
52
 
 
4
  from langgraph.graph.message import add_messages
5
  from langchain_community.vectorstores import SupabaseVectorStore
6
  from supabase.client import create_client
7
+ from langchain_huggingface import HuggingFaceEmbeddings, ChatHuggingFace, HuggingFaceEndpoint
8
  from serpapi import GoogleSearch
9
  from dotenv import load_dotenv
10
  import os
 
46
  )
47
  return context if context else "No results found."
48
 
49
+ llm = ChatHuggingFace(llm = HuggingFaceEndpoint(
50
+ url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
51
+ temperature=0))
52
  tools = [add, subtract, divide, web_search]
53
  llm_with_tools = llm.bind_tools(tools, parallel_tool_calls = False)
54