From 3ede919c6668b72425b38b7a9f588e231a089662 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Mon, 15 May 2023 20:58:51 +0800 Subject: [PATCH] Log usage of /search, /chat, /update API endpoints to telemetry server --- src/khoj/routers/api.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py index 6aa72b38..d92da70f 100644 --- a/src/khoj/routers/api.py +++ b/src/khoj/routers/api.py @@ -14,7 +14,7 @@ from khoj.configure import configure_processor, configure_search from khoj.processor.conversation.gpt import converse, extract_questions from khoj.processor.conversation.utils import message_to_log, message_to_prompt from khoj.search_type import image_search, text_search -from khoj.utils.helpers import timer +from khoj.utils.helpers import log_telemetry, timer from khoj.utils.rawconfig import FullConfig, SearchResponse from khoj.utils.state import SearchType from khoj.utils import state, constants @@ -168,6 +168,8 @@ def search( # Cache results state.query_cache[query_cache_key] = results + log_telemetry(telemetry_type="api", api="search", app_config=state.config.app) + return results @@ -191,6 +193,8 @@ def update(t: Optional[SearchType] = None, force: Optional[bool] = False): else: logger.info("📬 Processor reconfigured via API") + log_telemetry(telemetry_type="api", api="update", app_config=state.config.app) + return {"status": "ok", "message": "khoj reloaded"} @@ -251,4 +255,6 @@ def chat(q: Optional[str] = None): conversation_log=meta_log.get("chat", []), ) + log_telemetry(telemetry_type="api", api="chat", app_config=state.config.app) + return {"status": status, "response": gpt_response, "context": compiled_references}