wt002 commited on
Commit
db3a218
·
verified ·
1 Parent(s): 0db0aa5

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -0
agent.py CHANGED
@@ -452,7 +452,11 @@ tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, vi
452
  #llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
453
  #llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
454
 
 
 
 
455
  from transformers import pipeline
 
456
 
457
  # Get the Hugging Face API token from the environment variable
458
  # Get the Hugging Face API token from the environment variable
 
452
  #llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
453
  #llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
454
 
455
+ from langchain.agents import initialize_agent, Tool, AgentType
456
+ from langchain_huggingface import HuggingFaceHub
457
+ from langchain.prompts import PromptTemplate
458
  from transformers import pipeline
459
+ from langchain_community.llms import HuggingFaceHub
460
 
461
  # Get the Hugging Face API token from the environment variable
462
  # Get the Hugging Face API token from the environment variable