From 7cb64cb2f9f801504adb58d7f02c603727b0252e Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Sun, 17 Dec 2023 18:25:03 +0530
Subject: [PATCH] Add telemetry for image generation conversation command

---
 src/khoj/routers/api.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py
index 235b6e7c..065020d3 100644
--- a/src/khoj/routers/api.py
+++ b/src/khoj/routers/api.py
@@ -714,6 +714,13 @@ async def chat(
                 status_code=200,
             )
     elif conversation_command == ConversationCommand.Image:
+        update_telemetry_state(
+            request=request,
+            telemetry_type="api",
+            api="chat",
+            metadata={"conversation_command": conversation_command.value},
+            **common.__dict__,
+        )
         image, status_code = await text_to_image(q)
         if image is None:
             content_obj = {