openclaw-server 0.1.0

package/src/weather/service.ts

@@ -0,0 +1,188 @@
+import { extractWeatherDay, extractWeatherLocation, parseWeatherQuery } from "./parser.js";
+import type {
+  WeatherChatResult,
+  WeatherConversationState,
+  WeatherDay,
+  WeatherMessageInspection,
+  WeatherMissingSlot,
+  WeatherProvider,
+} from "./types.js";
+
+function resolveMissingSlot(location?: string, day?: WeatherDay): WeatherMissingSlot | undefined {
+  if (!location && !day) {
+    return "both";
+  }
+  if (!location) {
+    return "location";
+  }
+  if (!day) {
+    return "day";
+  }
+  return undefined;
+}
+
+function dayLabel(day: WeatherDay): string {
+  switch (day) {
+    case "today":
+      return "今天";
+    case "tomorrow":
+      return "明天";
+    case "day_after_tomorrow":
+      return "后天";
+    case "in_three_days":
+      return "大后天";
+  }
+}
+
+function renderClarifyReply(
+  missing: WeatherMissingSlot,
+  params: { location?: string; day?: WeatherDay },
+): string {
+  if (missing === "both") {
+    return "你想查哪个城市、哪一天的天气？可以直接说“天津明天天气如何”。";
+  }
+  if (missing === "location") {
+    const dayText = params.day ? `${dayLabel(params.day)}` : "";
+    return `你想查${dayText}哪个城市的天气？`;
+  }
+  return `你想查${params.location ?? "这个城市"}哪一天的天气？可以说今天、明天、后天。`;
+}
+
+function renderForecastReply(params: {
+  location: string;
+  day: WeatherDay;
+  summary: string;
+  max: number;
+  min: number;
+  precipitationProbabilityMax?: number;
+  precipitationSum?: number;
+}): string {
+  const parts = [
+    `${params.location}${dayLabel(params.day)}${params.summary}`,
+    `最高 ${Math.round(params.max)}C`,
+    `最低 ${Math.round(params.min)}C`,
+  ];
+
+  if (typeof params.precipitationProbabilityMax === "number") {
+    parts.push(`降水概率 ${Math.round(params.precipitationProbabilityMax)}%`);
+  } else if (
+    typeof params.precipitationSum === "number" &&
+    Number.isFinite(params.precipitationSum)
+  ) {
+    parts.push(`预计降水 ${params.precipitationSum.toFixed(1)} mm`);
+  }
+
+  return `${parts.join("，")}。`;
+}
+
+function renderLookupError(location: string, day: WeatherDay, error: unknown): string {
+  const message = error instanceof Error ? error.message : "";
+  if (message.startsWith("Location not found:")) {
+    return `我还没识别出“${location}”是哪个城市，你可以换个更完整的地名再试一次。`;
+  }
+  return `暂时查不到${location}${dayLabel(day)}的天气，请稍后再试。`;
+}
+
+export class WeatherService {
+  private readonly pending = new Map<string, WeatherConversationState>();
+
+  constructor(
+    private readonly provider: WeatherProvider,
+    private readonly options: { pendingTtlMs?: number } = {},
+  ) {}
+
+  inspectMessage(params: { userId: string; text: string }): WeatherMessageInspection {
+    const pending = this.getPending(params.userId);
+    if (pending) {
+      const location = extractWeatherLocation(params.text) ?? pending.location;
+      const day = extractWeatherDay(params.text) ?? pending.day;
+      return {
+        shouldHandle: true,
+        reason: "conversation",
+        missing: resolveMissingSlot(location, day),
+        location,
+        day,
+      };
+    }
+
+    const query = parseWeatherQuery(params.text);
+    if (!query.isWeatherQuery) {
+      return {
+        shouldHandle: false,
+        reason: "no_match",
+      };
+    }
+
+    return {
+      shouldHandle: true,
+      reason: "query",
+      missing: resolveMissingSlot(query.location, query.day),
+      location: query.location,
+      day: query.day,
+    };
+  }
+
+  async processMessage(params: { userId: string; text: string }): Promise<WeatherChatResult> {
+    const pending = this.getPending(params.userId);
+    const query = parseWeatherQuery(params.text);
+    const location = extractWeatherLocation(params.text) ?? pending?.location ?? query.location;
+    const day = extractWeatherDay(params.text) ?? pending?.day ?? query.day;
+    const missing = resolveMissingSlot(location, day);
+
+    if (missing) {
+      this.pending.set(params.userId, {
+        userId: params.userId,
+        location,
+        day,
+        updatedAt: Date.now(),
+      });
+      return {
+        reply: renderClarifyReply(missing, { location, day }),
+        intent: "clarify",
+        missing,
+      };
+    }
+
+    this.pending.delete(params.userId);
+
+    try {
+      const forecast = await this.provider.lookupForecast({
+        locationQuery: location!,
+        day: day!,
+      });
+      return {
+        reply: renderForecastReply({
+          location: forecast.location.name,
+          day: forecast.day,
+          summary: forecast.summary,
+          max: forecast.temperatureMax,
+          min: forecast.temperatureMin,
+          precipitationProbabilityMax: forecast.precipitationProbabilityMax,
+          precipitationSum: forecast.precipitationSum,
+        }),
+        intent: "weather_forecast",
+        forecast,
+      };
+    } catch (error) {
+      return {
+        reply: renderLookupError(location!, day!, error),
+        intent: "clarify",
+      };
+    }
+  }
+
+  private getPending(userId: string): WeatherConversationState | undefined {
+    const pending = this.pending.get(userId);
+    if (!pending) {
+      return undefined;
+    }
+
+    const ttlMs = this.options.pendingTtlMs ?? 30 * 60_000;
+    if (Date.now() - pending.updatedAt <= ttlMs) {
+      return pending;
+    }
+
+    this.pending.delete(userId);
+    return undefined;
+  }
+}

package/src/weather/types.ts

@@ -0,0 +1,56 @@
+export type WeatherDay = "today" | "tomorrow" | "day_after_tomorrow" | "in_three_days";
+
+export type WeatherLocation = {
+  name: string;
+  latitude: number;
+  longitude: number;
+  timezone?: string;
+  country?: string;
+  admin1?: string;
+};
+
+export type WeatherForecast = {
+  day: WeatherDay;
+  localDate: string;
+  summary: string;
+  temperatureMin: number;
+  temperatureMax: number;
+  precipitationProbabilityMax?: number;
+  precipitationSum?: number;
+  location: WeatherLocation;
+  source: "open-meteo";
+};
+
+export type WeatherQuery = {
+  isWeatherQuery: boolean;
+  location?: string;
+  day?: WeatherDay;
+};
+
+export type WeatherMissingSlot = "location" | "day" | "both";
+
+export type WeatherMessageInspection = {
+  shouldHandle: boolean;
+  reason: "conversation" | "query" | "no_match";
+  missing?: WeatherMissingSlot;
+  location?: string;
+  day?: WeatherDay;
+};
+
+export type WeatherChatResult = {
+  reply: string;
+  intent: "weather_forecast" | "clarify";
+  forecast?: WeatherForecast;
+  missing?: WeatherMissingSlot;
+};
+
+export type WeatherProvider = {
+  lookupForecast(params: { locationQuery: string; day: WeatherDay }): Promise<WeatherForecast>;
+};
+
+export type WeatherConversationState = {
+  userId: string;
+  location?: string;
+  day?: WeatherDay;
+  updatedAt: number;
+};