File size: 1,330 Bytes
b1cbef8 c758355 b1cbef8 c758355 b1cbef8 c5addb2 b1cbef8 c758355 c0538ec b1cbef8 |
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 |
import express from 'express';
const app = express()
const port = 8080
import { Client } from "@gradio/client";
let HfToken = process.env.HF_TOKEN;
process.on('uncaughtException', err => log('Exception:', err))
app.use(express.json());
function log(m){
let d = new Date();
let diso = d.toISOString();
console.log(`${diso} ${m}`)
}
let SpaceSql = null;
app.post('/v1/embeddings', async (req, res) => {
console.log(req.body);
let model = req.body.model;
let text = req.body.input;
if(SpaceSql == null){
SpaceSql = await Client.connect("rrg92/sqlserver-lib-assistant", {hf_token: HfToken})
}
let result = await SpaceSql.predict("/embed",{
text: text
}
)
console.log(result)
res.json({
object: "list"
,data: [{
object: "embeddings"
,embedding: JSON.parse(result.data[0])
,index:0
}]
,model
,usage:{
prompt_tokens: 0
,total_tokens: 0
}
})
})
app.get("/test", async (req,res) =>{
})
app.get('/', async (req, res) => {
res.send('on!')
})
app.use(function(err, req, res, next) {
console.log('error!');
console.log(err);
console.error(err.stack);
res.json({error:'Server error, admin must check logs',status:res.status})
});
app.listen(port, () => {
log(`Server started`)
}) |