refactor: generate code viewer

This commit is contained in:
jialin
2024-12-09 15:28:18 +08:00
parent a76560c38a
commit acf99a6a9e
20 changed files with 490 additions and 1708 deletions
+122
View File
@@ -0,0 +1,122 @@
import { fomatNodeJsParams, formatCurlArgs, formatPyParams } from './utils';
export const speechToTextCode = ({ api, parameters }: Record<string, any>) => {
const host = window.location.origin;
// ========================= Curl =========================
const curlCode = `
curl ${host}${api} \\
-H "Content-Type: multipart/form-data" \\
-H "Authorization: Bearer $\{YOUR_GPUSTACK_API_KEY}" \\
-F file="@/path/to/file/audio.mp3;type=audio/mpeg" \\
${formatCurlArgs(parameters, true)}`
.trim()
.replace(/\\$/g, '');
// ========================= Python =========================
const pythonCode = `
from openai import OpenAI\n
audio_file = open("audio.mp3", "rb")
client = OpenAI(
base_url="${host}/v1-openai",
api_key="YOUR_GPUSTACK_API_KEY"
)
response = client.audio.transcriptions.create(\n${formatPyParams({
...parameters,
file: 'audio_file'
}).replace(/"audio_file"/, 'audio_file')})\n
print('response:', response.text)`.trim();
// ========================= Node.js =========================
const jsonParams = fomatNodeJsParams({
...parameters,
file: `fs.createReadStream(audio.mp3)`
});
const params = jsonParams.replace(
/"fs.createReadStream\(audio.mp3\)"/g,
'fs.createReadStream("audio.mp3")'
);
const nodeJsCode = `
const fs = require("fs")
const OpenAI = require("openai");
const openai = new OpenAI({
"apiKey": "YOUR_GPUSTACK_API_KEY",
"baseURL": "${host}/v1-openai"
});
async function main() {
const params = ${params};
const response = await openai.audio.transcriptions.create(params);
console.log(response.text);
}
main();`.trim();
return {
curlCode,
pythonCode,
nodeJsCode
};
};
export const TextToSpeechCode = ({ api, parameters }: Record<string, any>) => {
const host = window.location.origin;
// ========================= Curl =========================
const curlCode = `
curl ${host}${api} \\
-H "Content-Type: multipart/form-data" \\
-H "Authorization: Bearer $\{YOUR_GPUSTACK_API_KEY}" \\
${formatCurlArgs(parameters, false)} \\\n--output output.${parameters.response_format}`.trim();
// ========================= Python =========================
const pythonCode = `
from pathlib import Path
from openai import OpenAI\n
output_file_path = Path(__file__).parent / "output.mp3"
client = OpenAI(
base_url="${host}/v1-openai",
api_key="YOUR_GPUSTACK_API_KEY"
)
response = client.audio.speech.create(\n${formatPyParams({ ...parameters })})\n
with open(output_file_path, "wb") as f:
for chunk in response.iter_bytes():
f.write(chunk)
print(f"Audio saved to {output_file_path}")`.trim();
// ========================= Node.js =========================
const params = fomatNodeJsParams({
...parameters
});
const nodeJsCode = `
const fs = require("fs");
const path = require("path");
const OpenAI = require("openai");
const ouptFile = path.resolve("./output.mp3");
const openai = new OpenAI({
"apiKey": "YOUR_GPUSTACK_API_KEY",
"baseURL": "${host}/v1-openai"
});
async function main() {
const params = ${params};
const response = await openai.audio.speech.create(params);
console.log(ouptFile);
const buffer = Buffer.from(await response.arrayBuffer());
await fs.promises.writeFile(ouptFile, buffer);
}
main();`.trim();
return {
curlCode,
pythonCode,
nodeJsCode
};
};