diff --git a/plugins/wasm-go/extensions/ai-proxy/main.go b/plugins/wasm-go/extensions/ai-proxy/main.go index ed3789041..99c8641ee 100644 --- a/plugins/wasm-go/extensions/ai-proxy/main.go +++ b/plugins/wasm-go/extensions/ai-proxy/main.go @@ -358,6 +358,9 @@ func getApiName(path string) provider.ApiName { if strings.HasSuffix(path, "/v1/files") { return provider.ApiNameFiles } + if strings.HasSuffix(path, "/v1/models") { + return provider.ApiNameModels + } // cohere style if strings.HasSuffix(path, "/v1/rerank") { return provider.ApiNameCohereV1Rerank diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/moonshot.go b/plugins/wasm-go/extensions/ai-proxy/provider/moonshot.go index bd8cc1bbd..9d8b1995e 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/moonshot.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/moonshot.go @@ -20,6 +20,7 @@ import ( const ( moonshotDomain = "api.moonshot.cn" moonshotChatCompletionPath = "/v1/chat/completions" + moonshotModelsPath = "/v1/models" ) type moonshotProviderInitializer struct { @@ -38,6 +39,7 @@ func (m *moonshotProviderInitializer) ValidateConfig(config *ProviderConfig) err func (m *moonshotProviderInitializer) DefaultCapabilities() map[string]string { return map[string]string{ string(ApiNameChatCompletion): moonshotChatCompletionPath, + string(ApiNameModels): moonshotModelsPath, } } diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/ollama.go b/plugins/wasm-go/extensions/ai-proxy/provider/ollama.go index bfeb442fa..b9756666d 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/ollama.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/ollama.go @@ -30,6 +30,7 @@ func (m *ollamaProviderInitializer) DefaultCapabilities() map[string]string { // ollama的chat接口path和OpenAI的chat接口一样 string(ApiNameChatCompletion): PathOpenAIChatCompletions, string(ApiNameEmbeddings): PathOpenAIEmbeddings, + string(ApiNameModels): PathOpenAIModels, } } diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/openai.go b/plugins/wasm-go/extensions/ai-proxy/provider/openai.go index 8b8d8b133..154be2faa 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/openai.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/openai.go @@ -21,6 +21,7 @@ const ( defaultOpenaiEmbeddingsPath = "/v1/embeddings" defaultOpenaiAudioSpeech = "/v1/audio/speech" defaultOpenaiImageGeneration = "/v1/images/generations" + defaultOpenaiModels = "/v1/models" ) type openaiProviderInitializer struct { @@ -37,6 +38,7 @@ func (m *openaiProviderInitializer) DefaultCapabilities() map[string]string { string(ApiNameEmbeddings): defaultOpenaiEmbeddingsPath, string(ApiNameImageGeneration): defaultOpenaiImageGeneration, string(ApiNameAudioSpeech): defaultOpenaiAudioSpeech, + string(ApiNameModels): defaultOpenaiModels, } } diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/provider.go b/plugins/wasm-go/extensions/ai-proxy/provider/provider.go index 563a09e39..9a0a2bee7 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/provider.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/provider.go @@ -31,12 +31,14 @@ const ( ApiNameAudioSpeech ApiName = "openai/v1/audiospeech" ApiNameFiles ApiName = "openai/v1/files" ApiNameBatches ApiName = "openai/v1/batches" + ApiNameModels ApiName = "openai/v1/models" PathOpenAICompletions = "/v1/completions" PathOpenAIChatCompletions = "/v1/chat/completions" PathOpenAIEmbeddings = "/v1/embeddings" PathOpenAIFiles = "/v1/files" PathOpenAIBatches = "/v1/batches" + PathOpenAIModels = "/v1/models" // TODO: 以下是一些非标准的API名称,需要进一步确认是否支持 ApiNameCohereV1Rerank ApiName = "cohere/v1/rerank"