雷电芽衣
ce0c924c58
A retried POST /v1/auth/rotate loses the replacement token because the presented token is already in grace, and a retried webhook registration creates duplicates. Only idempotent methods now retry. Add previous_token_grace_until to RotateApiTokenResponse and percent-encode the agent id in request paths. Claude-Session: https://claude.ai/code/session_01Agoc9bRVE9KL7mmWtey9Vq
85 lines
2.6 KiB
Go
85 lines
2.6 KiB
Go
package elydora
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
// RegisterAgent registers a new AI agent within the organization.
|
|
func (c *Client) RegisterAgent(req *RegisterAgentRequest) (*RegisterAgentResponse, error) {
|
|
var result RegisterAgentResponse
|
|
if err := c.doPost("/v1/agents/register", req, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
// GetAgent retrieves an agent by ID.
|
|
func (c *Client) GetAgent(agentID string) (*GetAgentResponse, error) {
|
|
var result GetAgentResponse
|
|
if err := c.doGet(fmt.Sprintf("/v1/agents/%s", url.PathEscape(agentID)), &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
// FreezeAgent freezes an agent, preventing it from submitting operations.
|
|
func (c *Client) FreezeAgent(agentID, reason string) (*FreezeAgentResponse, error) {
|
|
var result FreezeAgentResponse
|
|
if err := c.doPost(fmt.Sprintf("/v1/agents/%s/freeze", url.PathEscape(agentID)), &FreezeAgentRequest{
|
|
Reason: reason,
|
|
}, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
// RevokeKey revokes a specific key for an agent.
|
|
func (c *Client) RevokeKey(agentID, kid, reason string) error {
|
|
return c.doPost(fmt.Sprintf("/v1/agents/%s/revoke", url.PathEscape(agentID)), &RevokeAgentRequest{
|
|
KID: kid,
|
|
Reason: reason,
|
|
}, nil)
|
|
}
|
|
|
|
// ListAgents retrieves all agents for the organization.
|
|
func (c *Client) ListAgents() (*ListAgentsResponse, error) {
|
|
var result ListAgentsResponse
|
|
if err := c.doGet("/v1/agents", &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
// UnfreezeAgent unfreezes a previously frozen agent.
|
|
func (c *Client) UnfreezeAgent(agentID, reason string) (*UnfreezeAgentResponse, error) {
|
|
var result UnfreezeAgentResponse
|
|
if err := c.doPost(fmt.Sprintf("/v1/agents/%s/unfreeze", url.PathEscape(agentID)), &UnfreezeAgentRequest{
|
|
Reason: reason,
|
|
}, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
// UpdateAgent updates an agent's integration type.
|
|
func (c *Client) UpdateAgent(agentID string, integrationType IntegrationType) (*UpdateAgentResponse, error) {
|
|
var result UpdateAgentResponse
|
|
if err := c.doRequest(http.MethodPatch, fmt.Sprintf("/v1/agents/%s", url.PathEscape(agentID)), &UpdateAgentRequest{
|
|
IntegrationType: integrationType,
|
|
}, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
// DeleteAgent permanently deletes an agent and all associated data.
|
|
func (c *Client) DeleteAgent(agentID string) (*DeleteAgentResponse, error) {
|
|
var result DeleteAgentResponse
|
|
if err := c.doRequest(http.MethodDelete, fmt.Sprintf("/v1/agents/%s", url.PathEscape(agentID)), nil, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|