| // 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 CONTENT_BROWSER_AI_ECHO_AI_REWRITER_H_ |
| #define CONTENT_BROWSER_AI_ECHO_AI_REWRITER_H_ |
| |
| #include "third_party/blink/public/mojom/ai/ai_rewriter.mojom.h" |
| |
| namespace content { |
| |
| // The implementation of `blink::mojom::AIRewriter` which only echoes back |
| // the input text used for testing. |
| class EchoAIRewriter : public blink::mojom::AIRewriter { |
| public: |
| EchoAIRewriter() = default; |
| EchoAIRewriter(const EchoAIRewriter&) = delete; |
| EchoAIRewriter& operator=(const EchoAIRewriter&) = delete; |
| |
| ~EchoAIRewriter() override = default; |
| |
| // `blink::mojom::AIRewriter` implementation. |
| void Rewrite(const std::string& input, |
| const std::optional<std::string>& context, |
| mojo::PendingRemote<blink::mojom::ModelStreamingResponder> |
| pending_responder) override; |
| void MeasureUsage(const std::string& input, |
| const std::string& context, |
| MeasureUsageCallback callback) override; |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_AI_ECHO_AI_REWRITER_H_ |