diff --git a/src/utils/chatting.py b/src/utils/chatting.py
index de33a00..e5cd638 100644
--- a/src/utils/chatting.py
+++ b/src/utils/chatting.py
@@ -44,7 +44,7 @@ def chatting(args):
         gc.collect()
         
         # Return the result
-        return {"status": "success", "response": response.response}
+        return {"status": "success", "api_key": args["api_key"], "response": response.response}
     else:
         print("invalide api key")