uncloseai.com/public/languages/javascript/nodejs/uncloseai.js

331 lines
10 KiB
JavaScript

// This is free software for the public good of a permacomputer hosted at
// permacomputer.com, an always-on computer by the people, for the people.
// One which is durable, easy to repair, & distributed like tap water
// for machine learning intelligence.
//
// The permacomputer is community-owned infrastructure optimized around
// four values:
//
// TRUTH First principles, math & science, open source code freely distributed
// FREEDOM Voluntary partnerships, freedom from tyranny & corporate control
// HARMONY Minimal waste, self-renewing systems with diverse thriving connections
// LOVE Be yourself without hurting others, cooperation through natural law
//
// This software contributes to that vision by making machine learning
// accessible to everyone through a free, open, embeddable chat interface.
// Code is seeds to sprout on any abandoned technology.
const https = require('https');
const fs = require('fs');
console.log('=== Node.js AI API Examples (Dynamic Model Discovery) ===\n');
function getJSON(url) {
return new Promise((resolve, reject) => {
const urlObj = new URL(url);
const options = {
hostname: urlObj.hostname,
port: 443,
path: urlObj.pathname,
method: 'GET',
timeout: 10000
};
const req = https.request(options, (res) => {
let body = '';
res.on('data', (chunk) => body += chunk);
res.on('end', () => {
if (res.statusCode === 200) {
resolve(JSON.parse(body));
} else {
reject(new Error(`HTTP ${res.statusCode}`));
}
});
});
req.on('error', reject);
req.on('timeout', () => {
req.destroy();
reject(new Error('Request timeout'));
});
req.end();
});
}
function postJSON(url, data) {
return new Promise((resolve, reject) => {
const jsonData = JSON.stringify(data);
const urlObj = new URL(url);
const options = {
hostname: urlObj.hostname,
port: 443,
path: urlObj.pathname,
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Content-Length': jsonData.length
},
timeout: 30000
};
const req = https.request(options, (res) => {
let body = '';
res.on('data', (chunk) => body += chunk);
res.on('end', () => {
if (res.headers['content-type']?.includes('application/json')) {
resolve(JSON.parse(body));
} else {
resolve(body);
}
});
});
req.on('error', reject);
req.on('timeout', () => {
req.destroy();
reject(new Error('Request timeout'));
});
req.write(jsonData);
req.end();
});
}
function postJSONBinary(url, data) {
return new Promise((resolve, reject) => {
const jsonData = JSON.stringify(data);
const urlObj = new URL(url);
const options = {
hostname: urlObj.hostname,
port: 443,
path: urlObj.pathname,
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Content-Length': jsonData.length
},
timeout: 30000
};
const req = https.request(options, (res) => {
const chunks = [];
res.on('data', (chunk) => chunks.push(chunk));
res.on('end', () => resolve(Buffer.concat(chunks)));
});
req.on('error', reject);
req.on('timeout', () => {
req.destroy();
reject(new Error('Request timeout'));
});
req.write(jsonData);
req.end();
});
}
async function discoverModels() {
console.log('Discovering models from environment variables...');
const models = [];
const ttsEndpoints = [];
// Discover chat/code models from MODEL_ENDPOINT_1..9999
for (let i = 1; i < 10000; i++) {
const endpoint = process.env[`MODEL_ENDPOINT_${i}`];
if (!endpoint) break;
console.log(`Discovering from: ${endpoint}`);
try {
const response = await getJSON(`${endpoint}/models`);
for (const model of response.data || []) {
// Filter out modelperm-* and chatcmpl-* entries
if (model.id.startsWith('modelperm-') || model.id.startsWith('chatcmpl-')) {
continue;
}
models.push({
id: model.id,
endpoint: endpoint,
max_tokens: model.max_model_len || 8192
});
}
} catch (error) {
console.log(` Error: ${error.message}`);
}
}
// Discover TTS endpoints from TTS_ENDPOINT_1..9999
for (let i = 1; i < 10000; i++) {
const endpoint = process.env[`TTS_ENDPOINT_${i}`];
if (!endpoint) break;
console.log(`Discovering TTS from: ${endpoint}`);
ttsEndpoints.push(endpoint);
}
console.log('');
console.log(`Discovered ${models.length} model(s) and ${ttsEndpoints.length} TTS endpoint(s)`);
console.log('');
return { models, ttsEndpoints };
}
async function chatExample(model, systemMsg, userMsg, maxTokens = 100) {
console.log('\n=== Non-Streaming Chat ===');
console.log(`Model: ${model.id}`);
console.log(`Endpoint: ${model.endpoint}`);
console.log('');
const request = {
model: model.id,
messages: [
{ role: 'system', content: systemMsg },
{ role: 'user', content: userMsg }
],
max_tokens: maxTokens
};
try {
const response = await postJSON(`${model.endpoint}/chat/completions`, request);
console.log('Response:');
console.log(response.choices[0].message.content);
} catch (error) {
console.log('Error:', error.message);
}
}
async function chatStreamExample(model, systemMsg, userMsg, maxTokens = 500) {
console.log('\n=== Streaming Chat ===');
console.log(`Model: ${model.id}`);
console.log(`Endpoint: ${model.endpoint}`);
console.log('');
const request = {
model: model.id,
messages: [
{ role: 'system', content: systemMsg },
{ role: 'user', content: userMsg }
],
max_tokens: maxTokens,
stream: true
};
return new Promise((resolve, reject) => {
const jsonData = JSON.stringify(request);
const urlObj = new URL(`${model.endpoint}/chat/completions`);
const options = {
hostname: urlObj.hostname,
port: 443,
path: urlObj.pathname,
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Content-Length': jsonData.length
},
timeout: 60000
};
const req = https.request(options, (res) => {
process.stdout.write('Response: ');
let buffer = '';
res.on('data', (chunk) => {
buffer += chunk.toString();
const lines = buffer.split('\n');
buffer = lines.pop() || '';
for (const line of lines) {
if (line.startsWith('data: ')) {
const data = line.slice(6).trim();
if (data === '[DONE]') {
process.stdout.write('\n');
resolve();
return;
}
try {
const parsed = JSON.parse(data);
if (parsed.choices?.[0]?.delta?.content) {
process.stdout.write(parsed.choices[0].delta.content);
}
} catch {
// Ignore parse errors
}
}
}
});
res.on('end', () => {
process.stdout.write('\n');
resolve();
});
});
req.on('error', reject);
req.on('timeout', () => {
req.destroy();
reject(new Error('Request timeout'));
});
req.write(jsonData);
req.end();
});
}
async function ttsExample(endpoint) {
console.log('');
console.log('---');
console.log('');
console.log('=== TTS Speech Generation Example ===');
console.log(`Endpoint: ${endpoint}`);
console.log('');
const request = {
model: 'tts-1',
voice: 'alloy',
input: 'Hello from Node.js! This is a text to speech example with dynamic model discovery and streaming support.'
};
try {
const audioData = await postJSONBinary(`${endpoint}/audio/speech`, request);
fs.writeFileSync('speech.mp3', audioData);
console.log(`✓ Speech file created: speech.mp3 (${audioData.length} bytes)`);
} catch (error) {
console.log('✗ Error:', error.message);
}
}
async function main() {
const { models, ttsEndpoints } = await discoverModels();
if (models.length === 0) {
console.log('ERROR: No models discovered. Set environment variables:');
console.log(' MODEL_ENDPOINT_1, MODEL_ENDPOINT_2, etc.');
process.exit(1);
}
await chatExample(models[0], 'You are a helpful AI assistant.', 'Explain quantum computing in one sentence.');
const modelIdx = models.length > 1 ? 1 : 0;
await chatStreamExample(models[modelIdx], 'You are a coding assistant.', 'Write a JavaScript function to check if a number is prime', 200);
if (ttsEndpoints.length > 0) {
await ttsExample(ttsEndpoints[0]);
} else {
console.log('\n=== TTS Speech Generation Example ===');
console.log('ERROR: No TTS endpoints available. Set TTS_ENDPOINT_1');
}
console.log('');
console.log('=== Examples Complete ===');
}
main().catch(console.error);