30行代码搭建属于自己的Ai聊天机器人Api接口

const Koa = require('koa');

const Router = require('koa-router');

const bodyParser = require('koa-bodyparser');

const Groq = require("groq-sdk");

const groq = new Groq({

    apiKey: $GROQ_API_KEY // 替换为自己的 API key

});

async function main(messages) {

    const chatCompletion = await getGroqChatCompletion(messages);

    return chatCompletion.choices[0]?.message || null;

}

async function getGroqChatCompletion(body) {

    return groq.chat.completions.create({

        messages:body.messages,

        model: body.model,

        max_tokens:body.maxTokens,

        temperature:body.temperature

    });

}

const app = new Koa();

const router = new Router();

app.use(bodyParser());

router.post('/api/endpoint', async (ctx) => {

  ctx.request.body.messages = ctx.request.body.messages.map(v=>{

    if(v.role == "user"){

        v.content+="  尽量使用中文进行回复"

    }

    return v;

  });

  const res = await main(ctx.request.body);

  ctx.body = { data:res, code: 200, msg: 'success' };

});

app.use(router.routes());

app.listen(3000, () => {

  console.log('Server listening on port 3000');

});