feat: attaching both to image

backend/agents/event_location_agent.go

@@ -283,26 +283,21 @@ func NewLocationEventAgent(locationModel models.LocationModel, eventModel models
 			return args, err
 		},
 		Fn: func(info ToolHandlerInfo, args CreateLocationArguments, call ToolCall) (model.Locations, error) {
-			return agent.locationModel.Save(context.Background(), info.userId, model.Locations{
+			ctx := context.Background()
+
+			location, err := agent.locationModel.Save(ctx, info.userId, model.Locations{
 				Name:        args.Name,
 				Address:     args.Address,
 				Coordinates: args.Coordinates,
 			})
-		},
-	}
 
-	// I'm not sure this one actually makes sense either.
-	// I think the earlier tool can do more.
-	toolHandler.Handlers["attachImageLocation"] = ToolHandler[AttachImageLocationArguments, model.ImageLocations]{
-		FunctionName: "attachImageLocation",
-		Parse: func(stringArgs string) (AttachImageLocationArguments, error) {
-			args := AttachImageLocationArguments{}
-			err := json.Unmarshal([]byte(stringArgs), &args)
+			if err != nil {
+				return location, err
+			}
 
-			return args, err
-		},
-		Fn: func(info ToolHandlerInfo, args AttachImageLocationArguments, call ToolCall) (model.ImageLocations, error) {
-			return agent.locationModel.SaveToImage(context.Background(), info.imageId, uuid.MustParse(args.LocationId))
+			_, err = agent.locationModel.SaveToImage(ctx, info.imageId, location.ID)
+
+			return location, err
 		},
 	}
 
@@ -325,6 +320,11 @@ func NewLocationEventAgent(locationModel models.LocationModel, eventModel models
 				return event, err
 			}
 
+			_, err = agent.eventModel.SaveToImage(ctx, info.imageId, event.ID)
+			if err != nil {
+				return event, err
+			}
+
 			locationId, err := uuid.Parse(args.LocationId)
 			if err != nil {
 				return event, err