Function buildSummaryDeps

packages/node-runtime/src/services/summary-service.ts:29–59

(db: DatabaseAdapter, llmConfig: LlmConfig, deps: SummaryServiceDeps)

}

function buildSummaryDeps(db: DatabaseAdapter, llmConfig: LlmConfig, deps: SummaryServiceDeps): SummaryDeps {
  const piModel = deps.buildPiModel(llmConfig)
  return {
    loadMessages(segmentId, limit = 500) {
      const data = getSegmentMessages(db, segmentId, limit)
      if (!data) return null
      return data.messages.map((m) => ({ senderName: m.senderName, content: m.content }))
    },
    saveSummary(segmentId, summary) {
      saveSegmentSummary(db, segmentId, summary)
    },
    getSummary(segmentId) {
      return getSegmentSummary(db, segmentId)
    },
    async llmComplete(systemPrompt, userPrompt, options) {
      const result = await completeSimple(
        piModel,
        {
          systemPrompt,
          messages: [{ role: 'user', content: [{ type: 'text', text: userPrompt }], timestamp: Date.now() }] as any,
        },
        { apiKey: llmConfig.apiKey, maxTokens: options?.maxTokens, temperature: options?.temperature }
      )
      return result.content
        .filter((item): item is PiTextContent => item.type === 'text')
        .map((item) => item.text)
        .join('')
    },
    t: (key: string) => key,
  }
}

export async function generateSummary(
  adapter: SessionRuntimeAdapter,