package com.correx.testing.fixtures.context

import com.correx.core.context.builder.DefaultContextPackBuilder
import com.correx.core.context.compression.CompressionStrategy
import com.correx.core.context.compression.ContextCompressor
import com.correx.core.context.model.ContextEntry
import com.correx.core.context.model.TokenBudget

object ContextFixtures {

    fun simpleCompressor(): ContextCompressor {
        return object : ContextCompressor {
            override fun compress(
                entries: List<ContextEntry>,
                budget: TokenBudget,
                strategy: CompressionStrategy
            ): List<ContextEntry> {
                return entries
            }
        }
    }

    fun simpleBuilder(): DefaultContextPackBuilder {
        return DefaultContextPackBuilder(simpleCompressor())
    }
}
