Hanson commited on
Commit
dbf30e9
·
1 Parent(s): f123926

feat: support dynamic max_steps via environment variable

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -21,7 +21,11 @@ def fake_gpu_function():
21
  class BasicAgent:
22
  def __init__(self):
23
  self.openrouter_key = os.getenv("OPENROUTER_API_KEY")
24
- self.model_id = os.getenv("HF_MODEL_ID", "google/gemini-2.5-flash")
 
 
 
 
25
 
26
  def __call__(self, question: str) -> str:
27
  try:
@@ -38,7 +42,7 @@ class BasicAgent:
38
  agent = CodeAgent(
39
  tools=tools,
40
  model=model,
41
- max_steps=6
42
  )
43
 
44
  return str(agent.run(question))
 
21
  class BasicAgent:
22
  def __init__(self):
23
  self.openrouter_key = os.getenv("OPENROUTER_API_KEY")
24
+ # 確保模型 ID 乾淨,避免 400 錯誤
25
+ self.model_id = "google/gemini-2.5-flash"
26
+
27
+ # 💡 從環境變數讀取 max_steps,如果沒設定,預設為 smolagents 原廠的 12 輪
28
+ self.max_steps = int(os.getenv("MAX_STEPS", "12"))
29
 
30
  def __call__(self, question: str) -> str:
31
  try:
 
42
  agent = CodeAgent(
43
  tools=tools,
44
  model=model,
45
+ max_steps=self.max_steps # 💡 帶入變數
46
  )
47
 
48
  return str(agent.run(question))