sqlserver / app /index.js
rrg92's picture
Add hf token
c758355
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`)
})