APEX/BrowserUse_and_ComputerUse_skills
5
0
Fork 0
Code Issues 6 Pull requests 1 Projects 1 Releases Packages Wiki Activity Actions

refactor: remove unused compression model variable in AIAgent

Browse source 
- Eliminated the `compression_model` variable from the AIAgent class, as it was not being utilized.
- Cleaned up the context compressor initialization for improved clarity and maintainability.
This commit is contained in:
teknium1 2026-02-22 02:17:33 -08:00
parent ededaaa874
commit f072801f38
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
run_agent.py
View file

@ -390,15 +390,13 @@ class AIAgent:
# Compresses conversation when approaching model's context limit # Compresses conversation when approaching model's context limit
# Configuration via environment variables (can be set in .env or cli-config.yaml) # Configuration via environment variables (can be set in .env or cli-config.yaml)
compression_threshold = float(os.getenv("CONTEXT_COMPRESSION_THRESHOLD", "0.85")) compression_threshold = float(os.getenv("CONTEXT_COMPRESSION_THRESHOLD", "0.85"))
compression_model = os.getenv("CONTEXT_COMPRESSION_MODEL", "google/gemini-3-flash-preview")
compression_enabled = os.getenv("CONTEXT_COMPRESSION_ENABLED", "true").lower() in ("true", "1", "yes") compression_enabled = os.getenv("CONTEXT_COMPRESSION_ENABLED", "true").lower() in ("true", "1", "yes")
self.context_compressor = ContextCompressor( self.context_compressor = ContextCompressor(
model=self.model, model=self.model,
threshold_percent=compression_threshold, threshold_percent=compression_threshold,
summary_model=compression_model, protect_first_n=3,
protect_first_n=3, # Keep system, first user, first assistant protect_last_n=4,
protect_last_n=4, # Keep recent context
summary_target_tokens=500, summary_target_tokens=500,
quiet_mode=self.quiet_mode, quiet_mode=self.quiet_mode,
) )