Spaces:
Sleeping
Sleeping
Commit
·
61857d6
1
Parent(s):
6b8cb8c
removed tokens
Browse files
agent.py
CHANGED
@@ -6,14 +6,12 @@ from langchain_huggingface import HuggingFaceEndpoint
|
|
6 |
|
7 |
llm = HuggingFaceEndpoint(
|
8 |
repo_id="microsoft/DialoGPT-small",
|
9 |
-
huggingfacehub_api_token=
|
10 |
task="text-generation",
|
11 |
temperature=0.7
|
12 |
)
|
13 |
# Optional: Use web search tool
|
14 |
-
|
15 |
-
search = SerpAPIWrapper(serpapi_api_key="308082f0c98a96a1b8dac63242122d5df8a96b65678b17cee9e8462c857ec5c8")
|
16 |
-
# # Requires SERPAPI_API_KEY env var
|
17 |
|
18 |
tools = [
|
19 |
Tool(
|
|
|
6 |
|
7 |
llm = HuggingFaceEndpoint(
|
8 |
repo_id="microsoft/DialoGPT-small",
|
9 |
+
huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
|
10 |
task="text-generation",
|
11 |
temperature=0.7
|
12 |
)
|
13 |
# Optional: Use web search tool
|
14 |
+
search = SerpAPIWrapper(serpapi_api_key=os.environ["SERPAPI_API_KEY"])
|
|
|
|
|
15 |
|
16 |
tools = [
|
17 |
Tool(
|