Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 25 additions & 8 deletions src/providers/anthropic/anthropic_factory.cpp
Original file line number Diff line number Diff line change
@@ -1,33 +1,50 @@
#include "anthropic_factory.h"

#include "ai/errors.h"
#include "anthropic_client.h"

#include <cstdlib>
#include <memory>
#include <optional>
#include <stdexcept>

namespace ai {
namespace anthropic {

namespace {
constexpr const char* kDefaultBaseUrl = "https://api.anthropic.com";

std::string get_api_key_or_default(const std::string& api_key) {
if (!api_key.empty()) {
return api_key;
}

const char* env_api_key = std::getenv("ANTHROPIC_API_KEY");
if (!env_api_key) {
throw ConfigurationError(
"API key not provided and ANTHROPIC_API_KEY environment variable not "
"set");
}
return env_api_key;
}

std::string get_base_url_or_default(const std::string& base_url) {
return base_url.empty() ? kDefaultBaseUrl : base_url;
}
} // namespace

Client create_client() {
const char* api_key = std::getenv("ANTHROPIC_API_KEY");
if (!api_key) {
throw std::runtime_error("ANTHROPIC_API_KEY environment variable not set");
}
return Client(std::make_unique<AnthropicClient>(api_key, kDefaultBaseUrl));
return Client(std::make_unique<AnthropicClient>(get_api_key_or_default(""),
kDefaultBaseUrl));
}

Client create_client(const std::string& api_key) {
return Client(std::make_unique<AnthropicClient>(api_key, kDefaultBaseUrl));
return Client(std::make_unique<AnthropicClient>(
get_api_key_or_default(api_key), kDefaultBaseUrl));
}

Client create_client(const std::string& api_key, const std::string& base_url) {
return Client(std::make_unique<AnthropicClient>(api_key, base_url));
return Client(std::make_unique<AnthropicClient>(
get_api_key_or_default(api_key), get_base_url_or_default(base_url)));
}

std::optional<Client> try_create_client() {
Expand Down
37 changes: 27 additions & 10 deletions src/providers/openai/openai_factory.cpp
Original file line number Diff line number Diff line change
@@ -1,40 +1,57 @@
#include "openai_factory.h"

#include "ai/errors.h"
#include "openai_client.h"

#include <cstdlib>
#include <memory>
#include <optional>
#include <stdexcept>

namespace ai {
namespace openai {

namespace {
constexpr const char* kDefaultBaseUrl = "https://api.openai.com";

std::string get_api_key_or_default(const std::string& api_key) {
if (!api_key.empty()) {
return api_key;
}

const char* env_api_key = std::getenv("OPENAI_API_KEY");
if (!env_api_key) {
throw ConfigurationError(
"API key not provided and OPENAI_API_KEY environment variable not set");
}
return env_api_key;
}

std::string get_base_url_or_default(const std::string& base_url) {
return base_url.empty() ? kDefaultBaseUrl : base_url;
}
} // namespace

Client create_client() {
const char* api_key = std::getenv("OPENAI_API_KEY");
if (!api_key) {
throw std::runtime_error("OPENAI_API_KEY environment variable not set");
}
return Client(std::make_unique<OpenAIClient>(api_key, kDefaultBaseUrl));
return Client(std::make_unique<OpenAIClient>(get_api_key_or_default(""),
kDefaultBaseUrl));
}

Client create_client(const std::string& api_key) {
return Client(std::make_unique<OpenAIClient>(api_key, kDefaultBaseUrl));
return Client(std::make_unique<OpenAIClient>(get_api_key_or_default(api_key),
kDefaultBaseUrl));
}

Client create_client(const std::string& api_key, const std::string& base_url) {
return Client(std::make_unique<OpenAIClient>(api_key, base_url));
return Client(std::make_unique<OpenAIClient>(
get_api_key_or_default(api_key), get_base_url_or_default(base_url)));
}

Client create_client(const std::string& api_key,
const std::string& base_url,
const retry::RetryConfig& retry_config) {
return Client(
std::make_unique<OpenAIClient>(api_key, base_url, retry_config));
return Client(std::make_unique<OpenAIClient>(
get_api_key_or_default(api_key), get_base_url_or_default(base_url),
retry_config));
}

std::optional<Client> try_create_client() {
Expand Down