pi-requesty-discover 1.1.0 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-requesty-discover",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Discover available Requesty models for the Pi Coding Agent",
   "repository": {
     "type": "git",

package/src/health-check.ts

@@ -2,6 +2,7 @@ import type { ProviderModelConfig } from '@earendil-works/pi-coding-agent'
 import fs from 'node:fs'
 import path from 'node:path'
 import { getEnv, type Env } from './env'
+import { formatModelsDiffSummary, type ModelsDiff } from './models-json'
 
 const HEALTH_CHECK_CONCURRENCY = 10
 const HEALTH_CHECK_TIMEOUT_MS = 15_000
@@ -98,7 +99,12 @@ export function formatHealthSummary(results: HealthCheckResult[]): string {
   return `Health check: ${passed.length} OK, ${failed.length} failed:\n${failedModels}\n`
 }
 
-export function writeHealthCheckLog(provider: Provider, results: HealthCheckResult[], envConfig: Env = getEnv()): void {
+export function writeHealthCheckLog(
+  provider: Provider,
+  results: HealthCheckResult[],
+  diff: ModelsDiff,
+  envConfig: Env = getEnv(),
+): void {
   const passed = results.filter(r => r.ok)
   const failed = results.filter(r => !r.ok)
   const lines = [
@@ -110,6 +116,8 @@ export function writeHealthCheckLog(provider: Provider, results: HealthCheckResu
     `Passed: ${passed.length}`,
     `Failed: ${failed.length}`,
     '',
+    formatModelsDiffSummary(diff),
+    '',
   ]
 
   if (failed.length === 0) {

package/src/index.ts

@@ -1,6 +1,12 @@
 import { ExtensionAPI, ExtensionCommandContext, ProviderModelConfig } from '@earendil-works/pi-coding-agent'
 import { getEnv } from './env'
-import { getRequestyConfig, updateModelsJson } from './models-json'
+import {
+  diffModels,
+  formatModelsDiffSummary,
+  getRequestyConfig,
+  updateModelsJson,
+  type ModelsDiff,
+} from './models-json'
 import { discoverModels } from './requesty-api'
 import { checkModels, formatHealthSummary, writeHealthCheckLog } from './health-check'
 import { RequestyStatusLoader } from './ui/requesty-status-loader.ts'
@@ -67,10 +73,11 @@ export async function runCommand(args: string, status: StatusReporter, notifier:
 
   try {
     const env = getEnv()
-    const { data, provider } = getRequestyConfig(env)
+    const { data, provider, existingModelIds } = getRequestyConfig(env)
     const models = await discoverModels(provider)
     const modelsMap = new Map(models.map(m => [m.id, m]))
 
+    let diff: ModelsDiff
     let failed = []
     let passing: ProviderModelConfig[]
     let logNote = ''
@@ -89,11 +96,13 @@ export async function runCommand(args: string, status: StatusReporter, notifier:
         const model = modelsMap.get(r.modelId)
         return r.ok && model ? [model] : []
       })
+      diff = diffModels(existingModelIds, passing)
       healthCheckSummary = formatHealthSummary(sortedResults)
-      writeHealthCheckLog(provider, sortedResults, env)
+      writeHealthCheckLog(provider, sortedResults, diff, env)
       logNote = `Full health check log: ${env.health_check_log_path}\n`
     } else {
       passing = models
+      diff = diffModels(existingModelIds, passing)
     }
 
     const shouldUpdate = passing.length > 0 && !dryRun
@@ -102,7 +111,7 @@ export async function runCommand(args: string, status: StatusReporter, notifier:
     }
 
     const writeNote = shouldUpdate ? 'Run /reload to use models.json changes.' : 'models.json was not updated.'
-    const message = `Discovered ${models.length} Requesty model(s).\n${healthCheckSummary}${logNote}${writeNote}`
+    const message = `Discovered ${models.length} Requesty model(s).\n${healthCheckSummary}${formatModelsDiffSummary(diff)}\n${logNote}${writeNote}`
 
     if (failed.length === 0) {
       notifier.notify(message, 'info')

package/src/models-json.ts

@@ -12,6 +12,7 @@ const ProviderSchema = z
     name: z.string().optional(),
     baseUrl: z.string().optional(),
     apiKey: z.string().optional(),
+    models: z.array(z.object({ id: z.string() }).catchall(z.unknown())).optional(),
   })
   .catchall(z.unknown())
 
@@ -33,6 +34,12 @@ type RequestyProvider = ModelsJsonProvider & {
 type RequestyConfig = {
   data: ModelsJson
   provider: RequestyProvider
+  existingModelIds: string[]
+}
+
+export type ModelsDiff = {
+  added: string[]
+  removed: string[]
 }
 
 export function getRequestyConfig(envConfig: Env = getEnv()): RequestyConfig {
@@ -46,6 +53,7 @@ export function getRequestyConfig(envConfig: Env = getEnv()): RequestyConfig {
   const apiKey = envConfig.requesty_api_key
   return {
     data,
+    existingModelIds: (provider.models ?? []).map(m => m.id),
     provider: {
       ...provider,
       name: nonEmptyString(provider.name) ?? DEFAULT_NAME,
@@ -55,6 +63,24 @@ export function getRequestyConfig(envConfig: Env = getEnv()): RequestyConfig {
   }
 }
 
+export function diffModels(previousIds: string[], nextModels: ProviderModelConfig[]): ModelsDiff {
+  const previous = new Set(previousIds)
+  const next = new Set(nextModels.map(m => m.id))
+  return {
+    added: [...next].filter(id => !previous.has(id)).toSorted(),
+    removed: [...previous].filter(id => !next.has(id)).toSorted(),
+  }
+}
+
+export function formatModelsDiffSummary(diff: ModelsDiff): string {
+  return [
+    diff.added.length === 0 ? 'No added models.' : 'Added models:',
+    ...diff.added.map(id => `- ${id}`),
+    diff.removed.length === 0 ? 'No removed models.' : 'Removed models:',
+    ...diff.removed.map(id => `- ${id}`),
+  ].join('\n')
+}
+
 export function updateModelsJson(data: ModelsJson, models: ProviderModelConfig[], envConfig: Env = getEnv()): void {
   const provider = data.providers[envConfig.provider_id]
   data.providers[envConfig.provider_id] = {