feat: backend servis iskeletini ve yönetim uçlarını ekle
This commit is contained in:
17
backend/README.md
Normal file
17
backend/README.md
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
# WiseClaw Backend
|
||||||
|
|
||||||
|
FastAPI service for WiseClaw. The backend now includes:
|
||||||
|
|
||||||
|
- SQLite persistence through SQLAlchemy
|
||||||
|
- runtime/admin settings endpoints
|
||||||
|
- Ollama integration status endpoint
|
||||||
|
- Telegram polling runtime scaffold
|
||||||
|
|
||||||
|
## Run locally
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python3.12 -m venv .venv312
|
||||||
|
source .venv312/bin/activate
|
||||||
|
pip install .
|
||||||
|
uvicorn app.main:app --reload
|
||||||
|
```
|
||||||
1
backend/app/__init__.py
Normal file
1
backend/app/__init__.py
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
1
backend/app/admin/__init__.py
Normal file
1
backend/app/admin/__init__.py
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
78
backend/app/admin/routes.py
Normal file
78
backend/app/admin/routes.py
Normal file
@@ -0,0 +1,78 @@
|
|||||||
|
from fastapi import APIRouter, Depends
|
||||||
|
from pydantic import BaseModel
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from app.admin.services import AdminService
|
||||||
|
from app.db import get_session
|
||||||
|
from app.llm.ollama_client import OllamaClient
|
||||||
|
from app.models import MemoryRecord, OllamaStatus, RuntimeSettings, TelegramStatus, UserRecord
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/admin", tags=["admin"])
|
||||||
|
|
||||||
|
|
||||||
|
class SecretPayload(BaseModel):
|
||||||
|
key: str
|
||||||
|
value: str
|
||||||
|
|
||||||
|
|
||||||
|
def get_admin_service(session: Session = Depends(get_session)) -> AdminService:
|
||||||
|
return AdminService(session)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/dashboard")
|
||||||
|
def get_dashboard(service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.dashboard()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/settings", response_model=RuntimeSettings)
|
||||||
|
def get_settings(service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.get_runtime_settings()
|
||||||
|
|
||||||
|
|
||||||
|
@router.put("/settings", response_model=RuntimeSettings)
|
||||||
|
def put_settings(payload: RuntimeSettings, service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.update_runtime_settings(payload)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/users", response_model=list[UserRecord])
|
||||||
|
def get_users(service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.list_users()
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/users", response_model=UserRecord)
|
||||||
|
def post_user(payload: UserRecord, service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.save_user(payload)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/memory", response_model=list[MemoryRecord])
|
||||||
|
def get_memory(service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.list_memory()
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete("/memory")
|
||||||
|
def delete_memory(service: AdminService = Depends(get_admin_service)):
|
||||||
|
service.clear_memory()
|
||||||
|
return {"status": "ok"}
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/secrets/{key}")
|
||||||
|
def get_secret(key: str, service: AdminService = Depends(get_admin_service)):
|
||||||
|
return {"key": key, "masked": service.get_secret_mask(key)}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/secrets")
|
||||||
|
def post_secret(payload: SecretPayload, service: AdminService = Depends(get_admin_service)):
|
||||||
|
service.save_secret(payload.key, payload.value)
|
||||||
|
return {"status": "ok"}
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/integrations/ollama", response_model=OllamaStatus)
|
||||||
|
async def get_ollama_status(service: AdminService = Depends(get_admin_service)):
|
||||||
|
runtime = service.get_runtime_settings()
|
||||||
|
client = OllamaClient(runtime.ollama_base_url)
|
||||||
|
return await client.status(runtime.default_model)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/integrations/telegram", response_model=TelegramStatus)
|
||||||
|
def get_telegram_status(service: AdminService = Depends(get_admin_service)):
|
||||||
|
return service.telegram_status()
|
||||||
142
backend/app/admin/services.py
Normal file
142
backend/app/admin/services.py
Normal file
@@ -0,0 +1,142 @@
|
|||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
from sqlalchemy import func, select
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from app.db import (
|
||||||
|
AuditLogORM,
|
||||||
|
AuthorizedUserORM,
|
||||||
|
MemoryItemORM,
|
||||||
|
SecretORM,
|
||||||
|
SettingORM,
|
||||||
|
ToolStateORM,
|
||||||
|
list_recent_logs,
|
||||||
|
)
|
||||||
|
from app.config import get_settings
|
||||||
|
from app.models import DashboardSnapshot, MemoryRecord, RuntimeSettings, TelegramStatus, ToolToggle, UserRecord
|
||||||
|
|
||||||
|
|
||||||
|
class AdminService:
|
||||||
|
def __init__(self, session: Session) -> None:
|
||||||
|
self.session = session
|
||||||
|
|
||||||
|
def get_runtime_settings(self) -> RuntimeSettings:
|
||||||
|
settings = {
|
||||||
|
item.key: item.value for item in self.session.scalars(select(SettingORM))
|
||||||
|
}
|
||||||
|
tools = list(self.session.scalars(select(ToolStateORM).order_by(ToolStateORM.name.asc())))
|
||||||
|
return RuntimeSettings(
|
||||||
|
terminal_mode=int(settings["terminal_mode"]),
|
||||||
|
search_provider=settings["search_provider"],
|
||||||
|
ollama_base_url=settings["ollama_base_url"],
|
||||||
|
default_model=settings["default_model"],
|
||||||
|
tools=[ToolToggle(name=tool.name, enabled=tool.enabled) for tool in tools],
|
||||||
|
)
|
||||||
|
|
||||||
|
def update_runtime_settings(self, payload: RuntimeSettings) -> RuntimeSettings:
|
||||||
|
self._save_setting("terminal_mode", str(payload.terminal_mode))
|
||||||
|
self._save_setting("search_provider", payload.search_provider)
|
||||||
|
self._save_setting("ollama_base_url", payload.ollama_base_url)
|
||||||
|
self._save_setting("default_model", payload.default_model)
|
||||||
|
|
||||||
|
for tool in payload.tools:
|
||||||
|
record = self.session.get(ToolStateORM, tool.name)
|
||||||
|
if record is None:
|
||||||
|
self.session.add(ToolStateORM(name=tool.name, enabled=tool.enabled, updated_at=datetime.utcnow()))
|
||||||
|
else:
|
||||||
|
record.enabled = tool.enabled
|
||||||
|
record.updated_at = datetime.utcnow()
|
||||||
|
|
||||||
|
self.session.add(AuditLogORM(category="settings", message="settings:runtime-updated"))
|
||||||
|
self.session.commit()
|
||||||
|
return self.get_runtime_settings()
|
||||||
|
|
||||||
|
def dashboard(self) -> DashboardSnapshot:
|
||||||
|
return DashboardSnapshot(
|
||||||
|
settings=self.get_runtime_settings(),
|
||||||
|
whitelist_count=self.session.scalar(select(func.count()).select_from(AuthorizedUserORM)) or 0,
|
||||||
|
memory_items=self.session.scalar(select(func.count()).select_from(MemoryItemORM)) or 0,
|
||||||
|
recent_logs=list_recent_logs(self.session, limit=10),
|
||||||
|
)
|
||||||
|
|
||||||
|
def list_users(self) -> list[UserRecord]:
|
||||||
|
stmt = select(AuthorizedUserORM).order_by(AuthorizedUserORM.created_at.desc())
|
||||||
|
return [
|
||||||
|
UserRecord(
|
||||||
|
telegram_user_id=user.telegram_user_id,
|
||||||
|
username=user.username,
|
||||||
|
display_name=user.display_name,
|
||||||
|
is_active=user.is_active,
|
||||||
|
)
|
||||||
|
for user in self.session.scalars(stmt)
|
||||||
|
]
|
||||||
|
|
||||||
|
def save_user(self, user: UserRecord) -> UserRecord:
|
||||||
|
record = self.session.get(AuthorizedUserORM, user.telegram_user_id)
|
||||||
|
if record is None:
|
||||||
|
record = AuthorizedUserORM(
|
||||||
|
telegram_user_id=user.telegram_user_id,
|
||||||
|
username=user.username,
|
||||||
|
display_name=user.display_name,
|
||||||
|
is_active=user.is_active,
|
||||||
|
created_at=datetime.utcnow(),
|
||||||
|
updated_at=datetime.utcnow(),
|
||||||
|
)
|
||||||
|
self.session.add(record)
|
||||||
|
else:
|
||||||
|
record.username = user.username
|
||||||
|
record.display_name = user.display_name
|
||||||
|
record.is_active = user.is_active
|
||||||
|
record.updated_at = datetime.utcnow()
|
||||||
|
self.session.add(AuditLogORM(category="users", message=f"users:upsert:{user.telegram_user_id}"))
|
||||||
|
self.session.commit()
|
||||||
|
return user
|
||||||
|
|
||||||
|
def list_memory(self) -> list[MemoryRecord]:
|
||||||
|
stmt = select(MemoryItemORM).order_by(MemoryItemORM.created_at.desc(), MemoryItemORM.id.desc()).limit(50)
|
||||||
|
return [
|
||||||
|
MemoryRecord(id=item.id, content=item.content, kind=item.kind, created_at=item.created_at)
|
||||||
|
for item in self.session.scalars(stmt)
|
||||||
|
]
|
||||||
|
|
||||||
|
def clear_memory(self) -> None:
|
||||||
|
for item in self.session.scalars(select(MemoryItemORM)):
|
||||||
|
self.session.delete(item)
|
||||||
|
self.session.add(AuditLogORM(category="memory", message="memory:cleared"))
|
||||||
|
self.session.commit()
|
||||||
|
|
||||||
|
def get_secret_mask(self, key: str) -> str:
|
||||||
|
record = self.session.get(SecretORM, key)
|
||||||
|
value = record.value if record else ""
|
||||||
|
if len(value) < 4:
|
||||||
|
return ""
|
||||||
|
return f"{value[:2]}***{value[-2:]}"
|
||||||
|
|
||||||
|
def save_secret(self, key: str, value: str) -> None:
|
||||||
|
record = self.session.get(SecretORM, key)
|
||||||
|
if record is None:
|
||||||
|
self.session.add(SecretORM(key=key, value=value, updated_at=datetime.utcnow()))
|
||||||
|
else:
|
||||||
|
record.value = value
|
||||||
|
record.updated_at = datetime.utcnow()
|
||||||
|
self.session.add(AuditLogORM(category="secrets", message=f"secrets:updated:{key}"))
|
||||||
|
self.session.commit()
|
||||||
|
|
||||||
|
def _save_setting(self, key: str, value: str) -> None:
|
||||||
|
record = self.session.get(SettingORM, key)
|
||||||
|
if record is None:
|
||||||
|
self.session.add(SettingORM(key=key, value=value, updated_at=datetime.utcnow()))
|
||||||
|
else:
|
||||||
|
record.value = value
|
||||||
|
record.updated_at = datetime.utcnow()
|
||||||
|
|
||||||
|
def telegram_status(self) -> TelegramStatus:
|
||||||
|
settings = get_settings()
|
||||||
|
configured = bool(settings.telegram_bot_token)
|
||||||
|
return TelegramStatus(
|
||||||
|
configured=configured,
|
||||||
|
polling_active=False,
|
||||||
|
message="Telegram token is configured. Polling starts when the backend boots."
|
||||||
|
if configured
|
||||||
|
else "Telegram token is not configured.",
|
||||||
|
)
|
||||||
28
backend/app/config.py
Normal file
28
backend/app/config.py
Normal file
@@ -0,0 +1,28 @@
|
|||||||
|
from functools import lru_cache
|
||||||
|
|
||||||
|
from pydantic import Field
|
||||||
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||||
|
|
||||||
|
|
||||||
|
class Settings(BaseSettings):
|
||||||
|
model_config = SettingsConfigDict(
|
||||||
|
env_file=".env",
|
||||||
|
env_prefix="WISECLAW_",
|
||||||
|
extra="ignore",
|
||||||
|
)
|
||||||
|
|
||||||
|
env: str = "development"
|
||||||
|
db_url: str = "sqlite:///./wiseclaw.db"
|
||||||
|
admin_host: str = "127.0.0.1"
|
||||||
|
admin_port: int = 8000
|
||||||
|
ollama_base_url: str = "http://127.0.0.1:11434"
|
||||||
|
default_model: str = "qwen3.5:4b"
|
||||||
|
search_provider: str = "brave"
|
||||||
|
telegram_bot_token: str = Field(default="", repr=False)
|
||||||
|
brave_api_key: str = Field(default="", repr=False)
|
||||||
|
|
||||||
|
|
||||||
|
@lru_cache
|
||||||
|
def get_settings() -> Settings:
|
||||||
|
return Settings()
|
||||||
|
|
||||||
133
backend/app/db.py
Normal file
133
backend/app/db.py
Normal file
@@ -0,0 +1,133 @@
|
|||||||
|
from collections.abc import Iterator
|
||||||
|
from contextlib import contextmanager
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
from sqlalchemy import Boolean, DateTime, Integer, String, Text, create_engine, select
|
||||||
|
from sqlalchemy.orm import DeclarativeBase, Mapped, Session, mapped_column, sessionmaker
|
||||||
|
|
||||||
|
from app.config import get_settings
|
||||||
|
|
||||||
|
|
||||||
|
DEFAULT_SETTINGS = {
|
||||||
|
"terminal_mode": "3",
|
||||||
|
"search_provider": "brave",
|
||||||
|
"ollama_base_url": "http://127.0.0.1:11434",
|
||||||
|
"default_model": "qwen3.5:4b",
|
||||||
|
}
|
||||||
|
|
||||||
|
DEFAULT_TOOLS = {
|
||||||
|
"brave_search": True,
|
||||||
|
"searxng_search": False,
|
||||||
|
"web_fetch": True,
|
||||||
|
"apple_notes": True,
|
||||||
|
"files": True,
|
||||||
|
"terminal": True,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class Base(DeclarativeBase):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class SettingORM(Base):
|
||||||
|
__tablename__ = "settings"
|
||||||
|
|
||||||
|
key: Mapped[str] = mapped_column(String(100), primary_key=True)
|
||||||
|
value: Mapped[str] = mapped_column(Text, nullable=False)
|
||||||
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
|
||||||
|
|
||||||
|
class ToolStateORM(Base):
|
||||||
|
__tablename__ = "tool_states"
|
||||||
|
|
||||||
|
name: Mapped[str] = mapped_column(String(100), primary_key=True)
|
||||||
|
enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||||||
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
|
||||||
|
|
||||||
|
class AuthorizedUserORM(Base):
|
||||||
|
__tablename__ = "authorized_users"
|
||||||
|
|
||||||
|
telegram_user_id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||||
|
username: Mapped[str | None] = mapped_column(String(255))
|
||||||
|
display_name: Mapped[str | None] = mapped_column(String(255))
|
||||||
|
is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
||||||
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
|
||||||
|
|
||||||
|
class MemoryItemORM(Base):
|
||||||
|
__tablename__ = "memory_items"
|
||||||
|
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
content: Mapped[str] = mapped_column(Text, nullable=False)
|
||||||
|
kind: Mapped[str] = mapped_column(String(50), nullable=False, default="message")
|
||||||
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
|
||||||
|
|
||||||
|
class AuditLogORM(Base):
|
||||||
|
__tablename__ = "audit_logs"
|
||||||
|
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
category: Mapped[str] = mapped_column(String(50), nullable=False)
|
||||||
|
message: Mapped[str] = mapped_column(Text, nullable=False)
|
||||||
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
|
||||||
|
|
||||||
|
class SecretORM(Base):
|
||||||
|
__tablename__ = "secrets"
|
||||||
|
|
||||||
|
key: Mapped[str] = mapped_column(String(100), primary_key=True)
|
||||||
|
value: Mapped[str] = mapped_column(Text, nullable=False)
|
||||||
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||||
|
|
||||||
|
|
||||||
|
settings = get_settings()
|
||||||
|
engine = create_engine(
|
||||||
|
settings.db_url,
|
||||||
|
connect_args={"check_same_thread": False} if settings.db_url.startswith("sqlite") else {},
|
||||||
|
)
|
||||||
|
SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False)
|
||||||
|
|
||||||
|
|
||||||
|
def init_db() -> None:
|
||||||
|
Base.metadata.create_all(bind=engine)
|
||||||
|
with session_scope() as session:
|
||||||
|
_seed_defaults(session)
|
||||||
|
|
||||||
|
|
||||||
|
def _seed_defaults(session: Session) -> None:
|
||||||
|
for key, value in DEFAULT_SETTINGS.items():
|
||||||
|
if session.get(SettingORM, key) is None:
|
||||||
|
session.add(SettingORM(key=key, value=value))
|
||||||
|
|
||||||
|
for name, enabled in DEFAULT_TOOLS.items():
|
||||||
|
if session.get(ToolStateORM, name) is None:
|
||||||
|
session.add(ToolStateORM(name=name, enabled=enabled))
|
||||||
|
|
||||||
|
|
||||||
|
def get_session() -> Iterator[Session]:
|
||||||
|
session = SessionLocal()
|
||||||
|
try:
|
||||||
|
yield session
|
||||||
|
finally:
|
||||||
|
session.close()
|
||||||
|
|
||||||
|
|
||||||
|
@contextmanager
|
||||||
|
def session_scope() -> Iterator[Session]:
|
||||||
|
session = SessionLocal()
|
||||||
|
try:
|
||||||
|
yield session
|
||||||
|
session.commit()
|
||||||
|
except Exception:
|
||||||
|
session.rollback()
|
||||||
|
raise
|
||||||
|
finally:
|
||||||
|
session.close()
|
||||||
|
|
||||||
|
|
||||||
|
def list_recent_logs(session: Session, limit: int = 10) -> list[str]:
|
||||||
|
stmt = select(AuditLogORM).order_by(AuditLogORM.created_at.desc(), AuditLogORM.id.desc()).limit(limit)
|
||||||
|
return [row.message for row in session.scalars(stmt)]
|
||||||
|
|
||||||
1
backend/app/llm/__init__.py
Normal file
1
backend/app/llm/__init__.py
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
37
backend/app/llm/ollama_client.py
Normal file
37
backend/app/llm/ollama_client.py
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
import httpx
|
||||||
|
from httpx import HTTPError
|
||||||
|
|
||||||
|
from app.models import OllamaStatus
|
||||||
|
|
||||||
|
class OllamaClient:
|
||||||
|
def __init__(self, base_url: str) -> None:
|
||||||
|
self.base_url = base_url.rstrip("/")
|
||||||
|
|
||||||
|
async def health(self) -> bool:
|
||||||
|
async with httpx.AsyncClient(timeout=5.0) as client:
|
||||||
|
response = await client.get(f"{self.base_url}/api/tags")
|
||||||
|
return response.is_success
|
||||||
|
|
||||||
|
async def status(self, model: str) -> OllamaStatus:
|
||||||
|
try:
|
||||||
|
async with httpx.AsyncClient(timeout=5.0) as client:
|
||||||
|
response = await client.get(f"{self.base_url}/api/tags")
|
||||||
|
response.raise_for_status()
|
||||||
|
except HTTPError as exc:
|
||||||
|
return OllamaStatus(
|
||||||
|
reachable=False,
|
||||||
|
base_url=self.base_url,
|
||||||
|
model=model,
|
||||||
|
message=f"Ollama unreachable: {exc}",
|
||||||
|
)
|
||||||
|
|
||||||
|
payload = response.json()
|
||||||
|
installed_models = [item.get("name", "") for item in payload.get("models", []) if item.get("name")]
|
||||||
|
has_model = model in installed_models
|
||||||
|
return OllamaStatus(
|
||||||
|
reachable=True,
|
||||||
|
base_url=self.base_url,
|
||||||
|
model=model,
|
||||||
|
installed_models=installed_models,
|
||||||
|
message="Model found." if has_model else "Ollama reachable but model is not installed.",
|
||||||
|
)
|
||||||
15
backend/app/llm/planner.py
Normal file
15
backend/app/llm/planner.py
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
from app.models import RuntimeSettings
|
||||||
|
|
||||||
|
|
||||||
|
def build_prompt_context(message: str, runtime: RuntimeSettings, memory: list[str]) -> dict[str, object]:
|
||||||
|
return {
|
||||||
|
"system": (
|
||||||
|
"You are WiseClaw, a local-first assistant running on macOS. "
|
||||||
|
"Use tools carefully and obey terminal safety mode."
|
||||||
|
),
|
||||||
|
"message": message,
|
||||||
|
"model": runtime.default_model,
|
||||||
|
"memory": memory,
|
||||||
|
"available_tools": [tool.name for tool in runtime.tools if tool.enabled],
|
||||||
|
}
|
||||||
|
|
||||||
56
backend/app/main.py
Normal file
56
backend/app/main.py
Normal file
@@ -0,0 +1,56 @@
|
|||||||
|
from contextlib import asynccontextmanager
|
||||||
|
|
||||||
|
from fastapi import FastAPI
|
||||||
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
|
|
||||||
|
from app.admin.routes import router as admin_router
|
||||||
|
from app.config import get_settings
|
||||||
|
from app.db import init_db, session_scope
|
||||||
|
from app.models import HealthStatus
|
||||||
|
from app.orchestrator import WiseClawOrchestrator
|
||||||
|
from app.runtime import RuntimeServices
|
||||||
|
from app.telegram.bot import TelegramBotService
|
||||||
|
|
||||||
|
settings = get_settings()
|
||||||
|
runtime_services = RuntimeServices()
|
||||||
|
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def lifespan(_: FastAPI):
|
||||||
|
init_db()
|
||||||
|
if settings.telegram_bot_token:
|
||||||
|
runtime_services.telegram_bot = TelegramBotService(settings.telegram_bot_token, session_scope)
|
||||||
|
await runtime_services.telegram_bot.start()
|
||||||
|
yield
|
||||||
|
await runtime_services.shutdown()
|
||||||
|
|
||||||
|
|
||||||
|
app = FastAPI(title="WiseClaw", version="0.1.0", lifespan=lifespan)
|
||||||
|
|
||||||
|
app.add_middleware(
|
||||||
|
CORSMiddleware,
|
||||||
|
allow_origins=["http://127.0.0.1:5173", "http://localhost:5173"],
|
||||||
|
allow_credentials=True,
|
||||||
|
allow_methods=["*"],
|
||||||
|
allow_headers=["*"],
|
||||||
|
)
|
||||||
|
|
||||||
|
app.include_router(admin_router)
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/health", response_model=HealthStatus)
|
||||||
|
def health() -> HealthStatus:
|
||||||
|
return HealthStatus()
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/bootstrap")
|
||||||
|
def bootstrap() -> dict[str, object]:
|
||||||
|
with session_scope() as session:
|
||||||
|
orchestrator = WiseClawOrchestrator(session)
|
||||||
|
runtime = orchestrator.get_runtime_settings().model_dump()
|
||||||
|
return {
|
||||||
|
"env": settings.env,
|
||||||
|
"admin_host": settings.admin_host,
|
||||||
|
"admin_port": settings.admin_port,
|
||||||
|
"runtime": runtime,
|
||||||
|
}
|
||||||
1
backend/app/memory/__init__.py
Normal file
1
backend/app/memory/__init__.py
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
17
backend/app/memory/store.py
Normal file
17
backend/app/memory/store.py
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
from sqlalchemy import select
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from app.db import MemoryItemORM
|
||||||
|
|
||||||
|
|
||||||
|
class MemoryService:
|
||||||
|
def __init__(self, session: Session) -> None:
|
||||||
|
self.session = session
|
||||||
|
|
||||||
|
def add_item(self, content: str) -> None:
|
||||||
|
self.session.add(MemoryItemORM(content=content, kind="message"))
|
||||||
|
self.session.flush()
|
||||||
|
|
||||||
|
def latest_items(self, limit: int = 10) -> list[str]:
|
||||||
|
stmt = select(MemoryItemORM).order_by(MemoryItemORM.created_at.desc(), MemoryItemORM.id.desc()).limit(limit)
|
||||||
|
return [item.content for item in self.session.scalars(stmt)]
|
||||||
77
backend/app/models.py
Normal file
77
backend/app/models.py
Normal file
@@ -0,0 +1,77 @@
|
|||||||
|
from datetime import datetime
|
||||||
|
from typing import Literal
|
||||||
|
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
|
TerminalMode = Literal[1, 2, 3]
|
||||||
|
SearchProvider = Literal["brave", "searxng"]
|
||||||
|
|
||||||
|
|
||||||
|
class HealthStatus(BaseModel):
|
||||||
|
service: str = "wiseclaw"
|
||||||
|
status: str = "ok"
|
||||||
|
timestamp: datetime = Field(default_factory=datetime.utcnow)
|
||||||
|
|
||||||
|
|
||||||
|
class SettingRecord(BaseModel):
|
||||||
|
key: str
|
||||||
|
value: str
|
||||||
|
updated_at: datetime = Field(default_factory=datetime.utcnow)
|
||||||
|
|
||||||
|
|
||||||
|
class ToolToggle(BaseModel):
|
||||||
|
name: str
|
||||||
|
enabled: bool
|
||||||
|
|
||||||
|
|
||||||
|
class UserRecord(BaseModel):
|
||||||
|
telegram_user_id: int
|
||||||
|
username: str | None = None
|
||||||
|
display_name: str | None = None
|
||||||
|
is_active: bool = True
|
||||||
|
|
||||||
|
|
||||||
|
class RuntimeSettings(BaseModel):
|
||||||
|
terminal_mode: TerminalMode = 3
|
||||||
|
search_provider: SearchProvider = "brave"
|
||||||
|
ollama_base_url: str = "http://127.0.0.1:11434"
|
||||||
|
default_model: str = "qwen3.5:4b"
|
||||||
|
tools: list[ToolToggle] = Field(
|
||||||
|
default_factory=lambda: [
|
||||||
|
ToolToggle(name="brave_search", enabled=True),
|
||||||
|
ToolToggle(name="searxng_search", enabled=False),
|
||||||
|
ToolToggle(name="web_fetch", enabled=True),
|
||||||
|
ToolToggle(name="apple_notes", enabled=True),
|
||||||
|
ToolToggle(name="files", enabled=True),
|
||||||
|
ToolToggle(name="terminal", enabled=True),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class DashboardSnapshot(BaseModel):
|
||||||
|
settings: RuntimeSettings
|
||||||
|
whitelist_count: int
|
||||||
|
memory_items: int
|
||||||
|
recent_logs: list[str]
|
||||||
|
|
||||||
|
|
||||||
|
class MemoryRecord(BaseModel):
|
||||||
|
id: int
|
||||||
|
content: str
|
||||||
|
kind: str
|
||||||
|
created_at: datetime
|
||||||
|
|
||||||
|
|
||||||
|
class OllamaStatus(BaseModel):
|
||||||
|
reachable: bool
|
||||||
|
base_url: str
|
||||||
|
model: str
|
||||||
|
installed_models: list[str] = Field(default_factory=list)
|
||||||
|
message: str
|
||||||
|
|
||||||
|
|
||||||
|
class TelegramStatus(BaseModel):
|
||||||
|
configured: bool
|
||||||
|
polling_active: bool
|
||||||
|
message: str
|
||||||
46
backend/app/orchestrator.py
Normal file
46
backend/app/orchestrator.py
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
from sqlalchemy import select
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from app.db import AuditLogORM, SettingORM, ToolStateORM
|
||||||
|
from app.llm.planner import build_prompt_context
|
||||||
|
from app.memory.store import MemoryService
|
||||||
|
from app.models import RuntimeSettings
|
||||||
|
from app.telegram.auth import is_authorized
|
||||||
|
|
||||||
|
|
||||||
|
class WiseClawOrchestrator:
|
||||||
|
def __init__(self, session: Session) -> None:
|
||||||
|
self.session = session
|
||||||
|
self.memory = MemoryService(session)
|
||||||
|
|
||||||
|
def get_runtime_settings(self) -> RuntimeSettings:
|
||||||
|
settings = {
|
||||||
|
item.key: item.value for item in self.session.scalars(select(SettingORM))
|
||||||
|
}
|
||||||
|
tools = list(self.session.scalars(select(ToolStateORM).order_by(ToolStateORM.name.asc())))
|
||||||
|
return RuntimeSettings(
|
||||||
|
terminal_mode=int(settings["terminal_mode"]),
|
||||||
|
search_provider=settings["search_provider"],
|
||||||
|
ollama_base_url=settings["ollama_base_url"],
|
||||||
|
default_model=settings["default_model"],
|
||||||
|
tools=[{"name": tool.name, "enabled": tool.enabled} for tool in tools],
|
||||||
|
)
|
||||||
|
|
||||||
|
def handle_text_message(self, telegram_user_id: int, text: str) -> str:
|
||||||
|
if not is_authorized(self.session, telegram_user_id):
|
||||||
|
return "This Telegram user is not authorized for WiseClaw."
|
||||||
|
|
||||||
|
self.memory.add_item(f"user:{telegram_user_id}:{text}")
|
||||||
|
context = build_prompt_context(
|
||||||
|
message=text,
|
||||||
|
runtime=self.get_runtime_settings(),
|
||||||
|
memory=self.memory.latest_items(limit=5),
|
||||||
|
)
|
||||||
|
response = (
|
||||||
|
"WiseClaw scaffold received your message.\n\n"
|
||||||
|
f"Prompt context prepared for model `{context['model']}` with "
|
||||||
|
f"{len(context['memory'])} memory items."
|
||||||
|
)
|
||||||
|
self.session.add(AuditLogORM(category="telegram", message=f"telegram:{telegram_user_id}:{text}"))
|
||||||
|
self.session.commit()
|
||||||
|
return response
|
||||||
14
backend/app/runtime.py
Normal file
14
backend/app/runtime.py
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
from contextlib import suppress
|
||||||
|
|
||||||
|
from app.telegram.bot import TelegramBotService
|
||||||
|
|
||||||
|
|
||||||
|
class RuntimeServices:
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self.telegram_bot: TelegramBotService | None = None
|
||||||
|
|
||||||
|
async def shutdown(self) -> None:
|
||||||
|
if self.telegram_bot is not None:
|
||||||
|
with suppress(Exception):
|
||||||
|
await self.telegram_bot.stop()
|
||||||
|
|
||||||
77
backend/app/security.py
Normal file
77
backend/app/security.py
Normal file
@@ -0,0 +1,77 @@
|
|||||||
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
|
||||||
|
SAFE_COMMAND_PREFIXES = (
|
||||||
|
"pwd",
|
||||||
|
"ls",
|
||||||
|
"cat",
|
||||||
|
"head",
|
||||||
|
"tail",
|
||||||
|
"find",
|
||||||
|
"rg",
|
||||||
|
"wc",
|
||||||
|
"stat",
|
||||||
|
"git status",
|
||||||
|
"git diff",
|
||||||
|
"git log",
|
||||||
|
"git show",
|
||||||
|
"date",
|
||||||
|
"whoami",
|
||||||
|
"uname",
|
||||||
|
"ps",
|
||||||
|
)
|
||||||
|
|
||||||
|
APPROVAL_REQUIRED_PREFIXES = (
|
||||||
|
"curl",
|
||||||
|
"wget",
|
||||||
|
"pip",
|
||||||
|
"npm",
|
||||||
|
"python",
|
||||||
|
"python3",
|
||||||
|
"node",
|
||||||
|
"git commit",
|
||||||
|
"git push",
|
||||||
|
"pkill",
|
||||||
|
"kill",
|
||||||
|
"touch",
|
||||||
|
"echo ",
|
||||||
|
)
|
||||||
|
|
||||||
|
BLOCKED_PATTERNS = (
|
||||||
|
"sudo ",
|
||||||
|
"rm -rf",
|
||||||
|
"chmod ",
|
||||||
|
"chown ",
|
||||||
|
";",
|
||||||
|
"&&",
|
||||||
|
"||",
|
||||||
|
"$(",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class TerminalDecision:
|
||||||
|
decision: str
|
||||||
|
reason: str
|
||||||
|
|
||||||
|
|
||||||
|
def evaluate_terminal_command(command: str, mode: int) -> TerminalDecision:
|
||||||
|
normalized = command.strip()
|
||||||
|
|
||||||
|
if any(pattern in normalized for pattern in BLOCKED_PATTERNS):
|
||||||
|
return TerminalDecision(decision="blocked", reason="Blocked by hard policy.")
|
||||||
|
|
||||||
|
if mode == 1:
|
||||||
|
return TerminalDecision(decision="allow", reason="Terminal mode 1 auto-runs commands.")
|
||||||
|
|
||||||
|
if mode == 2:
|
||||||
|
return TerminalDecision(decision="approval", reason="Terminal mode 2 requires approval.")
|
||||||
|
|
||||||
|
if normalized.startswith(SAFE_COMMAND_PREFIXES):
|
||||||
|
return TerminalDecision(decision="allow", reason="Safe read-only command.")
|
||||||
|
|
||||||
|
if normalized.startswith(APPROVAL_REQUIRED_PREFIXES):
|
||||||
|
return TerminalDecision(decision="approval", reason="Command needs approval.")
|
||||||
|
|
||||||
|
return TerminalDecision(decision="approval", reason="Unknown command defaults to approval.")
|
||||||
|
|
||||||
1
backend/app/telegram/__init__.py
Normal file
1
backend/app/telegram/__init__.py
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
8
backend/app/telegram/auth.py
Normal file
8
backend/app/telegram/auth.py
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from app.db import AuthorizedUserORM
|
||||||
|
|
||||||
|
|
||||||
|
def is_authorized(session: Session, telegram_user_id: int) -> bool:
|
||||||
|
record = session.get(AuthorizedUserORM, telegram_user_id)
|
||||||
|
return bool(record and record.is_active)
|
||||||
51
backend/app/telegram/bot.py
Normal file
51
backend/app/telegram/bot.py
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from telegram import Update
|
||||||
|
from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters
|
||||||
|
|
||||||
|
from app.orchestrator import WiseClawOrchestrator
|
||||||
|
|
||||||
|
|
||||||
|
class TelegramBotService:
|
||||||
|
def __init__(self, token: str, orchestrator_factory: Any) -> None:
|
||||||
|
self.token = token
|
||||||
|
self.orchestrator_factory = orchestrator_factory
|
||||||
|
self.application: Application | None = None
|
||||||
|
|
||||||
|
async def process_message(self, telegram_user_id: int, text: str) -> str:
|
||||||
|
with self.orchestrator_factory() as session:
|
||||||
|
orchestrator = WiseClawOrchestrator(session)
|
||||||
|
return orchestrator.handle_text_message(telegram_user_id=telegram_user_id, text=text)
|
||||||
|
|
||||||
|
async def start(self) -> None:
|
||||||
|
if not self.token:
|
||||||
|
return
|
||||||
|
self.application = Application.builder().token(self.token).build()
|
||||||
|
self.application.add_handler(CommandHandler("start", self._on_start))
|
||||||
|
self.application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self._on_text))
|
||||||
|
await self.application.initialize()
|
||||||
|
await self.application.start()
|
||||||
|
await self.application.updater.start_polling(drop_pending_updates=True)
|
||||||
|
|
||||||
|
async def stop(self) -> None:
|
||||||
|
if self.application is None:
|
||||||
|
return
|
||||||
|
await self.application.updater.stop()
|
||||||
|
await self.application.stop()
|
||||||
|
await self.application.shutdown()
|
||||||
|
self.application = None
|
||||||
|
|
||||||
|
async def _on_start(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||||
|
del context
|
||||||
|
if update.message is None or update.effective_user is None:
|
||||||
|
return
|
||||||
|
await update.message.reply_text(
|
||||||
|
"WiseClaw is online. If your Telegram user is whitelisted, send a message to start."
|
||||||
|
)
|
||||||
|
|
||||||
|
async def _on_text(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||||
|
del context
|
||||||
|
if update.message is None or update.effective_user is None or update.message.text is None:
|
||||||
|
return
|
||||||
|
reply = await self.process_message(update.effective_user.id, update.message.text)
|
||||||
|
await update.message.reply_text(reply)
|
||||||
1
backend/app/tools/__init__.py
Normal file
1
backend/app/tools/__init__.py
Normal file
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
18
backend/app/tools/apple_notes.py
Normal file
18
backend/app/tools/apple_notes.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from app.tools.base import Tool
|
||||||
|
|
||||||
|
|
||||||
|
class AppleNotesTool(Tool):
|
||||||
|
name = "apple_notes"
|
||||||
|
description = "Create notes in Apple Notes through AppleScript."
|
||||||
|
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
title = str(payload.get("title", "")).strip()
|
||||||
|
return {
|
||||||
|
"tool": self.name,
|
||||||
|
"status": "stub",
|
||||||
|
"title": title,
|
||||||
|
"message": "Apple Notes integration is not wired yet.",
|
||||||
|
}
|
||||||
|
|
||||||
12
backend/app/tools/base.py
Normal file
12
backend/app/tools/base.py
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
from abc import ABC, abstractmethod
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
|
||||||
|
class Tool(ABC):
|
||||||
|
name: str
|
||||||
|
description: str
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
18
backend/app/tools/brave_search.py
Normal file
18
backend/app/tools/brave_search.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from app.tools.base import Tool
|
||||||
|
|
||||||
|
|
||||||
|
class BraveSearchTool(Tool):
|
||||||
|
name = "brave_search"
|
||||||
|
description = "Search the web with Brave Search."
|
||||||
|
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
query = str(payload.get("query", "")).strip()
|
||||||
|
return {
|
||||||
|
"tool": self.name,
|
||||||
|
"status": "stub",
|
||||||
|
"query": query,
|
||||||
|
"message": "Brave Search integration is not wired yet.",
|
||||||
|
}
|
||||||
|
|
||||||
21
backend/app/tools/files.py
Normal file
21
backend/app/tools/files.py
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from app.tools.base import Tool
|
||||||
|
|
||||||
|
|
||||||
|
class FilesTool(Tool):
|
||||||
|
name = "files"
|
||||||
|
description = "Read and write files within allowed paths."
|
||||||
|
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
action = str(payload.get("action", "read")).strip()
|
||||||
|
path = Path(str(payload.get("path", "")).strip()).expanduser()
|
||||||
|
return {
|
||||||
|
"tool": self.name,
|
||||||
|
"status": "stub",
|
||||||
|
"action": action,
|
||||||
|
"path": str(path),
|
||||||
|
"message": "File integration is not wired yet.",
|
||||||
|
}
|
||||||
|
|
||||||
18
backend/app/tools/searxng_search.py
Normal file
18
backend/app/tools/searxng_search.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from app.tools.base import Tool
|
||||||
|
|
||||||
|
|
||||||
|
class SearXNGSearchTool(Tool):
|
||||||
|
name = "searxng_search"
|
||||||
|
description = "Search the web through a SearXNG instance."
|
||||||
|
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
query = str(payload.get("query", "")).strip()
|
||||||
|
return {
|
||||||
|
"tool": self.name,
|
||||||
|
"status": "stub",
|
||||||
|
"query": query,
|
||||||
|
"message": "SearXNG integration is not wired yet.",
|
||||||
|
}
|
||||||
|
|
||||||
24
backend/app/tools/terminal.py
Normal file
24
backend/app/tools/terminal.py
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from app.security import evaluate_terminal_command
|
||||||
|
from app.tools.base import Tool
|
||||||
|
|
||||||
|
|
||||||
|
class TerminalTool(Tool):
|
||||||
|
name = "terminal"
|
||||||
|
description = "Run terminal commands under WiseClaw policy."
|
||||||
|
|
||||||
|
def __init__(self, terminal_mode: int) -> None:
|
||||||
|
self.terminal_mode = terminal_mode
|
||||||
|
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
command = str(payload.get("command", "")).strip()
|
||||||
|
decision = evaluate_terminal_command(command, self.terminal_mode)
|
||||||
|
return {
|
||||||
|
"tool": self.name,
|
||||||
|
"status": "stub",
|
||||||
|
"command": command,
|
||||||
|
"decision": decision.decision,
|
||||||
|
"reason": decision.reason,
|
||||||
|
}
|
||||||
|
|
||||||
18
backend/app/tools/web_fetch.py
Normal file
18
backend/app/tools/web_fetch.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from app.tools.base import Tool
|
||||||
|
|
||||||
|
|
||||||
|
class WebFetchTool(Tool):
|
||||||
|
name = "web_fetch"
|
||||||
|
description = "Fetch a webpage and return simplified content."
|
||||||
|
|
||||||
|
async def run(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||||
|
url = str(payload.get("url", "")).strip()
|
||||||
|
return {
|
||||||
|
"tool": self.name,
|
||||||
|
"status": "stub",
|
||||||
|
"url": url,
|
||||||
|
"message": "Web fetch integration is not wired yet.",
|
||||||
|
}
|
||||||
|
|
||||||
22
backend/pyproject.toml
Normal file
22
backend/pyproject.toml
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
[build-system]
|
||||||
|
requires = ["setuptools>=68", "wheel"]
|
||||||
|
build-backend = "setuptools.build_meta"
|
||||||
|
|
||||||
|
[project]
|
||||||
|
name = "wiseclaw-backend"
|
||||||
|
version = "0.1.0"
|
||||||
|
description = "FastAPI backend for WiseClaw"
|
||||||
|
readme = "README.md"
|
||||||
|
requires-python = ">=3.11"
|
||||||
|
dependencies = [
|
||||||
|
"fastapi>=0.116.0,<1.0.0",
|
||||||
|
"uvicorn[standard]>=0.35.0,<1.0.0",
|
||||||
|
"pydantic-settings>=2.10.0,<3.0.0",
|
||||||
|
"sqlalchemy>=2.0.39,<3.0.0",
|
||||||
|
"httpx>=0.28.0,<1.0.0",
|
||||||
|
"python-telegram-bot>=22.0,<23.0",
|
||||||
|
]
|
||||||
|
|
||||||
|
[tool.setuptools.packages.find]
|
||||||
|
where = ["."]
|
||||||
|
|
||||||
Reference in New Issue
Block a user