diff --git a/src/Cli/src/Helpers.cs b/src/Cli/src/Helpers.cs index 9d7a2eca..525de965 100644 --- a/src/Cli/src/Helpers.cs +++ b/src/Cli/src/Helpers.cs @@ -107,6 +107,7 @@ public static IChatClient GetChatModel( { Provider.Free or Provider.OpenRouter => new Uri(tryAGI.OpenAI.CustomProviders.OpenRouterBaseUrl), Provider.Requesty => new Uri("https://router.requesty.ai/v1"), + Provider.ApiRoute => new Uri("https://global.api-route.com/v1"), _ => null, }; @@ -115,6 +116,7 @@ public static IChatClient GetChatModel( Provider.OpenAi or null => "OPENAI_API_KEY", Provider.OpenRouter or Provider.Free => "OPENROUTER_API_KEY", Provider.Requesty => "REQUESTY_API_KEY", + Provider.ApiRoute => "API_ROUTE_API_KEY", _ => throw new NotImplementedException(), }; } diff --git a/src/Cli/src/Models/Provider.cs b/src/Cli/src/Models/Provider.cs index ee560002..382ad784 100644 --- a/src/Cli/src/Models/Provider.cs +++ b/src/Cli/src/Models/Provider.cs @@ -5,6 +5,7 @@ internal enum Provider OpenAi, OpenRouter, Requesty, + ApiRoute, Anthropic, Free, -} \ No newline at end of file +} diff --git a/src/Cli/test/LangChain.Cli.IntegrationTests/HelpersTests.cs b/src/Cli/test/LangChain.Cli.IntegrationTests/HelpersTests.cs index 1c9abf04..202d665c 100644 --- a/src/Cli/test/LangChain.Cli.IntegrationTests/HelpersTests.cs +++ b/src/Cli/test/LangChain.Cli.IntegrationTests/HelpersTests.cs @@ -11,4 +11,11 @@ public void Requesty_UsesExpectedEndpointAndApiKeyEnvironmentVariable() Helpers.GetEndpoint(Provider.Requesty).Should().Be(new Uri("https://router.requesty.ai/v1")); Helpers.GetApiKeyEnvironmentVariable(Provider.Requesty).Should().Be("REQUESTY_API_KEY"); } + + [Test] + public void ApiRoute_UsesExpectedEndpointAndApiKeyEnvironmentVariable() + { + Helpers.GetEndpoint(Provider.ApiRoute).Should().Be(new Uri("https://global.api-route.com/v1")); + Helpers.GetApiKeyEnvironmentVariable(Provider.ApiRoute).Should().Be("API_ROUTE_API_KEY"); + } }