store embedding meta data on start

This way errors during the a new embedding run will not trigger a clear
index on retry
This commit is contained in:
Andreas Gohr
2024-07-22 12:58:30 +02:00
parent 8502e301c3
commit 22a3672c7d

View File

@ -366,15 +366,14 @@ class cli_plugin_aichat extends AbstractCLI
$clear = true;
}
$data['embed ran at'] = dformat();
$data['embed used'] = (string) $this->helper->getEmbeddingModel();
$this->helper->setRunData($data);
$start = time();
$this->helper->getEmbeddings()->createNewIndex($skipRE, $matchRE, $clear);
$this->notice('Peak memory used: {memory}', ['memory' => filesize_h(memory_get_peak_usage(true))]);
$this->notice('Spent time: {time}min', ['time' => round((time() - $start) / 60, 2)]);
$data['embed ran at'] = dformat();
$data['embed used'] = (string) $this->helper->getEmbeddingModel();
$this->helper->setRunData($data);
}
/**