diff options
Diffstat (limited to 'Omni/Jr.hs')
| -rwxr-xr-x | Omni/Jr.hs | 15 |
1 files changed, 12 insertions, 3 deletions
@@ -53,7 +53,7 @@ jr Usage: jr task [<args>...] - jr work [<task-id>] + jr work [<task-id>] [--engine=ENGINE] jr prompt <task-id> jr web [--port=PORT] jr review [<task-id>] [--auto] @@ -77,6 +77,7 @@ Commands: Options: -h --help Show this help --port=PORT Port for web server [default: 8080] + --engine=ENGINE LLM engine: openrouter, ollama, amp [default: openrouter] --auto Auto-review: accept if tests pass, reject if they fail --delay=SECONDS Delay between loop iterations [default: 5] --project=PROJECT Filter facts by project @@ -119,13 +120,20 @@ move args absPath <- Directory.getCurrentDirectory let name = Text.pack (takeFileName absPath) + -- Parse engine flag + let engineType = case Cli.getArg args (Cli.longOption "engine") of + Just "ollama" -> AgentCore.EngineOllama + Just "amp" -> AgentCore.EngineAmp + _ -> AgentCore.EngineOpenRouter + let worker = AgentCore.Worker { AgentCore.workerName = name, AgentCore.workerPid = Nothing, AgentCore.workerStatus = AgentCore.Idle, AgentCore.workerPath = path, - AgentCore.workerQuiet = False -- Show ANSI status bar for manual work + AgentCore.workerQuiet = False, -- Show ANSI status bar for manual work + AgentCore.workerEngine = engineType } let taskId = fmap Text.pack (Cli.getArg args (Cli.argument "task-id")) @@ -183,7 +191,8 @@ runLoop delaySec = do AgentCore.workerPid = Nothing, AgentCore.workerStatus = AgentCore.Idle, AgentCore.workerPath = ".", - AgentCore.workerQuiet = True -- No ANSI status bar in loop mode + AgentCore.workerQuiet = True, -- No ANSI status bar in loop mode + AgentCore.workerEngine = AgentCore.EngineOpenRouter -- Default for loop } putText "[loop] Starting worker..." AgentWorker.start worker (Just (TaskCore.taskId task)) |