Spaces:
Sleeping
Sleeping
Hanson commited on
Commit ·
dbf30e9
1
Parent(s): f123926
feat: support dynamic max_steps via environment variable
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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=
|
| 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))
|