sjs/Peekaboo
1
0
Fork 0
mirror of https://github.com/samsonjs/Peekaboo.git synced 2026-04-27 15:07:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Report analyze time

Browse source
This commit is contained in:
Peter Steinberger 2025-05-25 02:26:45 +02:00
parent acca3ba4a4
commit b5f1fa7ded
1 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/tools/analyze.ts
View file

@ -14,7 +14,7 @@ export const analyzeToolSchema = z.object({
question: z.string().describe("Required. Question for the AI about the image."), question: z.string().describe("Required. Question for the AI about the image."),
provider_config: z.object({ provider_config: z.object({
type: z.enum(["auto", "ollama", "openai"]).default("auto") type: z.enum(["auto", "ollama", "openai"]).default("auto")
.describe("AI provider. 'auto' uses server's PEEKABOO_AI_PROVIDERS ENV preference. Specific provider must be enabled in server's PEEKABOO_AI_PROVIDERS."), .describe("AI provider, default: auto. 'auto' uses server's PEEKABOO_AI_PROVIDERS environment preference. Specific provider must be enabled in server's PEEKABOO_AI_PROVIDERS."),
model: z.string().optional().describe("Optional. Model name. If omitted, uses model from server's PEEKABOO_AI_PROVIDERS for chosen provider, or an internal default for that provider.") model: z.string().optional().describe("Optional. Model name. If omitted, uses model from server's PEEKABOO_AI_PROVIDERS for chosen provider, or an internal default for that provider.")
}).optional().describe("Optional. Explicit provider/model. Validated against server's PEEKABOO_AI_PROVIDERS.") }).optional().describe("Optional. Explicit provider/model. Validated against server's PEEKABOO_AI_PROVIDERS.")
}); });
@ -101,6 +101,7 @@ export async function analyzeToolHandler(
// Analyze image // Analyze image
let analysisResult: string; let analysisResult: string;
const startTime = Date.now(); // Record start time
try { try {
analysisResult = await analyzeImageWithProvider( analysisResult = await analyzeImageWithProvider(
{ provider, model }, { provider, model },
@ -123,11 +124,23 @@ export async function analyzeToolHandler(
}; };
} }
const endTime = Date.now(); // Record end time
const durationMs = endTime - startTime;
const durationSeconds = (durationMs / 1000).toFixed(2);
const analysisTimeMessage = `👻 Peekaboo: Analyzed image with ${provider}/${model} in ${durationSeconds}s.`;
return { return {
content: [{ content: [
type: 'text', {
text: analysisResult type: 'text',
}], text: analysisResult
},
{
type: 'text',
text: analysisTimeMessage // Add the timing message
}
],
analysis_text: analysisResult, analysis_text: analysisResult,
model_used: `${provider}/${model}` model_used: `${provider}/${model}`
}; };