File size: 791 Bytes
7f611c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | """Minimal database for the Claude Code baseline.
Claude Code handles its own internal iteration loop, so this database
just stores whatever the controller adds (typically one final result).
"""
from skydiscover.search.base_database import Program, ProgramDatabase
class ClaudeCodeDatabase(ProgramDatabase):
def add(self, program: Program, iteration=None, **kwargs) -> str:
self.programs[program.id] = program
if iteration is not None:
self.last_iteration = max(self.last_iteration, iteration)
if self.config.db_path:
self._save_program(program)
self._update_best_program(program)
return program.id
def sample(self, num_context_programs=4, **kwargs):
best = self.get_best_program()
return best, []
|