From ecdd45b638b046edfb275a653c35970b8f93a70a Mon Sep 17 00:00:00 2001 From: Rushil Umaretiya Date: Sat, 4 Mar 2023 21:56:03 -0500 Subject: [PATCH 1/2] chore: chatjs formatting --- src/chat.js | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/chat.js b/src/chat.js index 16b8699..9202b78 100644 --- a/src/chat.js +++ b/src/chat.js @@ -11,15 +11,15 @@ const configuration = new Configuration({ const openai = new OpenAIApi(configuration); const convertRole = (role) => { - switch(role){ + switch (role) { case Role.BOT: - return "assistant" + return "assistant"; case Role.USER: - return "user" + return "user"; default: - return "user" + return "user"; } -} +}; export const chat = async (callId) => { const msgs = await getMessages(callId); @@ -38,12 +38,12 @@ export const chat = async (callId) => { }); } - console.log(messages) + console.log(messages); const res = await openai.createChatCompletion({ model: "gpt-3.5-turbo", messages: messages, }); - return res.data.choices[0].message.content + return res.data.choices[0].message.content; }; From b18eba70df199c985ef248f697427bb0f799cf34 Mon Sep 17 00:00:00 2001 From: Rushil Umaretiya Date: Sun, 5 Mar 2023 02:03:02 -0500 Subject: [PATCH 2/2] feat: added summarize, finalized api, and cleaned up code --- package.json | 1 + prisma/schema.prisma | 2 + src/call.js | 92 ++++++++++++++++++++++++++++++++++++-------- src/chat.js | 41 ++++++++++++++++++-- src/session.js | 31 +++++++++++++++ yarn.lock | 5 +++ 6 files changed, 151 insertions(+), 21 deletions(-) diff --git a/package.json b/package.json index 448bde6..3c16171 100644 --- a/package.json +++ b/package.json @@ -13,6 +13,7 @@ "dependencies": { "@prisma/client": "^4.11.0", "express": "^4.18.2", + "loglevel": "^1.8.1", "openai": "^3.1.0", "twilio": "^4.8.0" }, diff --git a/prisma/schema.prisma b/prisma/schema.prisma index 1aa7ba9..7d88c8c 100644 --- a/prisma/schema.prisma +++ b/prisma/schema.prisma @@ -19,6 +19,8 @@ model Session { callId String @unique callerPhone String + + summary String? operatorPhone String? operator Operator? @relation(fields: [operatorPhone], references: [phoneNumber]) diff --git a/src/call.js b/src/call.js index 921f8c5..da6ad27 100644 --- a/src/call.js +++ b/src/call.js @@ -1,60 +1,118 @@ import * as dotenv from "dotenv"; -import { Router } from "express"; dotenv.config(); -import twilio from "twilio"; +import { Router, json } from "express"; +import log from "loglevel"; +import twilio from "twilio"; const accountSid = process.env.TWILIO_ACCOUNT_SID; const authToken = process.env.TWILIO_AUTH_TOKEN; const phone_number = process.env.TWILIO_PHONE_NUMBER; - -const client = new twilio.Twilio(accountSid, authToken); const VoiceResponse = twilio.twiml.VoiceResponse; -const app = Router(); +import { Role } from "@prisma/client"; -app.post("/receive", (req, res) => { - // Use the Twilio Node.js SDK to build an XML response +import { + addMessage, + createSession, + findSessionByCallId, + transferSession, + checkOperatorReady, + postSummary, +} from "./session.js"; +import { chat, summarize } from "./chat.js"; + +const app = Router(); +app.use(json()); + +app.post("/receive", async (req, res) => { + let callId = req.body.CallSid; + log.info(`CallSid: ${callId}`); const twiml = new VoiceResponse(); - // Use to record and transcribe the caller's message + await createSession(req.body.CallSid, req.body.From); + const gather = twiml.gather({ - action: "/respond", + action: "/call/respond", method: "POST", input: "speech", language: "en-US", speechTimeout: "auto", + model: "experimental_conversations", }); - gather.say("Tell us what makes you sad."); + gather.say( + "Welcome to the suicide hotline. Tell me your name and what's going on?." + ); - twiml.redirect("/call"); + log.info(twiml.toString()); + + twiml.redirect("/call/receive"); res.type("text/xml"); res.send(twiml.toString()); }); app.post("/respond", async (req, res) => { + let callId = req.body.CallSid; + log.info(`CallSid: ${callId}`); const twiml = new VoiceResponse(); let transcription = req.body.SpeechResult; - console.log(transcription); + await addMessage(callId, Role.USER, transcription); - twiml.say(`You said, ${transcription}`); - twiml.say(`Don't kill yourself.`); + let operatorReady = await checkOperatorReady(callId); + + if (operatorReady) { + let session = await findSessionByCallId(callId); + let operatorPhone = session.operatorPhone; + log.info(`transferring call ${callId} to ${operatorPhone}`); + + transferSession(req.body.CallSid, operatorPhone); + twiml.say("We're connecting you to a counselor now."); + + await addMessage(callId, Role.BOT, ""); + let summary = await summarize(callId); + log.info(summary); + + const dial = twiml.dial({}); + dial.number(operatorPhone); + res.type("text/xml"); + res.send(twiml.toString()); + return; + } const gather = twiml.gather({ - action: "/respond", + action: "/call/respond", method: "POST", input: "speech", language: "en-US", speechTimeout: "auto", + model: "experimental_conversations", }); - gather.say("Tell us what makes you sad."); + let response = await chat(callId); + + gather.say(response); + await addMessage(callId, Role.BOT, response); + + twiml.redirect("/call/respond"); res.type("text/xml"); res.send(twiml.toString()); }); -export default app \ No newline at end of file +app.post("/summarize", async (req, res) => { + let sessionId = req.body.SessionId; + log.info(`summarizing ${sessionId}`); + + let summary = await summarize(sessionId); + log.info(summary); + + await postSummary(sessionId, summary); + + res.type("application/json"); + res.send(JSON.stringify({ SessionId: sessionId, summary: summary })); +}); + +export default app; diff --git a/src/chat.js b/src/chat.js index 9202b78..1f390fe 100644 --- a/src/chat.js +++ b/src/chat.js @@ -1,15 +1,17 @@ -import { Configuration, OpenAIApi } from "openai"; import * as dotenv from "dotenv"; -import { getMessages } from "./session.js"; -import { Role } from "@prisma/client"; dotenv.config(); +import { Configuration, OpenAIApi } from "openai"; const configuration = new Configuration({ apiKey: process.env.OPENAI_API_KEY, }); const openai = new OpenAIApi(configuration); +import { Role } from "@prisma/client"; + +import { getMessages, getMessagesBySession } from "./session.js"; + const convertRole = (role) => { switch (role) { case Role.BOT: @@ -27,7 +29,7 @@ export const chat = async (callId) => { { role: "system", content: - "ChatGPT, for the following conversation, please pretend to be a therapist working at a suicide. Respond as if I've called you.", + "ChatGPT, for the following conversation, please pretend to be a therapist working at a suicide prevention hotline. Respond as if I've called you. Limit your responses to 20 seconds, and don't recommend that they seek other help. Make sure you continue the conversation by ending every response with a question.", }, ]; @@ -47,3 +49,34 @@ export const chat = async (callId) => { return res.data.choices[0].message.content; }; + +export const summarize = async (sessionId) => { + const msgs = await getMessagesBySession(sessionId); + const context = [ + { + role: "system", + content: + "This is a conversation between ChatGPT and someone calling the suicide prevention hotline. Please summarize the main issues that the caller is facing and highlight important points in the conversation. Keep your summary short.", + }, + ]; + + for (let msg of msgs) { + context.push({ + role: convertRole(msg.role), + content: msg.content, + }); + } + + context.push({ + role: convertRole(Role.USER), + content: + "This conversation is now being shown to a mental health professional. Please summarize the main issues that the caller is facing and highlight important points in the conversation. Keep your summary short.", + }); + + const res = await openai.createChatCompletion({ + model: "gpt-3.5-turbo", + messages: context, + }); + + return res.data.choices[0].message.content; +}; diff --git a/src/session.js b/src/session.js index 11c1c61..832bc4f 100644 --- a/src/session.js +++ b/src/session.js @@ -30,6 +30,22 @@ export const transferSession = async (callId, operatorPhone) => { }); }; +export const postSummary = async (sessionId, summary) => { + return await db.session.update({ + where: { + id: sessionId, + }, + data: { + summary: summary, + }, + }); +}; + +export const checkOperatorReady = async (callId) => { + const session = await findSessionByCallId(callId); + return session.operatorPhone != null; +}; + export const endSession = async (callId) => { return await db.session.update({ where: { @@ -66,3 +82,18 @@ export const getMessages = async (callId) => { ], }); }; + +export const getMessagesBySession = async (sessionId) => { + return await db.message.findMany({ + where: { + session: { + id: sessionId, + }, + }, + orderBy: [ + { + createdAt: "asc", + }, + ], + }); +}; diff --git a/yarn.lock b/yarn.lock index 8aa6a6f..58322fb 100644 --- a/yarn.lock +++ b/yarn.lock @@ -350,6 +350,11 @@ lodash@^4.17.21: resolved "https://registry.yarnpkg.com/lodash/-/lodash-4.17.21.tgz#679591c564c3bffaae8454cf0b3df370c3d6911c" integrity sha512-v2kDEe57lecTulaDIuNTPy3Ry4gLGJ6Z1O3vE1krgXZNrsQ+LFTGHVxVjcXPs17LhbZVGedAJv8XZ1tvj5FvSg== +loglevel@^1.8.1: + version "1.8.1" + resolved "https://registry.yarnpkg.com/loglevel/-/loglevel-1.8.1.tgz#5c621f83d5b48c54ae93b6156353f555963377b4" + integrity sha512-tCRIJM51SHjAayKwC+QAg8hT8vg6z7GSgLJKGvzuPb1Wc+hLzqtuVLxp6/HzSPOozuK+8ErAhy7U/sVzw8Dgfg== + lru-cache@^6.0.0: version "6.0.0" resolved "https://registry.yarnpkg.com/lru-cache/-/lru-cache-6.0.0.tgz#6d6fe6570ebd96aaf90fcad1dafa3b2566db3a94"