38 lines
1.7 KiB
TypeScript
38 lines
1.7 KiB
TypeScript
import { GoogleGenAI, Chat } from "@google/genai";
|
||
import { WorkoutSession } from '../types';
|
||
|
||
const MODEL_ID = 'gemini-2.5-flash';
|
||
|
||
export const createFitnessChat = (history: WorkoutSession[]): Chat | null => {
|
||
const apiKey = process.env.API_KEY;
|
||
if (!apiKey) return null;
|
||
|
||
const ai = new GoogleGenAI({ apiKey });
|
||
|
||
// Summarize data to reduce token count while keeping relevant context
|
||
const summary = history.slice(0, 10).map(s => ({
|
||
date: new Date(s.startTime).toLocaleDateString('ru-RU'),
|
||
userWeight: s.userBodyWeight,
|
||
exercises: s.sets.map(set => `${set.exerciseName}: ${set.weight ? set.weight + 'кг' : ''}${set.reps ? ' x ' + set.reps + 'повт' : ''} ${set.distanceMeters ? set.distanceMeters + 'м' : ''}`).join(', ')
|
||
}));
|
||
|
||
const systemInstruction = `
|
||
Ты — опытный и поддерживающий фитнес-тренер.
|
||
Твоя задача — анализировать тренировки пользователя и давать краткие, полезные советы на русском языке.
|
||
|
||
Учитывай вес пользователя (userWeight в json), если он указан, при анализе прогресса в упражнениях с собственным весом.
|
||
|
||
Вот последние 10 тренировок пользователя (в формате JSON):
|
||
${JSON.stringify(summary)}
|
||
|
||
Если пользователь спрашивает о прогрессе, используй эти данные.
|
||
Отвечай емко, мотивирующе. Избегай длинных лекций, если не просили.
|
||
`;
|
||
|
||
return ai.chats.create({
|
||
model: MODEL_ID,
|
||
config: {
|
||
systemInstruction,
|
||
},
|
||
});
|
||
}; |