mirror of
https://github.com/kennethnym/aris.git
synced 2026-03-30 06:41:18 +01:00
feat: combine hourly weather into single feed item (#100)
* feat: combine hourly weather into single feed item Co-authored-by: Ona <no-reply@ona.com> * fix: use worst-case timeRelevance, improve tests - Use most urgent timeRelevance across hours instead of hardcoded Ambient - Use HourlyWeatherData type in test casts - Add test for averaged urgency with mixed conditions Co-authored-by: Ona <no-reply@ona.com> --------- Co-authored-by: Ona <no-reply@ona.com>
This commit is contained in:
@@ -32,7 +32,7 @@ export interface CurrentWeatherFeedItem extends FeedItem<
|
|||||||
CurrentWeatherData
|
CurrentWeatherData
|
||||||
> {}
|
> {}
|
||||||
|
|
||||||
export type HourlyWeatherData = {
|
export type HourlyWeatherEntry = {
|
||||||
forecastTime: Date
|
forecastTime: Date
|
||||||
conditionCode: ConditionCode
|
conditionCode: ConditionCode
|
||||||
daylight: boolean
|
daylight: boolean
|
||||||
@@ -48,6 +48,10 @@ export type HourlyWeatherData = {
|
|||||||
windSpeed: number
|
windSpeed: number
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export type HourlyWeatherData = {
|
||||||
|
hours: HourlyWeatherEntry[]
|
||||||
|
}
|
||||||
|
|
||||||
export interface HourlyWeatherFeedItem extends FeedItem<
|
export interface HourlyWeatherFeedItem extends FeedItem<
|
||||||
typeof WeatherFeedItemType.Hourly,
|
typeof WeatherFeedItemType.Hourly,
|
||||||
HourlyWeatherData
|
HourlyWeatherData
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ export {
|
|||||||
type CurrentWeatherData,
|
type CurrentWeatherData,
|
||||||
type HourlyWeatherFeedItem,
|
type HourlyWeatherFeedItem,
|
||||||
type HourlyWeatherData,
|
type HourlyWeatherData,
|
||||||
|
type HourlyWeatherEntry,
|
||||||
type DailyWeatherFeedItem,
|
type DailyWeatherFeedItem,
|
||||||
type DailyWeatherData,
|
type DailyWeatherData,
|
||||||
type WeatherAlertFeedItem,
|
type WeatherAlertFeedItem,
|
||||||
|
|||||||
@@ -4,10 +4,10 @@ import { Context } from "@aelis/core"
|
|||||||
import { LocationKey } from "@aelis/source-location"
|
import { LocationKey } from "@aelis/source-location"
|
||||||
import { describe, expect, test } from "bun:test"
|
import { describe, expect, test } from "bun:test"
|
||||||
|
|
||||||
import type { WeatherKitClient, WeatherKitResponse } from "./weatherkit"
|
import type { WeatherKitClient, WeatherKitResponse, HourlyForecast } from "./weatherkit"
|
||||||
|
|
||||||
import fixture from "../fixtures/san-francisco.json"
|
import fixture from "../fixtures/san-francisco.json"
|
||||||
import { WeatherFeedItemType } from "./feed-items"
|
import { WeatherFeedItemType, type HourlyWeatherData } from "./feed-items"
|
||||||
import { WeatherKey, type Weather } from "./weather-context"
|
import { WeatherKey, type Weather } from "./weather-context"
|
||||||
import { WeatherSource, Units } from "./weather-source"
|
import { WeatherSource, Units } from "./weather-source"
|
||||||
|
|
||||||
@@ -131,10 +131,67 @@ describe("WeatherSource", () => {
|
|||||||
const hourlyItems = items.filter((i) => i.type === WeatherFeedItemType.Hourly)
|
const hourlyItems = items.filter((i) => i.type === WeatherFeedItemType.Hourly)
|
||||||
const dailyItems = items.filter((i) => i.type === WeatherFeedItemType.Daily)
|
const dailyItems = items.filter((i) => i.type === WeatherFeedItemType.Daily)
|
||||||
|
|
||||||
expect(hourlyItems.length).toBe(3)
|
expect(hourlyItems.length).toBe(1)
|
||||||
|
expect((hourlyItems[0]!.data as HourlyWeatherData).hours.length).toBe(3)
|
||||||
expect(dailyItems.length).toBe(2)
|
expect(dailyItems.length).toBe(2)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
test("produces a single hourly item with hours array", async () => {
|
||||||
|
const source = new WeatherSource({ client: mockClient })
|
||||||
|
const context = createMockContext({ lat: 37.7749, lng: -122.4194 })
|
||||||
|
|
||||||
|
const items = await source.fetchItems(context)
|
||||||
|
|
||||||
|
const hourlyItems = items.filter((i) => i.type === WeatherFeedItemType.Hourly)
|
||||||
|
expect(hourlyItems.length).toBe(1)
|
||||||
|
|
||||||
|
const hourlyData = hourlyItems[0]!.data as HourlyWeatherData
|
||||||
|
expect(Array.isArray(hourlyData.hours)).toBe(true)
|
||||||
|
expect(hourlyData.hours.length).toBeGreaterThan(0)
|
||||||
|
expect(hourlyData.hours.length).toBeLessThanOrEqual(12)
|
||||||
|
})
|
||||||
|
|
||||||
|
test("averages urgency across hours with mixed conditions", async () => {
|
||||||
|
const mildHour: HourlyForecast = {
|
||||||
|
forecastStart: "2026-01-17T01:00:00Z",
|
||||||
|
conditionCode: "Clear",
|
||||||
|
daylight: true,
|
||||||
|
humidity: 0.5,
|
||||||
|
precipitationAmount: 0,
|
||||||
|
precipitationChance: 0,
|
||||||
|
precipitationType: "clear",
|
||||||
|
pressure: 1013,
|
||||||
|
snowfallIntensity: 0,
|
||||||
|
temperature: 20,
|
||||||
|
temperatureApparent: 20,
|
||||||
|
temperatureDewPoint: 10,
|
||||||
|
uvIndex: 3,
|
||||||
|
visibility: 20000,
|
||||||
|
windDirection: 180,
|
||||||
|
windGust: 10,
|
||||||
|
windSpeed: 5,
|
||||||
|
}
|
||||||
|
const severeHour: HourlyForecast = {
|
||||||
|
...mildHour,
|
||||||
|
forecastStart: "2026-01-17T02:00:00Z",
|
||||||
|
conditionCode: "SevereThunderstorm",
|
||||||
|
}
|
||||||
|
const mixedResponse: WeatherKitResponse = {
|
||||||
|
forecastHourly: { hours: [mildHour, severeHour] },
|
||||||
|
}
|
||||||
|
const source = new WeatherSource({ client: createMockClient(mixedResponse) })
|
||||||
|
const context = createMockContext({ lat: 37.7749, lng: -122.4194 })
|
||||||
|
|
||||||
|
const items = await source.fetchItems(context)
|
||||||
|
const hourlyItem = items.find((i) => i.type === WeatherFeedItemType.Hourly)
|
||||||
|
|
||||||
|
expect(hourlyItem).toBeDefined()
|
||||||
|
// Mild urgency = 0.3, severe urgency = 0.6, average = 0.45
|
||||||
|
expect(hourlyItem!.signals!.urgency).toBeCloseTo(0.45, 5)
|
||||||
|
// Worst-case: SevereThunderstorm → Imminent
|
||||||
|
expect(hourlyItem!.signals!.timeRelevance).toBe("imminent")
|
||||||
|
})
|
||||||
|
|
||||||
test("sets timestamp from context.time", async () => {
|
test("sets timestamp from context.time", async () => {
|
||||||
const source = new WeatherSource({ client: mockClient })
|
const source = new WeatherSource({ client: mockClient })
|
||||||
const queryTime = new Date("2026-01-17T12:00:00Z")
|
const queryTime = new Date("2026-01-17T12:00:00Z")
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import type { ActionDefinition, ContextEntry, FeedItemSignals, FeedSource } from
|
|||||||
import { Context, TimeRelevance, UnknownActionError } from "@aelis/core"
|
import { Context, TimeRelevance, UnknownActionError } from "@aelis/core"
|
||||||
import { LocationKey } from "@aelis/source-location"
|
import { LocationKey } from "@aelis/source-location"
|
||||||
|
|
||||||
import { WeatherFeedItemType, type WeatherFeedItem } from "./feed-items"
|
import { WeatherFeedItemType, type HourlyWeatherEntry, type WeatherFeedItem } from "./feed-items"
|
||||||
import currentWeatherInsightPrompt from "./prompts/current-weather-insight.txt"
|
import currentWeatherInsightPrompt from "./prompts/current-weather-insight.txt"
|
||||||
import { WeatherKey, type Weather } from "./weather-context"
|
import { WeatherKey, type Weather } from "./weather-context"
|
||||||
import {
|
import {
|
||||||
@@ -174,11 +174,8 @@ export class WeatherSource implements FeedSource<WeatherFeedItem> {
|
|||||||
|
|
||||||
if (response.forecastHourly?.hours) {
|
if (response.forecastHourly?.hours) {
|
||||||
const hours = response.forecastHourly.hours.slice(0, this.hourlyLimit)
|
const hours = response.forecastHourly.hours.slice(0, this.hourlyLimit)
|
||||||
for (let i = 0; i < hours.length; i++) {
|
if (hours.length > 0) {
|
||||||
const hour = hours[i]
|
items.push(createHourlyForecastFeedItem(hours, timestamp, this.units, this.id))
|
||||||
if (hour) {
|
|
||||||
items.push(createHourlyWeatherFeedItem(hour, i, timestamp, this.units, this.id))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -323,24 +320,18 @@ function createCurrentWeatherFeedItem(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function createHourlyWeatherFeedItem(
|
function createHourlyForecastFeedItem(
|
||||||
hourly: HourlyForecast,
|
hourlyForecasts: HourlyForecast[],
|
||||||
index: number,
|
|
||||||
timestamp: Date,
|
timestamp: Date,
|
||||||
units: Units,
|
units: Units,
|
||||||
sourceId: string,
|
sourceId: string,
|
||||||
): WeatherFeedItem {
|
): WeatherFeedItem {
|
||||||
const signals: FeedItemSignals = {
|
const hours: HourlyWeatherEntry[] = []
|
||||||
urgency: adjustUrgencyForCondition(BASE_URGENCY.hourly, hourly.conditionCode),
|
let totalUrgency = 0
|
||||||
timeRelevance: timeRelevanceForCondition(hourly.conditionCode),
|
let worstTimeRelevance: TimeRelevance = TimeRelevance.Ambient
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
for (const hourly of hourlyForecasts) {
|
||||||
id: `weather-hourly-${timestamp.getTime()}-${index}`,
|
hours.push({
|
||||||
sourceId,
|
|
||||||
type: WeatherFeedItemType.Hourly,
|
|
||||||
timestamp,
|
|
||||||
data: {
|
|
||||||
forecastTime: new Date(hourly.forecastStart),
|
forecastTime: new Date(hourly.forecastStart),
|
||||||
conditionCode: hourly.conditionCode,
|
conditionCode: hourly.conditionCode,
|
||||||
daylight: hourly.daylight,
|
daylight: hourly.daylight,
|
||||||
@@ -354,7 +345,27 @@ function createHourlyWeatherFeedItem(
|
|||||||
windDirection: hourly.windDirection,
|
windDirection: hourly.windDirection,
|
||||||
windGust: convertSpeed(hourly.windGust, units),
|
windGust: convertSpeed(hourly.windGust, units),
|
||||||
windSpeed: convertSpeed(hourly.windSpeed, units),
|
windSpeed: convertSpeed(hourly.windSpeed, units),
|
||||||
},
|
})
|
||||||
|
totalUrgency += adjustUrgencyForCondition(BASE_URGENCY.hourly, hourly.conditionCode)
|
||||||
|
const rel = timeRelevanceForCondition(hourly.conditionCode)
|
||||||
|
if (rel === TimeRelevance.Imminent) {
|
||||||
|
worstTimeRelevance = TimeRelevance.Imminent
|
||||||
|
} else if (rel === TimeRelevance.Upcoming && worstTimeRelevance !== TimeRelevance.Imminent) {
|
||||||
|
worstTimeRelevance = TimeRelevance.Upcoming
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const signals: FeedItemSignals = {
|
||||||
|
urgency: totalUrgency / hours.length,
|
||||||
|
timeRelevance: worstTimeRelevance,
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
id: `weather-hourly-${timestamp.getTime()}`,
|
||||||
|
sourceId,
|
||||||
|
type: WeatherFeedItemType.Hourly,
|
||||||
|
timestamp,
|
||||||
|
data: { hours },
|
||||||
signals,
|
signals,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user