feat: Enhance logging by including request payload in generation responses
This commit is contained in:
@@ -230,6 +230,12 @@ class Groq_AI_Ajax_Controller {
|
|||||||
]
|
]
|
||||||
);
|
);
|
||||||
|
|
||||||
|
$logged_parameters = $request_parameters;
|
||||||
|
if ( is_array( $result ) && isset( $result['request_payload'] ) ) {
|
||||||
|
$logged_parameters['http_request'] = $result['request_payload'];
|
||||||
|
unset( $result['request_payload'] );
|
||||||
|
}
|
||||||
|
|
||||||
if ( is_wp_error( $result ) ) {
|
if ( is_wp_error( $result ) ) {
|
||||||
if ( $logger ) {
|
if ( $logger ) {
|
||||||
$logger->log_generation_event(
|
$logger->log_generation_event(
|
||||||
@@ -242,7 +248,7 @@ class Groq_AI_Ajax_Controller {
|
|||||||
'status' => 'error',
|
'status' => 'error',
|
||||||
'error_message' => $result->get_error_message(),
|
'error_message' => $result->get_error_message(),
|
||||||
'post_id' => 0,
|
'post_id' => 0,
|
||||||
'parameters' => $request_parameters,
|
'parameters' => $logged_parameters,
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -272,7 +278,7 @@ class Groq_AI_Ajax_Controller {
|
|||||||
'status' => 'error',
|
'status' => 'error',
|
||||||
'error_message' => $parsed->get_error_message(),
|
'error_message' => $parsed->get_error_message(),
|
||||||
'post_id' => 0,
|
'post_id' => 0,
|
||||||
'parameters' => $request_parameters,
|
'parameters' => $logged_parameters,
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -294,7 +300,7 @@ class Groq_AI_Ajax_Controller {
|
|||||||
'usage' => $response_usage,
|
'usage' => $response_usage,
|
||||||
'status' => 'success',
|
'status' => 'success',
|
||||||
'post_id' => 0,
|
'post_id' => 0,
|
||||||
'parameters' => $request_parameters,
|
'parameters' => $logged_parameters,
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -544,6 +550,12 @@ class Groq_AI_Ajax_Controller {
|
|||||||
]
|
]
|
||||||
);
|
);
|
||||||
|
|
||||||
|
$logged_parameters = $request_parameters;
|
||||||
|
if ( is_array( $result ) && isset( $result['request_payload'] ) ) {
|
||||||
|
$logged_parameters['http_request'] = $result['request_payload'];
|
||||||
|
unset( $result['request_payload'] );
|
||||||
|
}
|
||||||
|
|
||||||
if ( is_wp_error( $result ) ) {
|
if ( is_wp_error( $result ) ) {
|
||||||
$this->plugin->get_generation_logger()->log_generation_event(
|
$this->plugin->get_generation_logger()->log_generation_event(
|
||||||
[
|
[
|
||||||
@@ -557,7 +569,7 @@ class Groq_AI_Ajax_Controller {
|
|||||||
'post_id' => $post_id,
|
'post_id' => $post_id,
|
||||||
'status' => 'error',
|
'status' => 'error',
|
||||||
'error_message' => $result->get_error_message(),
|
'error_message' => $result->get_error_message(),
|
||||||
'parameters' => $request_parameters,
|
'parameters' => $logged_parameters,
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
wp_send_json_error( [ 'message' => $result->get_error_message() ], 500 );
|
wp_send_json_error( [ 'message' => $result->get_error_message() ], 500 );
|
||||||
@@ -583,7 +595,7 @@ class Groq_AI_Ajax_Controller {
|
|||||||
'post_id' => $post_id,
|
'post_id' => $post_id,
|
||||||
'status' => 'error',
|
'status' => 'error',
|
||||||
'error_message' => $response->get_error_message(),
|
'error_message' => $response->get_error_message(),
|
||||||
'parameters' => $request_parameters,
|
'parameters' => $logged_parameters,
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
wp_send_json_error( [ 'message' => $response->get_error_message() ], 500 );
|
wp_send_json_error( [ 'message' => $response->get_error_message() ], 500 );
|
||||||
@@ -598,7 +610,7 @@ class Groq_AI_Ajax_Controller {
|
|||||||
'usage' => $response_usage,
|
'usage' => $response_usage,
|
||||||
'post_id' => $post_id,
|
'post_id' => $post_id,
|
||||||
'status' => 'success',
|
'status' => 'success',
|
||||||
'parameters' => $request_parameters,
|
'parameters' => $logged_parameters,
|
||||||
]
|
]
|
||||||
);
|
);
|
||||||
|
|
||||||
|
|||||||
@@ -100,8 +100,10 @@ abstract class Groq_AI_Abstract_OpenAI_Provider implements Groq_AI_Provider_Inte
|
|||||||
$request_body['response_format'] = $args['response_format'];
|
$request_body['response_format'] = $args['response_format'];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
$endpoint = $this->get_endpoint();
|
||||||
|
|
||||||
$response = wp_remote_post(
|
$response = wp_remote_post(
|
||||||
$this->get_endpoint(),
|
$endpoint,
|
||||||
[
|
[
|
||||||
'headers' => [
|
'headers' => [
|
||||||
'Authorization' => 'Bearer ' . $api_key,
|
'Authorization' => 'Bearer ' . $api_key,
|
||||||
@@ -140,6 +142,10 @@ abstract class Groq_AI_Abstract_OpenAI_Provider implements Groq_AI_Provider_Inte
|
|||||||
'content' => $content,
|
'content' => $content,
|
||||||
'usage' => $usage,
|
'usage' => $usage,
|
||||||
'raw_response' => $body,
|
'raw_response' => $body,
|
||||||
|
'request_payload' => [
|
||||||
|
'url' => $endpoint,
|
||||||
|
'body' => $request_body,
|
||||||
|
],
|
||||||
];
|
];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -231,12 +231,16 @@ class Groq_AI_Provider_Google implements Groq_AI_Provider_Interface {
|
|||||||
$usage['finish_reason'] = $finish_reason;
|
$usage['finish_reason'] = $finish_reason;
|
||||||
}
|
}
|
||||||
|
|
||||||
return [
|
return [
|
||||||
'content' => $content,
|
'content' => $content,
|
||||||
'usage' => $usage,
|
'usage' => $usage,
|
||||||
'raw_response' => $body,
|
'raw_response' => $body,
|
||||||
];
|
'request_payload' => [
|
||||||
}
|
'url' => $endpoint,
|
||||||
|
'body' => $payload,
|
||||||
|
],
|
||||||
|
];
|
||||||
|
}
|
||||||
|
|
||||||
private function build_safety_settings_payload( $settings ) {
|
private function build_safety_settings_payload( $settings ) {
|
||||||
if ( empty( $settings ) || ! is_array( $settings ) ) {
|
if ( empty( $settings ) || ! is_array( $settings ) ) {
|
||||||
|
|||||||
Reference in New Issue
Block a user