Added raw response if raw_yn is sent

beckn · Apr 16, 2024 · c37e9bc · c37e9bc
1 parent bdd1323
commit c37e9bc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/controllers/Agent.js b/controllers/Agent.js
@@ -7,7 +7,7 @@ import {
 const db = new DBService();
 
 async function getResponse(req, res) {
-    const { From, Body } = req.body
+    const { From, Body, raw_yn } = req.body
 
     if(!From || !Body){
         res.status(400).send("Bad Request")
@@ -38,14 +38,19 @@ async function getResponse(req, res) {
             ...session.text,
             { role: 'user', content: Body}
         ];
-        const response = await ai.getResponseFromOpenAI(messages)
+        const response = await ai.getResponseFromOpenAI(messages, raw_yn)
+
+        // prepare raw body if required
+        const responseBody = raw_yn ? response.raw : response.content;
+        delete response.raw;
+
         messages.push(response);
         session.text = messages; // Update session text (chat history)
 
         // save session
         await db.update_session(From, session)
 
-        res.send(response.content)
+        res.send(responseBody)
     }
 
 }

diff --git a/services/AI.js b/services/AI.js
@@ -810,7 +810,7 @@ class AI {
         return message;
     }
 
-    async getResponseFromOpenAI(messages){
+    async getResponseFromOpenAI(messages, raw_yn=false){
 
         const context = [
             {role: 'assistant', content : "You are a travel planner ai agent that is capable of performing actions. "},
@@ -852,6 +852,9 @@ class AI {
 
                         // call again to get the response
                         responseMessage = await this.getResponseFromOpenAI(messages);
+                        if(raw_yn) {
+                            responseMessage.raw = response.data;
+                        }
                     }
                 }
             }