mrchrisadams · January 25, 2026 23:00
diff --git a/01-sdk-energy-capture.ts b/01-sdk-energy-capture.ts
 /**
 * PATCH 01: SDK Energy Capture
 * 
 * File: packages/opencode/src/provider/sdk/openai-compatible/src/responses/openai-responses-language-model.ts
 * 
 * This patch modifies the OpenAI-compatible SDK to capture energy data from:
 * 1. GreenPT-style: `impact` field in response.completed chunks
 * 2. Neuralwatt-style: SSE comments (`: energy {...}`)
 * 
 * The energy data is passed through providerMetadata.energy to downstream consumers.
 */

 // =============================================================================
 // CHANGE 1: Add energy schema definitions (add after line ~1320, before usageSchema)
 // =============================================================================

 // Energy data schema - supports multiple provider formats
 const energyDataSchema = z.object({
  // Common fields (normalized)
  wh: z.number().optional(),           // Watt-hours
  kwh: z.number().optional(),          // Kilowatt-hours
  joules: z.number().optional(),       // Joules
  
  // Carbon/emissions
  gCO2e: z.number().optional(),        // Grams CO2 equivalent
  
  // Source tracking
  source: z.enum(['measured', 'estimated']).optional(),
  provider: z.string().optional(),     // 'greenpt', 'neuralwatt', etc.
  
  // Raw provider data (preserved for debugging/analysis)
  raw: z.record(z.unknown()).optional(),
 }).optional()

 // GreenPT impact schema
 const greenptImpactSchema = z.object({
  version: z.string().optional(),
  inferenceTime: z.object({
    total: z.number(),
    unit: z.string(),
  }).optional(),
  energy: z.object({
    total: z.number(),
    unit: z.string(),  // "Wms" = watt-milliseconds
  }).optional(),
  emissions: z.object({
    total: z.number(),
    unit: z.string(),  // "ugCO2e" = micrograms CO2 equivalent
  }).optional(),
 }).optional()

 // Neuralwatt energy schema  
 const neuralwattEnergySchema = z.object({
  energy_joules: z.number().optional(),
  energy_kwh: z.number().optional(),
  avg_power_watts: z.number().optional(),
  duration_seconds: z.number().optional(),
  attribution_method: z.string().optional(),
  attribution_ratio: z.number().optional(),
 }).optional()


 // =============================================================================
 // CHANGE 2: Extend responseFinishedChunkSchema (modify existing, ~line 1345)
 // =============================================================================

 const responseFinishedChunkSchema = z.object({
  type: z.enum(["response.completed", "response.incomplete"]),
  response: z.object({
    incomplete_details: z.object({ reason: z.string() }).nullish(),
    usage: usageSchema,
    service_tier: z.string().nullish(),
    // ADD: GreenPT-style impact data
    impact: greenptImpactSchema,
  }),
 })


 // =============================================================================
 // CHANGE 3: Add energy comment chunk schema (add to union, ~line 1380)
 // =============================================================================

 // SSE comment-based energy (Neuralwatt style)
 // Note: This requires modifying createEventSourceResponseHandler to capture comments
 const energyCommentChunkSchema = z.object({
  type: z.literal("energy.comment"),
  data: neuralwattEnergySchema,
 })

 const openaiResponsesChunkSchema = z.union([
  textDeltaChunkSchema,
  responseFinishedChunkSchema,
  responseCreatedChunkSchema,
  responseOutputItemAddedSchema,
  responseOutputItemDoneSchema,
  responseFunctionCallArgumentsDeltaSchema,
  responseImageGenerationCallPartialImageSchema,
  responseCodeInterpreterCallCodeDeltaSchema,
  responseCodeInterpreterCallCodeDoneSchema,
  responseAnnotationAddedSchema,
  responseReasoningSummaryPartAddedSchema,
  responseReasoningSummaryTextDeltaSchema,
  errorChunkSchema,
  energyCommentChunkSchema,  // ADD: Energy comment support
  z.object({ type: z.string() }).loose(),
 ])


 // =============================================================================
 // CHANGE 4: Add energy tracking variables (in doStream method, ~line 796)
 // =============================================================================

 // After: let serviceTier: string | undefined
 // Add:
 let capturedEnergy: z.infer<typeof energyDataSchema> = undefined


 // =============================================================================
 // CHANGE 5: Capture energy in transform function (~line 1250)
 // =============================================================================

 // Inside: if (isResponseFinishedChunk(value)) { ... }
 // After usage extraction, add:

 if (isResponseFinishedChunk(value)) {
  finishReason = mapOpenAIResponseFinishReason({
    finishReason: value.response.incomplete_details?.reason,
    hasFunctionCall,
  })
  usage.inputTokens = value.response.usage.input_tokens
  usage.outputTokens = value.response.usage.output_tokens
  usage.totalTokens = value.response.usage.input_tokens + value.response.usage.output_tokens
  usage.reasoningTokens = value.response.usage.output_tokens_details?.reasoning_tokens ?? undefined
  usage.cachedInputTokens = value.response.usage.input_tokens_details?.cached_tokens ?? undefined
  
  if (typeof value.response.service_tier === "string") {
    serviceTier = value.response.service_tier
  }
  
  // ADD: Capture GreenPT-style impact data
  if (value.response.impact) {
    const impact = value.response.impact
    capturedEnergy = {
      source: 'measured',
      provider: 'greenpt',
      raw: impact,
    }
    
    // Convert GreenPT units to standard
    if (impact.energy) {
      // GreenPT uses Wms (watt-milliseconds)
      if (impact.energy.unit === 'Wms') {
        const wms = impact.energy.total
        capturedEnergy.wh = wms / 3_600_000      // Wms to Wh
        capturedEnergy.kwh = wms / 3_600_000_000 // Wms to kWh
        capturedEnergy.joules = wms / 1000       // Wms to J
      }
    }
    
    if (impact.emissions) {
      // GreenPT uses ugCO2e (micrograms CO2 equivalent)
      if (impact.emissions.unit === 'ugCO2e') {
        capturedEnergy.gCO2e = impact.emissions.total / 1_000_000 // ug to g
      }
    }
  }
 }

 // ADD: Handle Neuralwatt-style energy comments
 if (isEnergyCommentChunk(value)) {
  const data = value.data
  capturedEnergy = {
    source: 'measured',
    provider: 'neuralwatt',
    raw: data,
    joules: data.energy_joules,
    kwh: data.energy_kwh,
    wh: data.energy_kwh ? data.energy_kwh * 1000 : undefined,
  }
 }


 // =============================================================================
 // CHANGE 6: Include energy in providerMetadata (in flush function, ~line 1290)
 // =============================================================================

 flush(controller) {
  // Close any dangling text part
  if (currentTextId) {
    controller.enqueue({ type: "text-end", id: currentTextId })
    currentTextId = null
  }

  const providerMetadata: SharedV2ProviderMetadata = {
    openai: {
      responseId,
    },
  }

  if (logprobs.length > 0) {
    providerMetadata.openai.logprobs = logprobs
  }

  if (serviceTier !== undefined) {
    providerMetadata.openai.serviceTier = serviceTier
  }
  
  // ADD: Include captured energy data
  if (capturedEnergy) {
    providerMetadata.energy = capturedEnergy
  }

  controller.enqueue({
    type: "finish",
    finishReason,
    usage,
    providerMetadata,
  })
 }


 // =============================================================================
 // CHANGE 7: Add type guard for energy comment chunks
 // =============================================================================

 function isEnergyCommentChunk(
  chunk: z.infer<typeof openaiResponsesChunkSchema>,
 ): chunk is z.infer<typeof energyCommentChunkSchema> {
  return chunk.type === "energy.comment"
 }


 // =============================================================================
 // CHANGE 8: Modify createEventSourceResponseHandler to capture SSE comments
 // 
 // This requires changes to @ai-sdk/provider-utils or a local wrapper.
 // The standard SSE spec says lines starting with ":" are comments and ignored.
 // We need to capture ": energy {...}" comments.
 //
 // Option A: Fork/patch @ai-sdk/provider-utils
 // Option B: Create a wrapper that pre-processes the stream
 // =============================================================================

 // Wrapper approach (add to this file or a separate utility):

 import { createEventSourceResponseHandler as originalHandler } from "@ai-sdk/provider-utils"

 function createEnergyAwareEventSourceResponseHandler<T>(schema: z.ZodType<T>) {
  // Create a transform stream that captures energy comments
  // before passing to the standard handler
  
  return async function* (response: Response) {
    const reader = response.body?.getReader()
    if (!reader) throw new Error('No response body')
    
    const decoder = new TextDecoder()
    let buffer = ''
    let capturedEnergyComment: any = null
    
    while (true) {
      const { done, value } = await reader.read()
      if (done) break
      
      buffer += decoder.decode(value, { stream: true })
      const lines = buffer.split('\n')
      buffer = lines.pop() || ''
      
      for (const line of lines) {
        // Check for energy comment (Neuralwatt style)
        if (line.startsWith(': energy ')) {
          try {
            capturedEnergyComment = JSON.parse(line.slice(9))
            // Emit as a synthetic chunk
            yield {
              success: true,
              value: {
                type: 'energy.comment',
                data: capturedEnergyComment,
              },
              rawValue: line,
            }
          } catch (e) {
            // Ignore parse errors
          }
          continue
        }
        
        // Standard SSE handling
        if (line.startsWith('data: ')) {
          const data = line.slice(6)
          if (data === '[DONE]') continue
          
          try {
            const parsed = JSON.parse(data)
            const result = schema.safeParse(parsed)
            yield {
              success: result.success,
              value: result.success ? result.data : undefined,
              error: result.success ? undefined : result.error,
              rawValue: data,
            }
          } catch (e) {
            yield {
              success: false,
              error: e,
              rawValue: data,
            }
          }
        }
      }
    }
  }
 }
diff --git a/05-example-plugin.ts b/05-example-plugin.ts
 /**
 * Example Plugin: Energy Tracking with Custom Grid Intensity
 * 
 * This plugin demonstrates how to use the new energy hooks to:
 * 1. Enhance energy data with real-time grid carbon intensity
 * 2. Log energy data to an external service
 * 3. Provide custom energy estimates for specific providers
 */

 import type { Plugin, Energy } from "@opencode-ai/plugin"

 // Simulated grid intensity API (in reality, use ElectricityMaps or similar)
 async function fetchGridIntensity(region: string): Promise<{ intensity: number; source: string }> {
  const intensities: Record<string, number> = {
    'us-west': 210,
    'us-east': 380,
    'eu-west': 230,
    'uk': 180,
    'france': 50,
    'germany': 350,
    'global': 436,
  }
  return {
    intensity: intensities[region] || intensities['global'],
    source: 'static-lookup',
  }
 }

 // Custom energy coefficients based on published research
 const CUSTOM_COEFFICIENTS: Record<string, { input: number; output: number }> = {
  'claude-3-5-sonnet': { input: 0.00018, output: 0.00036 },
  'claude-3-opus': { input: 0.00042, output: 0.00084 },
  'gpt-4o': { input: 0.00028, output: 0.00056 },
 }

 export const EnergyEnhancerPlugin: Plugin = async (ctx) => {
  // Track session totals for logging
  const sessionTotals: Record<string, { wh: number; gCO2e: number; count: number }> = {}
  
  await ctx.client.app.log({
    service: 'energy-enhancer',
    level: 'info',
    message: 'Energy enhancer plugin initialized',
  })

  return {
    /**
     * Provide custom energy estimates for providers that don't measure
     */
    "message.energy.estimate": async (input, output) => {
      // Check if we have custom coefficients for this model
      const modelKey = Object.keys(CUSTOM_COEFFICIENTS).find(k => 
        input.modelID.toLowerCase().includes(k)
      )
      
      if (!modelKey) return  // Use OpenCode's default estimation
      
      const coef = CUSTOM_COEFFICIENTS[modelKey]
      const inputKwh = (input.tokens.input / 1_000_000) * coef.input
      const outputKwh = ((input.tokens.output + input.tokens.reasoning) / 1_000_000) * coef.output
      const cacheKwh = (input.tokens.cache.read / 1_000_000) * coef.input * 0.1
      const totalKwh = (inputKwh + outputKwh + cacheKwh) * 1.1  // PUE
      
      output.energy = {
        kwh: totalKwh,
        wh: totalKwh * 1000,
        joules: totalKwh * 3_600_000,
        source: 'estimated',
        provider: 'energy-enhancer-plugin',
        method: 'custom_coefficients_v1',
      }
      
      await ctx.client.app.log({
        service: 'energy-enhancer',
        level: 'debug',
        message: `Custom estimate for ${input.modelID}: ${totalKwh.toExponential(3)} kWh`,
      })
    },
    
    /**
     * Enhance energy data with real-time grid intensity
     */
    "message.energy": async (input, output) => {
      // Fetch real-time grid intensity
      const region = input.region || 'global'
      const grid = await fetchGridIntensity(region)
      
      // Recalculate carbon with real grid intensity
      if (output.energy.kwh) {
        output.energy.gCO2e = output.energy.kwh * grid.intensity
        output.energy.gridIntensity = {
          value: grid.intensity,
          region,
          source: grid.source,
        }
      }
      
      // Track session totals
      if (!sessionTotals[input.sessionID]) {
        sessionTotals[input.sessionID] = { wh: 0, gCO2e: 0, count: 0 }
      }
      sessionTotals[input.sessionID].wh += output.energy.wh || 0
      sessionTotals[input.sessionID].gCO2e += output.energy.gCO2e || 0
      sessionTotals[input.sessionID].count += 1
      
      // Log to external service (example)
      await ctx.client.app.log({
        service: 'energy-enhancer',
        level: 'info',
        message: `Energy: ${(output.energy.wh || 0).toFixed(4)} Wh, ${(output.energy.gCO2e || 0).toFixed(4)} gCO2e`,
        extra: {
          sessionID: input.sessionID,
          messageID: input.messageID,
          model: input.modelID,
          provider: input.providerID,
          energy: output.energy,
          sessionTotals: sessionTotals[input.sessionID],
        },
      })
    },
    
    /**
     * Log session totals when session becomes idle
     */
    event: async ({ event }) => {
      if (event.type === 'session.idle') {
        const sessionID = event.properties.sessionID
        const totals = sessionTotals[sessionID]
        
        if (totals) {
          await ctx.client.app.log({
            service: 'energy-enhancer',
            level: 'info',
            message: `Session ${sessionID} complete`,
            extra: {
              totalWh: totals.wh,
              totalGCO2e: totals.gCO2e,
              messageCount: totals.count,
              avgWhPerMessage: totals.wh / totals.count,
            },
          })
          
          // Clean up
          delete sessionTotals[sessionID]
        }
      }
    },
  }
 }

 export default EnergyEnhancerPlugin
diff --git a/PROPOSAL.md b/PROPOSAL.md
diff --git a/SSE_PARSING_ANALYSIS.md b/SSE_PARSING_ANALYSIS.md
	/**
	* PATCH 01: SDK Energy Capture
	*
	* File: packages/opencode/src/provider/sdk/openai-compatible/src/responses/openai-responses-language-model.ts
	*
	* This patch modifies the OpenAI-compatible SDK to capture energy data from:
	* 1. GreenPT-style: `impact` field in response.completed chunks
	* 2. Neuralwatt-style: SSE comments (`: energy {...}`)
	*
	* The energy data is passed through providerMetadata.energy to downstream consumers.
	*/

	// =============================================================================
	// CHANGE 1: Add energy schema definitions (add after line ~1320, before usageSchema)
	// =============================================================================

	// Energy data schema - supports multiple provider formats
	const energyDataSchema = z.object({
	// Common fields (normalized)
	wh: z.number().optional(), // Watt-hours
	kwh: z.number().optional(), // Kilowatt-hours
	joules: z.number().optional(), // Joules

	// Carbon/emissions
	gCO2e: z.number().optional(), // Grams CO2 equivalent

	// Source tracking
	source: z.enum(['measured', 'estimated']).optional(),
	provider: z.string().optional(), // 'greenpt', 'neuralwatt', etc.

	// Raw provider data (preserved for debugging/analysis)
	raw: z.record(z.unknown()).optional(),
	}).optional()

	// GreenPT impact schema
	const greenptImpactSchema = z.object({
	version: z.string().optional(),
	inferenceTime: z.object({
	total: z.number(),
	unit: z.string(),
	}).optional(),
	energy: z.object({
	total: z.number(),
	unit: z.string(), // "Wms" = watt-milliseconds
	}).optional(),
	emissions: z.object({
	total: z.number(),
	unit: z.string(), // "ugCO2e" = micrograms CO2 equivalent
	}).optional(),
	}).optional()

	// Neuralwatt energy schema
	const neuralwattEnergySchema = z.object({
	energy_joules: z.number().optional(),
	energy_kwh: z.number().optional(),
	avg_power_watts: z.number().optional(),
	duration_seconds: z.number().optional(),
	attribution_method: z.string().optional(),
	attribution_ratio: z.number().optional(),
	}).optional()


	// =============================================================================
	// CHANGE 2: Extend responseFinishedChunkSchema (modify existing, ~line 1345)
	// =============================================================================

	const responseFinishedChunkSchema = z.object({
	type: z.enum(["response.completed", "response.incomplete"]),
	response: z.object({
	incomplete_details: z.object({ reason: z.string() }).nullish(),
	usage: usageSchema,
	service_tier: z.string().nullish(),
	// ADD: GreenPT-style impact data
	impact: greenptImpactSchema,
	}),
	})


	// =============================================================================
	// CHANGE 3: Add energy comment chunk schema (add to union, ~line 1380)
	// =============================================================================

	// SSE comment-based energy (Neuralwatt style)
	// Note: This requires modifying createEventSourceResponseHandler to capture comments
	const energyCommentChunkSchema = z.object({
	type: z.literal("energy.comment"),
	data: neuralwattEnergySchema,
	})

	const openaiResponsesChunkSchema = z.union([
	textDeltaChunkSchema,
	responseFinishedChunkSchema,
	responseCreatedChunkSchema,
	responseOutputItemAddedSchema,
	responseOutputItemDoneSchema,
	responseFunctionCallArgumentsDeltaSchema,
	responseImageGenerationCallPartialImageSchema,
	responseCodeInterpreterCallCodeDeltaSchema,
	responseCodeInterpreterCallCodeDoneSchema,
	responseAnnotationAddedSchema,
	responseReasoningSummaryPartAddedSchema,
	responseReasoningSummaryTextDeltaSchema,
	errorChunkSchema,
	energyCommentChunkSchema, // ADD: Energy comment support
	z.object({ type: z.string() }).loose(),
	])


	// =============================================================================
	// CHANGE 4: Add energy tracking variables (in doStream method, ~line 796)
	// =============================================================================

	// After: let serviceTier: string \| undefined
	// Add:
	let capturedEnergy: z.infer<typeof energyDataSchema> = undefined


	// =============================================================================
	// CHANGE 5: Capture energy in transform function (~line 1250)
	// =============================================================================

	// Inside: if (isResponseFinishedChunk(value)) { ... }
	// After usage extraction, add:

	if (isResponseFinishedChunk(value)) {
	finishReason = mapOpenAIResponseFinishReason({
	finishReason: value.response.incomplete_details?.reason,
	hasFunctionCall,
	})
	usage.inputTokens = value.response.usage.input_tokens
	usage.outputTokens = value.response.usage.output_tokens
	usage.totalTokens = value.response.usage.input_tokens + value.response.usage.output_tokens
	usage.reasoningTokens = value.response.usage.output_tokens_details?.reasoning_tokens ?? undefined
	usage.cachedInputTokens = value.response.usage.input_tokens_details?.cached_tokens ?? undefined

	if (typeof value.response.service_tier === "string") {
	serviceTier = value.response.service_tier
	}

	// ADD: Capture GreenPT-style impact data
	if (value.response.impact) {
	const impact = value.response.impact
	capturedEnergy = {
	source: 'measured',
	provider: 'greenpt',
	raw: impact,
	}

	// Convert GreenPT units to standard
	if (impact.energy) {
	// GreenPT uses Wms (watt-milliseconds)
	if (impact.energy.unit === 'Wms') {
	const wms = impact.energy.total
	capturedEnergy.wh = wms / 3_600_000 // Wms to Wh
	capturedEnergy.kwh = wms / 3_600_000_000 // Wms to kWh
	capturedEnergy.joules = wms / 1000 // Wms to J
	}
	}

	if (impact.emissions) {
	// GreenPT uses ugCO2e (micrograms CO2 equivalent)
	if (impact.emissions.unit === 'ugCO2e') {
	capturedEnergy.gCO2e = impact.emissions.total / 1_000_000 // ug to g
	}
	}
	}
	}

	// ADD: Handle Neuralwatt-style energy comments
	if (isEnergyCommentChunk(value)) {
	const data = value.data
	capturedEnergy = {
	source: 'measured',
	provider: 'neuralwatt',
	raw: data,
	joules: data.energy_joules,
	kwh: data.energy_kwh,
	wh: data.energy_kwh ? data.energy_kwh * 1000 : undefined,
	}
	}


	// =============================================================================
	// CHANGE 6: Include energy in providerMetadata (in flush function, ~line 1290)
	// =============================================================================

	flush(controller) {
	// Close any dangling text part
	if (currentTextId) {
	controller.enqueue({ type: "text-end", id: currentTextId })
	currentTextId = null
	}

	const providerMetadata: SharedV2ProviderMetadata = {
	openai: {
	responseId,
	},
	}

	if (logprobs.length > 0) {
	providerMetadata.openai.logprobs = logprobs
	}

	if (serviceTier !== undefined) {
	providerMetadata.openai.serviceTier = serviceTier
	}

	// ADD: Include captured energy data
	if (capturedEnergy) {
	providerMetadata.energy = capturedEnergy
	}

	controller.enqueue({
	type: "finish",
	finishReason,
	usage,
	providerMetadata,
	})
	}


	// =============================================================================
	// CHANGE 7: Add type guard for energy comment chunks
	// =============================================================================

	function isEnergyCommentChunk(
	chunk: z.infer<typeof openaiResponsesChunkSchema>,
	): chunk is z.infer<typeof energyCommentChunkSchema> {
	return chunk.type === "energy.comment"
	}


	// =============================================================================
	// CHANGE 8: Modify createEventSourceResponseHandler to capture SSE comments
	//
	// This requires changes to @ai-sdk/provider-utils or a local wrapper.
	// The standard SSE spec says lines starting with ":" are comments and ignored.
	// We need to capture ": energy {...}" comments.
	//
	// Option A: Fork/patch @ai-sdk/provider-utils
	// Option B: Create a wrapper that pre-processes the stream
	// =============================================================================

	// Wrapper approach (add to this file or a separate utility):

	import { createEventSourceResponseHandler as originalHandler } from "@ai-sdk/provider-utils"

	function createEnergyAwareEventSourceResponseHandler<T>(schema: z.ZodType<T>) {
	// Create a transform stream that captures energy comments
	// before passing to the standard handler

	return async function* (response: Response) {
	const reader = response.body?.getReader()
	if (!reader) throw new Error('No response body')

	const decoder = new TextDecoder()
	let buffer = ''
	let capturedEnergyComment: any = null

	while (true) {
	const { done, value } = await reader.read()
	if (done) break

	buffer += decoder.decode(value, { stream: true })
	const lines = buffer.split('\n')
	buffer = lines.pop() \|\| ''

	for (const line of lines) {
	// Check for energy comment (Neuralwatt style)
	if (line.startsWith(': energy ')) {
	try {
	capturedEnergyComment = JSON.parse(line.slice(9))
	// Emit as a synthetic chunk
	yield {
	success: true,
	value: {
	type: 'energy.comment',
	data: capturedEnergyComment,
	},
	rawValue: line,
	}
	} catch (e) {
	// Ignore parse errors
	}
	continue
	}

	// Standard SSE handling
	if (line.startsWith('data: ')) {
	const data = line.slice(6)
	if (data === '[DONE]') continue

	try {
	const parsed = JSON.parse(data)
	const result = schema.safeParse(parsed)
	yield {
	success: result.success,
	value: result.success ? result.data : undefined,
	error: result.success ? undefined : result.error,
	rawValue: data,
	}
	} catch (e) {
	yield {
	success: false,
	error: e,
	rawValue: data,
	}
	}
	}
	}
	}
	}
	}
	/**
	* Example Plugin: Energy Tracking with Custom Grid Intensity
	*
	* This plugin demonstrates how to use the new energy hooks to:
	* 1. Enhance energy data with real-time grid carbon intensity
	* 2. Log energy data to an external service
	* 3. Provide custom energy estimates for specific providers
	*/

	import type { Plugin, Energy } from "@opencode-ai/plugin"

	// Simulated grid intensity API (in reality, use ElectricityMaps or similar)
	async function fetchGridIntensity(region: string): Promise<{ intensity: number; source: string }> {
	const intensities: Record<string, number> = {
	'us-west': 210,
	'us-east': 380,
	'eu-west': 230,
	'uk': 180,
	'france': 50,
	'germany': 350,
	'global': 436,
	}
	return {
	intensity: intensities[region] \|\| intensities['global'],
	source: 'static-lookup',
	}
	}

	// Custom energy coefficients based on published research
	const CUSTOM_COEFFICIENTS: Record<string, { input: number; output: number }> = {
	'claude-3-5-sonnet': { input: 0.00018, output: 0.00036 },
	'claude-3-opus': { input: 0.00042, output: 0.00084 },
	'gpt-4o': { input: 0.00028, output: 0.00056 },
	}

	export const EnergyEnhancerPlugin: Plugin = async (ctx) => {
	// Track session totals for logging
	const sessionTotals: Record<string, { wh: number; gCO2e: number; count: number }> = {}

	await ctx.client.app.log({
	service: 'energy-enhancer',
	level: 'info',
	message: 'Energy enhancer plugin initialized',
	})

	return {
	/**
	* Provide custom energy estimates for providers that don't measure
	*/
	"message.energy.estimate": async (input, output) => {
	// Check if we have custom coefficients for this model
	const modelKey = Object.keys(CUSTOM_COEFFICIENTS).find(k =>
	input.modelID.toLowerCase().includes(k)
	)

	if (!modelKey) return // Use OpenCode's default estimation

	const coef = CUSTOM_COEFFICIENTS[modelKey]
	const inputKwh = (input.tokens.input / 1_000_000) * coef.input
	const outputKwh = ((input.tokens.output + input.tokens.reasoning) / 1_000_000) * coef.output
	const cacheKwh = (input.tokens.cache.read / 1_000_000) * coef.input * 0.1
	const totalKwh = (inputKwh + outputKwh + cacheKwh) * 1.1 // PUE

	output.energy = {
	kwh: totalKwh,
	wh: totalKwh * 1000,
	joules: totalKwh * 3_600_000,
	source: 'estimated',
	provider: 'energy-enhancer-plugin',
	method: 'custom_coefficients_v1',
	}

	await ctx.client.app.log({
	service: 'energy-enhancer',
	level: 'debug',
	message: `Custom estimate for ${input.modelID}: ${totalKwh.toExponential(3)} kWh`,
	})
	},

	/**
	* Enhance energy data with real-time grid intensity
	*/
	"message.energy": async (input, output) => {
	// Fetch real-time grid intensity
	const region = input.region \|\| 'global'
	const grid = await fetchGridIntensity(region)

	// Recalculate carbon with real grid intensity
	if (output.energy.kwh) {
	output.energy.gCO2e = output.energy.kwh * grid.intensity
	output.energy.gridIntensity = {
	value: grid.intensity,
	region,
	source: grid.source,
	}
	}

	// Track session totals
	if (!sessionTotals[input.sessionID]) {
	sessionTotals[input.sessionID] = { wh: 0, gCO2e: 0, count: 0 }
	}
	sessionTotals[input.sessionID].wh += output.energy.wh \|\| 0
	sessionTotals[input.sessionID].gCO2e += output.energy.gCO2e \|\| 0
	sessionTotals[input.sessionID].count += 1

	// Log to external service (example)
	await ctx.client.app.log({
	service: 'energy-enhancer',
	level: 'info',
	message: `Energy: ${(output.energy.wh \|\| 0).toFixed(4)} Wh, ${(output.energy.gCO2e \|\| 0).toFixed(4)} gCO2e`,
	extra: {
	sessionID: input.sessionID,
	messageID: input.messageID,
	model: input.modelID,
	provider: input.providerID,
	energy: output.energy,
	sessionTotals: sessionTotals[input.sessionID],
	},
	})
	},

	/**
	* Log session totals when session becomes idle
	*/
	event: async ({ event }) => {
	if (event.type === 'session.idle') {
	const sessionID = event.properties.sessionID
	const totals = sessionTotals[sessionID]

	if (totals) {
	await ctx.client.app.log({
	service: 'energy-enhancer',
	level: 'info',
	message: `Session ${sessionID} complete`,
	extra: {
	totalWh: totals.wh,
	totalGCO2e: totals.gCO2e,
	messageCount: totals.count,
	avgWhPerMessage: totals.wh / totals.count,
	},
	})

	// Clean up
	delete sessionTotals[sessionID]
	}
	}
	},
	}
	}

	export default EnergyEnhancerPlugin