diff --git a/src/routes/geminiRoutes.js b/src/routes/geminiRoutes.js index 223a66d4..abd4379e 100644 --- a/src/routes/geminiRoutes.js +++ b/src/routes/geminiRoutes.js @@ -568,11 +568,11 @@ router.post('/v1internal\\:countTokens', authenticateApiKey, handleCountTokens); router.post('/v1internal\\:generateContent', authenticateApiKey, handleGenerateContent); router.post('/v1internal\\:streamGenerateContent', authenticateApiKey, handleStreamGenerateContent); -// v1beta 版本的端点 -router.post('/v1beta/models\\:loadCodeAssist', authenticateApiKey, handleLoadCodeAssist); -router.post('/v1beta/models\\:onboardUser', authenticateApiKey, handleOnboardUser); -router.post('/v1beta/models\\:countTokens', authenticateApiKey, handleCountTokens); -router.post('/v1beta/models\\:generateContent', authenticateApiKey, handleGenerateContent); -router.post('/v1beta/models\\:streamGenerateContent', authenticateApiKey, handleStreamGenerateContent); +// v1beta 版本的端点 - 支持动态模型名称 +router.post('/v1beta/models/:modelName\\:loadCodeAssist', authenticateApiKey, handleLoadCodeAssist); +router.post('/v1beta/models/:modelName\\:onboardUser', authenticateApiKey, handleOnboardUser); +router.post('/v1beta/models/:modelName\\:countTokens', authenticateApiKey, handleCountTokens); +router.post('/v1beta/models/:modelName\\:generateContent', authenticateApiKey, handleGenerateContent); +router.post('/v1beta/models/:modelName\\:streamGenerateContent', authenticateApiKey, handleStreamGenerateContent); module.exports = router; \ No newline at end of file