GLECO commited on
Commit
64358a0
·
1 Parent(s): 6c4bc49

Ajout env variable

Browse files
Files changed (1) hide show
  1. 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)