diff --git a/clarifai/runners/models/model_runner.py b/clarifai/runners/models/model_runner.py index 9f75756a..47e13919 100644 --- a/clarifai/runners/models/model_runner.py +++ b/clarifai/runners/models/model_runner.py @@ -132,8 +132,8 @@ def runner_item_predict( start_time = time.time() req_id = get_req_id_from_context() status_str = STATUS_UNKNOWN - # Endpoint is always POST /v2/.../outputs for this runner - endpoint = "POST /v2/.../outputs " + # Operation name for logging purposes + endpoint = "model_predict" # if method_name == '_GET_SIGNATURES' then the request is for getting signatures and we don't want to log it. # This is a workaround to avoid logging the _GET_SIGNATURES method call. @@ -204,7 +204,7 @@ def runner_item_generate( start_time = time.time() req_id = get_req_id_from_context() status_str = STATUS_UNKNOWN - endpoint = "POST /v2/.../outputs/generate" + endpoint = "model_generate" # Use req_secrets_context to temporarily set request-type secrets as environment variables with req_secrets_context(request): @@ -260,7 +260,7 @@ def runner_item_stream( start_time = time.time() req_id = get_req_id_from_context() status_str = STATUS_UNKNOWN - endpoint = "POST /v2/.../outputs/stream " + endpoint = "model_stream" # Get the first request to establish secrets context first_request = None