anthropic_ai_service.h

Go to the documentation of this file.

#ifndef YAZE_SRC_CLI_ANTHROPIC_AI_SERVICE_H_
#define YAZE_SRC_CLI_ANTHROPIC_AI_SERVICE_H_
 
#include <string>
#include <vector>
 
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "cli/service/ai/ai_service.h"
#include "cli/service/ai/provider_ids.h"
 
#ifdef YAZE_AI_RUNTIME_AVAILABLE
#include "cli/service/ai/prompt_builder.h"
#endif
 
namespace yaze {
namespace cli {
 
struct AnthropicConfig {
  std::string api_key;
  std::string model = "claude-3-5-sonnet-20241022";  // Default to 3.5 Sonnet
  float temperature = 0.7f;
  int max_output_tokens = 4096;  // Anthropic supports larger output context
  std::string system_instruction;
  bool use_function_calling = true;
  std::string prompt_version = "v3";
  bool verbose = false;
 
  AnthropicConfig() = default;
  explicit AnthropicConfig(const std::string& key) : api_key(key) {}
};
 
#ifdef YAZE_AI_RUNTIME_AVAILABLE
 
class AnthropicAIService : public AIService {
 public:
  explicit AnthropicAIService(const AnthropicConfig& config);
  void SetRomContext(Rom* rom) override;
 
  // Primary interface
  absl::StatusOr<AgentResponse> GenerateResponse(
      const std::string& prompt) override;
  absl::StatusOr<AgentResponse> GenerateResponse(
      const std::vector<agent::ChatMessage>& history) override;
 
  // Health check
  absl::Status CheckAvailability();
 
  // List available models from Anthropic API (simulated, as list endpoint is sparse)
  absl::StatusOr<std::vector<ModelInfo>> ListAvailableModels() override;
 
  std::string GetProviderName() const override { return kProviderAnthropic; }
 
  // Function calling support
  void EnableFunctionCalling(bool enable = true);
  std::vector<std::string> GetAvailableTools() const;
 
 private:
  std::string BuildSystemInstruction();
  absl::StatusOr<AgentResponse> ParseAnthropicResponse(
      const std::string& response_body);
 
  bool function_calling_enabled_ = true;
 
  AnthropicConfig config_;
  PromptBuilder prompt_builder_;
};
 
#else  // !YAZE_AI_RUNTIME_AVAILABLE
 
// Stub implementation when AI runtime is disabled
class AnthropicAIService : public AIService {
 public:
  explicit AnthropicAIService(const AnthropicConfig&) {}
  void SetRomContext(Rom*) override {}
  absl::StatusOr<AgentResponse> GenerateResponse(
      const std::string& prompt) override {
    return absl::FailedPreconditionError(
        "Anthropic AI runtime is disabled (prompt: " + prompt + ")");
  }
  absl::StatusOr<AgentResponse> GenerateResponse(
      const std::vector<agent::ChatMessage>&) override {
    return absl::FailedPreconditionError("Anthropic AI runtime is disabled");
  }
  absl::Status CheckAvailability() {
    return absl::FailedPreconditionError("Anthropic AI runtime is disabled");
  }
  void EnableFunctionCalling(bool) {}
  std::vector<std::string> GetAvailableTools() const { return {}; }
  absl::StatusOr<std::vector<ModelInfo>> ListAvailableModels() override {
    return absl::FailedPreconditionError("Anthropic AI runtime is disabled");
  }
  std::string GetProviderName() const override { return kProviderAnthropic; }
};
 
#endif  // YAZE_AI_RUNTIME_AVAILABLE
 
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_SRC_CLI_ANTHROPIC_AI_SERVICE_H_