Spaces:
Sleeping
Sleeping
Update AgentGPT.py
Browse files- AgentGPT.py +1 -1
AgentGPT.py
CHANGED
@@ -35,7 +35,7 @@ class AgentsGPT:
|
|
35 |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
|
36 |
|
37 |
llm = Fireworks(model="accounts/fireworks/models/llama-v2-70b-chat", model_kwargs={"temperature":0, "max_tokens":1500, "top_p":1.0}, streaming=True)
|
38 |
-
tools = load_tools(["google-search"
|
39 |
agent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True, return_intermediate_steps=True)
|
40 |
|
41 |
response = agent({"input": question})
|
|
|
35 |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
|
36 |
|
37 |
llm = Fireworks(model="accounts/fireworks/models/llama-v2-70b-chat", model_kwargs={"temperature":0, "max_tokens":1500, "top_p":1.0}, streaming=True)
|
38 |
+
tools = load_tools(["google-search"], llm=llm)
|
39 |
agent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True, return_intermediate_steps=True)
|
40 |
|
41 |
response = agent({"input": question})
|