IamRulo commited on
Commit
e6dd859
·
verified ·
1 Parent(s): ae44e28

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -692,7 +692,9 @@ def build_graph(provider: str = "huggingface"):
692
  llm=HuggingFaceEndpoint(
693
  #repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
694
  #endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
695
- endpoint_url="https://api-inference.huggingface.co/models/Qwen/Qwen3-4B",
 
 
696
  #task="text-generation", # for chat‐style use “text-generation”
697
  #max_new_tokens=1024,
698
  #do_sample=False,
 
692
  llm=HuggingFaceEndpoint(
693
  #repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
694
  #endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
695
+ #endpoint_url="https://api-inference.huggingface.co/models/Qwen/Qwen3-4B",
696
+ url="https://api-inference.huggingface.co/models/microsoft/phi-4",
697
+
698
  #task="text-generation", # for chat‐style use “text-generation”
699
  #max_new_tokens=1024,
700
  #do_sample=False,