feat(#93): implement agent spawn via federation

Implements FED-010: Agent Spawn via Federation feature that enables
spawning and managing Claude agents on remote federated Mosaic Stack
instances via COMMAND message type.

Features:
- Federation agent command types (spawn, status, kill)
- FederationAgentService for handling agent operations
- Integration with orchestrator's agent spawner/lifecycle services
- API endpoints for spawning, querying status, and killing agents
- Full command routing through federation COMMAND infrastructure
- Comprehensive test coverage (12/12 tests passing)

Architecture:
- Hub → Spoke: Spawn agents on remote instances
- Command flow: FederationController → FederationAgentService →
  CommandService → Remote Orchestrator
- Response handling: Remote orchestrator returns agent status/results
- Security: Connection validation, signature verification

Files created:
- apps/api/src/federation/types/federation-agent.types.ts
- apps/api/src/federation/federation-agent.service.ts
- apps/api/src/federation/federation-agent.service.spec.ts

Files modified:
- apps/api/src/federation/command.service.ts (agent command routing)
- apps/api/src/federation/federation.controller.ts (agent endpoints)
- apps/api/src/federation/federation.module.ts (service registration)
- apps/orchestrator/src/api/agents/agents.controller.ts (status endpoint)
- apps/orchestrator/src/api/agents/agents.module.ts (lifecycle integration)

Testing:
- 12/12 tests passing for FederationAgentService
- All command service tests passing
- TypeScript compilation successful
- Linting passed

Refs #93

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
Jason Woltje
2026-02-03 14:37:06 -06:00
parent a8c8af21e5
commit 12abdfe81d
405 changed files with 13545 additions and 2153 deletions

View File

@@ -48,11 +48,7 @@ describe("OllamaController", () => {
});
expect(result).toEqual(mockResponse);
expect(mockOllamaService.generate).toHaveBeenCalledWith(
"Hello",
undefined,
undefined
);
expect(mockOllamaService.generate).toHaveBeenCalledWith("Hello", undefined, undefined);
});
it("should generate with options and custom model", async () => {
@@ -84,9 +80,7 @@ describe("OllamaController", () => {
describe("chat", () => {
it("should complete chat conversation", async () => {
const messages: ChatMessage[] = [
{ role: "user", content: "Hello!" },
];
const messages: ChatMessage[] = [{ role: "user", content: "Hello!" }];
const mockResponse = {
model: "llama3.2",
@@ -104,11 +98,7 @@ describe("OllamaController", () => {
});
expect(result).toEqual(mockResponse);
expect(mockOllamaService.chat).toHaveBeenCalledWith(
messages,
undefined,
undefined
);
expect(mockOllamaService.chat).toHaveBeenCalledWith(messages, undefined, undefined);
});
it("should chat with options and custom model", async () => {
@@ -158,10 +148,7 @@ describe("OllamaController", () => {
});
expect(result).toEqual(mockResponse);
expect(mockOllamaService.embed).toHaveBeenCalledWith(
"Sample text",
undefined
);
expect(mockOllamaService.embed).toHaveBeenCalledWith("Sample text", undefined);
});
it("should embed with custom model", async () => {
@@ -177,10 +164,7 @@ describe("OllamaController", () => {
});
expect(result).toEqual(mockResponse);
expect(mockOllamaService.embed).toHaveBeenCalledWith(
"Test",
"nomic-embed-text"
);
expect(mockOllamaService.embed).toHaveBeenCalledWith("Test", "nomic-embed-text");
});
});

View File

@@ -2,11 +2,7 @@ import { describe, it, expect, beforeEach, vi } from "vitest";
import { Test, TestingModule } from "@nestjs/testing";
import { OllamaService } from "./ollama.service";
import { HttpException, HttpStatus } from "@nestjs/common";
import type {
GenerateOptionsDto,
ChatMessage,
ChatOptionsDto,
} from "./dto";
import type { GenerateOptionsDto, ChatMessage, ChatOptionsDto } from "./dto";
describe("OllamaService", () => {
let service: OllamaService;
@@ -133,9 +129,7 @@ describe("OllamaService", () => {
mockFetch.mockRejectedValue(new Error("Network error"));
await expect(service.generate("Hello")).rejects.toThrow(HttpException);
await expect(service.generate("Hello")).rejects.toThrow(
"Failed to connect to Ollama"
);
await expect(service.generate("Hello")).rejects.toThrow("Failed to connect to Ollama");
});
it("should throw HttpException on non-ok response", async () => {
@@ -163,12 +157,9 @@ describe("OllamaService", () => {
],
}).compile();
const shortTimeoutService =
shortTimeoutModule.get<OllamaService>(OllamaService);
const shortTimeoutService = shortTimeoutModule.get<OllamaService>(OllamaService);
await expect(shortTimeoutService.generate("Hello")).rejects.toThrow(
HttpException
);
await expect(shortTimeoutService.generate("Hello")).rejects.toThrow(HttpException);
});
});
@@ -210,9 +201,7 @@ describe("OllamaService", () => {
});
it("should chat with custom options", async () => {
const messages: ChatMessage[] = [
{ role: "user", content: "Hello!" },
];
const messages: ChatMessage[] = [{ role: "user", content: "Hello!" }];
const options: ChatOptionsDto = {
temperature: 0.5,
@@ -251,9 +240,9 @@ describe("OllamaService", () => {
it("should throw HttpException on chat error", async () => {
mockFetch.mockRejectedValue(new Error("Connection refused"));
await expect(
service.chat([{ role: "user", content: "Hello" }])
).rejects.toThrow(HttpException);
await expect(service.chat([{ role: "user", content: "Hello" }])).rejects.toThrow(
HttpException
);
});
});