-
Notifications
You must be signed in to change notification settings - Fork 0
/
chatbot.js
55 lines (44 loc) · 1.26 KB
/
chatbot.js
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
async function query(data) {
const response = await fetch(
"https://api-inference.huggingface.co/models/facebook/blenderbot-3B",
{
headers: { Authorization: "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" },
method: "POST",
body: JSON.stringify(data),
}
);
const result = await response.json();
return result;
}
const { messages, user_text } = inputs
let past_user_inputs = [];
let generated_responses = [];
for (const message of messages){
past_user_inputs.push(message.user_input)
generated_responses.push(message.response)
}
// Prepare Query
const queryInput = {"inputs": {
"past_user_inputs": past_user_inputs,
"generated_responses": generated_responses,
"text": user_text
}
}
let newMessages = [...messages]
newMessages.push({
datetime: moment().format(),
})
const response = await query(queryInput);
console.log(JSON.stringify(response));
let result = []
console.log(response.conversation.generated_responses.length)
for (var i = 0; i < response.conversation.generated_responses.length; i++) {
result.push({
datetime: newMessages[i].datetime,
id: String(i + 1),
response: response.conversation.generated_responses[i],
user_input: response.conversation.past_user_inputs[i],
})
}
console.log("result", result)
return { result };