diff --git a/src/chat.js b/src/chat.js
index 9cf9f7b..0ef6674 100644
--- a/src/chat.js
+++ b/src/chat.js
@@ -1,6 +1,6 @@
 import { Configuration, OpenAIApi } from "openai";
 import * as dotenv from "dotenv";
-import { getMessages } from "./session";
+import { getMessages } from "./session.js";
 import { Role } from "@prisma/client";
 dotenv.config();
 
@@ -31,8 +31,6 @@ export const chat = async (callId) => {
     },
   ];
 
- 
-
   for (let msg of msgs) {
     messages.push({
       role: convertRole(msg.role),
@@ -40,8 +38,10 @@ export const chat = async (callId) => {
     });
   }
 
-  await openai.createChatCompletion({
+  const res = await openai.createChatCompletion({
     model: "gpt-3.5-turbo",
     messages: messages,
   });
+
+  return res.data.choices[0].message.content
 };
diff --git a/src/test.js b/src/test.js
index 77dded5..4de7fb9 100644
--- a/src/test.js
+++ b/src/test.js
@@ -1,11 +1,9 @@
-import { Conversation } from "./chat.js"
+import { chat } from "./chat.js"
+import { createSession } from "./session.js"
 
 const main = async () => {
-  let convo = new Conversation()
-  let res = await convo.message("I'm just feeling sad.")
-  console.log("RESPONSE", res)
-  res = await convo.message("Yeah. My parents got divorced last week, and I think it's my fault.")
-  console.log("RESPONSE", res)
+  // const session = await createSession("testid", "12345")
+  console.log(await chat("testid"))
 }
 
 main()