mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-16 12:03:23 +00:00
Update coding-agent dependencies to 0.6.0
This commit is contained in:
parent
bf1a7d8571
commit
09564f2940
4 changed files with 22 additions and 64 deletions
|
|
@ -1,5 +1,5 @@
|
|||
import { Agent, ProviderTransport, type ThinkingLevel } from "@mariozechner/pi-agent";
|
||||
import { getModel } from "@mariozechner/pi-ai";
|
||||
import { getModel, type KnownProvider } from "@mariozechner/pi-ai";
|
||||
import { ProcessTerminal, TUI } from "@mariozechner/pi-tui";
|
||||
import chalk from "chalk";
|
||||
import { readFileSync } from "fs";
|
||||
|
|
@ -16,6 +16,17 @@ const __dirname = dirname(__filename);
|
|||
const packageJson = JSON.parse(readFileSync(join(__dirname, "../package.json"), "utf-8"));
|
||||
const VERSION = packageJson.version;
|
||||
|
||||
const envApiKeyMap: Record<KnownProvider, string[]> = {
|
||||
google: ["GEMINI_API_KEY"],
|
||||
openai: ["OPENAI_API_KEY"],
|
||||
anthropic: ["ANTHROPIC_OAUTH_TOKEN", "ANTHROPIC_API_KEY"],
|
||||
xai: ["XAI_API_KEY"],
|
||||
groq: ["GROQ_API_KEY"],
|
||||
cerebras: ["CEREBRAS_API_KEY"],
|
||||
openrouter: ["OPENROUTER_API_KEY"],
|
||||
zai: ["ZAI_API_KEY"],
|
||||
};
|
||||
|
||||
interface Args {
|
||||
provider?: string;
|
||||
model?: string;
|
||||
|
|
@ -235,18 +246,7 @@ export async function main(args: string[]) {
|
|||
return parsed.apiKey;
|
||||
}
|
||||
|
||||
const envVarMap: Record<string, string[]> = {
|
||||
google: ["GEMINI_API_KEY"],
|
||||
openai: ["OPENAI_API_KEY"],
|
||||
anthropic: ["ANTHROPIC_OAUTH_TOKEN", "ANTHROPIC_API_KEY"],
|
||||
xai: ["XAI_API_KEY"],
|
||||
groq: ["GROQ_API_KEY"],
|
||||
cerebras: ["CEREBRAS_API_KEY"],
|
||||
zai: ["ZAI_API_KEY"],
|
||||
moonshotai: ["MOONSHOT_API_KEY"],
|
||||
};
|
||||
|
||||
const envVars = envVarMap[providerName] || [`${providerName.toUpperCase()}_API_KEY`];
|
||||
const envVars = envApiKeyMap[providerName as KnownProvider];
|
||||
|
||||
// Check each environment variable in priority order
|
||||
for (const envVar of envVars) {
|
||||
|
|
@ -262,17 +262,7 @@ export async function main(args: string[]) {
|
|||
// Get initial API key
|
||||
const initialApiKey = getApiKeyForProvider(provider);
|
||||
if (!initialApiKey) {
|
||||
const envVarMap: Record<string, string[]> = {
|
||||
google: ["GEMINI_API_KEY"],
|
||||
openai: ["OPENAI_API_KEY"],
|
||||
anthropic: ["ANTHROPIC_OAUTH_TOKEN", "ANTHROPIC_API_KEY"],
|
||||
xai: ["XAI_API_KEY"],
|
||||
groq: ["GROQ_API_KEY"],
|
||||
cerebras: ["CEREBRAS_API_KEY"],
|
||||
zai: ["ZAI_API_KEY"],
|
||||
moonshotai: ["MOONSHOT_API_KEY"],
|
||||
};
|
||||
const envVars = envVarMap[provider] || [`${provider.toUpperCase()}_API_KEY`];
|
||||
const envVars = envApiKeyMap[provider as KnownProvider];
|
||||
const envVarList = envVars.join(" or ");
|
||||
console.error(chalk.red(`Error: No API key found for provider "${provider}"`));
|
||||
console.error(chalk.dim(`Set ${envVarList} environment variable or use --api-key flag`));
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue