34 lines
1.4 KiB
Java
34 lines
1.4 KiB
Java
|
package com.company.agentfoundation.agent;
|
||
|
|
|
||
|
|
import com.company.agentfoundation.security.AgentAuthService;
|
||
|
|
import jakarta.validation.Valid;
|
||
|
|
import org.springframework.http.ResponseEntity;
|
||
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
||
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
||
|
|
import org.springframework.web.bind.annotation.RequestHeader;
|
||
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
||
|
|
import org.springframework.web.bind.annotation.RestController;
|
||
|
|
|
||
|
|
@RestController
|
||
|
|
@RequestMapping("/v1/agent")
|
||
|
|
public class AgentRuntimeController {
|
||
|
|
private final AgentAuthService authService;
|
||
|
|
private final AgentOrchestratorService agentOrchestratorService;
|
||
|
|
|
||
|
|
public AgentRuntimeController(AgentAuthService authService, AgentOrchestratorService agentOrchestratorService) {
|
||
|
|
this.authService = authService;
|
||
|
|
this.agentOrchestratorService = agentOrchestratorService;
|
||
|
|
}
|
||
|
|
|
||
|
|
@PostMapping("/messages")
|
||
|
|
public ResponseEntity<AgentMessageResponse> handleMessage(
|
||
|
|
@RequestHeader("X-Agent-Id") String agentId,
|
||
|
|
@RequestHeader("X-Internal-Key") String internalKey,
|
||
|
|
@Valid @RequestBody AgentMessageRequest request
|
||
|
|
) {
|
||
|
|
authService.authenticate(agentId, internalKey);
|
||
|
|
String reply = agentOrchestratorService.handleMessage(agentId, request.wecomUserId(), request.message());
|
||
|
|
return ResponseEntity.ok(AgentMessageResponse.of(agentId, request.wecomUserId(), reply));
|
||
|
|
}
|
||
|
|
}
|