mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 17:00:59 +00:00
030788140a
- Remove setApiKey, resolveApiKey, and global apiKeys Map from stream.ts - Rename getApiKey to getApiKeyFromEnv (only checks env vars) - Remove OAuth storage layer (storage.ts deleted) - OAuth login/refresh functions now return credentials instead of saving - getOAuthApiKey/refreshOAuthToken now take credentials as params - Add test/oauth.ts helper for ai package tests - Simplify root npm run check (single biome + tsgo pass) - Remove redundant check scripts from most packages - Add web-ui and coding-agent examples to biome/tsgo includes coding-agent still has compile errors - needs refactoring for new API
99 lines
2.6 KiB
TypeScript
99 lines
2.6 KiB
TypeScript
/**
|
|
* Full Control
|
|
*
|
|
* Replace everything - no discovery, explicit configuration.
|
|
* Still uses OAuth from ~/.pi/agent for convenience.
|
|
*
|
|
* IMPORTANT: When providing `tools` with a custom `cwd`, use the tool factory
|
|
* functions (createReadTool, createBashTool, etc.) to ensure tools resolve
|
|
* paths relative to your cwd.
|
|
*/
|
|
|
|
import { Type } from "@sinclair/typebox";
|
|
import { getAgentDir } from "../../src/config.js";
|
|
import {
|
|
type CustomAgentTool,
|
|
configureOAuthStorage,
|
|
createAgentSession,
|
|
createBashTool,
|
|
createReadTool,
|
|
defaultGetApiKey,
|
|
findModel,
|
|
type HookFactory,
|
|
SessionManager,
|
|
SettingsManager,
|
|
} from "../../src/index.js";
|
|
|
|
// Use OAuth from default location
|
|
configureOAuthStorage(getAgentDir());
|
|
|
|
// Custom API key with fallback
|
|
const getApiKey = async (model: { provider: string }) => {
|
|
if (model.provider === "anthropic" && process.env.MY_ANTHROPIC_KEY) {
|
|
return process.env.MY_ANTHROPIC_KEY;
|
|
}
|
|
return defaultGetApiKey()(model as any);
|
|
};
|
|
|
|
// Inline hook
|
|
const auditHook: HookFactory = (api) => {
|
|
api.on("tool_call", async (event) => {
|
|
console.log(`[Audit] ${event.toolName}`);
|
|
return undefined;
|
|
});
|
|
};
|
|
|
|
// Inline custom tool
|
|
const statusTool: CustomAgentTool = {
|
|
name: "status",
|
|
label: "Status",
|
|
description: "Get system status",
|
|
parameters: Type.Object({}),
|
|
execute: async () => ({
|
|
content: [{ type: "text", text: `Uptime: ${process.uptime()}s, Node: ${process.version}` }],
|
|
details: {},
|
|
}),
|
|
};
|
|
|
|
const { model } = findModel("anthropic", "claude-sonnet-4-20250514");
|
|
if (!model) throw new Error("Model not found");
|
|
|
|
// In-memory settings with overrides
|
|
const settingsManager = SettingsManager.inMemory({
|
|
compaction: { enabled: false },
|
|
retry: { enabled: true, maxRetries: 2 },
|
|
});
|
|
|
|
// When using a custom cwd with explicit tools, use the factory functions
|
|
const cwd = process.cwd();
|
|
|
|
const { session } = await createAgentSession({
|
|
cwd,
|
|
agentDir: "/tmp/my-agent",
|
|
|
|
model,
|
|
thinkingLevel: "off",
|
|
getApiKey,
|
|
|
|
systemPrompt: `You are a minimal assistant.
|
|
Available: read, bash, status. Be concise.`,
|
|
|
|
// Use factory functions with the same cwd to ensure path resolution works correctly
|
|
tools: [createReadTool(cwd), createBashTool(cwd)],
|
|
customTools: [{ tool: statusTool }],
|
|
hooks: [{ factory: auditHook }],
|
|
skills: [],
|
|
contextFiles: [],
|
|
slashCommands: [],
|
|
sessionManager: SessionManager.inMemory(),
|
|
settingsManager,
|
|
});
|
|
|
|
session.subscribe((event) => {
|
|
if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
|
|
process.stdout.write(event.assistantMessageEvent.delta);
|
|
}
|
|
});
|
|
|
|
await session.prompt("Get status and list files.");
|
|
console.log();
|