klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Log telemetry for index updates via push to API endpoint

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-10-17 04:09:33 -07:00
parent 05be6bd877
commit e347823ff4
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/khoj/routers/indexer.py
View File

@@ -3,8 +3,9 @@ import logging
from typing import Optional, Union, Dict from typing import Optional, Union, Dict
# External Packages # External Packages
from fastapi import APIRouter, HTTPException, Header, Response, UploadFile from fastapi import APIRouter, HTTPException, Header, Request, Response, UploadFile
from pydantic import BaseModel from pydantic import BaseModel
from khoj.routers.helpers import update_telemetry_state
# Internal Packages # Internal Packages
from khoj.utils import state, constants from khoj.utils import state, constants
@@ -57,10 +58,15 @@ class IndexerInput(BaseModel):
@indexer.post("/batch") @indexer.post("/batch")
async def index_batch( async def index_batch(
request: Request,
files: list[UploadFile], files: list[UploadFile],
x_api_key: str = Header(None), x_api_key: str = Header(None),
regenerate: bool = False, regenerate: bool = False,
search_type: Optional[Union[state.SearchType, str]] = None, search_type: Optional[Union[state.SearchType, str]] = None,
client: Optional[str] = None,
user_agent: Optional[str] = Header(None),
referer: Optional[str] = Header(None),
host: Optional[str] = Header(None),
): ):
if x_api_key != "secret": if x_api_key != "secret":
raise HTTPException(status_code=401, detail="Invalid API Key") raise HTTPException(status_code=401, detail="Invalid API Key")
@@ -135,6 +141,17 @@ async def index_batch(
logger.error(f"🚨 Failed to update content index via API: {e}", exc_info=True) logger.error(f"🚨 Failed to update content index via API: {e}", exc_info=True)
finally: finally:
state.config_lock.release() state.config_lock.release()
update_telemetry_state(
request=request,
telemetry_type="api",
api="index/update",
client=client,
user_agent=user_agent,
referer=referer,
host=host,
)
logger.info("📪 Content index updated via API") logger.info("📪 Content index updated via API")
return Response(content="OK", status_code=200) return Response(content="OK", status_code=200)