saswatdas123 commited on
Commit
b5a7a74
·
verified ·
1 Parent(s): a814633

Update pages/Intelligent Chatbot.py

Browse files
Files changed (1) hide show
  1. pages/Intelligent Chatbot.py +6 -3
pages/Intelligent Chatbot.py CHANGED
@@ -4,6 +4,8 @@ from langchain import hub
4
  from langchain.agents import AgentExecutor, create_react_agent, load_tools
5
  from langchain_community.chat_models.huggingface import ChatHuggingFace
6
  #from langchain_openai import OpenAI
 
 
7
 
8
  from langchain_community.callbacks.streamlit import (
9
  StreamlitCallbackHandler,
@@ -13,13 +15,14 @@ st_callback = StreamlitCallbackHandler(st.container())
13
 
14
  initdict={}
15
  initdict = ut.get_tokens()
16
- hf_token = initdict["hf_token"]
 
17
  reactstyle_prompt = initdict["reactstyle_prompt"]
18
  serpapi_api_key = initdict["serpapi_api_key"]
19
  llm_repoid = initdict["llm_repoid"]
20
 
21
- llm = HuggingFaceEndpoint(repo_id=llm_repoid,temperature=0.9,verbose=True)
22
- """llm = HuggingFaceEndpoint(repo_id=llm_repoid,huggingfacehub_api_token=hf_token,temperature=0.9,verbose=True)"""
23
 
24
  tools = load_tools(["serpapi"],llm=llm,serpapi_api_key=serpapi_api_key)
25
  prompt = hub.pull(reactstyle_prompt)
 
4
  from langchain.agents import AgentExecutor, create_react_agent, load_tools
5
  from langchain_community.chat_models.huggingface import ChatHuggingFace
6
  #from langchain_openai import OpenAI
7
+ import os
8
+ print('HF_TOKEN',os.getenv('HF_TOKEN'))
9
 
10
  from langchain_community.callbacks.streamlit import (
11
  StreamlitCallbackHandler,
 
15
 
16
  initdict={}
17
  initdict = ut.get_tokens()
18
+ #hf_token = initdict["hf_token"]
19
+ hf_token = os.getenv('HF_TOKEN')
20
  reactstyle_prompt = initdict["reactstyle_prompt"]
21
  serpapi_api_key = initdict["serpapi_api_key"]
22
  llm_repoid = initdict["llm_repoid"]
23
 
24
+ #llm = HuggingFaceEndpoint(repo_id=llm_repoid,temperature=0.9,verbose=True)
25
+ llm = HuggingFaceEndpoint(repo_id=llm_repoid,huggingfacehub_api_token=hf_token,temperature=0.9,verbose=True)
26
 
27
  tools = load_tools(["serpapi"],llm=llm,serpapi_api_key=serpapi_api_key)
28
  prompt = hub.pull(reactstyle_prompt)