1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
// 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 COMPONENTS_HISTORY_EMBEDDINGS_INTENT_CLASSIFIER_H_
#define COMPONENTS_HISTORY_EMBEDDINGS_INTENT_CLASSIFIER_H_
#include <string>
#include <vector>
#include "base/functional/callback.h"
namespace history_embeddings {
// The status of an intent classification attempt.
enum class ComputeIntentStatus {
// Not yet specified. This status means the intent isn't classified yet.
UNSPECIFIED,
// Intent classified successfully.
SUCCESS,
// The model files required for generation are not available.
MODEL_UNAVAILABLE,
// Failure occurred during model execution.
EXECUTION_FAILURE,
// Model execution cancelled.
EXECUTION_CANCELLED,
};
// Callback type to receive intent classification result.
using ComputeQueryIntentCallback =
base::OnceCallback<void(ComputeIntentStatus status,
bool is_query_answerable)>;
// Base class that hides implementation details for how intents are classified.
class IntentClassifier {
public:
virtual ~IntentClassifier() = default;
// Returns 0 if not ready, and the nonzero model version number when it's
// loaded and ready.
virtual int64_t GetModelVersion() = 0;
// Calls `callback` asynchronously with intent classification result.
virtual void ComputeQueryIntent(std::string query,
ComputeQueryIntentCallback callback) = 0;
};
} // namespace history_embeddings
#endif // COMPONENTS_HISTORY_EMBEDDINGS_INTENT_CLASSIFIER_H_
|
