feat: 更新 LiteLLM 客户端和 CLI 设置管理,支持自定义 API 路由和 CLI 工具集成

This commit is contained in:
catlog22
2026-01-12 10:28:42 +08:00
parent 1044886e7d
commit bdd545727b
5 changed files with 133 additions and 27 deletions

View File

@@ -122,17 +122,17 @@ export function saveEndpointSettings(request: SaveEndpointRequest): SettingsOper
saveIndex(index);
// Sync with cli-tools.json for ccw cli --tool integration
// API endpoints are added as tools with type: 'api-endpoint'
// Usage: ccw cli -p "..." --tool custom --model <endpoint-id> --mode analysis
// CLI Settings endpoints are added as tools with type: 'cli-wrapper'
// Usage: ccw cli -p "..." --tool <name> --mode analysis
try {
const projectDir = os.homedir(); // Use home dir as base for global config
addClaudeCustomEndpoint(projectDir, {
id: endpointId,
name: request.name,
enabled: request.enabled ?? true
// No cli-wrapper tag -> registers as type: 'api-endpoint'
enabled: request.enabled ?? true,
tags: ['cli-wrapper'] // cli-wrapper tag -> registers as type: 'cli-wrapper'
});
console.log(`[CliSettings] Synced endpoint ${endpointId} to cli-tools.json tools`);
console.log(`[CliSettings] Synced endpoint ${endpointId} to cli-tools.json tools (cli-wrapper)`);
} catch (syncError) {
console.warn(`[CliSettings] Failed to sync with cli-tools.json: ${syncError}`);
// Non-fatal: continue even if sync fails
@@ -303,14 +303,14 @@ export function toggleEndpointEnabled(endpointId: string, enabled: boolean): Set
index.set(endpointId, metadata);
saveIndex(index);
// Sync enabled status with cli-tools.json tools (api-endpoint type)
// Sync enabled status with cli-tools.json tools (cli-wrapper type)
try {
const projectDir = os.homedir();
addClaudeCustomEndpoint(projectDir, {
id: endpointId,
name: metadata.name,
enabled: enabled
// No cli-wrapper tag -> updates as type: 'api-endpoint'
enabled: enabled,
tags: ['cli-wrapper'] // cli-wrapper tag -> registers as type: 'cli-wrapper'
});
console.log(`[CliSettings] Synced endpoint ${endpointId} enabled=${enabled} to cli-tools.json tools`);
} catch (syncError) {

View File

@@ -18,6 +18,10 @@ import type {
CodexLensEmbeddingProvider,
EmbeddingPoolConfig,
} from '../types/litellm-api-config.js';
import {
addClaudeApiEndpoint,
removeClaudeApiEndpoint
} from '../tools/claude-cli-tools.js';
/**
* Default configuration
@@ -270,6 +274,20 @@ export function addEndpoint(
config.endpoints.push(endpoint);
saveConfig(baseDir, config);
// Sync to cli-tools.json as api-endpoint type
// Usage: ccw cli -p "..." --tool <endpoint-id>
try {
addClaudeApiEndpoint(homedir(), {
id: endpoint.id,
name: endpoint.id, // Use endpoint ID as tool name for CLI access
enabled: endpoint.enabled !== false
});
console.log(`[LiteLLM Config] Synced endpoint ${endpoint.id} to cli-tools.json (api-endpoint)`);
} catch (syncError) {
console.warn(`[LiteLLM Config] Failed to sync endpoint to cli-tools.json: ${syncError}`);
// Non-fatal: continue even if sync fails
}
return endpoint;
}
@@ -300,7 +318,21 @@ export function updateEndpoint(
};
saveConfig(baseDir, config);
return config.endpoints[endpointIndex];
// Sync enabled status to cli-tools.json
const updatedEndpoint = config.endpoints[endpointIndex];
try {
addClaudeApiEndpoint(homedir(), {
id: updatedEndpoint.id,
name: updatedEndpoint.id,
enabled: updatedEndpoint.enabled !== false
});
console.log(`[LiteLLM Config] Synced endpoint ${updatedEndpoint.id} update to cli-tools.json`);
} catch (syncError) {
console.warn(`[LiteLLM Config] Failed to sync endpoint update to cli-tools.json: ${syncError}`);
}
return updatedEndpoint;
}
/**
@@ -322,6 +354,15 @@ export function deleteEndpoint(baseDir: string, endpointId: string): boolean {
}
saveConfig(baseDir, config);
// Remove from cli-tools.json
try {
removeClaudeApiEndpoint(homedir(), endpointId);
console.log(`[LiteLLM Config] Removed endpoint ${endpointId} from cli-tools.json`);
} catch (syncError) {
console.warn(`[LiteLLM Config] Failed to remove endpoint from cli-tools.json: ${syncError}`);
}
return true;
}