mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-09-11 12:17:45 +02:00
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from typing import Optional, List
|
|
|
|
from backend.apps.modes.modes import get_mode_by_id
|
|
from backend.apps.modes.Mode import Mode
|
|
from backend.apps.settings.settings import load_settings
|
|
from backend.apps.settings.AppSettings import AppSettings
|
|
from backend.apps.agents.ResolvedModeConfig.compose_system_prompt import compose_system_prompt
|
|
from backend.core.tools.shared_structs.Toolkit import Toolkit
|
|
from typeguard import typechecked
|
|
from pydantic import BaseModel, Field
|
|
|
|
class ResolvedModeConfig(BaseModel):
|
|
system_prompt: Optional[str] = None
|
|
allowed_tools: List[str] = Field(default_factory=list)
|
|
disallowed_tools: List[str] = Field(default_factory=list)
|
|
cwd: Optional[str] = None
|
|
|
|
@typechecked
|
|
@classmethod
|
|
async def create(
|
|
cls,
|
|
mode_id: str,
|
|
toolkit: Toolkit,
|
|
session_prompt: Optional[str] = None,
|
|
) -> "ResolvedModeConfig":
|
|
|
|
settings: AppSettings = load_settings()
|
|
mode_def: Optional[Mode] = await get_mode_by_id(mode_id)
|
|
|
|
system_prompt: Optional[str] = compose_system_prompt(
|
|
global_default=settings.default_system_prompt,
|
|
mode_prompt=mode_def.system_prompt if mode_def else None,
|
|
session_prompt=session_prompt,
|
|
)
|
|
|
|
allowed_tools, disallowed_tools = toolkit.collect_tool_permissions()
|
|
if mode_def and mode_def.tools is not None:
|
|
mode_tool_set: set[str] = set[str](mode_def.tools)
|
|
disallowed_tools += [t for t in allowed_tools if t not in mode_tool_set]
|
|
allowed_tools = [t for t in allowed_tools if t in mode_tool_set]
|
|
|
|
cwd: Optional[str] = mode_def.default_folder if mode_def and mode_def.default_folder else None
|
|
|
|
return cls(
|
|
system_prompt=system_prompt,
|
|
allowed_tools=allowed_tools,
|
|
disallowed_tools=disallowed_tools,
|
|
cwd=cwd,
|
|
) |