.

Author: mattjohnsonpint

sdk/go/examples/textgeneration/toolcalling.go

@@ -130,6 +130,10 @@ func GenerateTextWithTools(prompt string) (string, error) {
 
 // This function will return the current time in a given time zone.
 func getCurrentTime(tz string) (string, error) {
+	if !localtime.IsValidTimeZone(tz) {
+		return "", errors.New("Invalid time zone.")
+	}
+
 	now, err := localtime.NowInZone(tz)
 	if err != nil {
 		return "", err

sdk/go/pkg/localtime/localtime.go

@@ -61,3 +61,9 @@ func GetLocation(tz string) (*time.Location, error) {
 	}
 	return hostGetTimeLocation(tz)
 }
+
+// IsValidTimeZone returns true if the given time zone is valid.
+func IsValidTimeZone(tz string) bool {
+	_, err := hostGetTimeInZone(tz)
+	return err == nil
+}