mirror of
https://github.com/zvx-echo6/meshai.git
synced 2026-06-11 01:14:45 +02:00
Remove dead modules: safety, rate_limiter, personality, webhook, web_status, announcements, log_setup
These modules were wired up but never actually functional in the running bot. Strips all imports and usage from main.py and router.py. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ubuntu
parent
9a628724ce
commit
10bc94b273
9 changed files with 22 additions and 1234 deletions
144
meshai/main.py
144
meshai/main.py
|
|
@ -11,8 +11,7 @@ from pathlib import Path
|
|||
from typing import Optional
|
||||
|
||||
from . import __version__
|
||||
from .announcements import AnnouncementScheduler
|
||||
from .backends import AnthropicBackend, FallbackBackend, GoogleBackend, LLMBackend, OpenAIBackend
|
||||
from .backends import AnthropicBackend, GoogleBackend, LLMBackend, OpenAIBackend
|
||||
from .cli import run_configurator
|
||||
from .commands import CommandDispatcher
|
||||
from .commands.dispatcher import create_dispatcher
|
||||
|
|
@ -21,13 +20,8 @@ from .config import Config, load_config
|
|||
from .connector import MeshConnector, MeshMessage
|
||||
from .history import ConversationHistory
|
||||
from .memory import ConversationSummary
|
||||
from .personality import PersonalityManager
|
||||
from .rate_limiter import RateLimiter
|
||||
from .responder import Responder
|
||||
from .router import MessageRouter, RouteType
|
||||
from .safety import SafetyFilter, UserFilter
|
||||
from .web_status import WebStatusServer, get_status_data
|
||||
from .webhook import WebhookClient
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
|
@ -43,13 +37,6 @@ class MeshAI:
|
|||
self.llm: Optional[LLMBackend] = None
|
||||
self.router: Optional[MessageRouter] = None
|
||||
self.responder: Optional[Responder] = None
|
||||
self.personality: Optional[PersonalityManager] = None
|
||||
self.safety_filter: Optional[SafetyFilter] = None
|
||||
self.user_filter: Optional[UserFilter] = None
|
||||
self.rate_limiter: Optional[RateLimiter] = None
|
||||
self.webhook: Optional[WebhookClient] = None
|
||||
self.web_status: Optional[WebStatusServer] = None
|
||||
self.announcements: Optional[AnnouncementScheduler] = None
|
||||
self._running = False
|
||||
self._loop: Optional[asyncio.AbstractEventLoop] = None
|
||||
self._last_cleanup: float = 0.0
|
||||
|
|
@ -70,14 +57,6 @@ class MeshAI:
|
|||
self._loop = asyncio.get_event_loop()
|
||||
self._last_cleanup = time.time()
|
||||
|
||||
# Start async services
|
||||
await self.webhook.start()
|
||||
await self.webhook.on_startup()
|
||||
await self.announcements.start()
|
||||
|
||||
# Start sync services
|
||||
self.web_status.start()
|
||||
|
||||
# Write PID file
|
||||
self._write_pid()
|
||||
|
||||
|
|
@ -97,16 +76,6 @@ class MeshAI:
|
|||
logger.info("Stopping MeshAI...")
|
||||
self._running = False
|
||||
|
||||
if self.webhook:
|
||||
await self.webhook.on_shutdown()
|
||||
await self.webhook.stop()
|
||||
|
||||
if self.announcements:
|
||||
await self.announcements.stop()
|
||||
|
||||
if self.web_status:
|
||||
self.web_status.stop()
|
||||
|
||||
if self.connector:
|
||||
self.connector.disconnect()
|
||||
|
||||
|
|
@ -125,30 +94,8 @@ class MeshAI:
|
|||
self.history = ConversationHistory(self.config.history)
|
||||
await self.history.initialize()
|
||||
|
||||
# Command dispatcher (2h: pass config)
|
||||
self.dispatcher = create_dispatcher(
|
||||
prefix=self.config.commands.prefix,
|
||||
disabled_commands=self.config.commands.disabled_commands,
|
||||
custom_commands=self.config.commands.custom_commands,
|
||||
)
|
||||
|
||||
# Safety and user filters (2a)
|
||||
self.user_filter = UserFilter(
|
||||
blocklist=self.config.users.blocklist,
|
||||
allowlist=self.config.users.allowlist,
|
||||
allowlist_only=self.config.users.allowlist_only,
|
||||
admin_nodes=self.config.users.admin_nodes,
|
||||
)
|
||||
self.safety_filter = SafetyFilter(self.config.safety)
|
||||
|
||||
# Rate limiter (2b)
|
||||
self.rate_limiter = RateLimiter(
|
||||
self.config.rate_limits,
|
||||
vip_nodes=self.config.users.vip_nodes,
|
||||
)
|
||||
|
||||
# Personality manager (2c)
|
||||
self.personality = PersonalityManager(self.config.personality)
|
||||
# Command dispatcher
|
||||
self.dispatcher = create_dispatcher()
|
||||
|
||||
# LLM backend
|
||||
api_key = self.config.resolve_api_key()
|
||||
|
|
@ -160,100 +107,52 @@ class MeshAI:
|
|||
window_size = mem_cfg.window_size if mem_cfg.enabled else 0
|
||||
summarize_threshold = mem_cfg.summarize_threshold
|
||||
|
||||
# Create primary backend
|
||||
# Create backend
|
||||
backend = self.config.llm.backend.lower()
|
||||
if backend == "openai":
|
||||
primary = OpenAIBackend(
|
||||
self.llm = OpenAIBackend(
|
||||
self.config.llm, api_key, window_size, summarize_threshold
|
||||
)
|
||||
elif backend == "anthropic":
|
||||
primary = AnthropicBackend(
|
||||
self.llm = AnthropicBackend(
|
||||
self.config.llm, api_key, window_size, summarize_threshold
|
||||
)
|
||||
elif backend == "google":
|
||||
primary = GoogleBackend(
|
||||
self.llm = GoogleBackend(
|
||||
self.config.llm, api_key, window_size, summarize_threshold
|
||||
)
|
||||
else:
|
||||
logger.warning(f"Unknown backend '{backend}', defaulting to OpenAI")
|
||||
primary = OpenAIBackend(
|
||||
self.llm = OpenAIBackend(
|
||||
self.config.llm, api_key, window_size, summarize_threshold
|
||||
)
|
||||
|
||||
# Wrap in FallbackBackend if fallback is configured (2g)
|
||||
if self.config.llm.fallback:
|
||||
self.llm = FallbackBackend(
|
||||
self.config.llm, api_key, window_size, summarize_threshold
|
||||
)
|
||||
else:
|
||||
self.llm = primary
|
||||
|
||||
# Load persisted summaries into memory cache
|
||||
await self._load_summaries()
|
||||
|
||||
# Meshtastic connector
|
||||
self.connector = MeshConnector(self.config.connection)
|
||||
|
||||
# Message router (pass personality manager)
|
||||
# Message router
|
||||
self.router = MessageRouter(
|
||||
self.config, self.connector, self.history, self.dispatcher, self.llm,
|
||||
personality=self.personality,
|
||||
)
|
||||
|
||||
# Responder
|
||||
self.responder = Responder(self.config.response, self.connector)
|
||||
|
||||
# Webhook client (2d)
|
||||
self.webhook = WebhookClient(self.config.integrations.webhook)
|
||||
|
||||
# Web status server (2e)
|
||||
self.web_status = WebStatusServer(self.config.web_status)
|
||||
|
||||
# Announcement scheduler (2f)
|
||||
async def _send_announcement(text: str, channel: int) -> None:
|
||||
self.connector.send_message(text=text, channel=channel)
|
||||
|
||||
self.announcements = AnnouncementScheduler(
|
||||
self.config.announcements,
|
||||
send_callback=_send_announcement,
|
||||
)
|
||||
|
||||
async def _on_message(self, message: MeshMessage) -> None:
|
||||
"""Handle incoming message."""
|
||||
try:
|
||||
# Check user filter (2a)
|
||||
allowed, reason = self.user_filter.is_allowed(message.sender_id)
|
||||
if not allowed:
|
||||
logger.debug(f"Blocked message from {message.sender_id}: {reason}")
|
||||
return
|
||||
|
||||
# Check if we should respond
|
||||
if not self.router.should_respond(message):
|
||||
return
|
||||
|
||||
# Check rate limiter (2b)
|
||||
allowed, reason = self.rate_limiter.is_allowed(message.sender_id)
|
||||
if not allowed:
|
||||
logger.debug(f"Rate limited {message.sender_id}: {reason}")
|
||||
return
|
||||
|
||||
logger.info(
|
||||
f"Processing message from {message.sender_name} ({message.sender_id}): "
|
||||
f"{message.text[:50]}..."
|
||||
)
|
||||
|
||||
# Record in web status (2e)
|
||||
get_status_data().record_message(message.sender_id, message.sender_name)
|
||||
|
||||
# Send webhook event (2d)
|
||||
await self.webhook.on_message_received(
|
||||
sender_id=message.sender_id,
|
||||
sender_name=message.sender_name,
|
||||
text=message.text,
|
||||
channel=message.channel,
|
||||
is_dm=message.is_dm,
|
||||
)
|
||||
|
||||
# Route the message
|
||||
result = await self.router.route(message)
|
||||
|
||||
|
|
@ -271,10 +170,6 @@ class MeshAI:
|
|||
if not response:
|
||||
return
|
||||
|
||||
# Apply safety filter to LLM responses (2a)
|
||||
if result.route_type == RouteType.LLM:
|
||||
response = self.safety_filter.filter_response(response)
|
||||
|
||||
# Send response
|
||||
if message.is_dm:
|
||||
await self.responder.send_response(
|
||||
|
|
@ -292,21 +187,8 @@ class MeshAI:
|
|||
channel=message.channel,
|
||||
)
|
||||
|
||||
# Record response in rate limiter and status (2b, 2e)
|
||||
self.rate_limiter.record_message(message.sender_id)
|
||||
get_status_data().record_response()
|
||||
|
||||
# Send webhook event (2d)
|
||||
await self.webhook.on_response_sent(
|
||||
recipient_id=message.sender_id if message.is_dm else None,
|
||||
text=response,
|
||||
channel=message.channel,
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error handling message: {e}", exc_info=True)
|
||||
get_status_data().record_error(str(e))
|
||||
await self.webhook.on_error(str(e))
|
||||
|
||||
async def _load_summaries(self) -> None:
|
||||
"""Load persisted summaries from database into memory cache."""
|
||||
|
|
@ -417,14 +299,6 @@ def main() -> None:
|
|||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
# Handle SIGHUP for config reload
|
||||
def reload_handler(sig, frame):
|
||||
logger.info("Received SIGHUP - reloading config")
|
||||
# For now, just log - full reload would require more work
|
||||
# Could reload config and reinitialize components
|
||||
|
||||
signal.signal(signal.SIGHUP, reload_handler)
|
||||
|
||||
try:
|
||||
loop.run_until_complete(bot.start())
|
||||
except KeyboardInterrupt:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue