gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: adds log usage decorator to main api endpoints

Browse source
This commit is contained in:
hajdul88 2026-01-15 16:04:09 +01:00
parent bc8c6e8bae
commit c0a7b14ff3
3 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/api/v1/add/routers/get_add_router.py
View file

@ -10,6 +10,7 @@ from cognee.modules.users.methods import get_authenticated_user
from cognee.shared.utils import send_telemetry from cognee.shared.utils import send_telemetry
from cognee.modules.pipelines.models import PipelineRunErrored from cognee.modules.pipelines.models import PipelineRunErrored
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
from cognee.shared.usage_logger import log_usage
from cognee import __version__ as cognee_version from cognee import __version__ as cognee_version
logger = get_logger() logger = get_logger()
@ -19,6 +20,7 @@ def get_add_router() -> APIRouter:
router = APIRouter() router = APIRouter()
@router.post("", response_model=dict) @router.post("", response_model=dict)
@log_usage(function_name="POST /v1/add", log_type="api_endpoint")
async def add( async def add(
data: List[UploadFile] = File(default=None), data: List[UploadFile] = File(default=None),
datasetName: Optional[str] = Form(default=None), datasetName: Optional[str] = Form(default=None),

2
cognee/api/v1/cognify/routers/get_cognify_router.py
View file

@ -29,6 +29,7 @@ from cognee.modules.pipelines.queues.pipeline_run_info_queues import (
) )
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
from cognee.shared.utils import send_telemetry from cognee.shared.utils import send_telemetry
from cognee.shared.usage_logger import log_usage
from cognee import __version__ as cognee_version from cognee import __version__ as cognee_version
logger = get_logger("api.cognify") logger = get_logger("api.cognify")
@ -52,6 +53,7 @@ def get_cognify_router() -> APIRouter:
router = APIRouter() router = APIRouter()
@router.post("", response_model=dict) @router.post("", response_model=dict)
@log_usage(function_name="POST /v1/cognify", log_type="api_endpoint")
async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)): async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)):
""" """
Transform datasets into structured knowledge graphs through cognitive processing. Transform datasets into structured knowledge graphs through cognitive processing.

2
cognee/api/v1/search/routers/get_search_router.py
View file

@ -13,6 +13,7 @@ from cognee.modules.users.models import User
from cognee.modules.search.operations import get_history from cognee.modules.search.operations import get_history
from cognee.modules.users.methods import get_authenticated_user from cognee.modules.users.methods import get_authenticated_user
from cognee.shared.utils import send_telemetry from cognee.shared.utils import send_telemetry
from cognee.shared.usage_logger import log_usage
from cognee import __version__ as cognee_version from cognee import __version__ as cognee_version
from cognee.infrastructure.databases.exceptions import DatabaseNotCreatedError from cognee.infrastructure.databases.exceptions import DatabaseNotCreatedError
from cognee.exceptions import CogneeValidationError from cognee.exceptions import CogneeValidationError
@ -75,6 +76,7 @@ def get_search_router() -> APIRouter:
return JSONResponse(status_code=500, content={"error": str(error)}) return JSONResponse(status_code=500, content={"error": str(error)})
@router.post("", response_model=Union[List[SearchResult], List]) @router.post("", response_model=Union[List[SearchResult], List])
@log_usage(function_name="POST /v1/search", log_type="api_endpoint")
async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user)): async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user)):
""" """
Search for nodes in the graph database. Search for nodes in the graph database.