forked from ctavolazzi/NovaSystem
-
Notifications
You must be signed in to change notification settings - Fork 0
/
stream_ai_response.js
106 lines (86 loc) · 2.66 KB
/
stream_ai_response.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
let responseMessage = '';
function streamToConsole(message) {
for (let i = 0; i < message.length; i++) {
setTimeout(() => {
process.stdout.write(message[i]);
}, 35 * i);
}
}
async function getAIResponse(userInput) {
// Your logic for calling the AI API.
// For now, just return a placeholder response.
return `AI's response: ${userInput}`;
}
function streamAIResponse(userInput) {
getAIResponse(userInput)
.then((response) => {
responseMessage = response;
streamToConsole(responseMessage);
})
.catch((err) => {
console.error(`Error: ${err}`);
});
}
streamAIResponse("Hello, world!");
function streamToHandler(message, outputHandler) {
for (let i = 0; i < message.length; i++) {
setTimeout(() => {
outputHandler(message[i]);
}, 35 * i);
}
}
function consoleOutputHandler(character) {
process.stdout.write(character);
}
// Define more output handlers as needed...
async function getAIResponse(userInput) {
// Your logic for calling the AI API.
// For now, just return a placeholder response.
return `AI's response: ${userInput}`;
}
function streamAIResponse(userInput, outputHandler) {
getAIResponse(userInput)
.then((response) => {
streamToHandler(response, outputHandler);
})
.catch((err) => {
console.error(`Error: ${err}`);
});
}
streamAIResponse("Hello, world!", consoleOutputHandler);
let responseMessage = '';
// The default handler prints to the console
function defaultOutputHandler(character) {
process.stdout.write(character);
}
function streamToHandler(message, outputHandler = defaultOutputHandler) {
for (let i = 0; i < message.length; i++) {
setTimeout(() => {
outputHandler(message[i]);
}, 35 * i);
}
}
async function getAIResponse(userInput) {
// Your logic for calling the AI API.
// For now, just return a placeholder response.
return `AI's response: ${userInput}`;
}
// If no handler is provided, it defaults to the console
function streamAIResponse(userInput, outputHandler) {
getAIResponse(userInput)
.then((response) => {
responseMessage = response;
streamToHandler(responseMessage, outputHandler);
})
.catch((err) => {
console.error(`Error: ${err}`);
});
}
// Default case, streaming to console
streamAIResponse("Hello, world!");
// Custom case, streaming to a custom handler
function customOutputHandler(character) {
// Custom logic here
console.log(`Received character: ${character}`);
}
streamAIResponse("Hello, world!", customOutputHandler);