mirror of
https://github.com/vrana/adminer.git
synced 2025-12-27 10:51:01 +01:00
95 lines
3.4 KiB
PHP
95 lines
3.4 KiB
PHP
<?php
|
|
|
|
/** AI prompt in SQL command generating the queries with Google Gemini
|
|
* Beware that this sends your whole database structure (not data) to Google Gemini.
|
|
* @link https://www.adminer.org/static/plugins/sql-gemini.gif
|
|
* @link https://gemini.google.com/
|
|
* @link https://www.adminer.org/plugins/#use
|
|
* @author Jakub Vrana, https://www.vrana.cz/
|
|
* @license https://www.apache.org/licenses/LICENSE-2.0 Apache License, Version 2.0
|
|
* @license https://www.gnu.org/licenses/gpl-2.0.html GNU General Public License, version 2 (one or other)
|
|
*/
|
|
class AdminerSqlGemini {
|
|
private $apiKey;
|
|
private $model;
|
|
|
|
/**
|
|
* @param string $apiKey The default key is shared with all users and may run out of quota; get your own API key at: https://aistudio.google.com/apikey
|
|
* @param string $model Available models: https://ai.google.dev/gemini-api/docs/models#available-models
|
|
*/
|
|
function __construct(string $apiKey = 'AIzaSyDWDbPjmvH9_hphsnY_yJGdue42qRMG3do', string $model = "gemini-2.0-flash") {
|
|
$this->apiKey = $apiKey;
|
|
$this->model = $model;
|
|
}
|
|
|
|
function headers() {
|
|
if (isset($_POST["gemini"]) && !isset($_POST["query"])) {
|
|
$prompt = "I have a " . Adminer\get_driver(Adminer\DRIVER) . " database with this structure:\n\n";
|
|
foreach (Adminer\tables_list() as $table => $type) {
|
|
$prompt .= Adminer\create_sql($table, false, "CREATE") . ";\n\n";
|
|
}
|
|
$prompt .= "Prefer returning relevant columns including primary key.\n\n";
|
|
$prompt .= "Give me this SQL query and nothing else:\n\n$_POST[gemini]\n\n";
|
|
//~ echo $prompt; exit;
|
|
$context = stream_context_create(array("http" => array(
|
|
"method" => "POST",
|
|
"header" => array("User-Agent: AdminerSqlGemini", "Content-Type: application/json"),
|
|
"content" => '{"contents": [{"parts":[{"text": ' . json_encode($prompt) . '}]}]}',
|
|
"ignore_errors" => true,
|
|
)));
|
|
$response = json_decode(file_get_contents("https://generativelanguage.googleapis.com/v1beta/models/$this->model:generateContent?key=$this->apiKey", false, $context));
|
|
if (isset($response->error)) {
|
|
echo "-- " . $response->error->message;
|
|
} else {
|
|
$text = $response->candidates[0]->content->parts[0]->text;
|
|
$text = preg_replace('~(\n|^)```sql\n(.+)\n```(\n|$)~sU', "*/\n\n\\2\n\n/*", "/*\n$text*/\n");
|
|
echo preg_replace('~/\*\s*\*/\n*~', '', $text);
|
|
}
|
|
exit;
|
|
}
|
|
}
|
|
|
|
function sqlPrintAfter() {
|
|
echo "<p><textarea name='gemini' rows='5' cols='50' placeholder='Ask Gemini'>" . Adminer\h($_POST["gemini"]) . "</textarea>\n";
|
|
?>
|
|
<p><input type='button' value='Gemini'>
|
|
<script <?php echo Adminer\nonce(); ?>>
|
|
const geminiText = qsl('textarea');
|
|
const geminiButton = qsl('input');
|
|
|
|
function setSqlareaValue(value) {
|
|
const sqlarea = qs('textarea.sqlarea');
|
|
sqlarea.value = value;
|
|
sqlarea.onchange && sqlarea.onchange();
|
|
}
|
|
|
|
geminiButton.onclick = () => {
|
|
setSqlareaValue('-- Just a sec...'); // this is the phrase used by Google Gemini
|
|
ajax(
|
|
'',
|
|
req => setSqlareaValue(req.responseText),
|
|
'gemini=' + encodeURIComponent(geminiText.value)
|
|
);
|
|
};
|
|
|
|
geminiText.onfocus = event => {
|
|
alterClass(findDefaultSubmit(geminiText), 'default');
|
|
alterClass(geminiButton, 'default', true);
|
|
event.stopImmediatePropagation();
|
|
};
|
|
|
|
geminiText.onblur = () => {
|
|
alterClass(geminiButton, 'default');
|
|
};
|
|
|
|
geminiText.onkeydown = event => {
|
|
if (isCtrl(event) && (event.keyCode == 13 || event.keyCode == 10)) {
|
|
geminiButton.onclick();
|
|
event.stopPropagation();
|
|
}
|
|
};
|
|
</script>
|
|
<?php
|
|
}
|
|
}
|