116 lines
3.1 KiB
TypeScript
116 lines
3.1 KiB
TypeScript
import { get, omit } from "radash";
|
|
import { OpenAI } from "openai";
|
|
|
|
export default defineEventHandler(async (event) => {
|
|
const body = await readBody(event);
|
|
const db = useDrizzle();
|
|
const config = useRuntimeConfig();
|
|
|
|
const openai = new OpenAI({
|
|
apiKey: config.openaiApiKey,
|
|
});
|
|
|
|
const user = await db.query.users.findFirst({
|
|
where: (f, o) => o.eq(f.id, body.owner_id),
|
|
with: {
|
|
preferences: true,
|
|
},
|
|
});
|
|
|
|
if (!user?.preferences.data?.enabled) {
|
|
return;
|
|
}
|
|
|
|
const strava = await useStrava(body.owner_id);
|
|
|
|
const [, activity] = await strava!<any>(`/activities/${body.object_id}`);
|
|
|
|
const aiResponse = await openai.chat.completions.create({
|
|
model: "gpt-4o-mini",
|
|
messages: [
|
|
{
|
|
role: "user",
|
|
content: `
|
|
Generate a title and a short description for my strava activity. Use my preferred language.
|
|
Use ${user?.preferences.data.tone} tone to generate content.
|
|
Add emojis unless tone is set to minimalist.
|
|
|
|
My user profile:
|
|
Sex: ${user?.sex}
|
|
Weight: ${user?.weight}
|
|
Language: ${user?.preferences.data.language}
|
|
|
|
Activity notes:
|
|
distance is in meters, time is in seconds, don't include average speed. Convert time to hours or minutes, whatever's closer.
|
|
The activity data in json format from strava:
|
|
${JSON.stringify(
|
|
omit(activity, [
|
|
"laps",
|
|
"segment_efforts",
|
|
"splits_metric",
|
|
"splits_standard",
|
|
"hide_from_home",
|
|
"available_zones",
|
|
"map",
|
|
"start_date_local",
|
|
"gear",
|
|
"stats_visibility",
|
|
"embed_token",
|
|
"name",
|
|
"description",
|
|
]),
|
|
)}
|
|
`,
|
|
},
|
|
],
|
|
tools: [
|
|
{
|
|
type: "function",
|
|
function: {
|
|
name: "generate_strava_meta",
|
|
description: "Generates Strava metadata.",
|
|
parameters: {
|
|
type: "object",
|
|
properties: {
|
|
title: {
|
|
type: "string",
|
|
description: "The title of the activity",
|
|
},
|
|
description: {
|
|
type: "string",
|
|
description: "A short description of the activity",
|
|
},
|
|
},
|
|
required: ["title", "description"],
|
|
},
|
|
},
|
|
},
|
|
],
|
|
tool_choice: {
|
|
type: "function",
|
|
function: {
|
|
name: "generate_strava_meta",
|
|
},
|
|
},
|
|
});
|
|
|
|
const responseObject = JSON.parse(
|
|
get(aiResponse, "choices.0.message.tool_calls.0.function.arguments"),
|
|
) as { title: string; description: string };
|
|
|
|
const [stravaError] = await strava!(`activities/${body.object_id}`, {
|
|
method: "PUT",
|
|
body: {
|
|
name: responseObject.title,
|
|
description: responseObject.description,
|
|
},
|
|
});
|
|
|
|
if (stravaError) {
|
|
throw createError({
|
|
statusCode: 500,
|
|
message: `Strava API: ${stravaError.message}`,
|
|
});
|
|
}
|
|
});
|