From 5505d2b21706bb2f866bab6bd64b0c5cd1bc3e52 Mon Sep 17 00:00:00 2001
From: Lukas Parsons <lukasparsons2013@gmail.com>
Date: Sun, 22 Mar 2026 22:47:49 -0400
Subject: [PATCH] Fix compress endpoint to use request.messages correctly

---
 main.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 63d6e4e..a6dc493 100644
--- a/main.py
+++ b/main.py
@@ -19,7 +19,8 @@ from schemas import (
     Convention as ConventionSchema,
     MemoryBase,
     Memory as MemorySchema,
-    ContextBundle
+    ContextBundle,
+    CompressionRequest
 )
 from rag import build_context_bundle, clear_cache as clear_rag_cache
 from compression import compress_conversation, count_tokens
@@ -442,7 +443,7 @@ async def compress_messages_endpoint(
         ollama_model=config.compression.ollama_model,
         ollama_url=config.compression.ollama_url
     )
-    original_tokens = sum(count_tokens(m.get("content", "")) for m in messages)
+    original_tokens = sum(count_tokens(m.get("content", "")) for m in request.messages)
     compressed_tokens = sum(count_tokens(m.get("content", "")) for m in compressed)
     
     return {