Update agent.py
Browse files
agent.py
CHANGED
|
@@ -163,7 +163,7 @@ def build_graph(provider: str = "huggingface"):
|
|
| 163 |
# llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 164 |
if provider == "huggingface":
|
| 165 |
# TODO: Add huggingface endpoint
|
| 166 |
-
HUGGINGFACEHUB_API_TOKEN = os.getenv
|
| 167 |
repo_id = "llama-2-7b-chat-hf"
|
| 168 |
llm = ChatHuggingFace(
|
| 169 |
llm=HuggingFaceEndpoint(
|
|
|
|
| 163 |
# llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 164 |
if provider == "huggingface":
|
| 165 |
# TODO: Add huggingface endpoint
|
| 166 |
+
HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
|
| 167 |
repo_id = "llama-2-7b-chat-hf"
|
| 168 |
llm = ChatHuggingFace(
|
| 169 |
llm=HuggingFaceEndpoint(
|