Allow generic AI actions to expose metadata (e.g., token usage) from providers

I have drafted a module for a custom ECA action - yet untested. Because if you like the idea then it will probably be much better if it is included into your module. Also I would like to hear your considerations.

The code is just an idea, yet:

name: 'AI Chat with Full Data'
type: module
description: 'Provides an ECA Action for AI chat that returns the full API response and token count.'
core_version_requirement: ^10 || ^11
package: 'Artificial Intelligence'
# This dependency ensures this module loads after the modules it uses.
dependencies:
  - ai_integration_eca:ai_integration_eca
  - gemini_provider:gemini_provider

<?php

namespace Drupal\chat_full_data\Plugin\ECA\Action;

use Drupal\eca\Plugin\Action\ActionBase;
use Drupal\gemini_provider\ApiClient;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Provides the "AI Chat with full response" action.
 *
 * @Action(
 * id = "chat_full_data_action",
 * label = @Translation("AI Chat with full response"),
 * category = @Translation("AI"),
 * description = @Translation("This action calls an AI model with a prompt and provides multiple outputs.
 *
 * OUTPUTS:
 * The data from this action is available to subsequent tasks as tokens. The token pattern is [task_id:output_name], where 'task_id' is the unique ID you give this task in the BPMN modeler.
 *
 * For example, if you set this task's ID to 'security_check':
 * - Text answer: [security_check:answer]
 * - Full JSON response: [security_check:full_response]
 * - Token usage count: [security_check:token_usage]
 * "),
 * context_definitions = {
 * "prompt" = @ContextDefinition("string",
 * label = @Translation("Prompt"),
 * description = @Translation("The main text/question for the current turn."),
 * required = TRUE
 * ),
 * "history" = @ContextDefinition("string",
 * label = @Translation("Chat History (Token Name)"),
 * description = @Translation("The token name for a variable containing the JSON conversation history."),
 * required = FALSE
 * ),
 * "model" = @ContextDefinition("string",
 * label = @Translation("Model"),
 * description = @Translation("The specific AI model to use. Overrides the global setting."),
 * required = FALSE
 * ),
 * "config" = @ContextDefinition("string",
 * label = @Translation("Specific Configuration (YAML)"),
 * description = @Translation("YAML for extra settings like temperature or system_prompt."),
 * required = FALSE
 * )
 * }
 * )
 */
class ChatWithFullResponse extends ActionBase {

  /**
   * The Gemini Provider API client.
   *
   * @var \Drupal\gemini_provider\ApiClient
   */
  protected $geminiApiClient;

  /**
   * {@inheritdoc}
   */
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
    $instance = parent::create($container, $configuration, $plugin_id, $plugin_definition);
    $instance->geminiApiClient = $container->get('gemini_provider.api_client');
    return $instance;
  }

  /**
   * {@inheritdoc}
   */
  public function execute() {
    // Get all inputs from the ECA workflow's input mapping.
    $prompt = $this->getContextValue('prompt');
    $history_json = $this->getContextValue('history');
    $model = $this->getContextValue('model');
    $config_yaml = $this->getContextValue('config');

    if ($prompt) {
      // Pass all inputs to the intelligent service from the patched gemini_provider module.
      $result = $this->geminiApiClient->generateText($prompt, $history_json, $model, $config_yaml);

      if ($result) {
        $fullResponse = $this->geminiApiClient->getLastResponse();
        // Provide all outputs for the next tasks in the workflow.
        $this->setProvidedContext('answer', $result['answer']);
        $this->setProvidedContext('full_response', $fullResponse ? json_encode($fullResponse, JSON_PRETTY_PRINT) : NULL);
        $this->setProvidedContext('token_usage', (int) $result['usage']);
      }
    }
  }

  /**
   * {@inheritdoc}
   */
  public function getProvidedContext() {
    return [
      'answer' => @ContextDefinition("string", label: @Translation("Answer")),
      'full_response' => @ContextDefinition("string", label: @Translation("Full Response (JSON)")),
      'token_usage' => @ContextDefinition("integer", label: @Translation("Token Usage")),
    ];
  }

}

Yes I will try.
Can you please suggest to which files or locations you would prefer that in the structure of your module?
Thank you.