Fixed auto approver
This commit is contained in:
@@ -2,15 +2,16 @@ from aiohttp import web
|
||||
import json
|
||||
import asyncio
|
||||
|
||||
from ..auto_approver import AutoApprover
|
||||
from ..web_agent import WebAgent
|
||||
from ..web_io_buffer import WebIOBuffer
|
||||
from .util import wrap_async
|
||||
|
||||
class Api:
|
||||
def __init__(self, app: web.Application, agent: WebAgent, io_buffer: WebIOBuffer):
|
||||
def __init__(self, app: web.Application, agent: WebAgent, io_buffer: WebIOBuffer, auto_approver: AutoApprover):
|
||||
self._app = app
|
||||
self._agent = agent
|
||||
self._io_buffer = io_buffer
|
||||
self._auto_approver = auto_approver
|
||||
|
||||
self._init_routes()
|
||||
|
||||
@@ -23,10 +24,13 @@ class Api:
|
||||
self._app.router.add_post("/api/clear", self._clear_output)
|
||||
self._app.router.add_get("/api/output/{llm}", self._get_output)
|
||||
self._app.router.add_get("/api/llms", self._get_llms)
|
||||
|
||||
@property
|
||||
def app(self):
|
||||
return self._app
|
||||
self._app.router.add_get("/api/auto_approver/config", self._get_auto_approver_config)
|
||||
self._app.router.add_post("/api/auto_approver/config", self._set_auto_approver_config)
|
||||
self._app.router.add_post("/api/auto_approver/context_enabled", self._set_context_enabled)
|
||||
self._app.router.add_post("/api/auto_approver/response_enabled", self._set_response_enabled)
|
||||
self._app.router.add_post("/api/auto_approver/context_timeout", self._set_context_timeout)
|
||||
self._app.router.add_post("/api/auto_approver/response_timeout", self._set_response_timeout)
|
||||
self._app.router.add_post("/api/auto_approver/llm", self._set_llm_name)
|
||||
|
||||
async def _run_inference(self, request: web.Request) -> web.Response:
|
||||
"""Start inference on specified LLM."""
|
||||
@@ -95,3 +99,79 @@ class Api:
|
||||
}),
|
||||
content_type="application/json"
|
||||
)
|
||||
|
||||
async def _get_auto_approver_config(self, request: web.Request) -> web.Response:
|
||||
"""Get current auto approver configuration."""
|
||||
return web.Response(
|
||||
text=json.dumps(self._auto_approver.config),
|
||||
content_type="application/json"
|
||||
)
|
||||
|
||||
async def _set_auto_approver_config(self, request: web.Request) -> web.Response:
|
||||
"""Update auto approver configuration."""
|
||||
data = await request.json()
|
||||
try:
|
||||
self._auto_approver.set_config(data)
|
||||
return web.Response(status=200)
|
||||
except (ValueError, KeyError) as e:
|
||||
return web.Response(status=400, text=str(e))
|
||||
|
||||
async def _set_context_enabled(self, request: web.Request) -> web.Response:
|
||||
"""Set context auto-approval enabled state."""
|
||||
data = await request.json()
|
||||
enabled = data.get("enabled")
|
||||
if enabled is None:
|
||||
return web.Response(status=400, text="Missing enabled parameter")
|
||||
try:
|
||||
self._auto_approver.context_enabled = enabled
|
||||
return web.Response(status=200)
|
||||
except ValueError as e:
|
||||
return web.Response(status=400, text=str(e))
|
||||
|
||||
async def _set_response_enabled(self, request: web.Request) -> web.Response:
|
||||
"""Set response auto-approval enabled state."""
|
||||
data = await request.json()
|
||||
enabled = data.get("enabled")
|
||||
if enabled is None:
|
||||
return web.Response(status=400, text="Missing enabled parameter")
|
||||
try:
|
||||
self._auto_approver.response_enabled = enabled
|
||||
return web.Response(status=200)
|
||||
except ValueError as e:
|
||||
return web.Response(status=400, text=str(e))
|
||||
|
||||
async def _set_context_timeout(self, request: web.Request) -> web.Response:
|
||||
"""Set context auto-approval timeout."""
|
||||
data = await request.json()
|
||||
timeout = data.get("timeout")
|
||||
if timeout is None:
|
||||
return web.Response(status=400, text="Missing timeout parameter")
|
||||
try:
|
||||
self._auto_approver.context_timeout = float(timeout)
|
||||
return web.Response(status=200)
|
||||
except ValueError as e:
|
||||
return web.Response(status=400, text=str(e))
|
||||
|
||||
async def _set_response_timeout(self, request: web.Request) -> web.Response:
|
||||
"""Set response auto-approval timeout."""
|
||||
data = await request.json()
|
||||
timeout = data.get("timeout")
|
||||
if timeout is None:
|
||||
return web.Response(status=400, text="Missing timeout parameter")
|
||||
try:
|
||||
self._auto_approver.response_timeout = float(timeout)
|
||||
return web.Response(status=200)
|
||||
except ValueError as e:
|
||||
return web.Response(status=400, text=str(e))
|
||||
|
||||
async def _set_llm_name(self, request: web.Request) -> web.Response:
|
||||
"""Set LLM name for auto-approval."""
|
||||
data = await request.json()
|
||||
name = data.get("name")
|
||||
if name is None:
|
||||
return web.Response(status=400, text="Missing name parameter")
|
||||
try:
|
||||
self._auto_approver.llm_name = name
|
||||
return web.Response(status=200)
|
||||
except ValueError as e:
|
||||
return web.Response(status=400, text=str(e))
|
||||
Reference in New Issue
Block a user