Add weather tool to fetch weather data

src/lib/server/tools/index.ts

@@ -19,6 +19,7 @@ import calculator from "./calculator";
 import directlyAnswer from "./directlyAnswer";
 import fetchUrl from "./web/url";
 import websearch from "./web/search";
+import weather from "./weather";
 import { callSpace, getIpToken } from "./utils";
 import { uploadFile } from "../files/uploadFile";
 import type { MessageFile } from "$lib/types/Message";
@@ -127,7 +128,7 @@ export const configTools = z
 			}))
 	)
 	// add the extra hardcoded tools
-	.transform((val) => [...val, calculator, directlyAnswer, fetchUrl, websearch]);
+	.transform((val) => [...val, calculator, directlyAnswer, fetchUrl, websearch, weather]);
 
 export function getCallMethod(tool: Omit<BaseTool, "call">): BackendCall {
 	return async function* (params, ctx, uuid) {

src/lib/server/tools/outputs.ts

@@ -43,6 +43,10 @@ export const ToolOutputPaths: Record<
 		type: "str",
 		path: "$",
 	},
+	weather: {
+		type: "str",
+		path: "$",
+	},
 };
 
 export const isValidOutputComponent = (

src/lib/server/tools/utils.ts

@@ -111,3 +111,28 @@ export async function extractJson(text: string): Promise<unknown[]> {
 	}
 	return calls.flat();
 }
+
+export async function fetchWeatherData(latitude: number, longitude: number): Promise<any> {
+	const response = await fetch(
+		`https://api.open-meteo.com/v1/forecast?latitude=${latitude}&longitude=${longitude}&hourly=temperature_2m`
+	);
+	if (!response.ok) {
+		throw new Error("Failed to fetch weather data");
+	}
+	return response.json();
+}
+
+export async function fetchCoordinates(location: string): Promise<{ latitude: number; longitude: number }> {
+	const response = await fetch(
+		`https://geocoding-api.open-meteo.com/v1/search?name=${location}&count=1`
+	);
+	if (!response.ok) {
+		throw new Error("Failed to fetch coordinates");
+	}
+	const data = await response.json();
+	if (data.results.length === 0) {
+		throw new Error("Location not found");
+	}
+	const { latitude, longitude } = data.results[0];
+	return { latitude, longitude };
+}

src/lib/server/tools/weather.ts

@@ -0,0 +1,39 @@
+import type { ConfigTool } from "$lib/types/Tool";
+import { ObjectId } from "mongodb";
+import { fetchWeatherData, fetchCoordinates } from "./utils";
+
+const weather: ConfigTool = {
+  _id: new ObjectId("00000000000000000000000D"),
+  type: "config",
+  description: "Fetch the weather for a specified location",
+  color: "blue",
+  icon: "cloud",
+  displayName: "Weather",
+  name: "weather",
+  endpoint: null,
+  inputs: [
+    {
+      name: "location",
+      type: "str",
+      description: "The name of the location to fetch the weather for",
+      paramType: "required",
+    },
+  ],
+  outputComponent: null,
+  outputComponentIdx: null,
+  showOutput: false,
+  async *call({ location }) {
+    try {
+      const coordinates = await fetchCoordinates(location);
+      const weatherData = await fetchWeatherData(coordinates.latitude, coordinates.longitude);
+
+      return {
+        outputs: [{ weather: weatherData }],
+      };
+    } catch (error) {
+      throw new Error("Failed to fetch weather data", { cause: error });
+    }
+  },
+};
+
+export default weather;