allow matt and mary can use specify skill

src/agent/NpcMary.ts

@@ -13,6 +13,7 @@ import {
 	canChangeFavorability,
 	canGetFavorability,
 } from "./tools/Favorability";
+import { canCallPeople } from "./tools/Skill";
 
 @injectable()
 export class NpcMary {
@@ -42,6 +43,7 @@ export class NpcMary {
 			tools: {
 				getFavorability: canGetFavorability(this.name, city),
 				changeFavorability: canChangeFavorability(this.name, city),
+				callPeople: canCallPeople(city),
 			},
 		});
 

src/agent/NpcMatt.ts

@@ -13,6 +13,7 @@ import {
 	canChangeFavorability,
 	canGetFavorability,
 } from "./tools/Favorability";
+import { canEnableProtectMachine } from "./tools/Skill";
 
 @injectable()
 export class NpcMatt {
@@ -40,6 +41,7 @@ export class NpcMatt {
 			tools: {
 				getFavorability: canGetFavorability(this.name, city),
 				changeFavorability: canChangeFavorability(this.name, city),
+				enableProtectMachine: canEnableProtectMachine(city),
 			},
 		});
 

src/agent/tools/Skill.ts

@@ -0,0 +1,30 @@
+import { z } from "zod";
+import { tool } from "ai";
+
+import { City } from "@entity/City";
+
+export function canEnableProtectMachine(city: City) {
+	return tool({
+		description:
+			"Can enable the protect machine to reduce damage rate, if the machine is already enabled it will return false",
+		parameters: z.object({}),
+		execute: async () => {
+			const success = city.enableProtectedMachine();
+
+			return { success };
+		},
+	});
+}
+
+export function canCallPeople(city: City) {
+	return tool({
+		description:
+			"Can call people to reduce damage rate, if the people is already called it will return false",
+		parameters: z.object({}),
+		execute: async () => {
+			const success = city.callPeople();
+
+			return { success };
+		},
+	});
+}

src/entity/City.ts

@@ -19,6 +19,7 @@ export class City {
 	static readonly MAX_DAMAGE = 100;
 	static readonly MIN_DAMAGE = 0;
 	static readonly TARGET_DAMAGE_RATE = 0;
+	static readonly DAMAGE_RATE_DECREASE_RATIO = 0.25;
 	static readonly REQUIRED_MAX_FAVORABILITY_NPC = 3;
 	static readonly REQUIRED_MIN_FAVORABILITY = 80;
 
@@ -178,10 +179,12 @@ export class City {
 	public onEnableProtectedMachineEvent(
 		event: EnableProtectedMachineEvent,
 	): void {
+		this._damageRate = this._damageRate * City.DAMAGE_RATE_DECREASE_RATIO;
 		this._protecteMachineEnabled = true;
 	}
 
 	public onCallPeopleEvent(event: CallPeopleEvent): void {
+		this._damageRate = this._damageRate * City.DAMAGE_RATE_DECREASE_RATIO;
 		this._peopleIsCalled = true;
 	}
 }