Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -48,11 +48,11 @@ from langchain_community.chat_models import ChatOpenAI
|
|
48 |
from langchain_community.tools import Tool
|
49 |
import time
|
50 |
from huggingface_hub import InferenceClient
|
51 |
-
from
|
52 |
-
from langchain.llms import HuggingFaceHub
|
53 |
from langchain.prompts import PromptTemplate
|
54 |
from langchain.chains import LLMChain
|
55 |
from langchain.agents import initialize_agent, Tool, AgentType
|
|
|
56 |
|
57 |
load_dotenv()
|
58 |
|
@@ -457,10 +457,7 @@ tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, vi
|
|
457 |
#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
|
458 |
#llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
|
459 |
|
460 |
-
|
461 |
-
from langchain.prompts import PromptTemplate
|
462 |
-
from transformers import pipeline
|
463 |
-
from langchain_community.llms import HuggingFaceHub
|
464 |
|
465 |
# Get the Hugging Face API token from the environment variable
|
466 |
# Get the Hugging Face API token from the environment variable
|
|
|
48 |
from langchain_community.tools import Tool
|
49 |
import time
|
50 |
from huggingface_hub import InferenceClient
|
51 |
+
from langchain_community.llms import HuggingFaceHub
|
|
|
52 |
from langchain.prompts import PromptTemplate
|
53 |
from langchain.chains import LLMChain
|
54 |
from langchain.agents import initialize_agent, Tool, AgentType
|
55 |
+
from transformers import pipeline
|
56 |
|
57 |
load_dotenv()
|
58 |
|
|
|
457 |
#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
|
458 |
#llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
|
459 |
|
460 |
+
|
|
|
|
|
|
|
461 |
|
462 |
# Get the Hugging Face API token from the environment variable
|
463 |
# Get the Hugging Face API token from the environment variable
|