Initial commit: shiro-neko 0.1.0-beta.1
Agentic coding CLI on Bun, Ink, and the AI SDK. Core: streamText loop with SDK-level tool approval so a denied call provably never executes; endpoint fallback for OpenAI reasoning models; retry with backoff. Tools: read/write/edit/glob/grep/bash, path-jailed, gitignore-aware, ripgrep with a JS fallback, binary rejection, live bash streaming. Agents: five variants crossing thinking level with tool restriction; plan and review withhold mutating tools from the model. Extensibility: frontmatter skills with on-demand bodies, plugin host with blocking hooks, MCP stdio and HTTP, read-only subagents. State: durable per-project memory, session task lists, session persistence, compaction that repairs provider-item dependencies. Distribution: five-platform cross-compiled binaries with checksums, install scripts, CI on three operating systems. 404 tests, typecheck clean.
This commit is contained in:
+123
@@ -0,0 +1,123 @@
|
||||
import { createAnthropic } from '@ai-sdk/anthropic';
|
||||
import { createOpenAI } from '@ai-sdk/openai';
|
||||
import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
|
||||
import type { LanguageModel } from 'ai';
|
||||
import { homedir } from 'node:os';
|
||||
import { join } from 'node:path';
|
||||
import { withFallback, type FallbackEvent } from './fallback';
|
||||
import type { McpServerConfig } from './mcp';
|
||||
|
||||
export type ProviderName = 'anthropic' | 'openai';
|
||||
|
||||
export type Config = {
|
||||
provider: ProviderName;
|
||||
model: string;
|
||||
baseURL?: string;
|
||||
apiKey?: string;
|
||||
/** Preset id from providers.ts, kept so /provider can show what is configured. */
|
||||
presetId?: string;
|
||||
/** Retries per model call for transient failures. SDK default is 2. */
|
||||
maxRetries?: number;
|
||||
/** Default agent variant name. */
|
||||
agent?: string;
|
||||
/** Default thinking level. */
|
||||
thinking?: string;
|
||||
/** Plugin names to enable; omit for the default set. */
|
||||
plugins?: string[];
|
||||
mcpServers?: Record<string, McpServerConfig>;
|
||||
};
|
||||
|
||||
const configPath = () => join(process.env['SHIRO_HOME'] ?? homedir(), '.shiro-neko', 'config.json');
|
||||
|
||||
const DEFAULT_MODEL: Record<ProviderName, string> = {
|
||||
anthropic: 'claude-sonnet-4-5',
|
||||
openai: 'gpt-5',
|
||||
};
|
||||
|
||||
const DEFAULT_BASE_URL: Record<ProviderName, string> = {
|
||||
anthropic: 'https://api.anthropic.com/v1',
|
||||
openai: 'https://api.openai.com/v1',
|
||||
};
|
||||
|
||||
/** Env key checked per provider when no explicit apiKey is configured. */
|
||||
const ENV_KEY: Record<ProviderName, string> = {
|
||||
anthropic: 'ANTHROPIC_API_KEY',
|
||||
openai: 'OPENAI_API_KEY',
|
||||
};
|
||||
|
||||
function isProvider(v: unknown): v is ProviderName {
|
||||
return v === 'anthropic' || v === 'openai';
|
||||
}
|
||||
|
||||
/** Raw file contents, without env overlay. Used when rewriting the file. */
|
||||
export async function readConfigFile(): Promise<Partial<Config>> {
|
||||
const f = Bun.file(configPath());
|
||||
if (!(await f.exists())) return {};
|
||||
try {
|
||||
const parsed: unknown = await f.json();
|
||||
return parsed && typeof parsed === 'object' ? (parsed as Partial<Config>) : {};
|
||||
} catch {
|
||||
throw new Error(`${configPath()} is not valid JSON`);
|
||||
}
|
||||
}
|
||||
|
||||
/** Merges patch into the config file, preserving unrelated keys such as mcpServers. */
|
||||
export async function writeConfigFile(patch: Partial<Config>): Promise<string> {
|
||||
const merged = { ...(await readConfigFile()), ...patch };
|
||||
await Bun.write(configPath(), `${JSON.stringify(merged, null, 2)}\n`);
|
||||
return configPath();
|
||||
}
|
||||
|
||||
/** File config, then env overrides. Env wins so `SHIRO_MODEL=x shiro` works. */
|
||||
export async function loadConfig(): Promise<Config> {
|
||||
const file = await readConfigFile();
|
||||
|
||||
const envProvider = process.env['SHIRO_PROVIDER'];
|
||||
const provider = isProvider(envProvider) ? envProvider : isProvider(file.provider) ? file.provider : 'anthropic';
|
||||
|
||||
return {
|
||||
provider,
|
||||
model: process.env['SHIRO_MODEL'] ?? file.model ?? DEFAULT_MODEL[provider],
|
||||
baseURL: process.env['SHIRO_BASE_URL'] ?? file.baseURL ?? DEFAULT_BASE_URL[provider],
|
||||
apiKey: process.env['SHIRO_API_KEY'] ?? file.apiKey ?? process.env[ENV_KEY[provider]],
|
||||
...(file.presetId ? { presetId: file.presetId } : {}),
|
||||
...(file.maxRetries !== undefined ? { maxRetries: file.maxRetries } : {}),
|
||||
...(file.agent ? { agent: file.agent } : {}),
|
||||
...(file.thinking ? { thinking: file.thinking } : {}),
|
||||
...(Array.isArray(file.plugins) ? { plugins: file.plugins } : {}),
|
||||
...(file.mcpServers ? { mcpServers: file.mcpServers } : {}),
|
||||
};
|
||||
}
|
||||
|
||||
export function missingKeyMessage(provider: ProviderName): string {
|
||||
return `No API key for provider "${provider}". Run shiro and use /provider to set one, or set ${ENV_KEY[provider]} / SHIRO_API_KEY, or add "apiKey" to ${configPath()}`;
|
||||
}
|
||||
|
||||
const isOfficialOpenAI = (baseURL: string | undefined) =>
|
||||
!baseURL || /^https:\/\/api\.openai\.com(\/|$)/.test(baseURL);
|
||||
|
||||
/**
|
||||
* Newer OpenAI reasoning models refuse function tools on /v1/chat/completions and
|
||||
* demand /v1/responses. Rather than guess per model id, build both and let
|
||||
* withFallback switch when the endpoint rejects the request shape.
|
||||
*/
|
||||
export function resolveModel(cfg: Config, onFallback?: (e: FallbackEvent) => void): LanguageModel {
|
||||
if (!cfg.apiKey) throw new Error(missingKeyMessage(cfg.provider));
|
||||
|
||||
if (cfg.provider === 'anthropic') {
|
||||
return createAnthropic({ apiKey: cfg.apiKey, baseURL: cfg.baseURL })(cfg.model);
|
||||
}
|
||||
|
||||
const chat = createOpenAICompatible({
|
||||
name: 'openai',
|
||||
apiKey: cfg.apiKey,
|
||||
baseURL: cfg.baseURL ?? DEFAULT_BASE_URL.openai,
|
||||
})(cfg.model);
|
||||
|
||||
if (!isOfficialOpenAI(cfg.baseURL)) return chat;
|
||||
|
||||
const openai = createOpenAI({ apiKey: cfg.apiKey, baseURL: cfg.baseURL });
|
||||
return withFallback([chat, openai.responses(cfg.model)], onFallback);
|
||||
}
|
||||
|
||||
export { configPath, ENV_KEY };
|
||||
Reference in New Issue
Block a user