agent-bundle 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +161 -0
- package/dist/agent/agent.d.ts +32 -0
- package/dist/agent/agent.js +241 -0
- package/dist/agent/agent.js.map +1 -0
- package/dist/agent/agent.test-helpers.d.ts +63 -0
- package/dist/agent/agent.test-helpers.js +170 -0
- package/dist/agent/agent.test-helpers.js.map +1 -0
- package/dist/agent/define-agent.d.ts +2 -0
- package/dist/agent/define-agent.js +18 -0
- package/dist/agent/define-agent.js.map +1 -0
- package/dist/agent/dependencies.d.ts +15 -0
- package/dist/agent/dependencies.js +14 -0
- package/dist/agent/dependencies.js.map +1 -0
- package/dist/agent/index.d.ts +4 -0
- package/dist/agent/index.js +5 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/internals.d.ts +28 -0
- package/dist/agent/internals.js +160 -0
- package/dist/agent/internals.js.map +1 -0
- package/dist/agent/session.d.ts +4 -0
- package/dist/agent/session.js +2 -0
- package/dist/agent/session.js.map +1 -0
- package/dist/agent/types.d.ts +52 -0
- package/dist/agent/types.js +2 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/agent-loop/agent-loop.d.ts +27 -0
- package/dist/agent-loop/agent-loop.js +2 -0
- package/dist/agent-loop/agent-loop.js.map +1 -0
- package/dist/agent-loop/index.d.ts +4 -0
- package/dist/agent-loop/index.js +5 -0
- package/dist/agent-loop/index.js.map +1 -0
- package/dist/agent-loop/pi-mono/events.d.ts +3 -0
- package/dist/agent-loop/pi-mono/events.js +73 -0
- package/dist/agent-loop/pi-mono/events.js.map +1 -0
- package/dist/agent-loop/pi-mono/index.d.ts +1 -0
- package/dist/agent-loop/pi-mono/index.js +2 -0
- package/dist/agent-loop/pi-mono/index.js.map +1 -0
- package/dist/agent-loop/pi-mono/input.d.ts +6 -0
- package/dist/agent-loop/pi-mono/input.js +120 -0
- package/dist/agent-loop/pi-mono/input.js.map +1 -0
- package/dist/agent-loop/pi-mono/model.d.ts +7 -0
- package/dist/agent-loop/pi-mono/model.js +72 -0
- package/dist/agent-loop/pi-mono/model.js.map +1 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.d.ts +11 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.js +167 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.js.map +1 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.test-helpers.d.ts +28 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.test-helpers.js +76 -0
- package/dist/agent-loop/pi-mono/pi-mono-loop.test-helpers.js.map +1 -0
- package/dist/agent-loop/pi-mono/queue.d.ts +9 -0
- package/dist/agent-loop/pi-mono/queue.js +37 -0
- package/dist/agent-loop/pi-mono/queue.js.map +1 -0
- package/dist/agent-loop/pi-mono/tools.d.ts +4 -0
- package/dist/agent-loop/pi-mono/tools.js +164 -0
- package/dist/agent-loop/pi-mono/tools.js.map +1 -0
- package/dist/agent-loop/pi-mono/utils.d.ts +14 -0
- package/dist/agent-loop/pi-mono/utils.js +83 -0
- package/dist/agent-loop/pi-mono/utils.js.map +1 -0
- package/dist/agent-loop/system-prompt/fill.d.ts +1 -0
- package/dist/agent-loop/system-prompt/fill.js +19 -0
- package/dist/agent-loop/system-prompt/fill.js.map +1 -0
- package/dist/agent-loop/system-prompt/generate.d.ts +11 -0
- package/dist/agent-loop/system-prompt/generate.js +21 -0
- package/dist/agent-loop/system-prompt/generate.js.map +1 -0
- package/dist/agent-loop/system-prompt/index.d.ts +2 -0
- package/dist/agent-loop/system-prompt/index.js +3 -0
- package/dist/agent-loop/system-prompt/index.js.map +1 -0
- package/dist/agent-loop/types.d.ts +77 -0
- package/dist/agent-loop/types.js +2 -0
- package/dist/agent-loop/types.js.map +1 -0
- package/dist/cli/build/build.d.ts +35 -0
- package/dist/cli/build/build.js +174 -0
- package/dist/cli/build/build.js.map +1 -0
- package/dist/cli/build/build.test-helpers.d.ts +9 -0
- package/dist/cli/build/build.test-helpers.js +50 -0
- package/dist/cli/build/build.test-helpers.js.map +1 -0
- package/dist/cli/build/codegen-commands.d.ts +16 -0
- package/dist/cli/build/codegen-commands.js +81 -0
- package/dist/cli/build/codegen-commands.js.map +1 -0
- package/dist/cli/build/codegen.d.ts +55 -0
- package/dist/cli/build/codegen.js +212 -0
- package/dist/cli/build/codegen.js.map +1 -0
- package/dist/cli/build/e2b-template.d.ts +38 -0
- package/dist/cli/build/e2b-template.js +210 -0
- package/dist/cli/build/e2b-template.js.map +1 -0
- package/dist/cli/build/e2b-template.test-helpers.d.ts +19 -0
- package/dist/cli/build/e2b-template.test-helpers.js +80 -0
- package/dist/cli/build/e2b-template.test-helpers.js.map +1 -0
- package/dist/cli/build/sandbox-image.d.ts +35 -0
- package/dist/cli/build/sandbox-image.js +57 -0
- package/dist/cli/build/sandbox-image.js.map +1 -0
- package/dist/cli/build-codegen.d.ts +41 -0
- package/dist/cli/build-codegen.js +168 -0
- package/dist/cli/build-codegen.js.map +1 -0
- package/dist/cli/build-e2b-template.d.ts +38 -0
- package/dist/cli/build-e2b-template.js +223 -0
- package/dist/cli/build-e2b-template.js.map +1 -0
- package/dist/cli/build-e2b-template.test-helpers.d.ts +19 -0
- package/dist/cli/build-e2b-template.test-helpers.js +80 -0
- package/dist/cli/build-e2b-template.test-helpers.js.map +1 -0
- package/dist/cli/build-sandbox-image.d.ts +35 -0
- package/dist/cli/build-sandbox-image.js +57 -0
- package/dist/cli/build-sandbox-image.js.map +1 -0
- package/dist/cli/build.d.ts +30 -0
- package/dist/cli/build.js +151 -0
- package/dist/cli/build.js.map +1 -0
- package/dist/cli/build.test-helpers.d.ts +9 -0
- package/dist/cli/build.test-helpers.js +50 -0
- package/dist/cli/build.test-helpers.js.map +1 -0
- package/dist/cli/config/load-bundle-config.d.ts +2 -0
- package/dist/cli/config/load-bundle-config.js +20 -0
- package/dist/cli/config/load-bundle-config.js.map +1 -0
- package/dist/cli/config/resolve-project-root.d.ts +1 -0
- package/dist/cli/config/resolve-project-root.js +19 -0
- package/dist/cli/config/resolve-project-root.js.map +1 -0
- package/dist/cli/deploy/aws-cli.d.ts +42 -0
- package/dist/cli/deploy/aws-cli.js +95 -0
- package/dist/cli/deploy/aws-cli.js.map +1 -0
- package/dist/cli/deploy/aws-ecr.d.ts +20 -0
- package/dist/cli/deploy/aws-ecr.js +97 -0
- package/dist/cli/deploy/aws-ecr.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-infra.d.ts +39 -0
- package/dist/cli/deploy/aws-ecs-infra.js +169 -0
- package/dist/cli/deploy/aws-ecs-infra.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-role.d.ts +9 -0
- package/dist/cli/deploy/aws-ecs-role.js +114 -0
- package/dist/cli/deploy/aws-ecs-role.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-service.d.ts +37 -0
- package/dist/cli/deploy/aws-ecs-service.js +207 -0
- package/dist/cli/deploy/aws-ecs-service.js.map +1 -0
- package/dist/cli/deploy/aws-ecs-shared.d.ts +59 -0
- package/dist/cli/deploy/aws-ecs-shared.js +30 -0
- package/dist/cli/deploy/aws-ecs-shared.js.map +1 -0
- package/dist/cli/deploy/aws-ecs.d.ts +3 -0
- package/dist/cli/deploy/aws-ecs.js +141 -0
- package/dist/cli/deploy/aws-ecs.js.map +1 -0
- package/dist/cli/deploy/aws-prerequisites.d.ts +16 -0
- package/dist/cli/deploy/aws-prerequisites.js +28 -0
- package/dist/cli/deploy/aws-prerequisites.js.map +1 -0
- package/dist/cli/deploy/aws-teardown.d.ts +13 -0
- package/dist/cli/deploy/aws-teardown.js +204 -0
- package/dist/cli/deploy/aws-teardown.js.map +1 -0
- package/dist/cli/deploy/deploy.d.ts +36 -0
- package/dist/cli/deploy/deploy.js +209 -0
- package/dist/cli/deploy/deploy.js.map +1 -0
- package/dist/cli/error.d.ts +1 -0
- package/dist/cli/error.js +4 -0
- package/dist/cli/error.js.map +1 -0
- package/dist/cli/generate/generate.d.ts +37 -0
- package/dist/cli/generate/generate.js +121 -0
- package/dist/cli/generate/generate.js.map +1 -0
- package/dist/cli/generate.d.ts +32 -0
- package/dist/cli/generate.js +102 -0
- package/dist/cli/generate.js.map +1 -0
- package/dist/cli/index.d.ts +2 -0
- package/dist/cli/index.js +182 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/load-bundle-config.d.ts +2 -0
- package/dist/cli/load-bundle-config.js +20 -0
- package/dist/cli/load-bundle-config.js.map +1 -0
- package/dist/cli/resolve-project-root.d.ts +1 -0
- package/dist/cli/resolve-project-root.js +19 -0
- package/dist/cli/resolve-project-root.js.map +1 -0
- package/dist/cli/serve/dev.d.ts +6 -0
- package/dist/cli/serve/dev.js +73 -0
- package/dist/cli/serve/dev.js.map +1 -0
- package/dist/cli/serve/http.d.ts +12 -0
- package/dist/cli/serve/http.js +123 -0
- package/dist/cli/serve/http.js.map +1 -0
- package/dist/cli/serve/init.d.ts +42 -0
- package/dist/cli/serve/init.js +110 -0
- package/dist/cli/serve/init.js.map +1 -0
- package/dist/cli/serve/runtime.d.ts +25 -0
- package/dist/cli/serve/runtime.js +142 -0
- package/dist/cli/serve/runtime.js.map +1 -0
- package/dist/cli/serve/serve.d.ts +7 -0
- package/dist/cli/serve/serve.js +63 -0
- package/dist/cli/serve/serve.js.map +1 -0
- package/dist/cli/serve/serve.test-helpers.d.ts +36 -0
- package/dist/cli/serve/serve.test-helpers.js +205 -0
- package/dist/cli/serve/serve.test-helpers.js.map +1 -0
- package/dist/cli/serve/worktree-port.d.ts +36 -0
- package/dist/cli/serve/worktree-port.js +124 -0
- package/dist/cli/serve/worktree-port.js.map +1 -0
- package/dist/cli/serve-http.d.ts +12 -0
- package/dist/cli/serve-http.js +112 -0
- package/dist/cli/serve-http.js.map +1 -0
- package/dist/cli/serve-runtime.d.ts +16 -0
- package/dist/cli/serve-runtime.js +122 -0
- package/dist/cli/serve-runtime.js.map +1 -0
- package/dist/cli/serve.d.ts +33 -0
- package/dist/cli/serve.js +150 -0
- package/dist/cli/serve.js.map +1 -0
- package/dist/cli/serve.test-helpers.d.ts +30 -0
- package/dist/cli/serve.test-helpers.js +145 -0
- package/dist/cli/serve.test-helpers.js.map +1 -0
- package/dist/code-formatter/bundle.json +41 -0
- package/dist/code-formatter/index.ts +26 -0
- package/dist/code-formatter/package.json +10 -0
- package/dist/code-formatter/types.ts +2 -0
- package/dist/code-formatter-e2b/bundle.json +40 -0
- package/dist/code-formatter-e2b/index.ts +25 -0
- package/dist/code-formatter-e2b/package.json +10 -0
- package/dist/code-formatter-e2b/types.ts +2 -0
- package/dist/coding-assistant-ollama/bundle.json +43 -0
- package/dist/coding-assistant-ollama/index.ts +28 -0
- package/dist/coding-assistant-ollama/package.json +10 -0
- package/dist/coding-assistant-ollama/types.ts +2 -0
- package/dist/commands/find.d.ts +8 -0
- package/dist/commands/find.js +11 -0
- package/dist/commands/find.js.map +1 -0
- package/dist/commands/loader.d.ts +13 -0
- package/dist/commands/loader.js +163 -0
- package/dist/commands/loader.js.map +1 -0
- package/dist/commands/types.d.ts +7 -0
- package/dist/commands/types.js +2 -0
- package/dist/commands/types.js.map +1 -0
- package/dist/commands/with-commands.d.ts +6 -0
- package/dist/commands/with-commands.js +19 -0
- package/dist/commands/with-commands.js.map +1 -0
- package/dist/data-analyst/bundle.json +40 -0
- package/dist/data-analyst/index.ts +25 -0
- package/dist/data-analyst/package.json +10 -0
- package/dist/data-analyst/types.ts +2 -0
- package/dist/financial-analyst/bundle.json +40 -0
- package/dist/financial-analyst/index.ts +25 -0
- package/dist/financial-analyst/package.json +10 -0
- package/dist/financial-analyst/types.ts +2 -0
- package/dist/mcp/client-manager.d.ts +26 -0
- package/dist/mcp/client-manager.js +101 -0
- package/dist/mcp/client-manager.js.map +1 -0
- package/dist/mcp/connect-server.d.ts +21 -0
- package/dist/mcp/connect-server.js +170 -0
- package/dist/mcp/connect-server.js.map +1 -0
- package/dist/mcp/index.d.ts +2 -0
- package/dist/mcp/index.js +3 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/sandbox-stdio-transport.d.ts +32 -0
- package/dist/mcp/sandbox-stdio-transport.js +199 -0
- package/dist/mcp/sandbox-stdio-transport.js.map +1 -0
- package/dist/observability/hooks.d.ts +10 -0
- package/dist/observability/hooks.js +97 -0
- package/dist/observability/hooks.js.map +1 -0
- package/dist/observability/index.d.ts +6 -0
- package/dist/observability/index.js +7 -0
- package/dist/observability/index.js.map +1 -0
- package/dist/observability/metrics.d.ts +31 -0
- package/dist/observability/metrics.js +61 -0
- package/dist/observability/metrics.js.map +1 -0
- package/dist/observability/middleware.d.ts +12 -0
- package/dist/observability/middleware.js +45 -0
- package/dist/observability/middleware.js.map +1 -0
- package/dist/observability/otel-harness.test-util.d.ts +37 -0
- package/dist/observability/otel-harness.test-util.js +96 -0
- package/dist/observability/otel-harness.test-util.js.map +1 -0
- package/dist/observability/provider.d.ts +8 -0
- package/dist/observability/provider.js +19 -0
- package/dist/observability/provider.js.map +1 -0
- package/dist/observability/tracing.d.ts +14 -0
- package/dist/observability/tracing.js +40 -0
- package/dist/observability/tracing.js.map +1 -0
- package/dist/observability/types.d.ts +39 -0
- package/dist/observability/types.js +31 -0
- package/dist/observability/types.js.map +1 -0
- package/dist/personalized-recommend/bundle.json +64 -0
- package/dist/personalized-recommend/index.ts +41 -0
- package/dist/personalized-recommend/package.json +10 -0
- package/dist/personalized-recommend/types.ts +2 -0
- package/dist/plugins/loader.d.ts +11 -0
- package/dist/plugins/loader.js +100 -0
- package/dist/plugins/loader.js.map +1 -0
- package/dist/plugins/merge.d.ts +10 -0
- package/dist/plugins/merge.js +22 -0
- package/dist/plugins/merge.js.map +1 -0
- package/dist/plugins/parse.d.ts +10 -0
- package/dist/plugins/parse.js +139 -0
- package/dist/plugins/parse.js.map +1 -0
- package/dist/plugins/types.d.ts +28 -0
- package/dist/plugins/types.js +2 -0
- package/dist/plugins/types.js.map +1 -0
- package/dist/plugins/urls.d.ts +7 -0
- package/dist/plugins/urls.js +40 -0
- package/dist/plugins/urls.js.map +1 -0
- package/dist/runtime.d.ts +5 -0
- package/dist/runtime.js +3 -0
- package/dist/runtime.js.map +1 -0
- package/dist/sandbox/factory.d.ts +2 -0
- package/dist/sandbox/factory.js +27 -0
- package/dist/sandbox/factory.js.map +1 -0
- package/dist/sandbox/index.d.ts +4 -0
- package/dist/sandbox/index.js +5 -0
- package/dist/sandbox/index.js.map +1 -0
- package/dist/sandbox/providers/e2b.d.ts +28 -0
- package/dist/sandbox/providers/e2b.js +294 -0
- package/dist/sandbox/providers/e2b.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-command-run.d.ts +8 -0
- package/dist/sandbox/providers/kubernetes-command-run.js +195 -0
- package/dist/sandbox/providers/kubernetes-command-run.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-helpers.d.ts +31 -0
- package/dist/sandbox/providers/kubernetes-helpers.js +152 -0
- package/dist/sandbox/providers/kubernetes-helpers.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-kubeconfig.d.ts +7 -0
- package/dist/sandbox/providers/kubernetes-kubeconfig.js +51 -0
- package/dist/sandbox/providers/kubernetes-kubeconfig.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-spawn.d.ts +2 -0
- package/dist/sandbox/providers/kubernetes-spawn.js +42 -0
- package/dist/sandbox/providers/kubernetes-spawn.js.map +1 -0
- package/dist/sandbox/providers/kubernetes-spawn.utils.d.ts +32 -0
- package/dist/sandbox/providers/kubernetes-spawn.utils.js +249 -0
- package/dist/sandbox/providers/kubernetes-spawn.utils.js.map +1 -0
- package/dist/sandbox/providers/kubernetes.constants.d.ts +3 -0
- package/dist/sandbox/providers/kubernetes.constants.js +4 -0
- package/dist/sandbox/providers/kubernetes.constants.js.map +1 -0
- package/dist/sandbox/providers/kubernetes.d.ts +37 -0
- package/dist/sandbox/providers/kubernetes.js +258 -0
- package/dist/sandbox/providers/kubernetes.js.map +1 -0
- package/dist/sandbox/types.d.ts +57 -0
- package/dist/sandbox/types.js +2 -0
- package/dist/sandbox/types.js.map +1 -0
- package/dist/sandbox/utils.d.ts +1 -0
- package/dist/sandbox/utils.js +4 -0
- package/dist/sandbox/utils.js.map +1 -0
- package/dist/schema/bundle.d.ts +143 -0
- package/dist/schema/bundle.js +203 -0
- package/dist/schema/bundle.js.map +1 -0
- package/dist/service/command-routes.d.ts +8 -0
- package/dist/service/command-routes.js +44 -0
- package/dist/service/command-routes.js.map +1 -0
- package/dist/service/create-server.d.ts +9 -0
- package/dist/service/create-server.js +113 -0
- package/dist/service/create-server.js.map +1 -0
- package/dist/service/index.d.ts +1 -0
- package/dist/service/index.js +2 -0
- package/dist/service/index.js.map +1 -0
- package/dist/skills/index.d.ts +1 -0
- package/dist/skills/index.js +2 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/loader.d.ts +18 -0
- package/dist/skills/loader.js +142 -0
- package/dist/skills/loader.js.map +1 -0
- package/dist/skills/summaries.d.ts +8 -0
- package/dist/skills/summaries.js +9 -0
- package/dist/skills/summaries.js.map +1 -0
- package/dist/test-helpers/env.d.ts +2 -0
- package/dist/test-helpers/env.js +20 -0
- package/dist/test-helpers/env.js.map +1 -0
- package/dist/test-helpers/mock-agent.d.ts +56 -0
- package/dist/test-helpers/mock-agent.js +180 -0
- package/dist/test-helpers/mock-agent.js.map +1 -0
- package/dist/test-helpers/mock-sandbox.d.ts +33 -0
- package/dist/test-helpers/mock-sandbox.js +200 -0
- package/dist/test-helpers/mock-sandbox.js.map +1 -0
- package/dist/tui/index.d.ts +2 -0
- package/dist/tui/index.js +3 -0
- package/dist/tui/index.js.map +1 -0
- package/dist/tui/render.d.ts +8 -0
- package/dist/tui/render.js +40 -0
- package/dist/tui/render.js.map +1 -0
- package/dist/tui/tui.d.ts +19 -0
- package/dist/tui/tui.js +133 -0
- package/dist/tui/tui.js.map +1 -0
- package/dist/webui/create-webui-server.d.ts +22 -0
- package/dist/webui/create-webui-server.js +293 -0
- package/dist/webui/create-webui-server.js.map +1 -0
- package/dist/webui/event-bus.d.ts +16 -0
- package/dist/webui/event-bus.js +24 -0
- package/dist/webui/event-bus.js.map +1 -0
- package/dist/webui/index.d.ts +2 -0
- package/dist/webui/index.js +3 -0
- package/dist/webui/index.js.map +1 -0
- package/dist/webui/public/app.js +847 -0
- package/dist/webui/public/file-transfer.js +114 -0
- package/dist/webui/public/index.html +166 -0
- package/dist/webui/public/styles.css +1678 -0
- package/package.json +99 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"events.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/events.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAElD,SAAS,eAAe,CAAC,UAAkB,EAAE,QAAgB,EAAE,IAAa;IAC1E,OAAO;QACL,EAAE,EAAE,UAAU;QACd,IAAI,EAAE,oBAAoB,CAAC,QAAQ,CAAC;QACpC,KAAK,EAAE,aAAa,CAAC,IAAI,CAAC;KAC3B,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,UAAkB,EAAE,MAAe,EAAE,OAAgB;IAC9E,OAAO;QACL,UAAU;QACV,MAAM,EAAE,MAAM;QACd,OAAO;KACR,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,aAAsB;IAC/C,IACE,OAAO,aAAa,KAAK,QAAQ;WAC9B,aAAa,KAAK,IAAI;WACtB,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,SAAS,CAAC,EAC3C,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACtD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QACf,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAC9C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,QAAQ,CAAC;IAC/F,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI;YAC9C,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnC,CAAC,CAAC,EAAE,CAAC;IACT,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAiB;IAC/C,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,qBAAqB,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACzF,OAAO;YACL,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,KAAK,CAAC,qBAAqB,CAAC,KAAK;SACzC,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;QAC1C,OAAO;YACL,IAAI,EAAE,4BAA4B;YAClC,QAAQ,EAAE,eAAe,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC;SACxE,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QACrD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO;YACL,IAAI,EAAE,uBAAuB;YAC7B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,KAAK;SACN,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;QACxC,OAAO;YACL,IAAI,EAAE,yBAAyB;YAC/B,MAAM,EAAE,iBAAiB,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC;SACzE,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from "./pi-mono-loop.js";
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import type { AgentMessage } from "@mariozechner/pi-agent-core";
|
|
2
|
+
import type { Api, AssistantMessage, Model } from "@mariozechner/pi-ai";
|
|
3
|
+
import type { ResponseInput } from "../types.js";
|
|
4
|
+
export declare function toAgentMessages(input: ResponseInput, model: Model<Api>): AgentMessage[];
|
|
5
|
+
export declare function getLatestAssistantMessage(messages: AgentMessage[]): AssistantMessage | null;
|
|
6
|
+
export declare function toAssistantText(message: AssistantMessage | null): string;
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
import { isRecord, toToolContent } from "./utils.js";
|
|
2
|
+
function emptyUsage() {
|
|
3
|
+
return {
|
|
4
|
+
input: 0,
|
|
5
|
+
output: 0,
|
|
6
|
+
cacheRead: 0,
|
|
7
|
+
cacheWrite: 0,
|
|
8
|
+
totalTokens: 0,
|
|
9
|
+
cost: {
|
|
10
|
+
input: 0,
|
|
11
|
+
output: 0,
|
|
12
|
+
cacheRead: 0,
|
|
13
|
+
cacheWrite: 0,
|
|
14
|
+
total: 0,
|
|
15
|
+
},
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
function toPiToolName(toolName) {
|
|
19
|
+
const normalized = toolName.trim().toLowerCase();
|
|
20
|
+
if (normalized === "read" || normalized === "write" || normalized === "edit" || normalized === "bash") {
|
|
21
|
+
return normalized;
|
|
22
|
+
}
|
|
23
|
+
return toolName;
|
|
24
|
+
}
|
|
25
|
+
function toAssistantHistoryMessage(message, model, timestamp) {
|
|
26
|
+
const textContent = message.content.trim().length > 0
|
|
27
|
+
? [{ type: "text", text: message.content }]
|
|
28
|
+
: [];
|
|
29
|
+
const toolCallContent = (message.tool_calls ?? []).map((toolCall) => ({
|
|
30
|
+
type: "toolCall",
|
|
31
|
+
id: toolCall.id,
|
|
32
|
+
name: toPiToolName(toolCall.name),
|
|
33
|
+
arguments: toolCall.input,
|
|
34
|
+
}));
|
|
35
|
+
return {
|
|
36
|
+
role: "assistant",
|
|
37
|
+
content: textContent.length + toolCallContent.length > 0
|
|
38
|
+
? [...textContent, ...toolCallContent]
|
|
39
|
+
: [{ type: "text", text: "" }],
|
|
40
|
+
api: model.api,
|
|
41
|
+
provider: model.provider,
|
|
42
|
+
model: model.id,
|
|
43
|
+
usage: emptyUsage(),
|
|
44
|
+
stopReason: toolCallContent.length > 0 ? "toolUse" : "stop",
|
|
45
|
+
timestamp,
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
function toToolResultHistoryMessages(message, timestamp) {
|
|
49
|
+
if (message.tool_results.length === 0) {
|
|
50
|
+
return [
|
|
51
|
+
{
|
|
52
|
+
role: "toolResult",
|
|
53
|
+
toolCallId: `tool-${timestamp}`,
|
|
54
|
+
toolName: "tool",
|
|
55
|
+
content: [{ type: "text", text: toToolContent(undefined, message.content) }],
|
|
56
|
+
details: undefined,
|
|
57
|
+
isError: false,
|
|
58
|
+
timestamp,
|
|
59
|
+
},
|
|
60
|
+
];
|
|
61
|
+
}
|
|
62
|
+
return message.tool_results.map((toolResult, toolResultIndex) => ({
|
|
63
|
+
role: "toolResult",
|
|
64
|
+
toolCallId: toolResult.toolCallId,
|
|
65
|
+
toolName: "tool",
|
|
66
|
+
content: [{ type: "text", text: toToolContent(toolResult.output, message.content) }],
|
|
67
|
+
details: toolResult.output,
|
|
68
|
+
isError: Boolean(toolResult.isError),
|
|
69
|
+
timestamp: timestamp + toolResultIndex,
|
|
70
|
+
}));
|
|
71
|
+
}
|
|
72
|
+
function toAgentMessageArray(message, model, timestamp) {
|
|
73
|
+
if (message.role === "system") {
|
|
74
|
+
return [];
|
|
75
|
+
}
|
|
76
|
+
if (message.role === "user") {
|
|
77
|
+
return [{
|
|
78
|
+
role: "user",
|
|
79
|
+
content: [{ type: "text", text: message.content }],
|
|
80
|
+
timestamp,
|
|
81
|
+
}];
|
|
82
|
+
}
|
|
83
|
+
if (message.role === "assistant") {
|
|
84
|
+
return [toAssistantHistoryMessage(message, model, timestamp)];
|
|
85
|
+
}
|
|
86
|
+
if (message.role === "tool") {
|
|
87
|
+
return toToolResultHistoryMessages(message, timestamp);
|
|
88
|
+
}
|
|
89
|
+
return [];
|
|
90
|
+
}
|
|
91
|
+
function isAssistantMessage(message) {
|
|
92
|
+
if (!isRecord(message)) {
|
|
93
|
+
return false;
|
|
94
|
+
}
|
|
95
|
+
return (message.role === "assistant"
|
|
96
|
+
&& Array.isArray(message.content)
|
|
97
|
+
&& typeof message.model === "string"
|
|
98
|
+
&& typeof message.provider === "string");
|
|
99
|
+
}
|
|
100
|
+
export function toAgentMessages(input, model) {
|
|
101
|
+
const baseTimestamp = Date.now();
|
|
102
|
+
return input.flatMap((message, index) => {
|
|
103
|
+
const timestamp = baseTimestamp + index * 100;
|
|
104
|
+
return toAgentMessageArray(message, model, timestamp);
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
export function getLatestAssistantMessage(messages) {
|
|
108
|
+
const latestMessage = [...messages].reverse().find((message) => isAssistantMessage(message));
|
|
109
|
+
return latestMessage ?? null;
|
|
110
|
+
}
|
|
111
|
+
export function toAssistantText(message) {
|
|
112
|
+
if (message === null) {
|
|
113
|
+
return "";
|
|
114
|
+
}
|
|
115
|
+
return message.content
|
|
116
|
+
.filter((item) => item.type === "text")
|
|
117
|
+
.map((item) => item.text)
|
|
118
|
+
.join("");
|
|
119
|
+
}
|
|
120
|
+
//# sourceMappingURL=input.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"input.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/input.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAErD,SAAS,UAAU;IACjB,OAAO;QACL,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE;YACJ,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,CAAC;SACT;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,QAAgB;IACpC,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACjD,IAAI,UAAU,KAAK,MAAM,IAAI,UAAU,KAAK,OAAO,IAAI,UAAU,KAAK,MAAM,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;QACtG,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,yBAAyB,CAChC,OAA6D,EAC7D,KAAiB,EACjB,SAAiB;IAEjB,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;QACnD,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;QACpD,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,eAAe,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACpE,IAAI,EAAE,UAAmB;QACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,IAAI,EAAE,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC;QACjC,SAAS,EAAE,QAAQ,CAAC,KAAK;KAC1B,CAAC,CAAC,CAAC;IAEJ,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,WAAW,CAAC,MAAM,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC;YACtD,CAAC,CAAC,CAAC,GAAG,WAAW,EAAE,GAAG,eAAe,CAAC;YACtC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAChC,GAAG,EAAE,KAAK,CAAC,GAAG;QACd,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,KAAK,EAAE,KAAK,CAAC,EAAE;QACf,KAAK,EAAE,UAAU,EAAE;QACnB,UAAU,EAAE,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;QAC3D,SAAS;KACV,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAClC,OAAwD,EACxD,SAAiB;IAEjB,IAAI,OAAO,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtC,OAAO;YACL;gBACE,IAAI,EAAE,YAAY;gBAClB,UAAU,EAAE,QAAQ,SAAS,EAAE;gBAC/B,QAAQ,EAAE,MAAM;gBAChB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC5E,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,KAAK;gBACd,SAAS;aACV;SACF,CAAC;IACJ,CAAC;IAED,OAAO,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,eAAe,EAAE,EAAE,CAAC,CAAC;QAChE,IAAI,EAAE,YAAY;QAClB,UAAU,EAAE,UAAU,CAAC,UAAU;QACjC,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACpF,OAAO,EAAE,UAAU,CAAC,MAAM;QAC1B,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;QACpC,SAAS,EAAE,SAAS,GAAG,eAAe;KACvC,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,mBAAmB,CAC1B,OAA6B,EAC7B,KAAiB,EACjB,SAAiB;IAEjB,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC5B,OAAO,CAAC;gBACN,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClD,SAAS;aACV,CAAC,CAAC;IACL,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QACjC,OAAO,CAAC,yBAAyB,CAAC,OAAO,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC5B,OAAO,2BAA2B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAqB;IAC/C,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CACL,OAAO,CAAC,IAAI,KAAK,WAAW;WACzB,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;WAC9B,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ;WACjC,OAAO,OAAO,CAAC,QAAQ,KAAK,QAAQ,CACxC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAoB,EAAE,KAAiB;IACrE,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACjC,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QACtC,MAAM,SAAS,GAAG,aAAa,GAAG,KAAK,GAAG,GAAG,CAAC;QAC9C,OAAO,mBAAmB,CAAC,OAAO,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAwB;IAChE,MAAM,aAAa,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7F,OAAO,aAAa,IAAI,IAAI,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,OAAgC;IAC9D,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACrB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,OAAO,CAAC,OAAO;SACnB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SACtC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { type Api, type Model } from "@mariozechner/pi-ai";
|
|
2
|
+
import type { ModelConfig } from "../types.js";
|
|
3
|
+
type OllamaConfig = NonNullable<ModelConfig["ollama"]>;
|
|
4
|
+
export declare function normalizeOllamaBaseUrl(baseUrl: string): string;
|
|
5
|
+
export declare function resolveOllamaBaseUrl(config?: OllamaConfig): string;
|
|
6
|
+
export declare function resolvePiModel(config: ModelConfig): Model<Api>;
|
|
7
|
+
export {};
|
|
@@ -0,0 +1,72 @@
|
|
|
1
|
+
import { getModels, getProviders } from "@mariozechner/pi-ai";
|
|
2
|
+
const DEFAULT_OLLAMA_BASE_URL = "http://127.0.0.1:11434/v1";
|
|
3
|
+
const DEFAULT_OLLAMA_CONTEXT_WINDOW = 128_000;
|
|
4
|
+
const DEFAULT_OLLAMA_MAX_TOKENS = 32_000;
|
|
5
|
+
function toPiProvider(provider) {
|
|
6
|
+
if (provider === "gemini") {
|
|
7
|
+
return "google";
|
|
8
|
+
}
|
|
9
|
+
return provider;
|
|
10
|
+
}
|
|
11
|
+
export function normalizeOllamaBaseUrl(baseUrl) {
|
|
12
|
+
const normalizedBaseUrl = baseUrl.trim().replace(/\/+$/, "");
|
|
13
|
+
if (normalizedBaseUrl.endsWith("/v1")) {
|
|
14
|
+
return normalizedBaseUrl;
|
|
15
|
+
}
|
|
16
|
+
return `${normalizedBaseUrl}/v1`;
|
|
17
|
+
}
|
|
18
|
+
export function resolveOllamaBaseUrl(config) {
|
|
19
|
+
const configuredBaseUrl = config?.baseUrl ?? process.env.OLLAMA_BASE_URL ?? process.env.OLLAMA_HOST ?? DEFAULT_OLLAMA_BASE_URL;
|
|
20
|
+
return normalizeOllamaBaseUrl(configuredBaseUrl);
|
|
21
|
+
}
|
|
22
|
+
function createOllamaModel(modelId, config) {
|
|
23
|
+
const contextWindow = config?.contextWindow ?? DEFAULT_OLLAMA_CONTEXT_WINDOW;
|
|
24
|
+
const maxTokens = config?.maxTokens ?? DEFAULT_OLLAMA_MAX_TOKENS;
|
|
25
|
+
return {
|
|
26
|
+
id: modelId,
|
|
27
|
+
name: `${modelId} (Ollama)`,
|
|
28
|
+
api: "openai-completions",
|
|
29
|
+
provider: "ollama",
|
|
30
|
+
baseUrl: resolveOllamaBaseUrl(config),
|
|
31
|
+
reasoning: false,
|
|
32
|
+
input: ["text"],
|
|
33
|
+
cost: {
|
|
34
|
+
input: 0,
|
|
35
|
+
output: 0,
|
|
36
|
+
cacheRead: 0,
|
|
37
|
+
cacheWrite: 0,
|
|
38
|
+
},
|
|
39
|
+
contextWindow,
|
|
40
|
+
maxTokens,
|
|
41
|
+
compat: {
|
|
42
|
+
supportsStore: false,
|
|
43
|
+
supportsDeveloperRole: false,
|
|
44
|
+
supportsReasoningEffort: false,
|
|
45
|
+
supportsUsageInStreaming: false,
|
|
46
|
+
maxTokensField: "max_tokens",
|
|
47
|
+
supportsStrictMode: false,
|
|
48
|
+
},
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
export function resolvePiModel(config) {
|
|
52
|
+
if (config.provider === "ollama") {
|
|
53
|
+
return createOllamaModel(config.model, config.ollama);
|
|
54
|
+
}
|
|
55
|
+
const provider = toPiProvider(config.provider);
|
|
56
|
+
const knownProvider = provider;
|
|
57
|
+
const providers = getProviders();
|
|
58
|
+
if (!providers.includes(knownProvider)) {
|
|
59
|
+
throw new Error(`Unsupported model provider: ${config.provider}. Available providers: ${providers.join(", ")}`);
|
|
60
|
+
}
|
|
61
|
+
const availableModels = getModels(knownProvider);
|
|
62
|
+
const exactMatch = availableModels.find((model) => model.id === config.model);
|
|
63
|
+
if (!exactMatch) {
|
|
64
|
+
const sampledModelIds = availableModels
|
|
65
|
+
.slice(0, 8)
|
|
66
|
+
.map((model) => model.id)
|
|
67
|
+
.join(", ");
|
|
68
|
+
throw new Error(`Model ${config.model} is not available for provider ${config.provider}. Known models include: ${sampledModelIds}`);
|
|
69
|
+
}
|
|
70
|
+
return exactMatch;
|
|
71
|
+
}
|
|
72
|
+
//# sourceMappingURL=model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,YAAY,EAA4C,MAAM,qBAAqB,CAAC;AAKxG,MAAM,uBAAuB,GAAG,2BAA2B,CAAC;AAC5D,MAAM,6BAA6B,GAAG,OAAO,CAAC;AAC9C,MAAM,yBAAyB,GAAG,MAAM,CAAC;AAEzC,SAAS,YAAY,CAAC,QAA0C;IAC9D,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAID,MAAM,UAAU,sBAAsB,CAAC,OAAe;IACpD,MAAM,iBAAiB,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAE7D,IAAI,iBAAiB,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACtC,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,OAAO,GAAG,iBAAiB,KAAK,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAqB;IACxD,MAAM,iBAAiB,GAAG,MAAM,EAAE,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,uBAAuB,CAAC;IAC/H,OAAO,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAe,EAAE,MAAqB;IAC/D,MAAM,aAAa,GAAG,MAAM,EAAE,aAAa,IAAI,6BAA6B,CAAC;IAC7E,MAAM,SAAS,GAAG,MAAM,EAAE,SAAS,IAAI,yBAAyB,CAAC;IAEjE,OAAO;QACL,EAAE,EAAE,OAAO;QACX,IAAI,EAAE,GAAG,OAAO,WAAW;QAC3B,GAAG,EAAE,oBAAoB;QACzB,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,oBAAoB,CAAC,MAAM,CAAC;QACrC,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,CAAC,MAAM,CAAC;QACf,IAAI,EAAE;YACJ,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;SACd;QACD,aAAa;QACb,SAAS;QACT,MAAM,EAAE;YACN,aAAa,EAAE,KAAK;YACpB,qBAAqB,EAAE,KAAK;YAC5B,uBAAuB,EAAE,KAAK;YAC9B,wBAAwB,EAAE,KAAK;YAC/B,cAAc,EAAE,YAAY;YAC5B,kBAAkB,EAAE,KAAK;SAC1B;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAmB;IAChD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;IAED,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC/C,MAAM,aAAa,GAAkB,QAAQ,CAAC;IAE9C,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IACjC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CACb,+BAA+B,MAAM,CAAC,QAAQ,0BAA0B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/F,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GAAG,SAAS,CAAC,aAAa,CAAC,CAAC;IACjD,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9E,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,eAAe,GAAG,eAAe;aACpC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;aACX,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;aACxB,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,MAAM,IAAI,KAAK,CACb,SAAS,MAAM,CAAC,KAAK,kCAAkC,MAAM,CAAC,QAAQ,2BAA2B,eAAe,EAAE,CACnH,CAAC;IACJ,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { AgentLoop, AgentLoopConfig, RunOptions } from "../agent-loop.js";
|
|
2
|
+
import type { ResponseEvent, ResponseInput } from "../types.js";
|
|
3
|
+
export declare class PiMonoAgentLoop implements AgentLoop {
|
|
4
|
+
private agent;
|
|
5
|
+
private toolHandler;
|
|
6
|
+
private model;
|
|
7
|
+
init(config: AgentLoopConfig): Promise<void>;
|
|
8
|
+
run(input: ResponseInput, options?: RunOptions): AsyncIterable<ResponseEvent>;
|
|
9
|
+
dispose(): Promise<void>;
|
|
10
|
+
private executeRun;
|
|
11
|
+
}
|
|
@@ -0,0 +1,167 @@
|
|
|
1
|
+
import { randomUUID } from "node:crypto";
|
|
2
|
+
import { Agent } from "@mariozechner/pi-agent-core";
|
|
3
|
+
import { streamSimple } from "@mariozechner/pi-ai";
|
|
4
|
+
import { toResponseEvent } from "./events.js";
|
|
5
|
+
import { getLatestAssistantMessage, toAgentMessages, toAssistantText } from "./input.js";
|
|
6
|
+
import { resolvePiModel } from "./model.js";
|
|
7
|
+
import { AsyncEventQueue } from "./queue.js";
|
|
8
|
+
import { createPiTools } from "./tools.js";
|
|
9
|
+
import { toErrorMessage, toTokenUsage } from "./utils.js";
|
|
10
|
+
// Only Ollama needs a local placeholder key when unauthenticated.
|
|
11
|
+
// Other providers use pi-ai's built-in environment-based key resolution.
|
|
12
|
+
function resolveOllamaApiKey(provider) {
|
|
13
|
+
if (provider !== "ollama") {
|
|
14
|
+
return undefined;
|
|
15
|
+
}
|
|
16
|
+
const configuredKey = process.env.OLLAMA_API_KEY?.trim();
|
|
17
|
+
if (configuredKey && configuredKey.length > 0) {
|
|
18
|
+
return configuredKey;
|
|
19
|
+
}
|
|
20
|
+
return "ollama";
|
|
21
|
+
}
|
|
22
|
+
// Workaround: pi-ai hardcodes store: false in OpenAI Responses API requests,
|
|
23
|
+
// which breaks multi-turn reasoning models (gpt-5.x) because reasoning item IDs
|
|
24
|
+
// (rs_xxx) are not persisted and cannot be referenced in subsequent turns.
|
|
25
|
+
// See: https://github.com/openai/openai-agents-python/issues/919
|
|
26
|
+
// See: https://github.com/badlogic/pi-mono/pull/1504
|
|
27
|
+
const OPENAI_RESPONSES_APIS = new Set([
|
|
28
|
+
"openai-responses",
|
|
29
|
+
"openai-codex-responses",
|
|
30
|
+
"azure-openai-responses",
|
|
31
|
+
]);
|
|
32
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
33
|
+
const storeEnabledStreamFn = (model, context, options) => {
|
|
34
|
+
return streamSimple(model, context, {
|
|
35
|
+
...options,
|
|
36
|
+
onPayload: (params) => {
|
|
37
|
+
if (OPENAI_RESPONSES_APIS.has(model.api)) {
|
|
38
|
+
params.store = true;
|
|
39
|
+
}
|
|
40
|
+
options?.onPayload?.(params);
|
|
41
|
+
},
|
|
42
|
+
});
|
|
43
|
+
};
|
|
44
|
+
export class PiMonoAgentLoop {
|
|
45
|
+
agent = null;
|
|
46
|
+
toolHandler = null;
|
|
47
|
+
model = null;
|
|
48
|
+
async init(config) {
|
|
49
|
+
this.model = resolvePiModel(config.model);
|
|
50
|
+
this.toolHandler = config.toolHandler;
|
|
51
|
+
this.agent = new Agent({
|
|
52
|
+
initialState: {
|
|
53
|
+
systemPrompt: config.systemPrompt,
|
|
54
|
+
model: this.model,
|
|
55
|
+
},
|
|
56
|
+
getApiKey: resolveOllamaApiKey,
|
|
57
|
+
streamFn: storeEnabledStreamFn,
|
|
58
|
+
});
|
|
59
|
+
this.agent.setSystemPrompt(config.systemPrompt);
|
|
60
|
+
this.agent.setTools(createPiTools(config.toolHandler, config.externalTools ?? []));
|
|
61
|
+
}
|
|
62
|
+
async *run(input, options) {
|
|
63
|
+
if (!this.agent || !this.toolHandler || !this.model) {
|
|
64
|
+
throw new Error("PiMonoAgentLoop is not initialized.");
|
|
65
|
+
}
|
|
66
|
+
const agent = this.agent;
|
|
67
|
+
const signal = options?.signal;
|
|
68
|
+
const responseId = `resp-${randomUUID()}`;
|
|
69
|
+
const queue = new AsyncEventQueue();
|
|
70
|
+
const onAbort = signal
|
|
71
|
+
? () => {
|
|
72
|
+
agent.abort();
|
|
73
|
+
}
|
|
74
|
+
: undefined;
|
|
75
|
+
if (onAbort) {
|
|
76
|
+
if (signal.aborted) {
|
|
77
|
+
agent.abort();
|
|
78
|
+
}
|
|
79
|
+
else {
|
|
80
|
+
signal.addEventListener("abort", onAbort, { once: true });
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
const unsubscribe = agent.subscribe((event) => {
|
|
84
|
+
const responseEvent = toResponseEvent(event);
|
|
85
|
+
if (responseEvent) {
|
|
86
|
+
queue.push(responseEvent);
|
|
87
|
+
}
|
|
88
|
+
});
|
|
89
|
+
queue.push({
|
|
90
|
+
type: "response.created",
|
|
91
|
+
responseId,
|
|
92
|
+
});
|
|
93
|
+
const runPromise = this.executeRun(input, responseId, queue)
|
|
94
|
+
.finally(() => {
|
|
95
|
+
if (onAbort) {
|
|
96
|
+
signal.removeEventListener("abort", onAbort);
|
|
97
|
+
}
|
|
98
|
+
unsubscribe();
|
|
99
|
+
queue.close();
|
|
100
|
+
});
|
|
101
|
+
for await (const event of queue) {
|
|
102
|
+
yield event;
|
|
103
|
+
}
|
|
104
|
+
await runPromise;
|
|
105
|
+
}
|
|
106
|
+
async dispose() {
|
|
107
|
+
if (!this.agent) {
|
|
108
|
+
return;
|
|
109
|
+
}
|
|
110
|
+
this.agent.abort();
|
|
111
|
+
await this.agent.waitForIdle();
|
|
112
|
+
this.agent = null;
|
|
113
|
+
this.toolHandler = null;
|
|
114
|
+
this.model = null;
|
|
115
|
+
}
|
|
116
|
+
async executeRun(input, responseId, queue) {
|
|
117
|
+
if (!this.agent || !this.model) {
|
|
118
|
+
queue.push({
|
|
119
|
+
type: "response.error",
|
|
120
|
+
error: "PiMonoAgentLoop is not initialized.",
|
|
121
|
+
});
|
|
122
|
+
return;
|
|
123
|
+
}
|
|
124
|
+
try {
|
|
125
|
+
const messages = toAgentMessages(input, this.model);
|
|
126
|
+
if (messages.length === 0) {
|
|
127
|
+
throw new Error("ResponseInput must include at least one user, assistant, or tool message.");
|
|
128
|
+
}
|
|
129
|
+
this.agent.reset();
|
|
130
|
+
this.agent.replaceMessages(messages);
|
|
131
|
+
await this.agent.continue();
|
|
132
|
+
await this.agent.waitForIdle();
|
|
133
|
+
if (this.agent.state.error) {
|
|
134
|
+
queue.push({
|
|
135
|
+
type: "response.error",
|
|
136
|
+
error: this.agent.state.error,
|
|
137
|
+
});
|
|
138
|
+
return;
|
|
139
|
+
}
|
|
140
|
+
const latestAssistantMessage = getLatestAssistantMessage(this.agent.state.messages);
|
|
141
|
+
const outputText = toAssistantText(latestAssistantMessage);
|
|
142
|
+
const usage = toTokenUsage(latestAssistantMessage?.usage);
|
|
143
|
+
const output = {
|
|
144
|
+
id: responseId,
|
|
145
|
+
output: outputText,
|
|
146
|
+
};
|
|
147
|
+
if (usage) {
|
|
148
|
+
output.usage = usage;
|
|
149
|
+
}
|
|
150
|
+
queue.push({
|
|
151
|
+
type: "response.output_text.done",
|
|
152
|
+
text: outputText,
|
|
153
|
+
});
|
|
154
|
+
queue.push({
|
|
155
|
+
type: "response.completed",
|
|
156
|
+
output,
|
|
157
|
+
});
|
|
158
|
+
}
|
|
159
|
+
catch (error) {
|
|
160
|
+
queue.push({
|
|
161
|
+
type: "response.error",
|
|
162
|
+
error: toErrorMessage(error),
|
|
163
|
+
});
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
//# sourceMappingURL=pi-mono-loop.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pi-mono-loop.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/pi-mono-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,KAAK,EAAE,MAAM,6BAA6B,CAAC;AAEpD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAInD,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,yBAAyB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACzF,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1D,kEAAkE;AAClE,yEAAyE;AACzE,SAAS,mBAAmB,CAAC,QAAgB;IAC3C,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC;IACzD,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9C,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,6EAA6E;AAC7E,gFAAgF;AAChF,2EAA2E;AAC3E,iEAAiE;AACjE,qDAAqD;AACrD,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAC;IACpC,kBAAkB;IAClB,wBAAwB;IACxB,wBAAwB;CACzB,CAAC,CAAC;AAEH,8DAA8D;AAC9D,MAAM,oBAAoB,GAAG,CAAC,KAAU,EAAE,OAAY,EAAE,OAAY,EAAO,EAAE;IAC3E,OAAO,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE;QAClC,GAAG,OAAO;QACV,SAAS,EAAE,CAAC,MAA+B,EAAE,EAAE;YAC7C,IAAI,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,GAAa,CAAC,EAAE,CAAC;gBACnD,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC;YACtB,CAAC;YACD,OAAO,EAAE,SAAS,EAAE,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;KACF,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,OAAO,eAAe;IAClB,KAAK,GAAiB,IAAI,CAAC;IAC3B,WAAW,GAAuB,IAAI,CAAC;IACvC,KAAK,GAAsB,IAAI,CAAC;IAEjC,KAAK,CAAC,IAAI,CAAC,MAAuB;QACvC,IAAI,CAAC,KAAK,GAAG,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;YACrB,YAAY,EAAE;gBACZ,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,KAAK,EAAE,IAAI,CAAC,KAAK;aAClB;YACD,SAAS,EAAE,mBAAmB;YAC9B,QAAQ,EAAE,oBAAoB;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,CAAC;IACrF,CAAC;IAEM,KAAK,CAAC,CAAC,GAAG,CAAC,KAAoB,EAAE,OAAoB;QAC1D,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAC/B,MAAM,UAAU,GAAG,QAAQ,UAAU,EAAE,EAAE,CAAC;QAC1C,MAAM,KAAK,GAAG,IAAI,eAAe,EAAiB,CAAC;QAEnD,MAAM,OAAO,GAAG,MAAM;YACpB,CAAC,CAAC,GAAS,EAAE;gBACT,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,CAAC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,MAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,CAAC;iBAAM,CAAC;gBACN,MAAO,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;YAC5C,MAAM,aAAa,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;YAC7C,IAAI,aAAa,EAAE,CAAC;gBAClB,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,IAAI,CAAC;YACT,IAAI,EAAE,kBAAkB;YACxB,UAAU;SACX,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC;aACzD,OAAO,CAAC,GAAG,EAAE;YACZ,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAO,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAChD,CAAC;YACD,WAAW,EAAE,CAAC;YACd,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,CAAC,CAAC,CAAC;QAEL,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAChC,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,UAAU,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO;QAClB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;QAC/B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,KAAoB,EACpB,UAAkB,EAClB,KAAqC;QAErC,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,gBAAgB;gBACtB,KAAK,EAAE,qCAAqC;aAC7C,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YACpD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC,CAAC;YAC/F,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACnB,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YACrC,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC5B,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAE/B,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC3B,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,gBAAgB;oBACtB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK;iBAC9B,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,sBAAsB,GAAG,yBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACpF,MAAM,UAAU,GAAG,eAAe,CAAC,sBAAsB,CAAC,CAAC;YAC3D,MAAM,KAAK,GAAG,YAAY,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC1D,MAAM,MAAM,GAAmB;gBAC7B,EAAE,EAAE,UAAU;gBACd,MAAM,EAAE,UAAU;aACnB,CAAC;YAEF,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;YACvB,CAAC;YAED,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,2BAA2B;gBACjC,IAAI,EAAE,UAAU;aACjB,CAAC,CAAC;YACH,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,oBAAoB;gBAC1B,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,gBAAgB;gBACtB,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC;aAC7B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
export declare const getModelMock: import("vitest").Mock<(...args: any[]) => any>;
|
|
2
|
+
export declare const getModelsMock: import("vitest").Mock<(...args: any[]) => any>;
|
|
3
|
+
export declare const getProvidersMock: import("vitest").Mock<(...args: any[]) => any>;
|
|
4
|
+
type AgentEventListener = (event: unknown) => void;
|
|
5
|
+
export declare const agentInstances: MockAgent[];
|
|
6
|
+
export declare class MockAgent {
|
|
7
|
+
readonly options: unknown;
|
|
8
|
+
readonly setSystemPrompt: import("vitest").Mock<(prompt: string) => void>;
|
|
9
|
+
readonly setTools: import("vitest").Mock<(tools: unknown[]) => void>;
|
|
10
|
+
readonly replaceMessages: import("vitest").Mock<(messages: unknown[]) => void>;
|
|
11
|
+
readonly reset: import("vitest").Mock<() => void>;
|
|
12
|
+
readonly continue: import("vitest").Mock<() => Promise<undefined>>;
|
|
13
|
+
readonly waitForIdle: import("vitest").Mock<() => Promise<undefined>>;
|
|
14
|
+
readonly abort: import("vitest").Mock<() => undefined>;
|
|
15
|
+
readonly state: {
|
|
16
|
+
systemPrompt: string;
|
|
17
|
+
tools: unknown[];
|
|
18
|
+
messages: unknown[];
|
|
19
|
+
error?: string;
|
|
20
|
+
};
|
|
21
|
+
private listeners;
|
|
22
|
+
constructor(options: unknown);
|
|
23
|
+
subscribe(listener: AgentEventListener): () => void;
|
|
24
|
+
emit(event: unknown): void;
|
|
25
|
+
}
|
|
26
|
+
export declare function importPiMonoLoop(): Promise<typeof import("./pi-mono-loop.js")>;
|
|
27
|
+
export declare function resetPiMocks(): void;
|
|
28
|
+
export {};
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
import { vi } from "vitest";
|
|
2
|
+
export const getModelMock = vi.fn();
|
|
3
|
+
export const getModelsMock = vi.fn();
|
|
4
|
+
export const getProvidersMock = vi.fn();
|
|
5
|
+
export const agentInstances = [];
|
|
6
|
+
export class MockAgent {
|
|
7
|
+
options;
|
|
8
|
+
setSystemPrompt = vi.fn((prompt) => {
|
|
9
|
+
this.state.systemPrompt = prompt;
|
|
10
|
+
});
|
|
11
|
+
setTools = vi.fn((tools) => {
|
|
12
|
+
this.state.tools = tools;
|
|
13
|
+
});
|
|
14
|
+
replaceMessages = vi.fn((messages) => {
|
|
15
|
+
this.state.messages = [...messages];
|
|
16
|
+
});
|
|
17
|
+
reset = vi.fn(() => {
|
|
18
|
+
this.state.messages = [];
|
|
19
|
+
this.state.error = undefined;
|
|
20
|
+
});
|
|
21
|
+
continue = vi.fn(async () => undefined);
|
|
22
|
+
waitForIdle = vi.fn(async () => undefined);
|
|
23
|
+
abort = vi.fn(() => undefined);
|
|
24
|
+
state = {
|
|
25
|
+
systemPrompt: "",
|
|
26
|
+
tools: [],
|
|
27
|
+
messages: [],
|
|
28
|
+
error: undefined,
|
|
29
|
+
};
|
|
30
|
+
listeners = [];
|
|
31
|
+
constructor(options) {
|
|
32
|
+
this.options = options;
|
|
33
|
+
agentInstances.push(this);
|
|
34
|
+
}
|
|
35
|
+
subscribe(listener) {
|
|
36
|
+
this.listeners = [...this.listeners, listener];
|
|
37
|
+
return () => {
|
|
38
|
+
this.listeners = this.listeners.filter((candidate) => candidate !== listener);
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
emit(event) {
|
|
42
|
+
this.listeners.forEach((listener) => {
|
|
43
|
+
listener(event);
|
|
44
|
+
});
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
vi.mock("@mariozechner/pi-agent-core", () => ({
|
|
48
|
+
Agent: MockAgent,
|
|
49
|
+
}));
|
|
50
|
+
vi.mock("@mariozechner/pi-ai", () => ({
|
|
51
|
+
getModel: getModelMock,
|
|
52
|
+
getModels: getModelsMock,
|
|
53
|
+
getProviders: getProvidersMock,
|
|
54
|
+
}));
|
|
55
|
+
export async function importPiMonoLoop() {
|
|
56
|
+
return await import("./pi-mono-loop.js");
|
|
57
|
+
}
|
|
58
|
+
export function resetPiMocks() {
|
|
59
|
+
agentInstances.length = 0;
|
|
60
|
+
getProvidersMock.mockReset();
|
|
61
|
+
getModelsMock.mockReset();
|
|
62
|
+
getModelMock.mockReset();
|
|
63
|
+
getProvidersMock.mockReturnValue(["openai", "google", "openrouter", "anthropic"]);
|
|
64
|
+
getModelsMock.mockImplementation((provider) => {
|
|
65
|
+
if (provider === "google") {
|
|
66
|
+
return [{ id: "gemini-2.0-flash" }];
|
|
67
|
+
}
|
|
68
|
+
return [{ id: "gpt-5-mini" }];
|
|
69
|
+
});
|
|
70
|
+
getModelMock.mockImplementation((provider, model) => ({
|
|
71
|
+
id: model,
|
|
72
|
+
provider,
|
|
73
|
+
api: provider === "google" ? "google-generative-ai" : "openai-responses",
|
|
74
|
+
}));
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=pi-mono-loop.test-helpers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pi-mono-loop.test-helpers.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/pi-mono-loop.test-helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE5B,MAAM,CAAC,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACpC,MAAM,CAAC,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACrC,MAAM,CAAC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAIxC,MAAM,CAAC,MAAM,cAAc,GAAgB,EAAE,CAAC;AAE9C,MAAM,OAAO,SAAS;IAoCe;IAnCnB,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,MAAc,EAAE,EAAE;QACzD,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC;IACnC,CAAC,CAAC,CAAC;IAEa,QAAQ,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,KAAgB,EAAE,EAAE;QACpD,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEa,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,QAAmB,EAAE,EAAE;QAC9D,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEa,KAAK,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;QACjC,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEa,QAAQ,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC;IACxC,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC;IAC3C,KAAK,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAE/B,KAAK,GAKjB;QACA,YAAY,EAAE,EAAE;QAChB,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,EAAE;QACZ,KAAK,EAAE,SAAS;KACjB,CAAC;IAEI,SAAS,GAAyB,EAAE,CAAC;IAE7C,YAAmC,OAAgB;QAAhB,YAAO,GAAP,OAAO,CAAS;QACjD,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEM,SAAS,CAAC,QAA4B;QAC3C,IAAI,CAAC,SAAS,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC/C,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC;QAChF,CAAC,CAAC;IACJ,CAAC;IAEM,IAAI,CAAC,KAAc;QACxB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAClC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,EAAE,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,KAAK,EAAE,SAAS;CACjB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,QAAQ,EAAE,YAAY;IACtB,SAAS,EAAE,aAAa;IACxB,YAAY,EAAE,gBAAgB;CAC/B,CAAC,CAAC,CAAC;AAEJ,MAAM,CAAC,KAAK,UAAU,gBAAgB;IACpC,OAAO,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;IAC1B,gBAAgB,CAAC,SAAS,EAAE,CAAC;IAC7B,aAAa,CAAC,SAAS,EAAE,CAAC;IAC1B,YAAY,CAAC,SAAS,EAAE,CAAC;IAEzB,gBAAgB,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;IAClF,aAAa,CAAC,kBAAkB,CAAC,CAAC,QAAgB,EAAE,EAAE;QACpD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,CAAC,EAAE,EAAE,EAAE,kBAAkB,EAAE,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IACH,YAAY,CAAC,kBAAkB,CAAC,CAAC,QAAgB,EAAE,KAAa,EAAE,EAAE,CAAC,CAAC;QACpE,EAAE,EAAE,KAAK;QACT,QAAQ;QACR,GAAG,EAAE,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,kBAAkB;KACzE,CAAC,CAAC,CAAC;AACN,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export declare class AsyncEventQueue<T> implements AsyncIterable<T>, AsyncIterator<T> {
|
|
2
|
+
private readonly items;
|
|
3
|
+
private readonly resolvers;
|
|
4
|
+
private closed;
|
|
5
|
+
push(item: T): void;
|
|
6
|
+
close(): void;
|
|
7
|
+
next(): Promise<IteratorResult<T>>;
|
|
8
|
+
[Symbol.asyncIterator](): AsyncIterator<T>;
|
|
9
|
+
}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
export class AsyncEventQueue {
|
|
2
|
+
items = [];
|
|
3
|
+
resolvers = [];
|
|
4
|
+
closed = false;
|
|
5
|
+
push(item) {
|
|
6
|
+
const nextResolver = this.resolvers.shift();
|
|
7
|
+
if (nextResolver) {
|
|
8
|
+
nextResolver({ done: false, value: item });
|
|
9
|
+
return;
|
|
10
|
+
}
|
|
11
|
+
this.items.push(item);
|
|
12
|
+
}
|
|
13
|
+
close() {
|
|
14
|
+
this.closed = true;
|
|
15
|
+
let nextResolver = this.resolvers.shift();
|
|
16
|
+
while (nextResolver) {
|
|
17
|
+
nextResolver({ done: true, value: undefined });
|
|
18
|
+
nextResolver = this.resolvers.shift();
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
async next() {
|
|
22
|
+
const nextItem = this.items.shift();
|
|
23
|
+
if (nextItem !== undefined) {
|
|
24
|
+
return { done: false, value: nextItem };
|
|
25
|
+
}
|
|
26
|
+
if (this.closed) {
|
|
27
|
+
return { done: true, value: undefined };
|
|
28
|
+
}
|
|
29
|
+
return await new Promise((resolve) => {
|
|
30
|
+
this.resolvers.push(resolve);
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
[Symbol.asyncIterator]() {
|
|
34
|
+
return this;
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
//# sourceMappingURL=queue.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"queue.js","sourceRoot":"","sources":["../../../src/agent-loop/pi-mono/queue.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,eAAe;IACT,KAAK,GAAQ,EAAE,CAAC;IAChB,SAAS,GAAuB,EAAE,CAAC;IAC5C,MAAM,GAAG,KAAK,CAAC;IAEhB,IAAI,CAAC,IAAO;QACjB,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAC5C,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAC1C,OAAO,YAAY,EAAE,CAAC;YACpB,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/C,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACxC,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,IAAI;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;QAC1C,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC1C,CAAC;QAED,OAAO,MAAM,IAAI,OAAO,CAAoB,CAAC,OAAO,EAAE,EAAE;YACtD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,aAAa,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type { AgentTool } from "@mariozechner/pi-agent-core";
|
|
2
|
+
import type { AgentLoopTool, ToolHandler } from "../agent-loop.js";
|
|
3
|
+
export declare function createPiTools(toolHandler: ToolHandler, externalTools?: readonly AgentLoopTool[]): AgentTool[];
|
|
4
|
+
export declare function toBundleToolCallName(toolName: string): string;
|