| """First-Run Setup — naming the LLM on first initialization. |
| |
| On first run, the LLM greets the user: |
| "Hi, I am an Incentives Inc. LLM. What would you like to name me?" |
| |
| The user provides a name, which is stored persistently. |
| On subsequent runs, the LLM uses its name in greetings and system prompts. |
| |
| The name is stored in the config file and persists across restarts. |
| """ |
|
|
| from __future__ import annotations |
|
|
| import json |
| import logging |
| import os |
| from typing import Any |
|
|
| logger = logging.getLogger(__name__) |
|
|
|
|
| class FirstRunManager: |
| """Manages first-run setup and LLM naming. |
| |
| On first run: |
| 1. Detects no config exists → first run |
| 2. Greets user: "Hi, I am an Incentives Inc. LLM. What would you like to name me?" |
| 3. User provides a name |
| 4. Name is stored in config |
| |
| On subsequent runs: |
| - Name is loaded from config |
| - System prompt includes the name |
| - Greeting uses the name |
| """ |
|
|
| GREETING = "Hi, I am an Incentives Inc. LLM. What would you like to name me?" |
| WELCOME_BACK = "Hi, I am {name}. How can I help you?" |
|
|
| def __init__(self, data_dir: str) -> None: |
| self.data_dir = data_dir |
| self.config_path = os.path.join(data_dir, "identity.json") |
| self.name: str = "" |
| self.first_run: bool = True |
| self.created_at: float = 0.0 |
| self._load() |
|
|
| def _load(self) -> None: |
| """Load identity from config file.""" |
| if os.path.exists(self.config_path): |
| try: |
| with open(self.config_path, "r") as f: |
| data = json.load(f) |
| self.name = data.get("name", "") |
| self.first_run = data.get("first_run", True) |
| self.created_at = data.get("created_at", 0.0) |
| if self.name: |
| self.first_run = False |
| except Exception as e: |
| logger.warning("Failed to load identity: %s", e) |
| else: |
| self.first_run = True |
|
|
| def _save(self) -> None: |
| """Save identity to config file.""" |
| os.makedirs(self.data_dir, exist_ok=True) |
| data = { |
| "name": self.name, |
| "first_run": self.first_run, |
| "created_at": self.created_at, |
| "company": "Incentives Inc.", |
| } |
| with open(self.config_path, "w") as f: |
| json.dump(data, f, indent=2) |
|
|
| def is_first_run(self) -> bool: |
| """Check if this is the first run.""" |
| return self.first_run |
|
|
| def get_greeting(self) -> str: |
| """Get the appropriate greeting message.""" |
| if self.first_run: |
| return self.GREETING |
| return self.WELCOME_BACK.format(name=self.name) |
|
|
| def set_name(self, name: str) -> None: |
| """Set the LLM's name and mark first run as complete.""" |
| import time |
| self.name = name.strip() |
| self.first_run = False |
| self.created_at = time.time() |
| self._save() |
| logger.info("LLM named: %s", self.name) |
|
|
| def get_system_prompt_suffix(self) -> str: |
| """Get a system prompt suffix that includes the LLM's name.""" |
| if self.name: |
| return f" Your name is {self.name}. You are made by Incentives Inc." |
| return " You are an Incentives Inc. LLM." |
|
|
| def get_name(self) -> str: |
| """Get the LLM's name.""" |
| return self.name or "Incentives Inc. LLM" |
|
|
| def get_stats(self) -> dict[str, Any]: |
| return { |
| "name": self.name, |
| "first_run": self.first_run, |
| "created_at": self.created_at, |
| "company": "Incentives Inc.", |
| } |
|
|
