generated from coulomb/repo-seed
127 lines
5.6 KiB
Haskell
127 lines
5.6 KiB
Haskell
module Web.Controller.AgentRegistrations where
|
|
|
|
-- IHF Phase 11 — Advanced AI Federation (IHUB-WP-0012 T03)
|
|
|
|
import Web.Types
|
|
import Generated.Types
|
|
import IHP.Prelude
|
|
import IHP.ControllerPrelude
|
|
import IHP.ModelSupport (sqlQuery)
|
|
import Web.View.AgentRegistrations.Index
|
|
import Web.View.AgentRegistrations.Show
|
|
import Web.View.AgentRegistrations.New
|
|
import Web.View.AgentRegistrations.Edit
|
|
import Web.View.AgentRegistrations.Performance
|
|
|
|
instance Controller AgentRegistrationsController where
|
|
|
|
action AgentRegistrationsAction = do
|
|
agents <- query @AgentRegistration
|
|
|> orderByAsc #name
|
|
|> fetch
|
|
hubs <- query @Hub |> orderByAsc #name |> fetch
|
|
render IndexView { .. }
|
|
|
|
action ShowAgentRegistrationAction { agentRegistrationId } = do
|
|
agent <- fetch agentRegistrationId
|
|
policies <- query @ModelRoutingPolicy
|
|
|> filterWhere (#agentRegistrationId, agentRegistrationId)
|
|
|> orderByAsc #taskType
|
|
|> fetch
|
|
recentProposals <- query @AgentProposal
|
|
|> filterWhere (#agentRegistrationId, Just agentRegistrationId)
|
|
|> orderByDesc #createdAt
|
|
|> limit 10
|
|
|> fetch
|
|
mPerformance <- query @AgentPerformanceRecord
|
|
|> filterWhere (#agentRegistrationId, agentRegistrationId)
|
|
|> orderByDesc #computedAt
|
|
|> limit 1
|
|
|> fetchOneOrNothing
|
|
render ShowView { .. }
|
|
|
|
action NewAgentRegistrationAction = do
|
|
let agent = newRecord @AgentRegistration
|
|
hubs <- query @Hub |> orderByAsc #name |> fetch
|
|
render NewView { .. }
|
|
|
|
action CreateAgentRegistrationAction = do
|
|
let agent = newRecord @AgentRegistration
|
|
agent
|
|
|> fill @'["hubId","name","slug","description","provider","modelName","trustLevel","systemPrompt"]
|
|
|> set #capabilities (A.Array mempty)
|
|
|> validateField #name nonEmpty
|
|
|> validateField #slug nonEmpty
|
|
|> validateField #provider (isInList ["openrouter","gemini","openai","claude-code"])
|
|
|> validateField #modelName nonEmpty
|
|
|> validateField #trustLevel (isInList ["advisory","elevated","autonomous"])
|
|
|> ifValid \case
|
|
Left agent -> do
|
|
hubs <- query @Hub |> orderByAsc #name |> fetch
|
|
render NewView { .. }
|
|
Right agent -> do
|
|
agent <- createRecord agent
|
|
setSuccessMessage "Agent registered"
|
|
redirectTo AgentRegistrationsAction
|
|
|
|
action EditAgentRegistrationAction { agentRegistrationId } = do
|
|
agent <- fetch agentRegistrationId
|
|
hubs <- query @Hub |> orderByAsc #name |> fetch
|
|
render EditView { .. }
|
|
|
|
action UpdateAgentRegistrationAction { agentRegistrationId } = do
|
|
agent <- fetch agentRegistrationId
|
|
agent
|
|
|> fill @'["name","description","provider","modelName","trustLevel","systemPrompt"]
|
|
|> validateField #name nonEmpty
|
|
|> validateField #provider (isInList ["openrouter","gemini","openai","claude-code"])
|
|
|> validateField #modelName nonEmpty
|
|
|> validateField #trustLevel (isInList ["advisory","elevated","autonomous"])
|
|
|> ifValid \case
|
|
Left agent -> do
|
|
hubs <- query @Hub |> orderByAsc #name |> fetch
|
|
render EditView { .. }
|
|
Right agent -> do
|
|
updateRecord agent
|
|
setSuccessMessage "Agent updated"
|
|
redirectTo (ShowAgentRegistrationAction agentRegistrationId)
|
|
|
|
action DeactivateAgentAction { agentRegistrationId } = do
|
|
agent <- fetch agentRegistrationId
|
|
agent |> set #isActive False |> updateRecord
|
|
setSuccessMessage "Agent deactivated"
|
|
redirectTo (ShowAgentRegistrationAction agentRegistrationId)
|
|
|
|
action ComputeAgentPerformanceAction { agentRegistrationId } = do
|
|
agent <- fetch agentRegistrationId
|
|
rows <- sqlQuery
|
|
"SELECT \
|
|
\ COUNT(*) FILTER (WHERE ap.status = 'accepted')::int AS accepted, \
|
|
\ COUNT(*) FILTER (WHERE ap.status = 'rejected')::int AS rejected, \
|
|
\ COUNT(*) FILTER (WHERE ap.status NOT IN ('accepted','rejected'))::int AS other, \
|
|
\ COUNT(*)::int AS total, \
|
|
\ AVG(ca.score) AS mean_confidence \
|
|
\ FROM agent_proposals ap \
|
|
\ LEFT JOIN confidence_annotations ca ON ca.proposal_id = ap.id \
|
|
\ WHERE ap.agent_registration_id = ? \
|
|
\ AND ap.created_at >= NOW() - INTERVAL '30 days'"
|
|
[PersistUUID (toUUID agentRegistrationId)]
|
|
case rows of
|
|
[(accepted, rejected, _other, total, mConf)] -> do
|
|
now <- getCurrentTime
|
|
let periodStart = addUTCTime (negate $ 30 * 86400) now
|
|
newRecord @AgentPerformanceRecord
|
|
|> set #agentRegistrationId agentRegistrationId
|
|
|> set #hubId agent.hubId
|
|
|> set #periodStart periodStart
|
|
|> set #periodEnd now
|
|
|> set #proposalsGenerated total
|
|
|> set #proposalsAccepted accepted
|
|
|> set #proposalsRejected rejected
|
|
|> set #proposalsRevised 0
|
|
|> set #meanConfidence mConf
|
|
|> createRecord
|
|
setSuccessMessage "Performance snapshot computed"
|
|
_ -> setErrorMessage "Could not compute performance metrics"
|
|
redirectTo (ShowAgentRegistrationAction agentRegistrationId)
|