chromium/chrome/browser/ash/lobster/lobster_client_impl.h

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_ASH_LOBSTER_LOBSTER_CLIENT_IMPL_H_
#define CHROME_BROWSER_ASH_LOBSTER_LOBSTER_CLIENT_IMPL_H_

#include <string>

#include "ash/public/cpp/lobster/lobster_client.h"
#include "ash/public/cpp/lobster/lobster_session.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/ash/lobster/lobster_service.h"
#include "chrome/browser/ash/lobster/lobster_system_state_provider.h"

namespace ash {
struct LobsterSystemState;
}  // namespace ash

class LobsterClientImpl : public ash::LobsterClient {
 public:
  explicit LobsterClientImpl(LobsterService* service);
  ~LobsterClientImpl() override;

  // LobsterClient overrides
  void SetActiveSession(ash::LobsterSession* session) override;
  ash::LobsterSystemState GetSystemState() override;
  void RequestCandidates(const std::string& query,
                         int num_candidates,
                         ash::RequestCandidatesCallback) override;
  void InflateCandidate(uint32_t seed,
                        const std::string& query,
                        ash::InflateCandidateCallback callback) override;
  bool SubmitFeedback(const std::string& query,
                      const std::string& model_version,
                      const std::string& description,
                      const std::string& image_bytes) override;

 private:
  // Not owned by this class
  raw_ptr<LobsterService> service_;
};

#endif  // CHROME_BROWSER_ASH_LOBSTER_LOBSTER_CLIENT_IMPL_H_