Spaces:
Sleeping
Sleeping
Ajout env variable
Browse files- certificate_agent.py +3 -1
certificate_agent.py
CHANGED
|
@@ -6,6 +6,7 @@ import wikipedia
|
|
| 6 |
import json
|
| 7 |
from prompt import SYSTEM_PROMPT_MANAGER, SYSTEM_PROMPT_CLEANER
|
| 8 |
from langchain_community.tools import DuckDuckGoSearchResults
|
|
|
|
| 9 |
|
| 10 |
class GraphState(TypedDict):
|
| 11 |
history: List
|
|
@@ -13,7 +14,8 @@ class GraphState(TypedDict):
|
|
| 13 |
llm = HuggingFaceEndpoint(
|
| 14 |
repo_id="Qwen/Qwen2.5-72B-Instruct",
|
| 15 |
task='text-generation',
|
| 16 |
-
max_new_tokens=4096
|
|
|
|
| 17 |
)
|
| 18 |
manager_agent = ChatHuggingFace(llm=llm)
|
| 19 |
#cleaner_agent = ChatHuggingFace(llm=llm)
|
|
|
|
| 6 |
import json
|
| 7 |
from prompt import SYSTEM_PROMPT_MANAGER, SYSTEM_PROMPT_CLEANER
|
| 8 |
from langchain_community.tools import DuckDuckGoSearchResults
|
| 9 |
+
import os
|
| 10 |
|
| 11 |
class GraphState(TypedDict):
|
| 12 |
history: List
|
|
|
|
| 14 |
llm = HuggingFaceEndpoint(
|
| 15 |
repo_id="Qwen/Qwen2.5-72B-Instruct",
|
| 16 |
task='text-generation',
|
| 17 |
+
max_new_tokens=4096,
|
| 18 |
+
huggingfacehub_api_token=os.environ['hf_token']
|
| 19 |
)
|
| 20 |
manager_agent = ChatHuggingFace(llm=llm)
|
| 21 |
#cleaner_agent = ChatHuggingFace(llm=llm)
|