base.py•2.45 kB
"""
Base mode handler interface
"""
from abc import ABC, abstractmethod
from typing import Any, Dict
from providers.base import BaseProvider
class BaseMode(ABC):
"""Base class for all mode handlers"""
@abstractmethod
def get_system_prompt(self) -> str:
"""Get mode-specific system prompt"""
pass
async def handle(self, context: Dict[str, Any], provider: BaseProvider) -> str:
"""
Handle the mode execution with common workflow
Args:
context: Full context including prompt, files, conversation
provider: AI provider to use
Returns:
Response string
"""
messages = self.build_messages(context)
# Apply mode-specific prompt enhancement
mode_enhancement = self._get_mode_enhancement()
if mode_enhancement:
messages[-1]["content"] += f"\n\n{mode_enhancement}"
response = await provider.complete(
model=context["model"],
messages=messages,
temperature=context.get("temperature", self._get_default_temperature()),
)
return response
def _get_mode_enhancement(self) -> str:
"""Get mode-specific prompt enhancement (override in subclasses)"""
return ""
def _get_default_temperature(self) -> float:
"""Get mode-specific default temperature (override in subclasses)"""
return 0.7
def build_messages(self, context: Dict[str, Any]) -> list[dict]:
"""Build messages for AI provider"""
messages = []
# Add system prompt
messages.append({"role": "system", "content": self.get_system_prompt()})
# Add conversation history if present
if context.get("conversation"):
for turn in context["conversation"]["turns"][-10:]: # Last 10 turns
messages.append({"role": turn["role"], "content": turn["content"]})
# Build user message
user_content = context["prompt"]
# Add file contents if present
if context.get("files"):
file_section = "\n\n=== FILES ===\n"
for path, content in context["files"].items():
file_section += f"\n--- {path} ---\n{content}\n"
file_section += "\n=== END FILES ===\n"
user_content = f"{user_content}{file_section}"
messages.append({"role": "user", "content": user_content})
return messages