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})
|