feat: add config for how long to store usage

Signed-off-by: Lukas Schaefer <[email protected]>

Author: lukasdotcom

lib/Cron/CleanupQuotaDb.php

@@ -30,14 +30,15 @@ public function __construct(
 	protected function run($argument) {
 		$this->logger->debug('Run cleanup job for OpenAI quota db');
 		$quota = $this->openAiSettingsService->getQuotaPeriod();
-		$days = $quota['length'];
+		$quotaDays = $quota['length'];
 		if ($quota['unit'] == 'month') {
-			$days *= 30;
+			$quotaDays *= 30;
 		}
+		$days = $this->openAiSettingsService->getUsageStorageTime();
 		$this->quotaUsageMapper->cleanupQuotaUsages(
 			// The mimimum period is limited to DEFAULT_QUOTA_PERIOD to not lose
 			// the stored quota usage data below this limit.
-			max($days, Application::DEFAULT_QUOTA_PERIOD)
+			max($quotaDays, $days, Application::DEFAULT_QUOTA_PERIOD)
 		);
 
 	}

lib/Service/OpenAiSettingsService.php

@@ -37,6 +37,7 @@ class OpenAiSettingsService {
 		'llm_extra_params' => 'string',
 		'quota_period' => 'array',
 		'quotas' => 'array',
+		'usage_storage_time' => 'integer',
 		'translation_provider_enabled' => 'boolean',
 		'llm_provider_enabled' => 'boolean',
 		't2i_provider_enabled' => 'boolean',
@@ -310,6 +311,10 @@ public function getQuotas(): array {
 		return $quotas;
 	}
 
+	public function getUsageStorageTime() : int {
+		return $this->appConfig->getValueInt(Application::APP_ID, 'usage_storage_time', Application::DEFAULT_QUOTA_PERIOD, lazy: true);
+	}
+
 	/**
 	 * @return boolean
 	 */
@@ -397,6 +402,7 @@ public function getAdminConfig(): array {
 			// Updated to get quota period
 			'quotas' => $this->getQuotas(),
 			// Get quotas from the config value and return it
+			'usage_storage_time' => $this->getUsageStorageTime(),
 			'translation_provider_enabled' => $this->getTranslationProviderEnabled(),
 			'llm_provider_enabled' => $this->getLlmProviderEnabled(),
 			't2i_provider_enabled' => $this->getT2iProviderEnabled(),
@@ -524,6 +530,15 @@ public function setQuotas(array $quotas): void {
 		$cache->clear(Application::QUOTA_RULES_CACHE_PREFIX);
 	}
 
+	/**
+	 * @param int $usageStorageTime
+	 * @return void
+	 */
+	public function setUsageStorageTime(int $usageStorageTime): void {
+		$usageStorageTime = max(1, $usageStorageTime);
+		$this->appConfig->setValueInt(Application::APP_ID, 'usage_storage_time', $usageStorageTime, lazy: true);
+	}
+
 	/**
 	 * @param string $apiKey
 	 * @return void
@@ -833,6 +848,9 @@ public function setAdminConfig(array $adminConfig): void {
 		if (isset($adminConfig['quotas'])) {
 			$this->setQuotas($adminConfig['quotas']);
 		}
+		if (isset($adminConfig['usage_storage_time'])) {
+			$this->setUsageStorageTime(intval($adminConfig['usage_storage_time']));
+		}
 		if (isset($adminConfig['use_max_completion_tokens_param'])) {
 			$this->setUseMaxCompletionParam($adminConfig['use_max_completion_tokens_param']);
 		}

src/components/AdminSettings.vue

@@ -546,6 +546,15 @@
 						</tr>
 					</tbody>
 				</table>
+				<div class="line">
+					<NcInputField
+						id="openai-api-usage-storage-time"
+						v-model="state.usage_storage_time"
+						class="input"
+						type="number"
+						:label="t('integration_openai', 'Time period (days) for usage storage')"
+						@update:model-value="onInput()" />
+				</div>
 				<div class="line-gap">
 					<NcDateTimePickerNative
 						v-model="quota_usage.start_date"
@@ -907,6 +916,7 @@ export default {
 				quotas: this.state.quotas,
 				tts_voices: this.state.tts_voices,
 				default_tts_voice: this.state.default_tts_voice,
+				usage_storage_time: this.state.usage_storage_time,
 			}
 			await this.saveOptions(values, false)
 		}, 2000),