diff --git a/llm-mock-server/README.md b/llm-mock-server/README.md index f4248b4..ed466df 100644 --- a/llm-mock-server/README.md +++ b/llm-mock-server/README.md @@ -31,3 +31,4 @@ - 文心一言 - 智谱 AI - 阶跃星辰 +- Ollama diff --git a/llm-mock-server/pkg/provider/chat/provider.go b/llm-mock-server/pkg/provider/chat/provider.go index daf2d68..a240939 100644 --- a/llm-mock-server/pkg/provider/chat/provider.go +++ b/llm-mock-server/pkg/provider/chat/provider.go @@ -86,7 +86,7 @@ func SetupRoutes(server *gin.Engine, providerType string) { case "cloudflare": server.POST("/client/v4/accounts/:accountId/ai/v1/chat/completions", chatCompletionsHandlers["cloudflare"].HandleChatCompletions) // 其他 cases... - case "openai", "ai360", "deepseek", "together", "baichuan", "yi", "stepfun": + case "openai", "ai360", "deepseek", "together", "baichuan", "yi", "ollama", "stepfun": // 这些provider都使用OpenAI兼容的格式,调用openAiProvider server.POST("/v1/chat/completions", chatCompletionsHandlers["openai"].HandleChatCompletions) default: