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:
@@ -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");
|
||||
});
|
||||
});
|
||||
|
||||
|
||||
@@ -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
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
|
||||
Reference in New Issue
Block a user